em-apn 0.0.3

data/.gitignore

@@ -0,0 +1,6 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*
+certs/*.pem
+.rvmrc

data/.rspec

@@ -0,0 +1 @@
+--color

data/Gemfile

@@ -0,0 +1,4 @@
+source "http://rubygems.org"
+
+# Specify your gem's dependencies in em-apn.gemspec
+gemspec

data/LICENSE

@@ -0,0 +1,22 @@
+(The MIT-License)
+
+Copyright (c) 2011 GroupMe
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,78 @@
+# EM-APN - EventMachine'd Apple Push Notifications #
+
+We want:
+
+ * Streamlined for a persistent connection use-case
+ * Support for the enhanced protocol, with receipts
+
+## Usage ##
+
+In a nutshell:
+
+    require "em-apn"
+
+    # Inside a reactor...
+    notification = EM::APN::Notification.new(token, :alert => alert)
+    client = EM::APN::Client.connect
+    client.deliver(notification)
+
+Using this interface, the easiest way to configure the connection is by setting
+some environment variables so that EM::APN can find your SSL certificates:
+
+    ENV["APN_KEY"]  = "/path/to/key.pem"
+    ENV["APN_CERT"] = "/path/to/cert.pem"
+
+Also, by default, the library connects to Apple's sandbox push server. If you
+want to connect to the production server, simply set the `APN_ENV`
+environment variable to `production`:
+
+    ENV["APN_ENV"] = "production"
+
+The gateway and SSL certs can also be set directly when instantiating the object:
+
+    client = EM::APN::Client.connect(
+      :gateway => "some.host",
+      :key     => "/path/to/key.pem",
+      :cert    => "/path/to/cert.pem"
+    )
+
+The client manages an underlying `EM::Connection`, and it will automatically
+reconnect to the gateway when the connection is closed. Callbacks can be set
+on the client to handle error responses from the gateway and connection close
+events:
+
+    client = EM::APN::Client.connect
+    client.on_error do |response|
+      # See EM::APN::ErrorResponse
+    end
+
+    client.on_close do
+      # Do something.
+    end
+
+In our experience, we've found that Apple immediately closes the connection
+whenever an error is detected, so the error and close callbacks are nearly
+always called one-to-one. These methods exist as a convenience, and the
+callbacks can also be set directly to anything that responds to `#call`:
+
+    client.error_callback = Proc.new { |response| ... }
+    client.close_callback = Proc.new { ... }
+
+### Max Payload Size ###
+
+Apple enforces a limit of __256 bytes__ for the __entire payload__.
+
+We raise an `EM::APN::Notification::PayloadTooLarge` exception.
+
+How you truncate your payloads is up to you. Be especially careful when dealing with multi-byte data.
+
+## TODO ##
+
+ * Support the feedback API for dead tokens
+
+## Inspiration ##
+
+Much thanks to:
+
+ * https://github.com/kdonovan/apn_sender
+ * http://blog.technopathllc.com/2010/12/apples-push-notification-with-ruby.html

data/Rakefile

@@ -0,0 +1,9 @@
+require 'bundler/gem_tasks'
+
+desc "Run specs"
+task :spec do
+  spec_files = Dir["spec/**/*_spec.rb"]
+  ruby("-S bundle exec rspec #{spec_files.join(" ")}")
+end
+
+task :default => :spec

data/em-apn.gemspec

@@ -0,0 +1,24 @@
+# -*- mode: ruby; encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "em-apn/version"
+
+Gem::Specification.new do |s|
+  s.name        = "em-apn"
+  s.version     = EventMachine::APN::VERSION
+  s.authors     = ["Dave Yeu"]
+  s.email       = ["daveyeu@gmail.com"]
+  s.homepage    = ""
+  s.summary     = %q{EventMachine-driven Apple Push Notifications}
+
+  s.rubyforge_project = "em-apn"
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+
+  s.add_dependency "eventmachine", ">= 1.0.0.beta.3"
+  s.add_dependency "yajl-ruby",    ">= 0.8.2"
+
+  s.add_development_dependency "rspec", "~> 2.6.0"
+end

data/lib/em-apn.rb

@@ -0,0 +1,23 @@
+# encoding: UTF-8
+
+require "eventmachine"
+require "yajl"
+require "logger"
+require "em-apn/client"
+require "em-apn/connection"
+require "em-apn/notification"
+require "em-apn/log_message"
+require "em-apn/response"
+require "em-apn/error_response"
+
+module EventMachine
+  module APN
+    def self.logger
+      @logger ||= Logger.new(STDOUT)
+    end
+
+    def self.logger=(new_logger)
+      @logger = new_logger
+    end
+  end
+end

data/lib/em-apn/client.rb

@@ -0,0 +1,53 @@
+# encoding: UTF-8
+
+module EventMachine
+  module APN
+    class Client
+      SANDBOX_GATEWAY    = "gateway.sandbox.push.apple.com"
+      PRODUCTION_GATEWAY = "gateway.push.apple.com"
+      PORT               = 2195
+
+      attr_reader :gateway, :port, :key, :cert, :connection, :error_callback, :close_callback
+
+      # A convenience method for creating and connecting.
+      def self.connect(options = {})
+        new(options).tap do |client|
+          client.connect
+        end
+      end
+
+      def initialize(options = {})
+        @key  = options[:key]  || ENV["APN_KEY"]
+        @cert = options[:cert] || ENV["APN_CERT"]
+        @port = options[:port] || PORT
+
+        @gateway = options[:gateway] || ENV["APN_GATEWAY"]
+        @gateway ||= (ENV["APN_ENV"] == "production") ? PRODUCTION_GATEWAY : SANDBOX_GATEWAY
+
+        @connection = nil
+      end
+
+      def connect
+        @connection = EM.connect(gateway, port, Connection, self)
+      end
+
+      def deliver(notification)
+        connect if connection.nil? || connection.disconnected?
+        log(notification)
+        connection.send_data(notification.data)
+      end
+
+      def on_error(&block)
+        @error_callback = block
+      end
+
+      def on_close(&block)
+        @close_callback = block
+      end
+
+      def log(notification)
+        EM::APN.logger.info("TOKEN=#{notification.token} ALERT=#{notification.alert}")
+      end
+    end
+  end
+end

data/lib/em-apn/connection.rb

@@ -0,0 +1,46 @@
+module EventMachine
+  module APN
+    class Connection < EM::Connection
+      attr_reader :client
+
+      def initialize(*args)
+        super
+        @client = args.last
+        @disconnected = false
+      end
+
+      def disconnected?
+        @disconnected
+      end
+
+      def post_init
+        start_tls(
+          :private_key_file => client.key,
+          :cert_chain_file  => client.cert,
+          :verify_peer      => false
+        )
+      end
+
+      def connection_completed
+        EM::APN.logger.info("Connection completed")
+      end
+
+      def receive_data(data)
+        data_array = data.unpack("ccN")
+        error_response = ErrorResponse.new(*data_array)
+        EM::APN.logger.warn(error_response.to_s)
+
+        if client.error_callback
+          client.error_callback.call(error_response)
+        end
+      end
+
+      def unbind
+        @disconnected = true
+
+        EM::APN.logger.info("Connection closed")
+        client.close_callback.call if client.close_callback
+      end
+    end
+  end
+end

data/lib/em-apn/error_response.rb

@@ -0,0 +1,34 @@
+module EventMachine
+  module APN
+    class ErrorResponse
+      DESCRIPTION = {
+        0   => "No errors encountered",
+        1   => "Processing error",
+        2   => "Missing device token",
+        3   => "Missing topic",
+        4   => "Missing payload",
+        5   => "Invalid token size",
+        6   => "Invalid topic size",
+        7   => "Invalid payload size",
+        8   => "Invalid token",
+        255 => "None (unknown)"
+      }
+
+      attr_reader :command, :status_code, :identifier
+
+      def initialize(command, status_code, identifier)
+        @command     = command
+        @status_code = status_code
+        @identifier  = identifier
+      end
+
+      def to_s
+        "CODE=#{@status_code} ID=#{@identifier} DESC=#{description}"
+      end
+
+      def description
+        DESCRIPTION[@status_code] || "Missing description"
+      end
+    end
+  end
+end

data/lib/em-apn/log_message.rb

@@ -0,0 +1,13 @@
+module EventMachine
+  module APN
+    class LogMessage
+      def initialize(response)
+        @response = response
+      end
+
+      def log
+        EM::APN.logger.info(@response.to_s)
+      end
+    end
+  end
+end

data/lib/em-apn/notification.rb

@@ -0,0 +1,63 @@
+# encoding: UTF-8
+
+module EventMachine
+  module APN
+    class Notification
+      PAYLOAD_MAX_BYTES = 256
+      class PayloadTooLarge < StandardError;end
+
+      attr_reader :token
+      attr_accessor :identifier, :expiry
+
+      def initialize(token, aps = {}, custom = {}, options = {})
+        raise "Bad push token: #{token}" if token.nil? || (token.length != 64)
+
+        @token  = token
+        @aps    = aps
+        @custom = custom
+
+        self.identifier = options[:identifier] if options[:identifier]
+        self.expiry = options[:expiry] if options[:expiry]
+      end
+
+      def payload
+        @payload ||= build_payload
+      end
+
+      def data
+        @data ||= build_data
+      end
+
+      def identifier=(new_identifier)
+        @identifier = new_identifier.to_i
+      end
+
+      def alert
+        @aps["alert"][0..49] if @aps.include?("alert")
+      end
+
+      private
+
+      def build_payload
+        payload = @custom.merge(:aps => @aps)
+        Yajl::Encoder.encode(payload)
+      end
+
+      # Documentation about this format is here:
+      # http://developer.apple.com/library/ios/#documentation/NetworkingInternet/Conceptual/RemoteNotificationsPG/CommunicatingWIthAPS/CommunicatingWIthAPS.html
+      def build_data
+        identifier = @identifier || 0
+        expiry     = @expiry || 0
+
+        size = [payload].pack("a*").size
+        data_array = [1, identifier, expiry, 32, token, size, payload]
+        data = data_array.pack("cNNnH*na*")
+        if data.size > PAYLOAD_MAX_BYTES
+          error = "max is #{PAYLOAD_MAX_BYTES} bytes (got #{data.size})"
+          raise PayloadTooLarge.new(error)
+        end
+        data
+      end
+    end
+  end
+end

data/lib/em-apn/response.rb

@@ -0,0 +1,13 @@
+module EventMachine
+  module APN
+    class Response
+      def initialize(notification)
+        @notification = notification
+      end
+
+      def to_s
+        "TOKEN=#{@notification.token}"
+      end
+    end
+  end
+end

data/lib/em-apn/server.rb

@@ -0,0 +1,58 @@
+# encoding: UTF-8
+# Mock Apple push server... because we love to test
+module EventMachine
+  module APN
+    module Server
+      def post_init
+        EM::APN.logger.info("Received a new connection")
+        @data  = ""
+
+        start_tls(
+          :cert_chain_file  => ENV["APN_CERT"],
+          :private_key_file => ENV["APN_KEY"],
+          :verify_peer      => false
+        )
+      end
+
+      def ssl_handshake_completed
+        EM::APN.logger.info("SSL handshake completed")
+      end
+
+      def receive_data(data)
+        @data << data
+
+        # Try to extract the payload header
+        headers = @data.unpack("cNNnH64n")
+        return if headers.last.nil?
+
+        # Try to grab the payload
+        payload_size = headers.last
+        payload = @data[45, payload_size]
+        return if payload.length != payload_size
+
+        @data = @data[45 + payload_size, -1] || ""
+
+        process(headers, payload)
+      end
+
+      def process(headers, payload)
+        message = "APN RECV #{headers[4]} #{payload}"
+        EM::APN.logger.info(message)
+
+        args = Yajl::Parser.parse(payload)
+
+        # If the alert is 'DISCONNECT', then we fake a bad payload by replying
+        # with an error and disconnecting.
+        if args["aps"]["alert"] == "DISCONNECT"
+          EM::APN.logger.info("Disconnecting")
+          send_data([8, 1, 0].pack("ccN"))
+          close_connection_after_writing
+        end
+      end
+
+      def unbind
+        EM::APN.logger.info("Connection closed")
+      end
+    end
+  end
+end

data/lib/em-apn/test_helper.rb

@@ -0,0 +1,37 @@
+# Test helper for EM::APN
+#
+# To use this, start by simply requiring this file after EM::APN has already
+# been loaded
+#
+#     require "em-apn"
+#     require "em-apn/test_helper"
+#
+# This will nullify actual deliveries and instead, push them onto an accessible
+# list:
+#
+#     expect {
+#       client.deliver('notification)
+#     }.to change { EM::APN.deliveries.size }.by(1)
+#
+#     notification = EM::APN.deliveries.first
+#     notification.should be_an_instance_of(EM::APN::Notification)
+#     notification.payload.should == ...
+#
+module EventMachine
+  module APN
+    def self.deliveries
+      @deliveries ||= []
+    end
+
+    Client.class_eval do
+      def connect
+        # No-op
+      end
+
+      def deliver(notification)
+        log(notification)
+        EM::APN.deliveries << notification
+      end
+    end
+  end
+end

data/lib/em-apn/version.rb

@@ -0,0 +1,5 @@
+module EventMachine
+  module APN
+    VERSION = "0.0.3"
+  end
+end

data/script/push

@@ -0,0 +1,29 @@
+#!/usr/bin/env ruby
+#
+# Send a single push notification
+#
+#   $ script/push <TOKEN> "<ALERT>"
+#
+
+require "rubygems"
+require "bundler/setup"
+require "em-apn"
+
+ENV["APN_KEY"]  ||= File.join(File.dirname(__FILE__),  "..", "certs", "key.pem")
+ENV["APN_CERT"] ||= File.join(File.dirname(__FILE__), "..", "certs", "cert.pem")
+
+token, alert = ARGV
+
+if token.nil? || alert.nil?
+  puts "Usage: script/push <TOKEN> \"<ALERT>\""
+  exit 1
+end
+
+EM.run do
+  client = EM::APN::Client.connect
+  client.deliver(EM::APN::Notification.new(token, :alert => alert))
+
+  # Hopefully give ourselves enough time to receive a response on failure.
+  # Wish there was a better way to do this. Or at least a more timely way.
+  EM.add_timer(1) { EM.stop_event_loop }
+end

data/script/server

@@ -0,0 +1,17 @@
+#!/usr/bin/env ruby
+#
+# Run a mock Apple push server
+
+require "rubygems"
+require "bundler/setup"
+require "em-apn"
+require "em-apn/server"
+
+ENV["APN_KEY"]  = File.join(File.dirname(__FILE__), "..", "certs", "key.pem")
+ENV["APN_CERT"] = File.join(File.dirname(__FILE__), "..", "certs", "cert.pem")
+
+EM::APN.logger.info("Starting push server")
+
+EM.run do
+  EM.start_server("127.0.0.1", 2195, EM::APN::Server)
+end

data/script/test

@@ -0,0 +1,36 @@
+#!/usr/bin/env ruby
+
+require "rubygems"
+require "bundler/setup"
+require "em-apn"
+
+ENV["APN_KEY"]  = File.join(File.dirname(__FILE__), "..", "certs", "key.pem")
+ENV["APN_CERT"] = File.join(File.dirname(__FILE__), "..", "certs", "cert.pem")
+
+TOKEN = "fe9515ba7556cfdcfca6530235c95dff682fac765838e749a201a7f6cf3792e6"
+
+def notify(client, queue)
+  if queue.empty?
+    EM.add_periodic_timer(1) { EM.stop }
+  else
+    queue.pop do |alert|
+      notification = EM::APN::Notification.new(TOKEN, :alert => alert)
+      client.deliver(notification)
+
+      interval = rand(20).to_f / 100.0
+      EM.add_timer(interval) { notify(client, queue) }
+    end
+  end
+end
+
+EM.run do
+  client = EM::APN::Client.new(:gateway => "127.0.0.1")
+  queue = EM::Queue.new
+  queue.push *((1..5).to_a)
+  queue.push "DISCONNECT"
+  queue.push *((6..10).to_a)
+  queue.push "DISCONNECT"
+  queue.push *((11..15).to_a)
+
+  EM.next_tick { notify(client, queue) }
+end

data/spec/em-apn/client_spec.rb

@@ -0,0 +1,166 @@
+require "spec_helper"
+
+describe EventMachine::APN::Client do
+  def new_client(*args)
+    client = nil
+    EM.run_block {
+      client = EM.connect("localhost", 8888, EventMachine::APN::Client, *args)
+    }
+    client
+  end
+
+  describe ".new" do
+    it "creates a new client without a connection" do
+      client = EM::APN::Client.new
+      client.connection.should be_nil
+    end
+
+    context "configuring the gateway" do
+      before do
+        ENV["APN_GATEWAY"] = nil
+      end
+
+      let(:options) { {:key => ENV["APN_KEY"], :cert => ENV["APN_CERT"]} }
+
+      it "defaults to Apple's sandbox (gateway.sandbox.push.apple.com)" do
+        client = EM::APN::Client.new
+        client.gateway.should == "gateway.sandbox.push.apple.com"
+        client.port.should == 2195
+      end
+
+      it "uses an environment variable for the gateway host (APN_GATEWAY) if specified" do
+        ENV["APN_GATEWAY"] = "localhost"
+
+        client = EM::APN::Client.new
+        client.gateway.should == "localhost"
+        client.port.should == 2195
+      end
+
+      it "switches to the production gateway if APN_ENV is set to 'production'" do
+        ENV["APN_ENV"] = "production"
+
+        client = EM::APN::Client.new
+        client.gateway.should == "gateway.push.apple.com"
+        client.port.should == 2195
+
+        ENV["APN_ENV"] = nil
+      end
+
+      it "takes arguments for the gateway and port" do
+        client = EM::APN::Client.new(:gateway => "localhost", :port => 3333)
+        client.gateway.should == "localhost"
+        client.port.should == 3333
+      end
+    end
+
+    context "configuring SSL" do
+      it "falls back to environment variables for key and cert (APN_KEY and APN_CERT) if they are unspecified" do
+        ENV["APN_KEY"]  = "path/to/key.pem"
+        ENV["APN_CERT"] = "path/to/cert.pem"
+
+        client = EM::APN::Client.new
+        client.key.should == "path/to/key.pem"
+        client.cert.should == "path/to/cert.pem"
+      end
+
+      it "takes arguments for the key and cert" do
+        client = EM::APN::Client.new(:key => "key.pem", :cert => "cert.pem")
+        client.key.should == "key.pem"
+        client.cert.should == "cert.pem"
+      end
+    end
+  end
+
+  describe "#connect" do
+    it "creates a connection to the gateway" do
+      client = EM::APN::Client.new
+      client.connection.should be_nil
+
+      EM.run_block { client.connect }
+      client.connection.should be_an_instance_of(EM::APN::Connection)
+    end
+
+    it "passes the client to the new connection" do
+      client = EM::APN::Client.new
+      connection = double(EM::APN::Connection).as_null_object
+
+      EM::APN::Connection.should_receive(:new).with(instance_of(Fixnum), client).and_return(connection)
+      EM.run_block { client.connect }
+    end
+  end
+
+  describe "#deliver" do
+    let(:token) { "fe9515ba7556cfdcfca6530235c95dff682fac765838e749a201a7f6cf3792e6" }
+
+    it "sends a Notification object" do
+      notification = EM::APN::Notification.new(token, :alert => "Hello world")
+      delivered = nil
+
+      EM.run_block do
+        client = EM::APN::Client.new
+        client.connect
+        client.connection.stub(:send_data).and_return do |data|
+          delivered = data.unpack("cNNnH64na*")
+          nil
+        end
+        client.deliver(notification)
+      end
+
+      delivered[4].should == token
+      delivered[6].should == Yajl::Encoder.encode({:aps => {:alert => "Hello world"}})
+    end
+
+    it "logs a message" do
+      alert = "Hello world this is a long push notification to you"
+
+      test_log = StringIO.new
+      EM::APN.logger = Logger.new(test_log)
+
+      notification = EM::APN::Notification.new(token, "alert" => alert)
+
+      EM.run_block do
+        client = EM::APN::Client.new
+        client.deliver(notification)
+      end
+
+      test_log.rewind
+      test_log.read.should include("TOKEN=#{token} ALERT=#{alert[0..49]}")
+    end
+  end
+
+  describe "#on_error" do
+    it "sets a callback that is invoked when we receive data from Apple" do
+      error = nil
+
+      EM.run_block do
+        client = EM::APN::Client.new
+        client.connect
+        client.on_error { |e| error = e }
+        client.connection.receive_data([8, 8, 0].pack("ccN"))
+      end
+
+      error.should be
+
+      error.command.should == 8
+      error.status_code.should == 8
+      error.identifier.should == 0
+
+      error.description.should == "Invalid token"
+      error.to_s.should == "CODE=8 ID=0 DESC=Invalid token"
+    end
+  end
+
+  describe "#on_close" do
+    it "sets a callback that is invoked when the connection closes" do
+      called = false
+
+      EM.run_block do
+        client = EM::APN::Client.new
+        client.on_close { called = true }
+        client.connect # This should unbind immediately.
+      end
+
+      called.should be_true
+    end
+  end
+end

data/spec/em-apn/notification_spec.rb

@@ -0,0 +1,137 @@
+# encoding: UTF-8
+require "spec_helper"
+
+describe EventMachine::APN::Notification do
+  let(:token) { "fe9515ba7556cfdcfca6530235c95dff682fac765838e749a201a7f6cf3792e6" }
+
+  describe "#initialize" do
+    it "raises an exception if the token is blank" do
+      expect { EM::APN::Notification.new(nil) }.to raise_error
+      expect { EM::APN::Notification.new("") }.to raise_error
+    end
+
+    it "raises an exception if the token is less than or greater than 32 bytes" do
+      expect { EM::APN::Notification.new("0" * 63) }.to raise_error
+      expect { EM::APN::Notification.new("0" * 65) }.to raise_error
+    end
+  end
+
+  describe "#token" do
+    it "returns the token" do
+      notification = EM::APN::Notification.new(token)
+      notification.token.should == token
+    end
+  end
+
+  describe "#payload" do
+    it "returns aps properties encoded as JSON" do
+      notification = EM::APN::Notification.new(token, {
+        :alert => "Hello world",
+        :badge => 10,
+        :sound => "ding.aiff"
+      })
+      payload = Yajl::Parser.parse(notification.payload)
+      payload["aps"]["alert"].should == "Hello world"
+      payload["aps"]["badge"].should == 10
+      payload["aps"]["sound"].should == "ding.aiff"
+    end
+
+    it "returns custom properties as well" do
+      notification = EM::APN::Notification.new(token, {}, {:line => "I'm super bad"})
+      payload = Yajl::Parser.parse(notification.payload)
+      payload["line"].should == "I'm super bad"
+    end
+
+    it "handles string keys" do
+      notification = EM::APN::Notification.new(token,
+        {
+          "alert" => "Hello world",
+          "badge" => 10,
+          "sound" => "ding.aiff"
+        },
+        {
+          "custom" => "param"
+        }
+      )
+      payload = Yajl::Parser.parse(notification.payload)
+      payload["aps"]["alert"].should == "Hello world"
+      payload["aps"]["badge"].should == 10
+      payload["aps"]["sound"].should == "ding.aiff"
+      payload["custom"].should == "param"
+    end
+  end
+
+  describe "#data" do
+    it "returns the enhanced notification in the supported binary format" do
+      notification = EM::APN::Notification.new(token, {:alert => "Hello world"})
+      data = notification.data.unpack("cNNnH64na*")
+      data[4].should == token
+      data[5].should == notification.payload.length
+      data[6].should == notification.payload
+    end
+
+    it "handles UTF-8 payloads" do
+      string = "✓ Please"
+      notification = EM::APN::Notification.new(token, {:alert => string})
+      data = notification.data.unpack("cNNnH64na*")
+      data[4].should == token
+      data[5].should == [notification.payload].pack("a*").size
+      data[6].force_encoding("UTF-8").should == notification.payload
+    end
+
+    it "defaults the identifier and expiry to 0" do
+      notification = EM::APN::Notification.new(token, {:alert => "Hello world"})
+      data = notification.data.unpack("cNNnH64na*")
+      data[1].should == 0 # Identifier
+      data[2].should == 0 # Expiry
+    end
+
+    it "raises PayloadTooLarge error if PAYLOAD_MAX_BYTES exceeded" do
+      notification = EM::APN::Notification.new(token, {:alert => "X" * 512})
+
+      lambda {
+        notification.data
+      }.should raise_error(EM::APN::Notification::PayloadTooLarge)
+    end
+  end
+
+  describe "#identifier=" do
+    it "sets the identifier, which is returned in the binary data" do
+      notification = EM::APN::Notification.new(token)
+      notification.identifier = 12345
+      notification.identifier.should == 12345
+
+      data = notification.data.unpack("cNNnH64na*")
+      data[1].should == notification.identifier
+    end
+
+    it "converts everything to an integer" do
+      notification = EM::APN::Notification.new(token)
+      notification.identifier = "12345"
+      notification.identifier.should == 12345
+    end
+
+    it "can be set in the initializer" do
+      notification = EM::APN::Notification.new(token, {}, {}, {:identifier => 12345})
+      notification.identifier.should == 12345
+    end
+  end
+
+  describe "#expiry=" do
+    it "sets the expiry, which is returned in the binary data" do
+      epoch = Time.now.to_i
+      notification = EM::APN::Notification.new(token)
+      notification.expiry = epoch
+      notification.expiry.should == epoch
+
+      data = notification.data.unpack("cNNnH64na*")
+      data[2].should == epoch
+    end
+
+    it "can be set in the initializer" do
+      epoch = Time.now.to_i
+      notification = EM::APN::Notification.new(token, {}, {}, {:expiry => epoch})
+      notification.expiry.should == epoch
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,13 @@
+require "rubygems"
+require "bundler/setup"
+Bundler.require :default, :development
+
+RSpec.configure do |config|
+  config.before(:each) do
+    ENV["APN_KEY"]     = "spec/support/certs/key.pem"
+    ENV["APN_CERT"]    = "spec/support/certs/cert.pem"
+    ENV["APN_GATEWAY"] = "127.0.0.1"
+
+    EM::APN.logger = Logger.new("/dev/null")
+  end
+end

data/spec/support/certs/cert.pem

@@ -0,0 +1,17 @@
+-----BEGIN CERTIFICATE-----
+MIICqjCCAhOgAwIBAgIJAO3gD8Fv66MhMA0GCSqGSIb3DQEBBQUAMEMxCzAJBgNV
+BAYTAlVTMREwDwYDVQQIEwhOZXcgWW9yazERMA8GA1UEBxMITmV3IFlvcmsxDjAM
+BgNVBAoTBUR1bW15MB4XDTExMDYyMTE5MDAyOFoXDTIxMDYxODE5MDAyOFowQzEL
+MAkGA1UEBhMCVVMxETAPBgNVBAgTCE5ldyBZb3JrMREwDwYDVQQHEwhOZXcgWW9y
+azEOMAwGA1UEChMFRHVtbXkwgZ8wDQYJKoZIhvcNAQEBBQADgY0AMIGJAoGBAMsR
+05wNdOiFXil3/r4xGLXc7MYwU2LiZbnXAqaoGnLv3W8cN7i/0i63y8YYITNWm3ji
+lMiOkj2+TNVE/PiFizFBqhT28V/AsfOsiJrYQ4GXwLwJTsgDGTwAJgov0yPIffhK
+n9jU7NLNJjhrj0rVD++eNOT8dunvYRlI5S8lJWCNAgMBAAGjgaUwgaIwHQYDVR0O
+BBYEFISS0qC22So8BtQYTT4n/iLdYnZnMHMGA1UdIwRsMGqAFISS0qC22So8BtQY
+TT4n/iLdYnZnoUekRTBDMQswCQYDVQQGEwJVUzERMA8GA1UECBMITmV3IFlvcmsx
+ETAPBgNVBAcTCE5ldyBZb3JrMQ4wDAYDVQQKEwVEdW1teYIJAO3gD8Fv66MhMAwG
+A1UdEwQFMAMBAf8wDQYJKoZIhvcNAQEFBQADgYEAFBGom218pNCB1zYxhDDtOiYC
+DyU9rZkLXxMVjSHCvyBtQ3EC/21Zog05LSBMvOHFmfzPbyX/IYxSglJLJlQBzGtP
+trVx1slmva3/QoKYwAKdAe5oSY8eDqejBpKO12maSGsPwbw3oREmudeWkrFFVAlP
+hlkmwpq7901kIg4CCe8=
+-----END CERTIFICATE-----

data/spec/support/certs/key.pem

@@ -0,0 +1,15 @@
+-----BEGIN RSA PRIVATE KEY-----
+MIICXQIBAAKBgQDLEdOcDXTohV4pd/6+MRi13OzGMFNi4mW51wKmqBpy791vHDe4
+v9Iut8vGGCEzVpt44pTIjpI9vkzVRPz4hYsxQaoU9vFfwLHzrIia2EOBl8C8CU7I
+Axk8ACYKL9MjyH34Sp/Y1OzSzSY4a49K1Q/vnjTk/Hbp72EZSOUvJSVgjQIDAQAB
+AoGBAIHk8Ef076BAlz/Naty7yQOjwqzvgpdRHCLo3uA9zVVSC4GkOhxqTwblOGqJ
+SsttDdwgi21SjUcDcGBHVc2elq6SNx9A0mwhb9uPBUI74pJdbrN3/Ue1UnaKy7KZ
+1PGlMi8GtBL5cn5+NCs/IJUZIIp3oNxURJTF54tun6ygOeddAkEA74rcrXRvFAab
+mMwQuGlFCNU1ns6Zm19tBFBadTUJQl0cB0MAS9BKveXGOsxXVChR0JvDTnTnx5a4
+SwKz3t/irwJBANkFeJdZy9Ve5mzzY3rt6sCorOPo9+lER4K01HB6h24vWB1DIFRE
+YNNiK7YBVHrnb4mcKSdP93QXlYoAugP974MCQQDrVPACVI5ADVHV5j1S/tC8ocJg
+9zW/iBtxDoQf++/Ry+maVL+4u7SCJXf/EfuFiWr/V9ejf4Sp96+sucX+YtOvAkBs
+cVts5aYBHMavsn8HMlOXqbGawRMAMOo62fk9qzx5RpcVKDHDadeoSOnmrIt2Tqdh
+b/Lwffj8vbwvlWVeEUnZAkBwGrhn+BzcfsPLIQfyfC13c1oLJNTZ/ktQdiZIp9Sn
+Zmaf4yBl3NekmgdNQUGdbPBEUl32ukVhOkfOhI2qZcbA
+-----END RSA PRIVATE KEY-----

metadata

@@ -0,0 +1,109 @@
+--- !ruby/object:Gem::Specification
+name: em-apn
+version: !ruby/object:Gem::Version
+  version: 0.0.3
+  prerelease: 
+platform: ruby
+authors:
+- Dave Yeu
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2011-12-19 00:00:00.000000000Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: eventmachine
+  requirement: &70281393623880 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 1.0.0.beta.3
+  type: :runtime
+  prerelease: false
+  version_requirements: *70281393623880
+- !ruby/object:Gem::Dependency
+  name: yajl-ruby
+  requirement: &70281393622200 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 0.8.2
+  type: :runtime
+  prerelease: false
+  version_requirements: *70281393622200
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: &70281393609200 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 2.6.0
+  type: :development
+  prerelease: false
+  version_requirements: *70281393609200
+description: 
+email:
+- daveyeu@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- .rspec
+- Gemfile
+- LICENSE
+- README.md
+- Rakefile
+- certs/.gitignore
+- em-apn.gemspec
+- lib/em-apn.rb
+- lib/em-apn/client.rb
+- lib/em-apn/connection.rb
+- lib/em-apn/error_response.rb
+- lib/em-apn/log_message.rb
+- lib/em-apn/notification.rb
+- lib/em-apn/response.rb
+- lib/em-apn/server.rb
+- lib/em-apn/test_helper.rb
+- lib/em-apn/version.rb
+- script/push
+- script/server
+- script/test
+- spec/em-apn/client_spec.rb
+- spec/em-apn/notification_spec.rb
+- spec/spec_helper.rb
+- spec/support/certs/cert.pem
+- spec/support/certs/key.pem
+homepage: ''
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: em-apn
+rubygems_version: 1.8.10
+signing_key: 
+specification_version: 3
+summary: EventMachine-driven Apple Push Notifications
+test_files:
+- spec/em-apn/client_spec.rb
+- spec/em-apn/notification_spec.rb
+- spec/spec_helper.rb
+- spec/support/certs/cert.pem
+- spec/support/certs/key.pem