Updated with putDeviceToken method for deviceToken support

rudderlabs · Apr 3, 2020 · 0693cc4 · 0693cc4
1 parent 5f8d463
commit 0693cc4
Show file tree

Hide file tree

Showing 5 changed files with 13 additions and 2 deletions.
diff --git a/Example/Rudder/_AppDelegate.m b/Example/Rudder/_AppDelegate.m
@@ -9,13 +9,12 @@
 #import "_AppDelegate.h"
 #import <Rudder/Rudder.h>
 
-static NSString *DATA_PLANE_URL = @"https://89aef425.ngrok.io";
+static NSString *DATA_PLANE_URL = @"https://f1bc4b69.ngrok.io";
 static NSString *CONTROL_PLANE_URL = @"https://api.rudderlabs.com";
 static NSString *WRITE_KEY = @"1Xk22tE75wUqDqCSFvFHqeiYCdT";
 
 @implementation _AppDelegate
 
-
 - (BOOL)application:(UIApplication *)application didFinishLaunchingWithOptions:(NSDictionary *)launchOptions
 {
     // Override point for customization after application launch.
@@ -26,6 +25,9 @@ - (BOOL)application:(UIApplication *)application didFinishLaunchingWithOptions:(
     [builder withTrackLifecycleEvens:YES];
     [builder withRecordScreenViews:YES];
     [RudderClient getInstance:WRITE_KEY config:[builder build]];
+
+    [[[RudderClient sharedInstance] getContext] putDeviceToken:@"example_device_token"];
+
     return YES;
 }
 

diff --git a/Rudder/Classes/RudderContext.h b/Rudder/Classes/RudderContext.h
@@ -37,6 +37,7 @@ NS_ASSUME_NONNULL_BEGIN
 - (void) updateTraits: (RudderTraits* _Nullable) traits;
 - (void) persistTraits;
 - (void) updateTraitsDict: (NSMutableDictionary<NSString*, NSObject*>*) traitsDict;
+- (void) putDeviceToken: (NSString*) deviceToken;
 
 @end
 

diff --git a/Rudder/Classes/RudderContext.m b/Rudder/Classes/RudderContext.m
@@ -81,6 +81,10 @@ - (void)updateTraitsDict:(NSMutableDictionary<NSString *, NSObject *> *)traitsDi
     _traits = traitsDict;
 }
 
+- (void)putDeviceToken:(NSString *)deviceToken {
+    _device.token = deviceToken;
+}
+
 - (NSDictionary<NSString *,NSObject *> *)dict {
     NSMutableDictionary *tempDict = [[NSMutableDictionary alloc] init];
     [tempDict setObject:[_app dict] forKey:@"app"];

diff --git a/Rudder/Classes/RudderDeviceInfo.h b/Rudder/Classes/RudderDeviceInfo.h
@@ -19,6 +19,7 @@ NS_ASSUME_NONNULL_BEGIN
 @property (nonatomic, readwrite) NSString* model;
 @property (nonatomic, readwrite) NSString* name;
 @property (nonatomic, readwrite) NSString* type;
+@property (nonatomic, readwrite) NSString* token;
 
 @end
 

diff --git a/Rudder/Classes/RudderDeviceInfo.m b/Rudder/Classes/RudderDeviceInfo.m
@@ -32,6 +32,9 @@ - (instancetype)init
     [tempDict setValue:_model forKey:@"model"];
     [tempDict setValue:_name forKey:@"name"];
     [tempDict setValue:_type forKey:@"type"];
+    if (_token != nil) {
+        [tempDict setValue:_token forKey:@"token"];
+    }
 
     return [tempDict copy];
 }