require 'apnserver/payload'
require 'json'

module ApnServer
  
  class Config
    class << self
      attr_accessor :host, :port, :pem, :password
    end
  end
  
  
  class Notification
    include ApnServer::Payload
    
    attr_accessor :device_token, :alert, :badge, :sound, :custom
    
    
    def payload
      p = Hash.new
      [:badge, :alert, :sound, :custom].each do |k|
        p[k] = send(k) if send(k)        
      end
      create_payload(p)
    end
    
    def json_payload
      p = JSON.generate(payload)
      raise PayloadInvalid.new("Payload of #{p.size} is longer than 256") if p.size > 256
      p
    end
    
    def push
      if Config.pem.nil?
        socket = TCPSocket.new(Config.host || 'localhost', Config.port.to_i || 22195)
        socket.write(to_bytes)  
        socket.close
      else
        client = ApnServer::Client.new(Config.pem, Config.host || 'gateway.push.apple.com', Config.port.to_i || 2195)
        client.connect!
        client.write(self)
        client.disconnect!
      end      
    end
    
    def to_bytes
      json = json_payload
      [0, 0, device_token.size, device_token, 0, json.size, json].pack("ccca*cca*")
    end
    
    def self.valid?(payload)
      begin
        Notification.parse(payload)
      rescue RuntimeError
        false
      end
    end
    
    def self.parse(p)
      buffer = p.dup
      notification = Notification.new
      
      header = buffer.slice!(0, 3).unpack('ccc')
      if header[0] != 0 || header[1] != 0 || header[2] != 32
        raise RuntimeError.new("Header of notification is invalid: #{header.inspect}")
      end
      
      # parse token
      notification.device_token = buffer.slice!(0, 32).unpack('a*').first
      
      # parse json payload
      payload_len = buffer.slice!(0, 2).unpack('cc')
      result = JSON.parse(buffer.slice!(0, payload_len.last))
      
      ['alert', 'badge', 'sound'].each do |k|
        notification.send("#{k}=", result['aps'][k]) if result['aps'] && result['aps'][k]
      end
      result.delete('aps')
      notification.custom = result
      
      notification
    end
  end
end