diff options
Diffstat (limited to 'lib/oxidized/model')
-rw-r--r-- | lib/oxidized/model/model.rb | 31 | ||||
-rw-r--r-- | lib/oxidized/model/outputs.rb | 41 |
2 files changed, 62 insertions, 10 deletions
diff --git a/lib/oxidized/model/model.rb b/lib/oxidized/model/model.rb index c3d8760..d3f54b9 100644 --- a/lib/oxidized/model/model.rb +++ b/lib/oxidized/model/model.rb @@ -1,3 +1,5 @@ +require_relative 'outputs' + module Oxidized class Model include Oxidized::Config::Vars @@ -77,17 +79,16 @@ module Oxidized def cmd string, &block out = @input.cmd string return false unless out - out = Oxidized::String.new out self.class.cmds[:all].each do |all_block| - out = instance_exec out, string, &all_block + out = instance_exec Oxidized::String.new(out), string, &all_block end if vars :remove_secret self.class.cmds[:secret].each do |all_block| - out = instance_exec out, string, &all_block + out = instance_exec Oxidized::String.new(out), string, &all_block end end - out = instance_exec out, &block if block - out + out = instance_exec Oxidized::String.new(out), &block if block + process_cmd_output out, string end def output @@ -124,20 +125,20 @@ module Oxidized end def get - data, pre = '', '' + outputs = Outputs.new procs = self.class.procs self.class.cmds[:cmd].each do |command, block| out = cmd command, &block return false unless out - data << out.to_s + outputs << out end procs[:pre].each do |pre_proc| - pre << instance_eval(&pre_proc).to_s + outputs.unshift Oxidized::String.new(instance_eval(&pre_proc)) end procs[:post].each do |post_proc| - data << instance_eval(&post_proc).to_s + outputs << Oxidized::String.new(instance_eval(&post_proc)) end - pre + data + outputs end def comment _comment @@ -148,5 +149,15 @@ module Oxidized data end + private + + def process_cmd_output output, name + if output.class != Oxidized::String + output = Oxidized::String.new output + end + output.set_cmd(name) + output + end + end end diff --git a/lib/oxidized/model/outputs.rb b/lib/oxidized/model/outputs.rb new file mode 100644 index 0000000..a668e9d --- /dev/null +++ b/lib/oxidized/model/outputs.rb @@ -0,0 +1,41 @@ +module Oxidized + class Model + class Outputs + + def to_cfg + type_to_str(nil) + end + + def type_to_str want_type + type(want_type).map { |out| out }.join + end + + def << output + @outputs << output + end + + def unshift output + @outputs.unshift output + end + + def all + @outputs + end + + def type type + @outputs.select { |out| out.type==type } + end + + def types + @outputs.map { |out| out.type }.uniq.compact + end + + private + + def initialize + @outputs = [] + end + + end + end +end |