rocket_sms 0.1.1

data/.gitignore

@@ -0,0 +1,19 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp
+
+/vendor/smscsim/sim.*

data/.rspec

@@ -0,0 +1,2 @@
+--color
+--format progress

data/Gemfile

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

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2012 Marcelo Wiermann
+
+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 @@
+# LeanSms
+
+TODO: Write a gem description
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'lean_sms'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install lean_sms
+
+## 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 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/Rakefile

@@ -0,0 +1 @@
+require "bundler/gem_tasks"

data/bin/scheduler_runner.rb

@@ -0,0 +1,11 @@
+#!/usr/bin/env ruby
+
+require "rocket_sms"
+
+redis_url = ENV['REDIS_URL'] || ARGV[0]
+log_location = ENV['LOG_LOCATION'] || ARGV[1] || STDOUT
+
+scheduler = RocketSMS::Scheduler.instance
+scheduler.redis_url = redis_url
+scheduler.log_location = log_location
+scheduler.start

data/bin/transceiver_runner.rb

@@ -0,0 +1,10 @@
+#!/usr/bin/env ruby
+
+require "rocket_sms"
+
+id = ENV['TRANSCEIVER_ID'] || ARGV[0]
+redis_url = ENV['REDIS_URL'] || ARGV[1]
+log_location = ENV['LOG_LOCATION'] || ARGV[2] || STDOUT
+
+transceiver = RocketSMS::Transceiver.new(id, redis_url, log_location)
+transceiver.start

data/examples/gateway.rb

@@ -0,0 +1,7 @@
+require 'rocket_sms'
+
+RocketSMS.configure do |config|
+  config.settings = 'examples/gateway.yml'
+end
+
+RocketSMS.start

data/examples/gateway.yml

@@ -0,0 +1,40 @@
+redis:
+  url: nil
+log:
+  location: STDOUT
+transceivers:
+  bind1:
+    throughput: 50.0
+    connection:
+      host: '0.0.0.0'
+      port: 6000
+      system_id: 'hugo'
+      password: 'ggoohu'
+      system_type: '' 
+      interface_version: 52
+      source_ton: 0
+      source_npi: 1
+      destination_ton: 1
+      destination_npi: 1
+      source_address_range: ''
+      destination_address_range: ''
+      enquire_link_delay_secs: 10
+      registered_delivery: 1
+  bind2:
+    throughput: 5.0
+    connection:
+      host: '0.0.0.0'
+      port: 6001
+      system_id: 'hugo'
+      password: 'ggoohu'
+      system_type: '' 
+      interface_version: 52
+      source_ton: 0
+      source_npi: 1
+      destination_ton: 1
+      destination_npi: 1
+      source_address_range: ''
+      destination_address_range: ''
+      enquire_link_delay_secs: 10
+      registered_delivery: 1
+

data/examples/test.rb

@@ -0,0 +1,60 @@
+# encoding: utf-8
+require 'rubygems'
+require 'json'
+require 'redis'
+require 'securerandom'
+r = Redis.new
+
+r.del('gateway:queues:mt:pending')
+r.del('gateway:queues:mt:retry')
+r.del('gateway:queues:mt:dispatch')
+r.del('gateway:queues:mt:success')
+r.del('gateway:queues:mt:failure')
+
+keys = r.keys('gateway:dids*')
+keys.each{ |k| r.del(k) }
+r.del('gateway:sets:dispatch')
+
+dids = []
+1000000000.upto(1000001000) do |num|
+  did = { number: num, throughput: 1 }
+  r.set("gateway:dids:#{num}", did.to_json )
+  dids << did
+end
+
+t = Proc.new do
+  45.times do |i|
+    msgs = []
+    msgs << {id: "#{SecureRandom.hex(8)}", sender: dids.sample[:number], receiver: '9999999999', body: 'Attention LOs: Rob Huddle will be on vacation from Oct 4th - Oct 12th. In his absence, please contact Rachael Hawk with all urgent matters. Thank you' }
+    body = %q{ @$¥èéùìòÇØøÅå_^{}\[~]|ÆæßÉ !"#¤%&(')*+,-./0123456789:;<=>? ¡ABCDEFGHIJKLMNOPQRSTUVWXYZÄÖÑܧ¿abcdefghijklmnopqrstuvwxyzäöñüà"}
+    msgs << {id: "#{SecureRandom.hex(8)}", sender: dids.sample[:number], receiver: '9999999999', body: body } 
+    msgs.each do |m|
+      score = Time.now.to_i
+      r.zadd('gateway:queues:mt:pending', score, m.to_json)
+    end
+  end
+end
+
+threads = []
+threads << Thread.new do
+  while true do
+    t.call
+    sleep 1
+  end
+end
+
+threads << Thread.new do
+  l = r.llen('gateway:queues:mt:success')
+  while true do
+    nl = r.llen('gateway:queues:mt:success')
+    speed = l - nl
+    puts speed
+    l = nl
+    sleep 1
+  end
+end
+
+threads.each{ |t| t.join }
+
+
+

data/lib/rocket_sms.rb

@@ -0,0 +1,111 @@
+require 'rubygems'
+require 'bundler/setup'
+require 'eventmachine'
+require 'em-hiredis'
+require 'oj'
+require 'multi_json'
+require 'singleton'
+require 'securerandom'
+require 'ostruct'
+require 'forwardable'
+#require 'smpp'
+
+path = File.expand_path(__FILE__).split('/')
+path.delete_at(-1)
+path.delete_at(-1)
+path = path.join('/')
+
+require "#{path}/vendor/ruby-smpp/lib/smpp.rb"
+
+require "rocket_sms/version"
+
+module RocketSMS
+  extend self
+
+  # Disable ruby-smpp logging
+  require 'tempfile'
+  Smpp::Base.logger = Logger.new(Tempfile.new('ruby-smpp').path)
+
+  LIB_PATH = File.dirname(__FILE__) + '/rocket_sms/'
+
+  %w{ gateway did message transceiver scheduler lock }.each do |dep|
+    require LIB_PATH + dep
+  end
+
+  def start
+    @pid = Process.pid
+    #Process.daemon
+    gateway.start
+  end
+
+  def queues
+    @queues ||= {
+      mt: {
+        pending: 'gateway:queues:mt:pending',
+        success: 'gateway:queues:mt:success',
+        failure: 'gateway:queues:mt:failure'
+      },
+      mo: 'gateway:queues:mo:received',
+      dr: 'gateway:queues:dr'
+    }
+  end
+
+  def gateway
+    @gateway ||= RocketSMS::Gateway.instance
+  end
+
+  def redis
+    @redis ||= EM::Hiredis.connect(redis_url)
+  end
+
+  # Configuration and Setup
+  def configure
+    yield self
+  end
+
+  def settings=(yaml_file_location)
+    @settings = symbolize_keys(YAML.load(IO.read(yaml_file_location)))
+    redis_url = @settings[:redis] && @settings[:redis][:url]
+    log_location = @settings[:log] && @settings[:log][:location]
+  end
+
+  def settings
+    @settings
+  end
+
+  def redis_url
+    @redis_url ||= "redis://localhost:6379"
+  end
+
+  def redis_url=(url)
+    @redis_url = url unless url.nil?
+  end
+
+  def logger
+    @logger ||= Logger.new(log_location)
+  end
+
+  def log_location
+    @log_location ||= STDOUT
+  end
+
+  def log_location=(location)
+    @log_location = location unless location.nil?
+  end
+
+  def symbolize_keys(hash)
+    hash.inject({}){|result, (key, value)|
+      new_key = case key
+                when String then key.to_sym
+                else key
+                end
+      new_value = case value
+                  when Hash then symbolize_keys(value)
+                  else value
+                  end
+      result[new_key] = new_value
+      result
+    }
+  end
+
+end

data/lib/rocket_sms/did.rb

@@ -0,0 +1,25 @@
+module RocketSMS
+  class Did
+
+    attr_reader :params
+
+    def initialize(params)
+      @params = OpenStruct.new(params)
+    end
+
+    def to_json
+      MultiJson.dump(@params.marshal_dump)
+    end
+
+    def self.from_json(json)
+      params = MultiJson.load(json, symbolize_keys: true)
+      did = Did.new(params)
+      return did
+    end
+
+    def method_missing(sym, *args, &block)
+      @params.send(sym, *args, &block)
+    end
+
+  end
+end

data/lib/rocket_sms/gateway.rb

@@ -0,0 +1,123 @@
+module RocketSMS
+
+  class Gateway
+    include Singleton
+    extend Forwardable
+
+    def_delegators :RocketSMS, :settings, :redis, :logger, :redis_url, :log_location
+
+    def initialize
+      @scheduler = {}
+      @transceivers = {}
+      @path = Gem::Specification.find_by_name('rocket_sms').gem_dir
+    end
+
+    def log(msg, level = 'info')
+      if EM.reactor_running?
+        EM.defer{ logger.send(level, msg) }
+      else
+        logger.send(level, msg)
+      end
+    end
+
+    def start
+      EM.run do
+        log "Starting Gateway"
+        startup
+        # Trap exit-related signals
+        Signal.trap("INT") { |signal| stop(signal) }
+        Signal.trap("TERM") { |signal| stop(signal) }
+      end
+    end
+
+    def stop(signal = nil)
+      if @kill
+        log "Forcing Exit. Check your data for losses."
+        shutdown
+      else
+        log "Stopping. Waiting 5 seconds for pending operations to finish."
+        @kill = true
+        @active = false
+        if @scheduler[:pid]
+          Process.kill('TERM', @scheduler[:pid]) rescue nil
+          Process.wait(@scheduler[:pid]) 
+        end
+        if @transceivers
+          @transceivers.each_value do |t|
+            if t[:pid] 
+              Process.kill('TERM', t[:pid]) rescue nil
+              Process.wait(t[:pid])
+            end
+          end
+        end
+        EM::Timer.new(5){ shutdown }
+      end
+    end
+
+    def shutdown
+      if @scheduler[:pid]
+        Process.kill('TERM', @scheduler[:pid]) rescue nil
+      end
+      if @transceivers
+        @transceivers.each_value do |t|
+          if t[:pid]
+            Process.kill('TERM', t[:pid]) rescue nil
+          end
+        end
+      end
+      log "Gateway DOWN."
+      EM.stop
+    end
+
+    def startup
+      clean_up_stale_transceivers
+    end
+
+    def clean_up_stale_transceivers
+      redis.keys("gateway:transceivers:*") do |keys|
+        op = Proc.new do |key, iter|
+          redis.del(key) do |resp|
+            iter.next
+          end
+        end
+        cb = Proc.new do |responses|
+          setup_transceivers
+        end
+        EM::Iterator.new(keys).each(op,cb)
+      end
+    end
+
+    def setup_transceivers
+      # Clean up stale transceivers
+      op = Proc.new do |tid, iter|
+        tsettings = settings[:transceivers][tid]
+        redis.multi
+        redis.hset("gateway:transceivers:#{tid}", "throughput", tsettings[:throughput])
+        redis.hset("gateway:transceivers:#{tid}", "connection", MultiJson.dump(tsettings[:connection]))
+        redis.exec do |resp|
+          iter.next
+        end
+      end
+      cb = Proc.new do |responses|
+        start_scheduler
+        start_transceivers
+      end
+      EM::Iterator.new(settings[:transceivers].keys).each(op,cb)
+    end
+
+    def start_scheduler
+      cmd = "bundle exec ruby #{@path}/bin/scheduler_runner.rb"
+      @scheduler[:pid] = Process.spawn({ "REDIS_URL" => redis_url, "LOG_LOCATION" => (log_location == STDOUT ? nil : log_location) }, cmd)
+    end
+
+    def start_transceivers
+      settings[:transceivers].each do |tid, settings|
+        cmd = "bundle exec ruby #{@path}/bin/transceiver_runner.rb"
+        @transceivers[tid] = {}
+        @transceivers[tid][:pid] = Process.spawn({ "TRANSCEIVER_ID" => tid.to_s ,"REDIS_URL" => redis_url, "LOG_LOCATION" => (log_location == STDOUT ? nil : log_location) }, cmd)
+      end
+    end
+
+  end
+
+end