Add method to put device token

Analytics/Classes/Integrations/SEGIntegration.h

@@ -59,6 +59,7 @@ NS_ASSUME_NONNULL_BEGIN
 - (void)receivedRemoteNotification:(NSDictionary *)userInfo;
 - (void)failedToRegisterForRemoteNotificationsWithError:(NSError *)error;
 - (void)registeredForRemoteNotificationsWithDeviceToken:(NSData *)deviceToken;
+- (void)putDeviceToken:(NSString *)deviceToken;
 - (void)handleActionWithIdentifier:(NSString *)identifier forRemoteNotification:(NSDictionary *)userInfo;
 
 // Callbacks for app state changes

Analytics/Classes/Integrations/SEGIntegrationsManager.m

@@ -253,6 +253,11 @@ - (void)registeredForRemoteNotificationsWithDeviceToken:(NSData *)deviceToken
     [self callIntegrationsWithSelector:_cmd arguments:@[ deviceToken ] options:nil sync:true];
 }
 
+- (void)putDeviceToken:(NSString *)deviceToken
+{
+    [self callIntegrationsWithSelector:_cmd arguments:@[ deviceToken ] options:nil sync:true];
+}
+
 - (void)handleActionWithIdentifier:(NSString *)identifier forRemoteNotification:(NSDictionary *)userInfo
 {
     [self callIntegrationsWithSelector:_cmd arguments:@[ identifier, userInfo ] options:nil sync:true];
@@ -518,6 +523,8 @@ - (SEGEventType)eventTypeFromSelector:(SEL)selector
         result = SEGEventTypeFailedToRegisterForRemoteNotifications;
     } else if ([selectorString hasPrefix:@"registeredForRemoteNotificationsWithDeviceToken"]) {
         result = SEGEventTypeRegisteredForRemoteNotifications;
+    } else if ([selectorString hasPrefix:@"putDeviceToken"]) {
+        result = SEGEventTypePuttingDeviceToken;
     } else if ([selectorString hasPrefix:@"handleActionWithIdentifier"]) {
         result = SEGEventTypeHandleActionWithForRemoteNotification;
     } else if ([selectorString hasPrefix:@"continueUserActivity"]) {
@@ -679,6 +686,9 @@ - (void)context:(SEGContext *)context next:(void (^_Nonnull)(SEGContext *_Nullab
             [self registeredForRemoteNotificationsWithDeviceToken:
                       [(SEGRemoteNotificationPayload *)context.payload deviceToken]];
             break;
+        case SEGEventTypePuttingDeviceToken:
+            [self putDeviceToken:[(SEGPuttingDeviceTokenPayload *)context.payload deviceToken]];
+            break;
         case SEGEventTypeHandleActionWithForRemoteNotification: {
             SEGRemoteNotificationPayload *payload = (SEGRemoteNotificationPayload *)context.payload;
             [self handleActionWithIdentifier:payload.actionIdentifier

Analytics/Classes/Integrations/SEGPayload.h

@@ -58,3 +58,9 @@ NS_ASSUME_NONNULL_END
 @property (nonatomic, strong, nullable) NSData *deviceToken;
 
 @end
+
+@interface SEGPuttingDeviceTokenPayload : SEGPayload
+
+@property (nonatomic, strong, nonnull) NSString *deviceToken;
+
+@end

Analytics/Classes/Integrations/SEGPayload.m

@@ -24,6 +24,10 @@ @implementation SEGRemoteNotificationPayload
 
 @end
 
+@implementation SEGPuttingDeviceTokenPayload
+
+@end
+
 
 @implementation SEGContinueUserActivityPayload
 

Analytics/Classes/Internal/SEGSegmentIntegration.m

@@ -426,6 +426,11 @@ - (void)registeredForRemoteNotificationsWithDeviceToken:(NSData *)deviceToken
     [self.cachedStaticContext[@"device"] setObject:[token copy] forKey:@"token"];
 }
 
+- (void)putDeviceToken:(NSString *)deviceToken
+{
+    [self.cachedStaticContext[@"device"] setObject:deviceToken forKey:@"token"];
+}
+
 - (void)continueUserActivity:(NSUserActivity *)activity
 {
     if ([activity.activityType isEqualToString:NSUserActivityTypeBrowsingWeb]) {

Analytics/Classes/Middlewares/SEGContext.h

@@ -27,6 +27,7 @@ typedef NS_ENUM(NSInteger, SEGEventType) {
     SEGEventTypeReceivedRemoteNotification,
     SEGEventTypeFailedToRegisterForRemoteNotifications,
     SEGEventTypeRegisteredForRemoteNotifications,
+    SEGEventTypePuttingDeviceToken,
     SEGEventTypeHandleActionWithForRemoteNotification,
 
     // Application Lifecycle

Analytics/Classes/SEGAnalytics.h

@@ -148,6 +148,16 @@ NS_ASSUME_NONNULL_BEGIN
 - (void)receivedRemoteNotification:(NSDictionary *)userInfo;
 - (void)failedToRegisterForRemoteNotificationsWithError:(NSError *)error;
 - (void)registeredForRemoteNotificationsWithDeviceToken:(NSData *)deviceToken;
+
+/*!
+@method
+
+@abstract
+Manually set a device token (for example when using Firebase Cloud Messaging tokens).
+*/
+- (void)putDeviceToken:(NSString *)deviceToken;
+
+// todo: docs
 - (void)handleActionWithIdentifier:(NSString *)identifier forRemoteNotification:(NSDictionary *)userInfo;
 - (void)continueUserActivity:(NSUserActivity *)activity;
 - (void)openURL:(NSURL *)url options:(NSDictionary *)options;

Analytics/Classes/SEGAnalytics.m

@@ -342,6 +342,13 @@ - (void)registeredForRemoteNotificationsWithDeviceToken:(NSData *)deviceToken
     [self run:SEGEventTypeRegisteredForRemoteNotifications payload:payload];
 }
 
+- (void)putDeviceToken:(NSString *)deviceToken
+{
+    SEGPuttingDeviceTokenPayload *payload = [[SEGPuttingDeviceTokenPayload alloc] init];
+    payload.deviceToken = deviceToken;
+    [self run:SEGEventTypePuttingDeviceToken payload:payload];
+}
+
 - (void)handleActionWithIdentifier:(NSString *)identifier forRemoteNotification:(NSDictionary *)userInfo
 {
     SEGRemoteNotificationPayload *payload = [[SEGRemoteNotificationPayload alloc] init];