require 'notifiable'
require 'grocer'
require 'connection_pool'

module Notifiable
  module Apns
    module Grocer
  		class Stream < Notifiable::NotifierBase
        
        notifier_attribute :certificate, :passphrase, :sandbox, :connection_pool_size, :connection_pool_timeout
        
        attr_reader :certificate, :passphrase
        
        def connection_pool_size
          @connection_pool_size || 10
        end
        
        def connection_pool_timeout
          @connection_pool_timeout || 10
        end
        
        def sandbox?
          @sandbox == "1"
        end
                
        def close
          super
          @grocer_pusher = nil        
          @grocer_feedback = nil
        end
      
  			protected      
    			def enqueue(device, notification)        				
            raise "Certificate missing" if certificate.nil?
            
            grocer_notification = ::Grocer::Notification.new(grocer_payload(device, notification))
          
            pusher_pool.with do |pusher|
              pusher.push(grocer_notification)
            end
                      
            # assume processed. Errors will be receieved through a callback
            processed(device, 0)
    			end
      
          def flush
            process_feedback unless self.test_env?
          end

        private 
          def gateway_host
            self.sandbox? ? "gateway.sandbox.push.apple.com" : "gateway.push.apple.com"
          end
      
          def gateway_port
            2195
          end
      
          def feedback_host
            self.sandbox? ? "feedback.sandbox.push.apple.com" : "feedback.push.apple.com"
          end
      
          def feedback_port
            2196
          end
                          
          def gateway_config 
            {
              certificate: certificate,
              passphrase:  passphrase,
              gateway:     gateway_host,
              port:        gateway_port,
              retries:     3
            }
          end
          
          def feedback_config
            {
              certificate: certificate,
              passphrase:  passphrase,
              gateway:     feedback_host,
              port:        feedback_port,
              retries:     3
            }
          end
          
          def grocer_payload(device, notification)
            payload = {device_token: device.token, custom: notification.send_params}
            payload[:alert] = notification.title ? {title: notification.title, body: notification.message} : notification.message
            payload[:sound] = notification.sound if notification.sound
            payload[:badge] = notification.badge_count if notification.badge_count
            payload[:identifier] = notification.identifier if notification.identifier
            payload[:content_available] = notification.content_available if notification.content_available
            payload[:mutable_content] = notification.mutable_content if notification.mutable_content
            payload[:expiry] = notification.expiry if notification.expiry
            payload
          end
          
          def pusher_pool
            @pusher_pool ||= ConnectionPool.new(size: connection_pool_size, timeout: connection_pool_timeout) do
              ::Grocer.pusher(gateway_config)
            end
          end
      
          def grocer_feedback
    				@grocer_feedback ||= ::Grocer.feedback(feedback_config)
          end
      
          def process_feedback
    				grocer_feedback.each do |attempt|
    					token = attempt.device_token
    					device_token = DeviceToken.find_by_token(token)
    					if device_token
    						device_token.destroy if device_token.updated_at < attempt.timestamp
    						Rails.logger.info("Device #{token} removed at #{attempt.timestamp}")
    					end
    				end
          end
  		end
    end
	end
end