Add arguments to bin/chef-zero

1 files changed, 33 insertions, 1 deletions

diff --git a/bin/chef-zero b/bin/chef-zero
index 590d670..5f41dae 100755
--- a/bin/chef-zero
+++ b/bin/chef-zero
@@ -2,7 +2,39 @@
 
 require 'rubygems'
 $:.unshift(File.expand_path(File.join(File.dirname(__FILE__), "..", "lib")))
+
+require 'chef_zero/version'
 require 'chef_zero/server'
+require 'optparse'
+
+options = {
+  :Host => '127.0.0.1',
+  :Port => 8889,
+  :generate_real_keys => false
+}
+OptionParser.new do |opts|
+  opts.banner = "Usage: chef-zero [ARGS]"
+
+  opts.on("-H", "--host HOST", "Host to bind to (default: 127.0.0.1)") do |value|
+    options[:Host] = value
+  end
+  opts.on("-p", "--port PORT", Integer, "Port to listen on") do |value|
+    options[:Port] = value
+  end
+  opts.on("--[no-]generate-keys", "Whether to generate actual keys or fake it (faster).  Default: false.") do |value|
+    options[:generate_real_keys] = value
+  end
+
+  opts.on_tail("-h", "--help", "Show this message") do
+    puts opts
+    exit
+  end
+
+  opts.on_tail("--version", "Show version") do
+    puts ChefZero::VERSION
+    exit
+  end
+end.parse!
 
-server = ChefZero::Server.new(:Port => 8889)
+server = ChefZero::Server.new(options)
 server.start