em-apns 0.1.0

data/.gitignore

@@ -0,0 +1,23 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp
+Gemfile.lock
+pkg/*
+.rvmrc
+*.sock
+.project
+.DS_Store

data/.rspec

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

data/Gemfile

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

data/LICENSE

@@ -0,0 +1,22 @@
+Copyright (c) 2012 Yuri Barbashov
+
+MIT License
+
+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,29 @@
+# Em::Apns
+
+TODO: Write a gem description
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'em-apns'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install em-apns
+
+## Usage
+
+TODO: Write usage instructions here
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Added some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/Rakefile

@@ -0,0 +1,2 @@
+#!/usr/bin/env rake
+require "bundler/gem_tasks"

data/bin/em-apns

@@ -0,0 +1,7 @@
+#!/usr/bin/env ruby
+$LOAD_PATH.unshift File.expand_path(File.dirname(__FILE__) + '/../lib')
+require 'em-apns'
+require 'em-apns/server_runner'
+
+EM::APNS::ServerRunner.new(ARGV)
+

data/em-apns.gemspec

@@ -0,0 +1,22 @@
+# -*- encoding: utf-8 -*-
+require File.expand_path('../lib/em-apns/version', __FILE__)
+
+Gem::Specification.new do |gem|
+  gem.authors       = ["Yuri Barbashov"]
+  gem.email         = ["lolcoltd@gmail.com"]
+  gem.description   = %q{EventMachine-driven Apple Push Notifications Sender daemon}
+  gem.summary       = %q{EventMachine-driven Apple Push Notifications Sender daemon}
+  gem.homepage      = ""
+
+  gem.files         = `git ls-files`.split($\)
+  gem.executables   = gem.files.grep(%r{^bin/}).map{ |f| File.basename(f) }
+  gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
+  gem.name          = "em-apns"
+  gem.require_paths = ["lib"]
+  gem.version       = EM::APNS::VERSION
+  
+  gem.add_dependency "eventmachine", ">= 1.0.0.beta.3"
+  gem.add_dependency "json"
+
+  gem.add_development_dependency "rspec", "~> 2.6.0"
+end

data/lib/em-apns.rb

@@ -0,0 +1,67 @@
+require 'eventmachine'
+require 'socket'
+require 'logger'
+require 'json'
+require 'yaml'
+require 'em-apns/notification'
+require 'em-apns/error'
+require 'em-apns/connection'
+require 'em-apns/connection_pool'
+require 'em-apns/server'
+require "em-apns/version"
+
+module EM::APNS
+  class << self
+    attr_writer :sock, :logger
+    attr_accessor :key, :cert, :pool, :env, :host
+        
+    def sock
+      @sock ||= (defined?(Rails) && Rails.respond_to?(:root) ? File.join(Rails.root, 'tmp/sockets', "em-apns-#{Rails.env}.sock") : nil)
+    end    
+        
+    def logger
+      @logger ||= Logger.new(STDOUT)
+    end
+    
+    def send_notification(*args)
+      send_notifications([Notification.new(*args)])
+    end
+    
+    def send_notifications(notifications)
+      UNIXSocket.open(@sock) do |socket|
+        notifications.each do |n|
+          socket.puts(n.data)
+        end
+      end
+    rescue => e
+      logger.error(e)
+    end
+      
+    def config options
+      raise 'App root folder are not defined' unless options[:root]
+      if options[:config]
+        configurations = YAML.load_file options['config']
+      else
+        config_file = File.join(options[:root],"config/em-apns.yml")
+        configurations = YAML.load_file(config_file)
+      end
+      @env = options[:environment] || 'development'
+      @sock = File.join(options[:root], "tmp/sockets/em-apns-#{@env}.sock")
+      @pool = configurations['pool']
+      @key = configurations['key']
+      @cert = configurations['cert']
+      raise 'No key or certficate file' unless @key && @cert
+      @host = case @env
+      when 'development'
+        "gateway.sandbox.push.apple.com"
+      when 'production'
+        "gateway.push.apple.com"
+      else
+        "localhost"
+      end
+    end
+  
+  end
+  
+end
+

data/lib/em-apns/connection.rb

@@ -0,0 +1,33 @@
+module EM::APNS
+  class Connection < EM::Connection
+    attr_accessor :num
+    
+    def post_init
+      start_tls(
+        :private_key_file => EM::APNS.key,
+        :cert_chain_file  => EM::APNS.cert,
+        :verify_peer      => false
+      )
+    end
+
+    def connection_completed
+      EM::APNS.logger.info("Connection established")
+    end
+
+    def receive_data(data)
+      data_array = data.unpack("ccN")
+      error_response = Error.new(*data_array)
+      EM::APNS.logger.info("#{error_response}")
+    end
+    
+    def unbind
+      EM::APNS.logger.info("Connection terminated")
+      @unbind.call(self) if @unbind
+    end
+    
+    def on_unbind &block
+      @unbind = block
+    end
+    
+  end
+end

data/lib/em-apns/connection_pool.rb

@@ -0,0 +1,34 @@
+module EM::APNS
+  
+  #simple connection pool using EM queue, default size 4
+  class ConnectionPool
+    attr_reader :queue
+   
+    def initialize
+      @pool_size = EM::APNS.pool || 4
+      @connections = []
+      @queue = EM::Queue.new
+      @pool_size.times { add_connection }
+    end
+    
+    def queue_worker_loop
+      proc{ |connection|
+        @queue.pop do |notification|
+          connection.send_data(notification)
+          EM.next_tick { queue_worker_loop.call connection }
+        end
+      }
+    end
+    
+    def add_connection
+      connection = EM.connect(EM::APNS.host, 2195, Connection)
+      connection.on_unbind do |conn|
+        @connections.delete(conn)
+        EM.next_tick{ add_connection }
+      end
+      @connections << connection
+      queue_worker_loop.call connection
+    end
+    
+  end
+end

data/lib/em-apns/error.rb

@@ -0,0 +1,34 @@
+module EM
+  module APNS
+    class Error
+      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-apns/notification.rb

@@ -0,0 +1,59 @@
+# encoding: UTF-8
+
+module EventMachine
+  module APNS
+    class Notification
+      DATA_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
+        @custom.merge(:aps => @aps).to_json
+      end
+
+      # Documentation about this format is here:
+      # http://developer.apple.com/library/ios/#documentation/NetworkingInternet/Conceptual/RemoteNotificationsPG/CommunicatingWIthAPS/CommunicatingWIthAPS.html
+      def data
+        identifier = @identifier || 0
+        expiry     = @expiry || 0
+        size = [payload].pack("a*").size
+        data_array = [1, identifier, expiry, 32, token, size, payload]
+        data_array.pack("cNNnH*na*")
+      end
+
+      def validate!
+        if data.size > DATA_MAX_BYTES
+          error = "max is #{DATA_MAX_BYTES} bytes, but got #{data.size}: #{payload.inspect}"
+          raise PayloadTooLarge.new(error)
+        else
+          true
+        end
+      end
+
+      def identifier=(new_identifier)
+        @identifier = new_identifier.to_i
+      end
+
+      def truncate_alert!
+        while data.size > DATA_MAX_BYTES && !@aps["alert"].nil? && @aps["alert"].size > 0
+          @aps["alert"] = @aps["alert"][0..-2]
+        end
+      end
+
+    end
+  end
+end

data/lib/em-apns/server.rb

@@ -0,0 +1,15 @@
+module EM::APNS
+  module Server
+    def post_init
+      @@queue ||= ConnectionPool.new.queue 
+      @buf ||= ""
+    end
+  
+    def receive_data(data)
+      @buf << data
+      while line = @buf.slice!(/.+\r?\n/) do
+        @@queue.push(line.gsub(/\r?\n/, '')) 
+      end
+    end
+  end
+end

data/lib/em-apns/server_runner.rb

@@ -0,0 +1,89 @@
+module EM::APNS
+  class ServerRunner
+
+    def initialize(args)
+      require 'rubygems'
+      require 'optparse'
+      require 'logger'
+      
+      @options = { environment: 'development', root: File.expand_path('.') }
+      
+      optparse = OptionParser.new do |opts|
+        opts.banner = "Usage: em-apns [options] start|stop"
+
+        opts.on('-h', '--help', 'Show this message') do
+          puts opts
+          exit 1
+        end
+        
+        opts.on('-e', '--environment=NAME', 'Specifies the environment to run this apn sender under ([development]/production/test).') do |e|
+          @options[:environment] = e
+        end
+
+        opts.on('-p', '--pid_file=NAME', 'Specifies the pid file path.') do |path|
+          @pid_file = path
+        end
+
+        opts.on('-d', '--daemonize', 'Daemonize em-apnsender') do
+          @daemonize = true
+        end
+
+        opts.on('-c', '--config=CONFIG', 'Full path to configuration file.') do |path|
+          @options[:config] = path
+        end
+
+      end
+      
+      # If no arguments, give help screen
+      @args = optparse.parse!(args.empty? ? ['-h'] : args)
+      
+      case args[0]
+      when 'start'
+        start
+      when 'stop'
+        stop
+      end
+    end
+  
+    def start
+      if @daemonize
+        require 'daemons'
+        Daemons.run_proc("em-apns-#{@options[:environment]}", :dir => "#{@options[:root]}/tmp/pids", :dir_mode => :normal, :ARGV => @args) do |*args|
+          EM::APNS.logger = Logger.new(File.join(@options[:root], 'log', "em-apns-#{@options[:environment]}.log"))
+          run
+        end
+      else
+        run
+      end
+    end
+    
+    def stop
+      @pid_file ||= "#{@options[:root]}/tmp/pids/em-apns-#{@options[:environment]}.pid"
+      pid = File.open(@pid_file) {|h| h.read}.to_i
+      Process.kill('TERM', pid) 
+    rescue Errno => e
+      require 'fileutils'
+      puts "#{e} #{pid}"
+      puts "deleting pid-file..."
+      FileUtils.rm( f ) 
+    end
+    
+    def run
+      EM::APNS.config(@options)
+      file = EM::APNS.sock
+      File.unlink(file) if File.exists?(file)
+      EventMachine::run {
+        Signal.trap("INT") { EM::APNS.logger.fatal("Terminated"); EM.stop }
+        Signal.trap("TERM") { EM::APNS.logger.fatal("Terminated"); EM.stop }
+        EM::APNS.logger.info "Started"
+        EventMachine::start_unix_domain_server(file, EM::APNS::Server )
+      }
+    rescue => e
+      STDERR.puts e.message
+      EM::APNS.logger.fatal(e) if EM::APNS.logger.respond_to?(:fatal)
+      exit 1
+    end
+    
+  end
+end
+