bpoweski-apnserver 0.0.2 → 0.0.6

data/History.txt

@@ -0,0 +1,4 @@
+=== 0.0.1 2009-09-10
+
+* 1 major enhancement:
+  * Initial release

data/Manifest.txt

@@ -0,0 +1,23 @@
+History.txt
+Manifest.txt
+PostInstall.txt
+README
+README.rdoc
+Rakefile
+bin/apns
+bin/apnserverd
+lib/apnserver.rb
+lib/apnserver/client.rb
+lib/apnserver/notification.rb
+lib/apnserver/payload.rb
+lib/apnserver/server.rb
+script/console
+script/console.cmd
+script/destroy
+script/destroy.cmd
+script/generate
+script/generate.cmd
+test/notification_test.rb
+test/payload_test.rb
+test/test_apnserver.rb
+test/test_helper.rb

data/Rakefile

@@ -0,0 +1,29 @@
+require 'rubygems'
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gemspec|
+    gemspec.name = "apnserver"
+    gemspec.summary = "Apple Push Notification Server"
+    gemspec.description = "A toolkit for proxying and sending Apple Push Notifications"
+    gemspec.email = "bpoweski@3factors.com"
+    gemspec.homepage = "http://github.com/bpoweski/apnserver"
+    gemspec.authors = ["Ben Poweski"]
+    gemspec.add_dependency 'eventmachine'
+    gemspec.add_dependency 'daemons'
+    gemspec.files = FileList['lib/**/*.rb', 'bin/*', '[A-Z]*', 'test/**/*'].to_a
+  end
+rescue LoadError
+  puts "Jeweler not available. Install it with: sudo gem install technicalpickles-jeweler -s http://gems.github.com"
+end
+
+require 'rake/testtask'
+Rake::TestTask.new(:test) do |test|
+  test.test_files = FileList.new('test/**/test_*.rb') do |list|
+    list.exclude 'test/test_helper.rb'
+  end
+  test.libs << 'test'
+  test.verbose = true
+end
+
+task :default => [:test]

data/VERSION

@@ -0,0 +1 @@
+0.0.6

data/apnserver.gemspec

@@ -0,0 +1,70 @@
+# Generated by jeweler
+# DO NOT EDIT THIS FILE
+# Instead, edit Jeweler::Tasks in Rakefile, and run `rake gemspec`
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{apnserver}
+  s.version = "0.0.6"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Ben Poweski"]
+  s.date = %q{2009-09-14}
+  s.description = %q{A toolkit for proxying and sending Apple Push Notifications}
+  s.email = %q{bpoweski@3factors.com}
+  s.executables = ["apns", "apnsend", "apnserverd"]
+  s.extra_rdoc_files = [
+    "README.rdoc"
+  ]
+  s.files = [
+    "History.txt",
+     "Manifest.txt",
+     "README.rdoc",
+     "Rakefile",
+     "VERSION",
+     "apnserver.gemspec",
+     "bin/apns",
+     "bin/apnsend",
+     "bin/apnserverd",
+     "lib/apnserver.rb",
+     "lib/apnserver/client.rb",
+     "lib/apnserver/notification.rb",
+     "lib/apnserver/payload.rb",
+     "lib/apnserver/protocol.rb",
+     "lib/apnserver/server.rb",
+     "lib/apnserver/server_connection.rb",
+     "test/test_client.rb",
+     "test/test_helper.rb",
+     "test/test_notification.rb",
+     "test/test_payload.rb",
+     "test/test_protocol.rb"
+  ]
+  s.homepage = %q{http://github.com/bpoweski/apnserver}
+  s.rdoc_options = ["--charset=UTF-8"]
+  s.require_paths = ["lib"]
+  s.rubygems_version = %q{1.3.5}
+  s.summary = %q{Apple Push Notification Server}
+  s.test_files = [
+    "test/test_client.rb",
+     "test/test_helper.rb",
+     "test/test_notification.rb",
+     "test/test_payload.rb",
+     "test/test_protocol.rb"
+  ]
+
+  if s.respond_to? :specification_version then
+    current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::RubyGemsVersion) >= Gem::Version.new('1.2.0') then
+      s.add_runtime_dependency(%q<eventmachine>, [">= 0"])
+      s.add_runtime_dependency(%q<daemons>, [">= 0"])
+    else
+      s.add_dependency(%q<eventmachine>, [">= 0"])
+      s.add_dependency(%q<daemons>, [">= 0"])
+    end
+  else
+    s.add_dependency(%q<eventmachine>, [">= 0"])
+    s.add_dependency(%q<daemons>, [">= 0"])
+  end
+end

data/bin/apns

@@ -1,4 +1,6 @@
 #!/usr/bin/env ruby
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+
 require 'socket'
 require 'base64'
 require 'rubygems'

data/bin/apnsend

@@ -1,4 +1,6 @@
 #!/usr/bin/env ruby
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+
 require 'getoptlong'
 require 'rubygems'
 require 'apnserver'

data/bin/apnserverd

@@ -1,8 +1,10 @@
 #!/usr/bin/env ruby
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+
 require 'getoptlong'
 require 'rubygems'
-require 'apnserver'
 require 'daemons'
+require 'apnserver'
 
 def usage
   puts "Usage: apnserverd [switches] --pem <path>"
@@ -47,10 +49,15 @@ end
 
 if pem.nil?
   usage
+  exit 1
 else
-  # Become a daemon
-  Daemons.daemonize
-
+  Daemonize.daemonize('apnserverd', 'apnserverd')
+  @pid_file = '/var/run/apnserverd.pid'
+  open(@pid_file,"w") {|f| f.write(Process.pid) } # copied from mongrel
+  open(@pid_file,"w") do |f|
+	  f.write(Process.pid)
+      File.chmod(0644, @pid_file)
+  end 
   server = ApnServer::Server.new(pem, bind_address, proxy_port)
   server.start!
 end

data/lib/apnserver.rb

@@ -0,0 +1,7 @@
+require 'eventmachine'
+require 'apnserver/payload'
+require 'apnserver/notification'
+require 'apnserver/protocol'
+require 'apnserver/client'
+require 'apnserver/server_connection'
+require 'apnserver/server'

data/lib/apnserver/client.rb

@@ -0,0 +1,42 @@
+require 'openssl'
+require 'socket'
+
+module ApnServer
+  class Client
+    
+    attr_accessor :pem, :host, :port, :password
+    
+    def initialize(pem, host = 'gateway.push.apple.com', port = 2195, pass = nil)
+      @pem, @host, @port, @password = pem, host, port, pass
+    end
+    
+    def connect!
+      raise "The path to your pem file is not set." unless self.pem
+      raise "The path to your pem file does not exist!" unless File.exist?(self.pem)
+      
+      @context      = OpenSSL::SSL::SSLContext.new
+      @context.cert = OpenSSL::X509::Certificate.new(File.read(self.pem))
+      @context.key  = OpenSSL::PKey::RSA.new(File.read(self.pem), self.password)
+      
+      @sock         = TCPSocket.new(self.host, self.port)
+      @ssl          = OpenSSL::SSL::SSLSocket.new(@sock, @context)
+      @ssl.connect
+      
+      return @sock, @ssl
+    end
+    
+    def disconnect!
+      @ssl.close
+      @sock.close
+    end
+    
+    def write(notification)
+      puts "#{Time.now} [#{host}:#{port}] sending #{notification.alert}"
+      @ssl.write(notification.to_bytes)
+    end
+    
+    def connected?
+      @ssl
+    end
+  end
+end

data/lib/apnserver/notification.rb

@@ -0,0 +1,63 @@
+require 'apnserver/payload'
+require 'json'
+
+module ApnServer
+  
+  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 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

data/lib/apnserver/payload.rb

@@ -0,0 +1,23 @@
+module ApnServer
+  
+  module Payload
+    
+    class PayloadInvalid < RuntimeError
+    end
+    
+    def create_payload(payload)
+      case payload
+        when String then { :aps => { :alert =>  payload } }
+        when Hash then create_payload_from_hash(payload)
+      end
+    end
+    
+    def create_payload_from_hash(payload)
+      custom = payload.delete(:custom)
+      aps = {:aps => payload }
+      aps.merge!(custom) if custom
+      aps
+    end
+  end
+  
+end

data/lib/apnserver/protocol.rb

@@ -0,0 +1,21 @@
+module ApnServer
+  module Protocol
+    
+    def post_init
+      @address = Socket.unpack_sockaddr_in(self.get_peername)
+      puts "#{Time.now} [#{address.last}:#{address.first}] CONNECT"
+    end
+    
+    def unbind
+      puts "#{Time.now} [#{address.last}:#{address.first}] DISCONNECT"    
+    end
+    
+    def receive_data(data)
+      (@buf ||= "") << data
+      if notification = ApnServer::Notification.valid?(@buf)
+        queue.push(notification)
+      end
+    end
+    
+  end
+end

data/lib/apnserver/server.rb

@@ -0,0 +1,34 @@
+module ApnServer
+  
+  class Server
+    attr_accessor :client, :bind_address, :port
+    
+    def initialize(pem, bind_address = '0.0.0.0', port = 22195)
+      @queue = EM::Queue.new
+      @client = ApnServer::Client.new(pem)
+      @bind_address, @port = bind_address, port
+    end
+    
+    def start!
+      EventMachine::run do
+        puts "#{Time.now} Starting APN Server on #{bind_address}:#{port}"
+        
+        EM.start_server(bind_address, port, ApnServer::ServerConnection) do |s|
+          s.queue = @queue
+        end 
+         
+        EventMachine::PeriodicTimer.new(1) do
+          unless @queue.empty?
+            size = @queue.size
+            size.times do 
+              @queue.pop do |notification|
+                @client.connect! unless @client.connected?
+                @client.write(notification)
+              end
+            end
+          end
+        end
+      end   
+    end
+  end
+end

data/lib/apnserver/server_connection.rb

@@ -0,0 +1,9 @@
+require 'socket'
+require 'apnserver/protocol'
+
+module ApnServer
+  class ServerConnection < EventMachine::Connection
+    include Protocol
+    attr_accessor :queue, :address
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: bpoweski-apnserver
 version: !ruby/object:Gem::Version 
-  version: 0.0.2
+  version: 0.0.6
 platform: ruby
 authors: 
 - Ben Poweski
@@ -43,10 +43,29 @@ extensions: []
 extra_rdoc_files: 
 - README.rdoc
 files: 
+- History.txt
+- Manifest.txt
 - README.rdoc
-has_rdoc: true
+- Rakefile
+- VERSION
+- apnserver.gemspec
+- bin/apns
+- bin/apnsend
+- bin/apnserverd
+- lib/apnserver.rb
+- lib/apnserver/client.rb
+- lib/apnserver/notification.rb
+- lib/apnserver/payload.rb
+- lib/apnserver/protocol.rb
+- lib/apnserver/server.rb
+- lib/apnserver/server_connection.rb
+- test/test_client.rb
+- test/test_helper.rb
+- test/test_notification.rb
+- test/test_payload.rb
+- test/test_protocol.rb
+has_rdoc: false
 homepage: http://github.com/bpoweski/apnserver
-licenses: 
 post_install_message: 
 rdoc_options: 
 - --charset=UTF-8
@@ -67,7 +86,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
 requirements: []
 
 rubyforge_project: 
-rubygems_version: 1.3.5
+rubygems_version: 1.2.0
 signing_key: 
 specification_version: 3
 summary: Apple Push Notification Server