Added support for staging.

Author: bsneed

Sources/segmentcli/Commands/Profile.swift

@@ -77,7 +77,6 @@ class ProfileSetCommand: Command {
 }
 
 // MARK: - Global option for profile
-
 let specifiedProfileKey = Key<String>("-p", "--profile", description: "Specify a profile name to use for this operation")
 extension Command {
     var specifiedProfile: String? {

Sources/segmentcli/PAPI/PAPI.swift

@@ -6,8 +6,15 @@
 //
 
 import Foundation
+import SwiftCLI
 
-let PAPIEndpoint: String = "https://api.segmentapis.com/"
+var PAPIEndpoint: String {
+    if useStagingKey.value {
+        return "https://api.segmentapis.build/"
+    } else {
+        return "https://api.segmentapis.com/"
+    }
+}
 
 protocol PAPISection {
     static var pathEntry: String { get }
@@ -53,3 +60,11 @@ class PAPI {
     }
 
 }
+
+// MARK: - Global option to support staging
+let useStagingKey = Flag("--staging", description: "Use Segment staging for operations")
+extension Command {
+    var isStaging: Bool {
+        return useStagingKey.value
+    }
+}

Sources/segmentcli/main.swift

@@ -18,6 +18,7 @@ func main() {
                         SourcesGroup()
                       ])
 
+    segment.globalOptions.append(useStagingKey)
     segment.globalOptions.append(specifiedProfileKey)
 
     segment.go()