From e75ed38ac2edd80e34949382e03e657cc0da8e6d Mon Sep 17 00:00:00 2001
From: Matt Walsh <mwalsh@dramafever.com>
Date: Tue, 11 Jul 2017 21:47:44 -0400
Subject: [PATCH] Add setUsername method and change setUserProperties signature

---
 lib/angulartics-segment.js | 15 +++++++++++++--
 1 file changed, 13 insertions(+), 2 deletions(-)

diff --git a/lib/angulartics-segment.js b/lib/angulartics-segment.js
index ce9598a..647b227 100644
--- a/lib/angulartics-segment.js
+++ b/lib/angulartics-segment.js
@@ -62,12 +62,23 @@ angular.module('angulartics.segment', ['angulartics'])
         }
       }
     });
+    // https://segment.com/docs/libraries/analytics.js/#identify
+    // analytics.identify([userId], [traits], [options], [callback]);
+    $analyticsProvider.registerSetUsername(function (userId) {
+      try {
+        analytics.identify(userId);
+      } catch (e) {
+        if (!(e instanceof ReferenceError)) {
+          throw e;
+        }
+      }
+    });
 
     // https://segment.com/docs/libraries/analytics.js/#identify
     // analytics.identify([userId], [traits], [options], [callback]);
-    $analyticsProvider.registerSetUserProperties(function (userId, traits, options, callback) {
+    $analyticsProvider.registerSetUserProperties(function (traits, options, callback) {
       try {
-        analytics.identify(userId, traits, options, callback);
+        analytics.identify(traits, options, callback);
       } catch (e) {
         if (!(e instanceof ReferenceError)) {
           throw e;