From 7aa9299971a2cacad29ab86e82cc780354d659e0 Mon Sep 17 00:00:00 2001
From: Saku Ytti <saku@ytti.fi>
Date: Thu, 7 Aug 2014 14:33:18 +0300
Subject: Keep config as Oxidized::Config instead of hash

The model will look like this:

  cmd 'show ip cef' do |out|
    out.type = 'poop'
    out
  end

  cmd 'show process cpu' do |out|
    out.type = 'poop'
    out
  end

  cmd 'show memory statistics' do |out|
    out.type = 'poop'
    out
  end

I think it's cleaner.
---
 lib/oxidized/model/model.rb   | 17 ++++++++---------
 lib/oxidized/model/outputs.rb | 12 +++---------
 2 files changed, 11 insertions(+), 18 deletions(-)

(limited to 'lib/oxidized/model')

diff --git a/lib/oxidized/model/model.rb b/lib/oxidized/model/model.rb
index af86f1d..3dd3f98 100644
--- a/lib/oxidized/model/model.rb
+++ b/lib/oxidized/model/model.rb
@@ -79,16 +79,15 @@ 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 = instance_exec Oxidized::String.new(out), &block if block
       process_cmd_output out, string
     end
 
@@ -152,12 +151,12 @@ module Oxidized
 
     private
 
-    def process_cmd_output cmd, name
-      if Hash === cmd
-        cmd[:name] = name
-        return cmd
+    def process_cmd_output output, name
+      if output.class != Oxidized::String
+        output = Oxidized::String.new output
       end
-      {:output=>cmd, :type=>'cfg', :name=>name}
+      output.cmd = name
+      output
     end
 
   end
diff --git a/lib/oxidized/model/outputs.rb b/lib/oxidized/model/outputs.rb
index 5ef9bc4..406f886 100644
--- a/lib/oxidized/model/outputs.rb
+++ b/lib/oxidized/model/outputs.rb
@@ -7,13 +7,7 @@ module Oxidized
       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
+        type(want_type).map { |out| out }.join
       end
 
       def << output
@@ -25,11 +19,11 @@ module Oxidized
       end
 
       def type type
-        @outputs.select { |h| h[:type]==type }
+        @outputs.select { |out| out.type==type }
       end
 
       def types
-        @outputs.map { |h| h[:type] }.uniq
+        @outputs.map { |out| out.type }.uniq - ['cfg']
       end
 
       private
-- 
cgit v1.2.1