lib/bender/main.rb in bender-bot-0.2.4 vs lib/bender/main.rb in bender-bot-0.3.0

@@ -4,16 +4,20 @@
 require 'tilt/erb'
 require 'queryparams'
 
 require_relative 'metadata'
 require_relative 'mjolnir'
+require_relative 'helpers'
 require_relative 'web'
 require_relative 'bot'
 
+Thread.abort_on_exception = true
 
+
 module Bender
   class Main < Mjolnir
+    include Helpers
 
     desc 'version', 'Echo the application version'
     def version
       puts VERSION
     end
@@ -45,10 +49,20 @@
     option :hipchat_token, \
       type: :string,
       aliases: %w[ -t ],
       desc: 'Set HipChat v1 API token',
       required: true
+    option :hipchat_v2_token, \
+      type: :string,
+      aliases: %w[ -c ],
+      desc: 'Set HipChat v2 API token',
+      required: true
+    option :primary_room_id, \
+      type: :string,
+      aliases: %w[ -i ],
+      desc: 'Set HipChat primary room ID',
+      required: true
     option :jid, \
       type: :string,
       aliases: %w[ -j ],
       desc: 'Set HipChat JID',
       required: true
@@ -75,15 +89,10 @@
     option :database, \
       type: :string,
       aliases: %w[ -d ],
       desc: 'Set path to application database',
       required: true
-    option :rooms, \
-      type: :string,
-      aliases: %w[ -r ],
-      desc: 'Set HipChat rooms (comma-separated)',
-      required: true
     option :jira_user, \
       type: :string,
       aliases: %w[ -U ],
       desc: 'Set JIRA username',
       required: true
@@ -105,19 +114,25 @@
     option :jira_type, \
       type: :string,
       aliases: %w[ -T ],
       desc: 'Set JIRA issue type',
       required: true
-    option :refresh, \
+    option :user_refresh, \
       type: :numeric,
       aliases: %w[ -R ],
-      desc: 'Set JIRA refresh rate',
+      desc: 'Set JIRA user refresh rate',
       default: 300
+    option :issue_refresh, \
+      type: :numeric,
+      aliases: %w[ -S ],
+      desc: 'Set JIRA issue refresh rate',
+      default: 5
     include_common_options
     def start
       bot = start_bot
-      refresh_users bot
+      periodically_refresh_users bot
+      periodically_refresh_incidents bot
       serve_web bot
     end
 
 
 
@@ -139,11 +154,82 @@
 
       Bot.run! options
     end
 
 
-    def refresh_users bot
+    def set_room_name_and_topic room_id, incidents, hipchat, bot
+      room = hipchat[room_id]
+      new_room = room.get_room
+
+      open_incidents = incidents.select do |i|
+        status = normalize_value i['fields']['status']
+        !(status =~ /resolved|closed/i)
+      end
+
+      @room_name  ||= bot.store['primary_room_name']  || new_room['name']
+      @room_topic ||= bot.store['primary_room_topic'] || new_room['topic']
+
+      @open = false unless defined? @open
+
+      log.info \
+        primary_room_name: bot.store['primary_room_name'],
+        primary_room_topic: bot.store['primary_room_topic'],
+        new_room_name: new_room['name'],
+        new_room_topic: new_room['topic'],
+        room_name: @room_name,
+        room_topic: @room_topic,
+        open: @open
+
+      if open_incidents.empty?
+        if @open
+          new_room['name'] = @room_name
+          new_room['topic'] = @room_topic
+          room.update_room(new_room)
+        end
+
+        @open = false
+
+      else
+        unless @open
+          @room_name = new_room['name']
+          @room_topic = new_room['topic']
+          bot.store['primary_room_name'] = @room_name
+          bot.store['primary_room_topic'] = @room_topic
+          new_room['name'] = 'DANGER WILL ROBINSON'
+          new_room['topic'] = '%d open incicents!' % open_incidents.size
+          room.update_room(new_room)
+        end
+
+        @open = true
+      end
+    end
+
+
+    def periodically_refresh_incidents bot
+      Thread.new do
+        hipchat_v2 = HipChat::Client.new \
+          options.hipchat_v2_token, api_version: 'v2'
+
+        hipchat_v1 = HipChat::Client.new \
+          options.hipchat_token, api_version: 'v1'
+
+        room_id = options.primary_room_id
+
+        loop do
+          is = refresh_incidents bot
+          begin
+            set_room_name_and_topic room_id, is, hipchat_v2, bot
+            sleep options.issue_refresh
+          rescue NoMethodError
+            sleep 1
+          end
+        end
+      end
+    end
+
+
+    def periodically_refresh_users bot
       req_path = '/rest/api/2/user/assignable/search'
       req_params = QueryParams.encode \
         project: options.jira_project,
         startAt: 0,
         maxResults: 1_000_000
@@ -167,10 +253,10 @@
             } ; h
           end
 
           bot.store['users'] = users
 
-          sleep options.refresh
+          sleep options.user_refresh
         end
       end
     end