lib/wei-backend/base.rb in wei-backend-0.0.1 vs lib/wei-backend/base.rb in wei-backend-0.0.2

@@ -6,10 +6,14 @@
       @params = params
       results = send(:"handle_#{message_type.downcase}_message")
       create_model results
     end
 
+    def handle_event_message
+      send(:"handle_#{params[:Event].downcase}_message")
+    end
+
     def create_model data
       data.is_a?(Hash) || data.is_a?(Array) ? image_text_message(data) : text_message(data)
     end
 
     def text_message(data)
@@ -50,11 +54,19 @@
 
     def self.on_location &block
       define_method(:handle_location_message, &block)
     end
 
+    def self.on_subscribe &block
+      define_method(:handle_location_message, &block)
+    end
 
+    def self.on_unsubscribe &block
+      define_method(:handle_location_message, &block)
+    end
+
+
   end
 
   module Delegator
     def self.delegate(*methods)
       methods.each do |method_name|
@@ -63,10 +75,10 @@
         end
         private method_name
       end
     end
 
-    delegate :on_text, :on_event, :on_voice, :on_location
+    delegate :on_text, :on_event, :on_voice, :on_location, :on_subscribe, :on_unsubscribe
 
     class << self
       attr_accessor :target
     end