lib/suj/pusher/apn_notification.rb in suj-pusher-0.0.1 vs lib/suj/pusher/apn_notification.rb in suj-pusher-0.1.0

@@ -7,10 +7,11 @@
       class InvalidToken < StandardError; end
       class PayloadTooLarge < StandardError; end
 
       def initialize(options = {})
         @token = options[:token]
+        @ttl = options[:time_to_live] || 0
         @options = options
         raise InvalidToken if @token.nil? || (@token.length != 64)
         raise PayloadTooLarge if data.size > MAX_SIZE
       end
 
@@ -22,17 +23,24 @@
         @data ||= encode_data
       end
 
       private
 
+      def get_expiry
+        if @ttl.to_i == 0
+          return 0
+        else
+          return Time.now.to_i + @ttl.to_i
+        end
+      end
+
       def encode_data
         identifier = 0
-        expiry = 0
+        expiry = get_expiry
         size = [payload].pack("a*").size
         data_array = [1, identifier, expiry, 32, @token, size, payload]
         info("PAYLOAD: #{data_array}")
         data_array.pack("cNNnH*na*")
       end
-
     end
   end
 end