Merge pull request #32 from segment-integrations/add/setting_source_name

Add/setting source name

Author: ladanazita

circle.yml

@@ -4,8 +4,6 @@ machine:
   environment:
     NPM_CONFIG_PROGRESS: false
     NPM_CONFIG_SPIN: false
-    SAUCE_ACCESS_KEY: 80dcffc2-c9a2-45e2-9fb7-87b29a6cd986
-    SAUCE_USERNAME: segment-oss
     TEST_REPORTS_DIR: $CIRCLE_TEST_REPORTS
 
 dependencies:

lib/index.js

@@ -37,6 +37,7 @@ var Mixpanel = module.exports = integration('Mixpanel')
   .option('trackAllPages', false)
   .option('trackNamedPages', false)
   .option('trackCategorizedPages', false)
+  .option('sourceName', '')
   .tag('<script src="//cdn.mxpnl.com/libs/mixpanel-2-latest.min.js">');
 
 /**
@@ -184,7 +185,7 @@ Mixpanel.prototype.identify = function(identify) {
     var trait = traits[key];
     if (Array.isArray(trait) && trait.length > 0) {
       traitsToUnion[key] = trait;
-      // since mixpanel doesn't offer a union method for super properties we have to do it manually by retrieving the existing list super property 
+      // since mixpanel doesn't offer a union method for super properties we have to do it manually by retrieving the existing list super property
       // from mixpanel and manually unioning to it ourselves
       var existingTrait = window.mixpanel.get_property(key);
       if (existingTrait && Array.isArray(existingTrait)) {
@@ -235,6 +236,9 @@ Mixpanel.prototype.track = function(track) {
   var revenue = track.revenue();
   // Don't map traits, clients should use identify instead.
   var superProps = pick(this.options.superProperties, props);
+  var sourceName = this.options.sourceName;
+
+  if (sourceName) props.segment_source_name = sourceName;
 
   // delete mixpanel's reserved properties, so they don't conflict
   delete props.distinct_id;
@@ -382,7 +386,7 @@ function extendTraits(arr) {
  *
  * @api private
  * @param {Object} props
- * @example 
+ * @example
  * input: {products: [{sku: 32, revenue: 99}, {sku:2, revenue: 103}]}
  * output: {products_skus: [32, 2], products_revenues: [99, 103]}
  */
@@ -421,7 +425,7 @@ function invertObjectArray(propName, arr) {
         continue;
       }
       var attrKey = propName+'_'+key+'s';  // e.g. products_skus
-      
+
       // append to list if it exists or create new one if not
       if (attrKey in invertedArrays) {
         invertedArrays[attrKey].push(elem[key]);
@@ -464,4 +468,4 @@ function unionArrays(x, y) {
     obj[y[i]] = y[i];
   }
   return Object.keys(obj);
-}
+}

test/index.test.js

@@ -50,7 +50,8 @@ describe('Mixpanel', function() {
       .option('trackNamedPages', false)
       .option('consolidatedPageCalls', true)
       .option('setAllTraitsByDefault', true)
-      .option('trackCategorizedPages', false));
+      .option('trackCategorizedPages', false)
+      .option('sourceName', ''));
   });
 
   describe('before loading', function() {
@@ -492,6 +493,16 @@ describe('Mixpanel', function() {
         });
         analytics.called(window.mixpanel.people.increment, 'videos_watched', 3);
       });
+
+      it('should send sourceName if specified', function() {
+        mixpanel.options.sourceName = 'ios_prod';
+        analytics.track('event', {
+          segment_source_name: 'ios_prod'
+        });
+        analytics.called(window.mixpanel.track, 'event', {
+          segment_source_name: 'ios_prod'
+        });
+      });
     });
 
     describe('#alias', function() {