sms_twilio.rb in flapjack-1.2.2

- old
+ new

@@ -2,10 +2,12 @@
 
 require 'em-synchrony'
 require 'em-synchrony/em-http'
 require 'active_support/inflector'
 
+require 'flapjack/redis_pool'
+
 require 'flapjack/data/alert'
 require 'flapjack/utility'
 
 module Flapjack
   module Gateways
@@ -15,101 +17,127 @@
       # --data-urlencode 'To=+61414123456'  \
       # --data-urlencode 'From=+61414123456'  \
       # --data-urlencode 'Body=Sausage' \
       # -u [AccountSid]:[AuthToken]
 
-      class << self
+      include Flapjack::Utility
 
-        include Flapjack::Utility
+      def initialize(opts = {})
+        @config = opts[:config]
+        @logger = opts[:logger]
+        @redis_config = opts[:redis_config] || {}
+        @redis = Flapjack::RedisPool.new(:config => @redis_config, :size => 1, :logger => @logger)
 
-        def start
-          @sent = 0
-        end
+        @logger.info("starting")
+        @logger.debug("new sms_twilio gateway pikelet with the following options: #{@config.inspect}")
 
-        def perform(contents)
-          @logger.debug "Woo, got a notification to send out: #{contents.inspect}"
-          alert = Flapjack::Data::Alert.new(contents, :logger => @logger)
+        @sent = 0
+      end
 
-          account_sid = @config["account_sid"]
-          auth_token  = @config["auth_token"]
-          from        = @config["from"]
-          endpoint    = @config["endpoint"] || "https://api.twilio.com/2010-04-01/Accounts/#{account_sid}/Messages.json"
+      def stop
+        @logger.info("stopping")
+        @should_quit = true
 
-          address         = alert.address
-          notification_id = alert.notification_id
-          message_type    = alert.rollup ? 'rollup' : 'alert'
+        redis_uri = @redis_config[:path] ||
+          "redis://#{@redis_config[:host] || '127.0.0.1'}:#{@redis_config[:port] || '6379'}/#{@redis_config[:db] || '0'}"
+        shutdown_redis = EM::Hiredis.connect(redis_uri)
+        shutdown_redis.rpush(@config['queue'], Flapjack.dump_json('notification_type' => 'shutdown'))
+      end
 
-          my_dir = File.dirname(__FILE__)
-          sms_template_path = case
-          when @config.has_key?('templates') && @config['templates']["#{message_type}.text"]
-            @config['templates']["#{message_type}.text"]
-          else
-            my_dir + "/sms_twilio/#{message_type}.text.erb"
-          end
-          sms_template = ERB.new(File.read(sms_template_path), nil, '-')
+      def start
+        queue = @config['queue']
 
-          @alert  = alert
-          bnd     = binding
-
+        until @should_quit
           begin
-            message = sms_template.result(bnd).chomp
+            @logger.debug("sms_twilio gateway is going into blpop mode on #{queue}")
+            deliver( Flapjack::Data::Alert.next(queue, :redis => @redis, :logger => @logger) )
           rescue => e
-            @logger.error "Error while excuting the ERB for an sms: " +
-              "ERB being executed: #{sms_template_path}"
-            raise
+            @logger.error "Error generating or dispatching SMS Twilio message: #{e.class}: #{e.message}\n" +
+              e.backtrace.join("\n")
           end
+        end
+      end
 
-          if @config.nil? || (@config.respond_to?(:empty?) && @config.empty?)
-            @logger.error "sms_twilio config is missing"
-            return
-          end
+      def deliver(alert)
+        account_sid = @config["account_sid"]
+        auth_token  = @config["auth_token"]
+        from        = @config["from"]
+        endpoint    = @config["endpoint"] || "https://api.twilio.com/2010-04-01/Accounts/#{account_sid}/Messages.json"
 
-          errors = []
+        address         = alert.address
+        notification_id = alert.notification_id
+        message_type    = alert.rollup ? 'rollup' : 'alert'
 
-          safe_message = truncate(message, 159)
+        my_dir = File.dirname(__FILE__)
+        sms_template_path = case
+        when @config.has_key?('templates') && @config['templates']["#{message_type}.text"]
+          @config['templates']["#{message_type}.text"]
+        else
+          my_dir + "/sms_twilio/#{message_type}.text.erb"
+        end
+        sms_template = ERB.new(File.read(sms_template_path), nil, '-')
 
-          [[account_sid, "Twilio account_sid is missing"],
-           [auth_token, "Twilio auth_token is missing"],
-           [from,  "SMS from address is missing"],
-           [address,  "SMS address is missing"],
-           [notification_id, "Notification id is missing"]].each do |val_err|
+        @alert  = alert
+        bnd     = binding
 
-            next unless val_err.first.nil? || (val_err.first.respond_to?(:empty?) && val_err.first.empty?)
-            errors << val_err.last
-          end
+        begin
+          message = sms_template.result(bnd).chomp
+        rescue => e
+          @logger.error "Error while excuting the ERB for an sms: " +
+            "ERB being executed: #{sms_template_path}"
+          raise
+        end
 
-          unless errors.empty?
-            errors.each {|err| @logger.error err }
-            return
-          end
+        if @config.nil? || (@config.respond_to?(:empty?) && @config.empty?)
+          @logger.error "sms_twilio config is missing"
+          return
+        end
 
-          body_data = {'To'   => address,
-                   'From' => from,
-                   'Body' => safe_message}
-          @logger.debug "body_data: #{body_data.inspect}"
-          @logger.debug "authorization: [#{account_sid}, #{auth_token[0..2]}...#{auth_token[-3..-1]}]"
+        errors = []
 
-          http = EM::HttpRequest.new(endpoint).post(:body => body_data, :head => {'authorization' => [account_sid, auth_token]})
+        safe_message = truncate(message, 159)
 
-          @logger.debug "server response: #{http.response}"
+        [[account_sid, "Twilio account_sid is missing"],
+         [auth_token, "Twilio auth_token is missing"],
+         [from,  "SMS from address is missing"],
+         [address,  "SMS address is missing"],
+         [notification_id, "Notification id is missing"]].each do |val_err|
 
-          status = (http.nil? || http.response_header.nil?) ? nil : http.response_header.status
-          if (status >= 200) && (status <= 206)
-            @sent += 1
-            alert.record_send_success!
-            @logger.debug "Sent SMS via Twilio, response status is #{status}, " +
-              "notification_id: #{notification_id}"
-          else
-            @logger.error "Failed to send SMS via Twilio, response status is #{status}, " +
-              "notification_id: #{notification_id}"
-          end
-        rescue => e
-          @logger.error "Error generating or delivering sms to #{contents['address']}: #{e.class}: #{e.message}"
-          @logger.error e.backtrace.join("\n")
-          raise
+          next unless val_err.first.nil? || (val_err.first.respond_to?(:empty?) && val_err.first.empty?)
+          errors << val_err.last
         end
 
+        unless errors.empty?
+          errors.each {|err| @logger.error err }
+          return
+        end
+
+        body_data = {'To'   => address,
+                 'From' => from,
+                 'Body' => safe_message}
+        @logger.debug "body_data: #{body_data.inspect}"
+        @logger.debug "authorization: [#{account_sid}, #{auth_token[0..2]}...#{auth_token[-3..-1]}]"
+
+        http = EM::HttpRequest.new(endpoint).post(:body => body_data, :head => {'authorization' => [account_sid, auth_token]})
+
+        @logger.debug "server response: #{http.response}"
+
+        status = (http.nil? || http.response_header.nil?) ? nil : http.response_header.status
+        if (status >= 200) && (status <= 206)
+          @sent += 1
+          alert.record_send_success!
+          @logger.debug "Sent SMS via Twilio, response status is #{status}, " +
+            "notification_id: #{notification_id}"
+        else
+          @logger.error "Failed to send SMS via Twilio, response status is #{status}, " +
+            "notification_id: #{notification_id}"
+        end
+      rescue => e
+        @logger.error "Error generating or delivering sms to #{alert.address}: #{e.class}: #{e.message}"
+        @logger.error e.backtrace.join("\n")
+        raise
       end
+
     end
   end
 end
 