diff options
Diffstat (limited to 'lib/oxidized')
-rw-r--r-- | lib/oxidized/model/model.rb | 28 | ||||
-rw-r--r-- | lib/oxidized/model/outputs.rb | 43 | ||||
-rw-r--r-- | lib/oxidized/output/git.rb | 43 | ||||
-rw-r--r-- | lib/oxidized/output/output.rb | 5 |
4 files changed, 97 insertions, 22 deletions
diff --git a/lib/oxidized/model/model.rb b/lib/oxidized/model/model.rb index c3d8760..af86f1d 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 @@ -87,7 +89,7 @@ module Oxidized end end out = instance_exec out, &block if block - out + process_cmd_output out, string end def output @@ -124,20 +126,20 @@ module Oxidized end def get - data, pre = '', '' + outputs = Outputs.new procs = self.class.procs + procs[:pre].each do |pre_proc| + outputs << instance_eval(&pre_proc) + end self.class.cmds[:cmd].each do |command, block| out = cmd command, &block return false unless out - data << out.to_s - end - procs[:pre].each do |pre_proc| - pre << instance_eval(&pre_proc).to_s + outputs << out end procs[:post].each do |post_proc| - data << instance_eval(&post_proc).to_s + outputs << instance_eval(&post_proc) end - pre + data + outputs end def comment _comment @@ -148,5 +150,15 @@ module Oxidized data end + private + + def process_cmd_output cmd, name + if Hash === cmd + cmd[:name] = name + return cmd + end + {:output=>cmd, :type=>'cfg', :name=>name} + end + end end diff --git a/lib/oxidized/model/outputs.rb b/lib/oxidized/model/outputs.rb new file mode 100644 index 0000000..5ef9bc4 --- /dev/null +++ b/lib/oxidized/model/outputs.rb @@ -0,0 +1,43 @@ +module Oxidized + class Model + class Outputs + + def to_cfg + type_to_str('cfg') + end + + def type_to_str want_type + type(want_type).map { |h| h[:output] }.join + end + + def each_type &block + types.each do |want_type| + yield [want_type, type(want_type)] + end + end + + def << output + @outputs << output + end + + def all + @outputs + end + + def type type + @outputs.select { |h| h[:type]==type } + end + + def types + @outputs.map { |h| h[:type] }.uniq + end + + private + + def initialize + @outputs = [] + end + + end + end +end diff --git a/lib/oxidized/output/git.rb b/lib/oxidized/output/git.rb index 18c2cc1..59b6bd4 100644 --- a/lib/oxidized/output/git.rb +++ b/lib/oxidized/output/git.rb @@ -20,23 +20,27 @@ class Git < Output end end - def store file, data, opt={} - msg = opt[:msg] - user = (opt[:user] or @cfg.user) - email = (opt[:email] or @cfg.email) - repo = @cfg.repo - if opt[:group] - repo = File.join File.dirname(repo), opt[:group] + '.git' - end - begin - repo = Rugged::Repository.new repo - update_repo repo, file, data, msg, user, email - rescue Rugged::OSError, Rugged::RepositoryError - Rugged::Repository.init_at repo, :bare - retry + def store file, outputs, opt={} + data = outputs.to_cfg + @msg = opt[:msg] + @user = (opt[:user] or @cfg.user) + @email = (opt[:email] or @cfg.email) + @opt = opt + repo = @cfg.repo + + outputs.types.each do |type| + next if type == 'cfg' + type_repo = File.join File.dirname(repo), type + '.git' + outputs.type(type).each do |output| + type_file = file + '--' + output[:name].strip.gsub(/\s+/, '_') + update type_repo, type_file, output[:output] + end end + + update repo, file, outputs.to_cfg end + def fetch node, group begin repo = @cfg.repo @@ -54,6 +58,17 @@ class Git < Output private + def update repo, file, data + if @opt[:group] + repo = File.join File.dirname(repo), @opt[:group] + '.git' + end + repo = Rugged::Repository.new repo + update_repo repo, file, data, @msg, @user, @email + rescue Rugged::OSError, Rugged::RepositoryError + Rugged::Repository.init_at repo, :bare + retry + end + def update_repo repo, file, data, msg, user, email oid = repo.write data, :blob index = repo.index diff --git a/lib/oxidized/output/output.rb b/lib/oxidized/output/output.rb index eaf149a..1355d03 100644 --- a/lib/oxidized/output/output.rb +++ b/lib/oxidized/output/output.rb @@ -1,5 +1,10 @@ module Oxidized class Output class NoConfig < OxidizedError; end + + def cfg_to_str cfg + cfg.select{ |h| h[:type]=='cfg' }.map{ |h| h[:data] }.join + end + end end |