summaryrefslogtreecommitdiff
path: root/bin/chef-zero
blob: 33fc0e175077c6ad694338be4fa5dad08a4bb2cb (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
#!/usr/bin/env ruby

# Trap interrupts to quit cleanly.
Signal.trap('INT') { exit 1 }

require 'rubygems'
$:.unshift(File.expand_path(File.join(File.dirname(__FILE__), "..", "lib")))

require 'chef_zero/log'
require 'chef_zero/version'
require 'chef_zero/server'
require 'chef_zero/data_store/raw_file_store'
require 'optparse'

def parse_port(port)
  array = []
  port.split(',').each do |part|
    a,b = part.split('-',2)
    if b
      array = array.concat(a.to_i.upto(b.to_i).to_a)
    else
      array = array.concat([a.to_i])
    end
  end
  array
end

options = {}

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] ||= []
    options[:host] << value
  end

  opts.on("-p", "--port PORT", "Port to listen on (e.g. 8889, or 8500-8600 or 8885,8888)") do |value|
    options[:port] ||= []
    options[:port] += parse_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("-d", "--daemon", "Run as a daemon process") do |value|
    options[:daemon] = value
  end

  opts.on("-l", "--log-level LEVEL", "Set the output log level") do |value|
    options[:log_level] = value
  end

  opts.on("--log-file FILE", "Log to a file") do |value|
    options[:log_file] = value
  end

  opts.on("--enterprise", "Whether to run in enterprise mode") do |value|
    options[:single_org] = nil
    options[:osc_compat] = false
  end

  opts.on("--multi-org", "Whether to run in multi-org mode") do |value|
    options[:single_org] = nil
  end

  opts.on("--file-store PATH", "Persist data to files at the given path") do |value|
    options[:data_store] = ChefZero::DataStore::RawFileStore.new(value)
  end

  opts.on("--[no-]ssl", "Use SSL with self-signed certificate(Auto generate before every run).  Default: false.") do |value|
    options[:ssl] = 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!

if options[:data_store]
  options[:data_store] = ChefZero::DataStore::DefaultFacade.new(options[:data_store], options[:single_org], false)
end

if options[:log_file]
  ChefZero::Log.init(options[:log_file])
end

server = ChefZero::Server.new(options)

if options[:daemon]
  if Process.respond_to?(:daemon)
    Process.daemon(true)
    server.start(true)
  else
    if ENV['OS'] == 'Windows_NT' 
      abort 'Daemonization is not supported on Windows. Running 'start chef-zero' will fork the process.'
    else
      abort 'Process.daemon requires Ruby >= 1.9'
    end
  end
else
  server.start(true)
end