Run 'main' method if it exists

Gives alternative, maybe less awkward way to do conditional commands.
You can first use cmd methods to gather stuff you want, then in main
method you could conditional to things based on them, maybe futher call
methods per model.

1 files changed, 4 insertions, 4 deletions

diff --git a/lib/oxidized/model/junos.rb b/lib/oxidized/model/junos.rb
index d1f4de6..72f5275 100644
--- a/lib/oxidized/model/junos.rb
+++ b/lib/oxidized/model/junos.rb
@@ -15,12 +15,12 @@ class JunOS < Oxidized::Model
   cmd 'show configuration'
 
   cmd 'show version' do |cfg|
-    chassis = model $1 if cfg.match /^Model: (\S+)/
-    comment cfg << chassis.to_s
+    @model = $1 if cfg.match /^Model: (\S+)/
+    comment cfg
   end
 
-  def model chassis
-    case chassis
+  def main
+    case @model
     when 'mx960'
       cmd('show chassis fabric reachability')  { |cfg| comment cfg }
     end