Added API setSafetrackAPIURL (#1481)

Author: NidhiDixit09

Branch-TestBed/Branch-SDK-Tests/BNCAPIServerTest.m

@@ -409,4 +409,107 @@ - (void)testSetAPIURL_InvalidEmpty {
     XCTAssertEqualObjects(storedUrl, expectedUrl);
 }
 
+
+- (void)testSetSafeTrackServiceURLWithUserTrackingDomain {
+    NSString *url = @"https://links.toTestDomain.com";
+    NSString *safeTrackUrl = @"https://links.toTestDomain-safeTrack.com";
+    
+    [Branch setAPIUrl:url];
+    [Branch setSafetrackAPIURL:safeTrackUrl];
+    
+    BNCServerAPI *serverAPI = [BNCServerAPI sharedInstance];
+    serverAPI.automaticallyEnableTrackingDomain = NO;
+    serverAPI.useTrackingDomain = YES;
+    
+    NSString *storedUrl = [[BNCServerAPI sharedInstance] installServiceURL];
+    NSString *expectedUrl = @"https://links.toTestDomain-safeTrack.com/v1/install";
+    XCTAssertEqualObjects(storedUrl, expectedUrl);
+    
+    storedUrl = [[BNCServerAPI sharedInstance] openServiceURL];
+    expectedUrl = @"https://links.toTestDomain-safeTrack.com/v1/open";
+    XCTAssertEqualObjects(storedUrl, expectedUrl);
+    
+    storedUrl = [[BNCServerAPI sharedInstance] standardEventServiceURL];
+    expectedUrl = @"https://links.toTestDomain-safeTrack.com/v2/event/standard";
+    XCTAssertEqualObjects(storedUrl, expectedUrl);
+    
+    storedUrl = [[BNCServerAPI sharedInstance] customEventServiceURL];
+    expectedUrl = @"https://links.toTestDomain-safeTrack.com/v2/event/custom";
+    XCTAssertEqualObjects(storedUrl, expectedUrl);
+    
+    storedUrl = [[BNCServerAPI sharedInstance] linkServiceURL];
+    expectedUrl = @"https://links.toTestDomain.com/v1/url";
+    XCTAssertEqualObjects(storedUrl, expectedUrl);
+    
+    storedUrl = [[BNCServerAPI sharedInstance] qrcodeServiceURL];
+    expectedUrl = @"https://links.toTestDomain.com/v1/qr-code";
+    XCTAssertEqualObjects(storedUrl, expectedUrl);
+    
+    storedUrl = [[BNCServerAPI sharedInstance] latdServiceURL];
+    expectedUrl = @"https://links.toTestDomain.com/v1/cpid/latd";
+    XCTAssertEqualObjects(storedUrl, expectedUrl);
+    
+    storedUrl = [[BNCServerAPI sharedInstance] validationServiceURL];
+    expectedUrl = @"https://links.toTestDomain.com/v1/app-link-settings";
+    XCTAssertEqualObjects(storedUrl, expectedUrl);
+    
+    [BNCServerAPI sharedInstance].useTrackingDomain = NO;
+    [BNCServerAPI sharedInstance].useEUServers = NO;
+    [BNCServerAPI sharedInstance].automaticallyEnableTrackingDomain = YES;
+    [BNCServerAPI sharedInstance].customAPIURL = nil;
+    [BNCServerAPI sharedInstance].customSafeTrackAPIURL = nil;
+    
+}
+
+- (void)testSetSafeTrackServiceURLWithOutUserTrackingDomain {
+    NSString *url = @"https://links.toTestDomain.com";
+    NSString *safeTrackUrl = @"https://links.toTestDomain-safeTrack.com";
+    
+    [Branch setAPIUrl:url];
+    [Branch setSafetrackAPIURL:safeTrackUrl];
+    
+    BNCServerAPI *serverAPI = [BNCServerAPI sharedInstance];
+    serverAPI.automaticallyEnableTrackingDomain = NO;
+    serverAPI.useTrackingDomain = NO;
+    
+    NSString *storedUrl = [[BNCServerAPI sharedInstance] installServiceURL];
+    NSString *expectedUrl = @"https://links.toTestDomain.com/v1/install";
+    XCTAssertEqualObjects(storedUrl, expectedUrl);
+    
+    storedUrl = [[BNCServerAPI sharedInstance] openServiceURL];
+    expectedUrl = @"https://links.toTestDomain.com/v1/open";
+    XCTAssertEqualObjects(storedUrl, expectedUrl);
+    
+    storedUrl = [[BNCServerAPI sharedInstance] standardEventServiceURL];
+    expectedUrl = @"https://links.toTestDomain.com/v2/event/standard";
+    XCTAssertEqualObjects(storedUrl, expectedUrl);
+    
+    storedUrl = [[BNCServerAPI sharedInstance] customEventServiceURL];
+    expectedUrl = @"https://links.toTestDomain.com/v2/event/custom";
+    XCTAssertEqualObjects(storedUrl, expectedUrl);
+    
+    storedUrl = [[BNCServerAPI sharedInstance] linkServiceURL];
+    expectedUrl = @"https://links.toTestDomain.com/v1/url";
+    XCTAssertEqualObjects(storedUrl, expectedUrl);
+    
+    storedUrl = [[BNCServerAPI sharedInstance] qrcodeServiceURL];
+    expectedUrl = @"https://links.toTestDomain.com/v1/qr-code";
+    XCTAssertEqualObjects(storedUrl, expectedUrl);
+    
+    storedUrl = [[BNCServerAPI sharedInstance] latdServiceURL];
+    expectedUrl = @"https://links.toTestDomain.com/v1/cpid/latd";
+    XCTAssertEqualObjects(storedUrl, expectedUrl);
+    
+    storedUrl = [[BNCServerAPI sharedInstance] validationServiceURL];
+    expectedUrl = @"https://links.toTestDomain.com/v1/app-link-settings";
+    XCTAssertEqualObjects(storedUrl, expectedUrl);
+    
+    [BNCServerAPI sharedInstance].useTrackingDomain = NO;
+    [BNCServerAPI sharedInstance].useEUServers = NO;
+    [BNCServerAPI sharedInstance].automaticallyEnableTrackingDomain = YES;
+    [BNCServerAPI sharedInstance].customAPIURL = nil;
+    [BNCServerAPI sharedInstance].customSafeTrackAPIURL = nil;
+    
+}
+
 @end

Sources/BranchSDK/BNCServerAPI.m

@@ -94,15 +94,19 @@ - (NSString *)getBaseURLForLinkingEndpoints {
 }
 
 - (NSString *)getBaseURL {
-    //Check if user has set a custom API base URL
-    if (self.customAPIURL) {
-        return self.customAPIURL;
-    }
 
     if (self.automaticallyEnableTrackingDomain) {
         self.useTrackingDomain = [self optedIntoIDFA];
     }
 
+    //Check if user has set a custom API base URL / custom API safetrack base url
+    if (self.useTrackingDomain && self.customSafeTrackAPIURL){
+        return self.customSafeTrackAPIURL;
+    }
+    else if (self.customAPIURL) {
+        return self.customAPIURL;
+    }
+    
     NSString * urlString;
 
     if (self.useTrackingDomain && self.useEUServers){

Sources/BranchSDK/Branch.m

@@ -490,6 +490,14 @@ + (void)setAPIUrl:(NSString *)url {
     }
 }
 
++ (void)setSafetrackAPIURL:(NSString *)url {
+    if ([url hasPrefix:@"http://"] || [url hasPrefix:@"https://"] ){
+        [BNCServerAPI sharedInstance].customSafeTrackAPIURL = url;
+    } else {
+        [[BranchLogger shared] logWarning:@"Ignoring invalid custom Safe Track API URL" error:nil];
+    }
+}
+
 - (void)validateSDKIntegration {
     [self validateSDKIntegrationCore];
 }

Sources/BranchSDK/Private/BNCServerAPI.h

@@ -34,6 +34,7 @@ NS_ASSUME_NONNULL_BEGIN
 @property (nonatomic, assign, readwrite) BOOL automaticallyEnableTrackingDomain;
 
 @property (nonatomic, copy, readwrite, nullable) NSString *customAPIURL;
+@property (nonatomic, copy, readwrite, nullable) NSString *customSafeTrackAPIURL;
 
 @end
 

Sources/BranchSDK/Public/Branch.h

@@ -587,6 +587,13 @@ Sets a custom base URL for all calls to the Branch API.
 */
 + (void)setAPIUrl:(NSString *)url;
 
+/**
+Sets a custom base safetrack URL for non-linking calls to the Branch API.
+@param url  Base safetrack URL that the Branch API will use.
+ */
+
++ (void)setSafetrackAPIURL:(NSString *)url ;
+
 /**
   @brief        Use the `validateSDKIntegration` method as a debugging aid to assure that you've
                 integrated the Branch SDK correctly.