Add basic CLI

Author: rctcwyvrn

Sources/RegexBenchmark/Benchmark.swift

@@ -54,11 +54,18 @@ public struct BenchmarkRunner {
   // Register instances of Benchmark and run them
   let suiteName: String
   var suite: [any RegexBenchmark]
-  let samples: Int = 20
+  let samples: Int
 
-  public init(suiteName: String) {
+  public init(_ suiteName: String) {
     self.suiteName = suiteName
     self.suite = []
+    self.samples = 20
+  }
+  
+  public init(_ suiteName: String, _ n: Int) {
+    self.suiteName = suiteName
+    self.suite = []
+    self.samples = n
   }
 
   public mutating func register(_ new: some RegexBenchmark) {
@@ -101,6 +108,15 @@ public struct BenchmarkRunner {
       print("- \(b.name) \(measure(benchmark: b))")
     }
   }
+  
+  public func profile() {
+    print("Starting")
+    for b in suite {
+      print("- \(b.name)")
+      b.run()
+      print("- done")
+    }
+  }
 }
 
 // nom nom nom, consume the argument

Sources/RegexBenchmark/CLI.swift

@@ -0,0 +1,33 @@
+import ArgumentParser
+
+@main
+struct Runner: ParsableCommand {
+  @Argument(help: "Names of benchmarks to run")
+  var specificBenchmarks: [String] = []
+  
+  @Option(help: "Run only once for profiling purposes")
+  var profile = false
+  
+  @Option(help: "How many samples to collect for each benchmark")
+  var samples = 20
+    
+  func makeRunner() -> BenchmarkRunner {
+    var benchmark = BenchmarkRunner("RegexBench", samples)
+    benchmark.addReluctantQuant()
+    benchmark.addBacktracking()
+    benchmark.addCSS()
+    benchmark.addFirstMatch()
+    return benchmark
+  }
+  mutating func run() throws {
+    var runner = makeRunner()
+    if !self.specificBenchmarks.isEmpty {
+      runner.suite = runner.suite.filter { b in specificBenchmarks.contains(b.name) }
+    }
+    if profile {
+      runner.profile()
+    } else {
+      runner.run()
+    }
+  }
+}