Convert double-dash arguments into filters.

bin/bjob

@@ -2,6 +2,18 @@
 
 require_relative "../vendor/clap"
 
+class Args < Hash
+  def [](item)
+    return super(item) if include?(item)
+
+    if item.start_with?("--")
+      key = item[2..-1]
+
+      -> value { fetch(:*).call(key, value) }
+    end
+  end
+end
+
 if File.exist?(".bjobrc")
   ARGV.concat(File.read(".bjobrc").split("\n"))
 end
@@ -10,11 +22,15 @@ exit(1) if ARGV.empty?
 
 ENV["BATCH_INTERACTIVE"] = ENV["PS1"] ? "1" : "0"
 
-args = Clap.run ARGV,
+filter = {}
+
+args = Clap.run ARGV, Args[
   "-r" => -> file { require file },
   "-i" => -> { ENV["BATCH_INTERACTIVE"] = "1" },
-  "-s" => -> { ENV["BATCH_INTERACTIVE"] = "0" }
+  "-s" => -> { ENV["BATCH_INTERACTIVE"] = "0" },
+  :*   => -> key, value { filter[key.to_sym] = value }
+]
 
 constant = Module.const_get(args.first)
 
-constant.run
+constant.run((filter if filter.size > 0))

test/bin_test.rb

@@ -83,4 +83,10 @@ def assert_empty(str)
 
     assert_empty err
   end
+
+  test "parses command line arguments as a filter" do |bjob, _, capture, root|
+    out, err, status = bjob.(%w[-r ./test/jobs/ten Ten --count 5])
+
+    assert_equal 5, Dir["test/tmp/ten/*.txt"].size
+  end
 end

test/jobs/ten.rb

@@ -17,6 +17,10 @@ def process(item)
   end
 
   def items(filter = nil)
-    (0..9).to_a
+    if filter && filter.include?(:count)
+      filter.fetch(:count).to_i.times.to_a
+    else
+      (0..9).to_a
+    end
   end
 end