Added ability to adjust CDN url. (#3)

Author: migs647

SegmentExample/AppDelegate.swift

@@ -23,14 +23,14 @@ class AppDelegate: UIResponder, UIApplicationDelegate {
         let analytics = Analytics(configuration: config)
         self.analytics = analytics
 
-        analytics.add(plugin: AfterPlugin(name: "hello", analytics: analytics))
+        analytics.add(plugin: AfterPlugin(name: "AfterPlugin_EndOfTimeline", analytics: analytics))
 
         DispatchQueue.main.asyncAfter(deadline: DispatchTime.now() + 5) {
-            analytics.identify(userId: "Live Demo -- never breaks")
-            analytics.track(name: "I once tracked a cougar", properties: MyTraits(email: "Brandon@stinks.com"))
-            analytics.screen(screenTitle: "Screened the AppDelegate")
-            analytics.group(groupId: "Grouped By ID")
-            //analytics.alias(newId: "3333")
+            analytics.identify(userId: "Segment Spec: Identify")
+            analytics.track(name: "Segment Spec: Track", properties: MyTraits(email: "info@segment.com"))
+            analytics.screen(screenTitle: "Segment Spec: Screen")
+            analytics.group(groupId: "Segment Spec: Group")
+            //analytics.alias(newId: "Segment Spec: Alias")
         }
 
         return true

Sources/Segment/Configuration.swift

@@ -30,6 +30,7 @@ public class Configuration {
         var defaultSettings: Settings? = nil
         var autoAddSegmentDestination: Bool = true
         var apiHost: String = HTTPClient.getDefaultAPIHost()
+        var cdnHost: String = HTTPClient.getDefaultCDNHost()
     }
     internal var values: Values
 
@@ -83,5 +84,11 @@ public extension Configuration {
         values.apiHost = value
         return self
     }
+    
+    @discardableResult
+    func cdnHost(_ value: String) -> Configuration {
+        values.cdnHost = value
+        return self
+    }
 }
 

Sources/Segment/Plugins/SegmentDestination.swift

@@ -41,7 +41,7 @@ public class SegmentDestination: DestinationPlugin {
     }
 
     public func update(settings: Settings) {
-        let segmentInfo = settings.integrationSettings(for: "Segment.io")
+        let segmentInfo = settings.integrationSettings(for: Self.Constants.integrationName.rawValue)
         let apiKey = segmentInfo?[Self.Constants.apiKey.rawValue] as? String
         let apiHost = segmentInfo?[Self.Constants.apiHost.rawValue] as? String
         if (apiHost != nil && apiKey != nil) {

Sources/Segment/Settings.swift

@@ -14,9 +14,9 @@ public struct Settings: Codable {
 
     public init(writeKey: String, apiHost: String) {
         integrations = try! JSON([
-            "Segment.io": [
-                "apiKey": writeKey,
-                "apiHost": apiHost
+            SegmentDestination.Constants.integrationName.rawValue: [
+                SegmentDestination.Constants.apiKey.rawValue: writeKey,
+                SegmentDestination.Constants.apiHost.rawValue: apiHost
             ]
         ])
     }
@@ -50,7 +50,7 @@ public struct Settings: Codable {
 extension Analytics {
     func checkSettings() {
         let writeKey = self.configuration.values.writeKey
-        let httpClient = HTTPClient(analytics: self)
+        let httpClient = HTTPClient(analytics: self, cdnHost: configuration.values.cdnHost)
         httpClient.settingsFor(writeKey: writeKey) { (success, settings) in
             if success {
                 if let s = settings {

Sources/Segment/Utilities/HTTPClient.swift

@@ -20,9 +20,10 @@ public class HTTPClient {
     private var writeKeySessions = [String: URLSession]()
     private var apiHost: String
     private var apiKey: String
+    private var cdnHost: String
     private let analytics: Analytics
 
-    init(analytics: Analytics, apiKey: String? = nil, apiHost: String? = nil) {
+    init(analytics: Analytics, apiKey: String? = nil, apiHost: String? = nil, cdnHost: String? = nil) {
         self.analytics = analytics
 
         if let apiKey = apiKey {
@@ -36,6 +37,12 @@ public class HTTPClient {
         } else {
             self.apiHost = Self.defaultAPIHost
         }
+        
+        if let cdnHost = cdnHost {
+            self.cdnHost = cdnHost
+        } else {
+            self.cdnHost = Self.defaultCDNHost
+        }
     }
 
     func segmentURL(for host: String, path: String) -> URL? {
@@ -103,7 +110,7 @@ public class HTTPClient {
         // from write key sessions for uploading.
         let settingsKey = "\(writeKey)_settings"
 
-        guard let settingsURL = segmentURL(for: Self.defaultCDNHost, path: "/projects/\(writeKey)/settings") else {
+        guard let settingsURL = segmentURL(for: cdnHost, path: "/projects/\(writeKey)/settings") else {
             completion(false, nil)
             return
         }
@@ -165,6 +172,10 @@ extension HTTPClient {
     internal static func getDefaultAPIHost() -> String {
         return Self.defaultAPIHost
     }
+    
+    internal static func getDefaultCDNHost() -> String {
+        return Self.defaultCDNHost
+    }
 }