Update context object with referrer information (#575)

segmentio · Jul 10, 2016 · 5a62886 · 5a62886
1 parent 5b34049
commit 5a62886
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/Analytics/Classes/Internal/SEGSegmentIntegration.m b/Analytics/Classes/Internal/SEGSegmentIntegration.m
@@ -72,6 +72,7 @@ @interface SEGSegmentIntegration ()
 @property (nonatomic, strong) NSMutableDictionary *traits;
 @property (nonatomic, assign) SEGAnalytics *analytics;
 @property (nonatomic, assign) SEGAnalyticsConfiguration *configuration;
+@property (nonatomic, assign) NSDictionary *referrer;
 
 @end
 
@@ -258,6 +259,10 @@ - (NSDictionary *)liveContext
         traits;
     });
 
+    if (self.referrer) {
+        context[@"referrer"] = [self.referrer copy];
+    }
+
     return [context copy];
 }
 
@@ -403,6 +408,22 @@ - (void)registeredForRemoteNotificationsWithDeviceToken:(NSData *)deviceToken
     [self.context[@"device"] setObject:[token copy] forKey:@"token"];
 }
 
+- (void)continueUserActivity:(NSUserActivity *)activity
+{
+    if ([activity.activityType isEqualToString:NSUserActivityTypeBrowsingWeb]) {
+        self.referrer = @{
+            @"url" : activity.webpageURL,
+        };
+    }
+}
+
+- (void)openURL:(NSURL *)url options:(NSDictionary *)options
+{
+    self.referrer = @{
+        @"url" : url.absoluteString,
+    };
+}
+
 #pragma mark - Queueing
 
 - (NSDictionary *)integrationsDictionary:(NSDictionary *)integrations