lib/lotu/behaviors/system_user.rb in lotu-0.1.15 vs lib/lotu/behaviors/system_user.rb in lotu-0.1.16

@@ -1,11 +1,49 @@
 module Lotu
   module SystemUser
 
-    # Allows to activate a system in the host
-    def use( klass, opts={} )
+    attr_accessor :systems
+
+    def self.included base
+      base.extend ClassMethods
+    end
+
+    def init_behavior user_opts
+      super if defined? super
       @systems ||= Hash.new
-      @systems[klass] = klass.new( self, opts )
+
+      options_for_me = self.class.behavior_options[SystemUser]
+
+      options_for_me && options_for_me.each do |klass, options|
+        # add the behavior options to the user_opts hash
+        # in case we need access to some level class config param
+        user_opts.merge!(options)
+        @systems[klass] = klass.new( self, user_opts )
+      end
+    end
+
+    # Need to call this inside update
+    def update
+      super if defined? super
+      @systems.each_value do |system|
+        system.update
+      end
+    end
+
+    # Need to call this inside draw
+    def draw
+      super if defined? super
+      # Systems may report interesting stuff
+      @systems.each_value do |system|
+        system.draw
+      end
+    end
+
+    # Allows to activate a system in the host
+    module ClassMethods
+      def use( klass, opts={} )
+        behavior_options[SystemUser][klass] = opts
+      end
     end
 
   end
 end