1 files changed, 34 insertions, 8 deletions

diff --git a/lib/oxidized/model/model.rb b/lib/oxidized/model/model.rb
index 8f7b14f..2a6cbc4 100644
--- a/lib/oxidized/model/model.rb
+++ b/lib/oxidized/model/model.rb
@@ -2,22 +2,24 @@ module Oxidized
   class Model
     class << self
       def inherited klass
-        klass.instance_variable_set '@cmd', Hash.new { |h,k| h[k] = [] }
-        klass.instance_variable_set '@cfg', Hash.new { |h,k| h[k] = [] }
+        klass.instance_variable_set '@cmd',    Hash.new { |h,k| h[k] = [] }
+        klass.instance_variable_set '@cfg',    Hash.new { |h,k| h[k] = [] }
+        klass.instance_variable_set '@expect', []
+        klass.const_set :CFG, CFG
         Oxidized.mgr.loader = { :class => klass }
       end
       def comment _comment='# '
         return @comment if @comment
         @comment = block_given? ? yield : _comment
       end
+      def prompt _prompt=nil
+        @prompt or @prompt = _prompt
+      end
       def cfg *methods, &block
         [methods].flatten.each do |method|
           @cfg[method.to_s] << block
         end
       end
-      def prompt _prompt=nil
-        @prompt or @prompt = _prompt
-      end
       def cfgs
         @cfg
       end
@@ -31,8 +33,11 @@ module Oxidized
       def cmds
         @cmd
       end
-      def post_login &block
-        @post_login or @post_login = block
+      def expect re, &block
+        @expect << [re, block]
+      end
+      def expects
+        @expect
       end
     end
 
@@ -48,7 +53,15 @@ module Oxidized
       out
     end
 
-    def  cfg
+    def send data
+      @input.send data
+    end
+
+    def expect re, &block
+      self.class.expect re, &block
+    end
+
+    def cfg
       self.class.cfgs
     end
 
@@ -56,6 +69,19 @@ module Oxidized
       self.class.prompt
     end
 
+    def expects data
+      self.class.expects.each do |re, cb|
+        if data.match re
+          if cb.arity == 2
+            data = instance_exec [data, re], &cb
+          else
+            data = instance_exec data, &cb
+          end
+        end
+      end
+      data
+    end
+
     def get
       data = ''
       self.class.cmds[:cmd].each do |command, block|