slack.rb in flapjack-2.0.0b1

- old
+ new
@@ -1,124 +1,132 @@
 #!/usr/bin/env ruby
 
-require 'em-synchrony'
-require 'em-synchrony/em-http'
-require 'active_support/inflector'
+require 'erb'
 
-require 'flapjack/redis_pool'
+require 'flapjack/redis_proxy'
+require 'flapjack/record_queue'
+require 'flapjack/utility'
+require 'flapjack/exceptions'
 
 require 'flapjack/data/alert'
-require 'flapjack/utility'
 
 module Flapjack
   module Gateways
     class Slack
 
       include Flapjack::Utility
 
+      attr_accessor :sent
+
       def initialize(opts = {})
+        @lock = opts[:lock]
         @config = opts[:config]
-        @logger = opts[:logger]
-        @redis_config = opts[:redis_config] || {}
-        @redis = Flapjack::RedisPool.new(:config => @redis_config, :size => 1, :logger => @logger)
 
-        @logger.info("starting")
-        @logger.debug("new slack gateway pikelet with the following options: #{@config.inspect}")
+        # TODO support for config reloading
+        @queue = Flapjack::RecordQueue.new(@config['queue'] || 'slack_notifications',
+                   Flapjack::Data::Alert)
 
         @sent = 0
       end
 
-      def stop
-        @logger.info("stopping")
-        @should_quit = true
+      def start
+        Flapjack.logger.debug("new slack gateway pikelet with the following options: #{@config.inspect}")
 
-        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
+        begin
+          Zermelo.redis = Flapjack.redis
 
-      def start
-        queue = @config['queue']
+          loop do
+            @lock.synchronize do
+              @queue.foreach {|alert| handle_alert(alert) }
+            end
 
-        until @should_quit
-          begin
-            @logger.debug("slack gateway is going into blpop mode on #{queue}")
-            alert = Flapjack::Data::Alert.next(queue, :redis => @redis, :logger => @logger)
-            deliver(alert) unless alert.nil?
-          rescue => e
-            @logger.error "Error generating or dispatching Slack message: #{e.class}: #{e.message}\n" +
-              e.backtrace.join("\n")
+            @queue.wait
           end
+        ensure
+          Flapjack.redis.quit
         end
       end
 
-      def deliver(alert)
-        account_sid = @config["account_sid"]
-        endpoint    = @config["endpoint"]
-        icon_emoji  = @config["icon_emoji"] || ':ghost:'
+      def stop_type
+        :exception
+      end
 
-        channel         = "##{alert.address}"
+      private
+
+      def handle_alert(alert)
+        Flapjack.logger.debug "Woo, got an alert to send out: #{alert.inspect}"
+
+        account_sid = @config['account_sid']
+        endpoint    = @config['endpoint']
+        icon_emoji  = @config['icon_emoji'] || ':ghost:'
+
+        channel         = "##{alert.medium.address}"
         channel         = '#general' if (channel.size == 1)
         notification_id = alert.notification_id
         message_type    = alert.rollup ? 'rollup' : 'alert'
 
         slack_template_erb, slack_template =
-          load_template(@config['templates'], message_type, 'text',
-                        File.join(File.dirname(__FILE__), 'slack'))
+          load_template(@config['templates'], message_type,
+                        'text', File.join(File.dirname(__FILE__), 'slack'))
 
         @alert  = alert
         bnd     = binding
 
+        message = nil
         begin
           message = slack_template_erb.result(bnd).chomp
         rescue => e
-          @logger.error "Error while executing the ERB for an sms: " +
+          Flapjack.logger.error 'Error while executing the ERB for a slack message: ' \
             "ERB being executed: #{slack_template}"
           raise
         end
 
         errors = []
 
         [
-         [endpoint, "Slack endpoint is missing"],
-         [account_sid, "Slack account_sid is missing"]
+         [endpoint, 'Slack endpoint is missing'],
+         [account_sid, 'Slack account_sid is missing']
         ].each do |val_err|
 
           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 }
+          errors.each {|err| Flapjack.logger.error err }
           return
         end
 
         payload = Flapjack.dump_json(
           'channel'    => channel,
           'username'   => account_sid,
           'text'       => message,
           'icon_emoji' => icon_emoji
         )
-        @logger.debug "payload: #{payload.inspect}"
+        Flapjack.logger.debug "payload: #{payload.inspect}"
 
-        http = EM::HttpRequest.new("#{endpoint}").post(:body => {'payload' => payload})
+        uri = URI.parse(endpoint)
 
-        @logger.debug "server response: #{http.response}"
+        http = Net::HTTP.new(uri.host, uri.port)
+        http.use_ssl = true
+        http.verify_mode = OpenSSL::SSL::VERIFY_PEER
+        request = Net::HTTP::Post.new(uri.request_uri)
+        request.body = Flapjack.dump_json(:payload => payload)
+        http_response = http.request(request)
+        status   = http_response.code
 
-        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 message via Slack, response status is #{status}, " +
+          Flapjack.logger.debug "Sent message via Slack, response status is #{status}, " +
             "notification_id: #{notification_id}"
         else
-          @logger.error "Failed to send message via Slack, response status is #{status}, " +
+          Flapjack.logger.error "Failed to send message via Slack, 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")
+        Flapjack.logger.error "Error generating or delivering Slack message to #{alert.medium.address}: #{e.class}: #{e.message}"
+        Flapjack.logger.error e.backtrace.join("\n")
         raise
       end
 
     end
   end