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
|
# File 'lib/ak47/cli.rb', line 3
def self.run(*argv)
argv_opts, commands = if argv == ['--help'] or argv == ['-help']
[argv, []]
elsif divider_index = argv.index('--')
[argv[0...divider_index], argv[divider_index.succ...argv.size]]
else
[[], argv]
end
interval, maximum, error_time, interrupt = nil, nil, 5, true
optparse = OptionParser.new do |opts|
opts.banner = "Usage: ak47 [cmd] / ak47 [options] -- [cmd]"
opts.on( '-i', '--interval [FLOAT]', 'Interval before restarting' ) do |i|
interval = Float(i) rescue raise("Interval must be a valid floating point number (e.g. -i0.5)")
raise("Interval must be a positive number") unless interval >= 0
end
opts.on( '-m', '--maximum [FLOAT]', 'Maximum time to wait before restarting' ) do |m|
maximum = Float(m) rescue raise("Maximum must be a valid floating point number (e.g. -m60)")
raise("Maximum must be a positive number") unless maximum >= 0
end
opts.on( '-e', '--error-time [FLOAT]', 'Maximum time to wait before restarting if there was an abnormal status code' ) do |e|
error_time = Float(e) rescue raise("Error time must be a valid floating point number (e.g. -e10)")
raise("Maximum must be a positive number") unless error_time >= 0
end
opts.on( '-d', '--dont-interrupt', 'Don\'t interrupt a running process, wait for it to finish before reloading' ) do
interrupt = false
end
opts.on( '-h', '--help', 'Display this screen' ) do
puts opts
exit
end
end
optparse.parse!(argv_opts)
watch_dirs = argv_opts
watch_dirs << Dir.pwd if watch_dirs.empty?
watch_dirs.map! { |wd| File.expand_path(wd, Dir.pwd) }
command = ShellTools.escape(commands).strip
if command.empty?
puts optparse
puts
raise "No command supplied"
end
Runner.new(:watch_dirs => watch_dirs, :maximum => maximum, :interval => interval, :error_time => error_time, :command => command, :interrupt => interrupt) {
exec(command)
}.start
rescue
puts $!.backtrace.join("\n ") if ENV['DEBUG']
puts $!.message.red
exit 1
end
|