Implemented dynamic mapping of yaml values to command line arguments.

This commit is contained in:
Fabian Schlenz 2017-02-13 13:07:32 +01:00
parent 1b770d45d8
commit 7146cdb027

68
dup.rb
View File

@ -4,6 +4,25 @@ require 'pp'
require 'shellwords'
require 'getoptlong'
MAPPINGS = [
['stdin_open', '--interactive', {:type=>:switch}],
['tty', '--tty', {:type=>:switch}],
['detach', '--detach', {:type=>:switch}],
['remove', '--rm', {:type=>:switch}],
['container_name', '--name', {:type=>:name}],
['ports', '--publish'],
['restart', '--restart'],
['environment', '--env'],
['volumes', '--volume'],
['mem_limit', '--memory'],
['links', '--link'],
['stop_signal', '--stop-signal'],
['stop_grace_period', '--stop-timeout'],
['devices', '--device'],
['image', nil],
['command', nil, {:escape=>false}]
]
def get_sample(name="container")
return <<~HEREDOC
@ -70,6 +89,11 @@ def run(cmd, ignore_returnvalue=false)
raise "Command returned a non-zero exit value." if returnvalue!=true && !ignore_returnvalue
end
def esc(obj, escape=true)
return obj unless escape
return obj.to_s.shellescape
end
def action_create(container, file)
raise "File #{file} already exists" if File.exists?(file)
@ -95,27 +119,35 @@ def action_run(container, file)
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"]
MAPPINGS.each do |mapping|
yml_name, cmd_name, inspect_name, options = *mapping
options ||= {}
if !data[yml_name]
if options[:type]==:name
cmd << cmd_name << esc(key)
end
next
end
if options[:type]==:switch
cmd << cmd_name
next
end
if data[yml_name].is_a?(Array)
data[yml_name].each {|val| cmd << cmd_name << esc(val, options[:escape])}
elsif data[yml_name].is_a?(Hash)
data[yml_name].each {|key, val| cmd << cmd_name << "#{esc(key, options[:escape])}=#{esc(val, options[:escape])}"}
else
cmd << cmd_name << esc(data[yml_name], options[:escape])
end
end
run("docker rm -f #{(data["container_name"] || key).shellescape}", true)
run(cmd.join(" "))
run(cmd.compact.join(" "))
(data["after_run"] || []).each{|cmd| run(cmd)}
end