Revision control
Copy as Markdown
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
import Foundation
enum ArgumentProcessor {
static func initializeTooling(nimbus: NimbusInterface, args: CliArgs) {
if args.resetDatabase {
nimbus.resetEnrollmentsDatabase().waitUntilFinished()
}
if let experiments = args.experiments {
nimbus.setExperimentsLocally(experiments)
nimbus.applyPendingExperiments().waitUntilFinished()
// setExperimentsLocally and applyPendingExperiments run on the
// same single threaded dispatch queue, so we can run them in series,
// and wait for the apply.
nimbus.setFetchEnabled(false)
}
if args.logState {
nimbus.dumpStateToLog()
}
// We have isLauncher here doing nothing; this is to match the Android implementation.
// There is nothing to do at this point, because we're unable to affect the flow of the app.
if args.isLauncher {
() // NOOP.
}
}
static func createCommandLineArgs(url: URL) -> CliArgs? {
guard let components = URLComponents(url: url, resolvingAgainstBaseURL: false),
let scheme = components.scheme,
let queryItems = components.queryItems,
!["http", "https"].contains(scheme)
else {
return nil
}
var experiments: String?
var resetDatabase = false
var logState = false
var isLauncher = false
var meantForUs = false
func flag(_ v: String?) -> Bool {
guard let v = v else {
return true
}
return ["1", "true"].contains(v.lowercased())
}
for item in queryItems {
switch item.name {
case "--nimbus-cli":
meantForUs = flag(item.value)
case "--experiments":
experiments = item.value?.removingPercentEncoding
case "--reset-db":
resetDatabase = flag(item.value)
case "--log-state":
logState = flag(item.value)
case "--is-launcher":
isLauncher = flag(item.value)
default:
() // NOOP
}
}
if !meantForUs {
return nil
}
return check(args: CliArgs(
resetDatabase: resetDatabase,
experiments: experiments,
logState: logState,
isLauncher: isLauncher
))
}
static func createCommandLineArgs(args: [String]?) -> CliArgs? {
guard let args = args else {
return nil
}
if !args.contains("--nimbus-cli") {
return nil
}
var argMap = [String: String]()
var key: String?
var resetDatabase = false
var logState = false
for arg in args {
var value: String?
switch arg {
case "--version":
key = "version"
case "--experiments":
key = "experiments"
case "--reset-db":
resetDatabase = true
case "--log-state":
logState = true
default:
value = arg.replacingOccurrences(of: "'", with: "'")
}
if let k = key, let v = value {
argMap[k] = v
key = nil
value = nil
}
}
if argMap["version"] != "1" {
return nil
}
let experiments = argMap["experiments"]
return check(args: CliArgs(
resetDatabase: resetDatabase,
experiments: experiments,
logState: logState,
isLauncher: false
))
}
static func check(args: CliArgs) -> CliArgs? {
if let string = args.experiments {
guard let payload = try? Dictionary.parse(jsonString: string), payload["data"] is [Any]
else {
return nil
}
}
return args
}
}
struct CliArgs: Equatable {
let resetDatabase: Bool
let experiments: String?
let logState: Bool
let isLauncher: Bool
}
public extension NimbusInterface {
func initializeTooling(url: URL?) {
guard let url = url,
let args = ArgumentProcessor.createCommandLineArgs(url: url)
else {
return
}
ArgumentProcessor.initializeTooling(nimbus: self, args: args)
}
}