lib/xinge/ios.rb in ruby-for-xinge-0.1.2 vs lib/xinge/ios.rb in ruby-for-xinge-0.1.3

@@ -5,28 +5,28 @@
     ENV_MAP = {'production' => 1, 'development' => 2}
 
     def initialize(accessId = nil, secretKey = nil, options = {})
       super
     end
-    def pushToSingleDevice(token, title, content, params={})
-      self.push_single_device(token, 1, build_simple_message(title, content), params.merge({environment: ENV_MAP[Xinge.config[:env]]}))
+    def pushToSingleDevice(token, title, content, params={}, custom_content={})
+      self.push_single_device(token, 1, build_simple_message(title, content, custom_content), params.merge({environment: ENV_MAP[Xinge.config[:env]]}))
     end
-    def pushToAllDevice(title, content, params={})
-      self.push_all_device(1, build_simple_message(title, content), params.merge({environment: ENV_MAP[Xinge.config[:env]]}))
+    def pushToAllDevice(title, content, params={}, custom_content={})
+      self.push_all_device(1, build_simple_message(title, content, custom_content), params.merge({environment: ENV_MAP[Xinge.config[:env]]}))
     end
 
     protected
 
-    def build_simple_message(title,content)
+    def build_simple_message(title, content, custom_content)
       {
         aps: {
           alert: {
             title: title,
             body: content
           },
           sound: 'default',
           badge: 5
         }
-      }.to_json
+      }.merge(custom_content).to_json
     end
   end
 end