Parsing command line arguments via GetoptLong.
This commit is contained in:
parent
8900173016
commit
1b770d45d8
140
dup.rb
140
dup.rb
@ -2,6 +2,8 @@
|
||||
require 'yaml'
|
||||
require 'pp'
|
||||
require 'shellwords'
|
||||
require 'getoptlong'
|
||||
|
||||
|
||||
def get_sample(name="container")
|
||||
return <<~HEREDOC
|
||||
@ -49,75 +51,113 @@ def get_sample(name="container")
|
||||
HEREDOC
|
||||
end
|
||||
|
||||
if ARGV[0]=="--sample"
|
||||
def action_sample
|
||||
puts get_sample
|
||||
exit 1
|
||||
end
|
||||
|
||||
def action_help
|
||||
puts "-h, --help Show this help."
|
||||
puts "-n, --dry-run Don't execute any commands."
|
||||
puts "-s, --sample Outputs a sample yml file."
|
||||
puts "-c, --create Create a new container yml file."
|
||||
exit 1
|
||||
end
|
||||
|
||||
def run(cmd, ignore_returnvalue=false)
|
||||
puts "+ #{cmd}"
|
||||
returnvalue = true
|
||||
returnvalue = system(cmd)
|
||||
returnvalue = $dry_run ? true : system(cmd)
|
||||
raise "Command returned a non-zero exit value." if returnvalue!=true && !ignore_returnvalue
|
||||
end
|
||||
|
||||
if ARGV[0]=="--create"
|
||||
filename = ARGV[1]
|
||||
else
|
||||
filename = ARGV[0]
|
||||
end
|
||||
|
||||
if filename==nil || filename==""
|
||||
raise "No name given."
|
||||
end
|
||||
|
||||
file = "%s/%s.yml" % [ "/data/fabian/projects/dup", filename ]
|
||||
|
||||
if ARGV[0]=="--create"
|
||||
def action_create(container, file)
|
||||
raise "File #{file} already exists" if File.exists?(file)
|
||||
|
||||
File.open(file, "w") {|f| f.write(get_sample(filename))}
|
||||
File.open(file, "w") {|f| f.write(get_sample(container))}
|
||||
|
||||
exec("joe #{file.shellescape}")
|
||||
exit 1 # will never be reached because exec replaces this process. But just in case... ;-)
|
||||
end
|
||||
|
||||
raise "File #{file} not found." unless File.exists?(file)
|
||||
def action_run(container, file)
|
||||
raise "File #{file} not found." unless File.exists?(file)
|
||||
|
||||
data = File.open(file, "r") {|f| YAML.load(f.read)}
|
||||
data = File.open(file, "r") {|f| YAML.load(f.read)}
|
||||
|
||||
raise "Expected #{file} to define (at least) a container named #{filename}." unless data.has_key?(filename)
|
||||
raise "Expected #{file} to define (at least) a container named #{container}." unless data.has_key?(container)
|
||||
|
||||
data.each do |key, data|
|
||||
if data["build"]
|
||||
(data["before_build"] || []).each{|cmd| run(cmd)}
|
||||
cmd = ["docker", "build", "-t", data["image"].shellescape, data["build"].shellescape]
|
||||
data.each do |key, data|
|
||||
if data["build"]
|
||||
(data["before_build"] || []).each{|cmd| run(cmd)}
|
||||
cmd = ["docker", "build", "-t", data["image"].shellescape, data["build"].shellescape]
|
||||
run(cmd.join(" "))
|
||||
(data["after_build"] || []).each{|cmd| run(cmd)}
|
||||
end
|
||||
|
||||
(data["before_run"] || []).each{|cmd| run(cmd)}
|
||||
cmd = ["docker", "run"]
|
||||
data["interactive"] && cmd << "--tty" << "--interactive"
|
||||
data["detach"]===false || cmd << "--detach"
|
||||
data["remove"] && cmd << "--rm"
|
||||
cmd << "--name=#{(data["container_name"] || key).shellescape}"
|
||||
(data["ports"] || []).each {|p| cmd << "--publish" << p.to_s.shellescape}
|
||||
data["restart"] && cmd << "--restart=#{data["restart"].shellescape}"
|
||||
(data["environment"] || []).each {|key, val| cmd << "--env" << (val ? "#{key}=#{val.to_s.shellescape}" : key)}
|
||||
(data["volumes"] || []).each {|vol| cmd << "--volume" << vol.shellescape}
|
||||
data["mem_limit"] && cmd << "--memory=#{data["mem_limit"].shellescape}"
|
||||
(data["links"] || []).each {|link| cmd << "--link" << link.shellescape}
|
||||
data["stop_signal"] && cmd << "--stop-signal=#{data["stop_signal"].shellescape}"
|
||||
data["stop_grace_period"] && cmd << "--stop-timeout=#{data["stop_grace_period"].shellescape}"
|
||||
(data["devices"] || []).each {|dev| cmd << "--device" << dev.shellescape}
|
||||
|
||||
cmd << data["image"].shellescape
|
||||
|
||||
data["command"] && cmd << data["command"]
|
||||
|
||||
run("docker rm -f #{(data["container_name"] || key).shellescape}", true)
|
||||
run(cmd.join(" "))
|
||||
(data["after_build"] || []).each{|cmd| run(cmd)}
|
||||
|
||||
(data["after_run"] || []).each{|cmd| run(cmd)}
|
||||
end
|
||||
|
||||
(data["before_run"] || []).each{|cmd| run(cmd)}
|
||||
cmd = ["docker", "run"]
|
||||
data["interactive"] && cmd << "--tty" << "--interactive"
|
||||
data["detach"]===false || cmd << "--detach"
|
||||
data["remove"] && cmd << "--rm"
|
||||
cmd << "--name=#{(data["container_name"] || key).shellescape}"
|
||||
(data["ports"] || []).each {|p| cmd << "--publish" << p.to_s.shellescape}
|
||||
data["restart"] && cmd << "--restart=#{data["restart"].shellescape}"
|
||||
(data["environment"] || []).each {|key, val| cmd << "--env" << (val ? "#{key}=#{val.to_s.shellescape}" : key)}
|
||||
(data["volumes"] || []).each {|vol| cmd << "--volume" << vol.shellescape}
|
||||
data["mem_limit"] && cmd << "--memory=#{data["mem_limit"].shellescape}"
|
||||
(data["links"] || []).each {|link| cmd << "--link" << link.shellescape}
|
||||
data["stop_signal"] && cmd << "--stop-signal=#{data["stop_signal"].shellescape}"
|
||||
data["stop_grace_period"] && cmd << "--stop-timeout=#{data["stop_grace_period"].shellescape}"
|
||||
(data["devices"] || []).each {|dev| cmd << "--device" << dev.shellescape}
|
||||
|
||||
cmd << data["image"].shellescape
|
||||
|
||||
data["command"] && cmd << data["command"]
|
||||
|
||||
run("docker rm -f #{(data["container_name"] || key).shellescape}", true)
|
||||
run(cmd.join(" "))
|
||||
|
||||
(data["after_run"] || []).each{|cmd| run(cmd)}
|
||||
end
|
||||
|
||||
action = :run
|
||||
container = nil
|
||||
$dry_run = false
|
||||
|
||||
opts = GetoptLong.new(
|
||||
[ '--sample', '-s', GetoptLong::NO_ARGUMENT ],
|
||||
[ '--create', '-c', GetoptLong::NO_ARGUMENT ],
|
||||
[ '--help', '-h', GetoptLong::NO_ARGUMENT ],
|
||||
[ '--dry-run', '-n', GetoptLong::NO_ARGUMENT ]
|
||||
)
|
||||
|
||||
opts.each do |opt, arg|
|
||||
case opt
|
||||
when '--sample'
|
||||
action_sample
|
||||
when '--create'
|
||||
action = :create
|
||||
when '--help'
|
||||
action_help
|
||||
when '--dry-run'
|
||||
puts "Dry-run. Not going to execute any command."
|
||||
$dry_run = true
|
||||
end
|
||||
end
|
||||
|
||||
container = ARGV.shift
|
||||
|
||||
if container==nil || container==""
|
||||
raise "No container given."
|
||||
end
|
||||
|
||||
file = "%s/%s.yml" % [ "/data/fabian/projects/dup", container ]
|
||||
|
||||
if action == :create
|
||||
action_create(container, file)
|
||||
elsif action == :run
|
||||
action_run(container, file)
|
||||
end
|
||||
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user