bender-bot 0.2.4 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 702cbc0ab493486d62b7c9db41dfc1191a5c9868
-  data.tar.gz: 3e6535a990ed42e03aa8d6b93c4a24bd83643791
+  metadata.gz: 50e516ad8b6f747483302fc79ab811a8cccf2368
+  data.tar.gz: bd931995c92e814c96bbcea6f305534c2941cede
 SHA512:
-  metadata.gz: 10f0310b8a034da123d18f08da7b22698b57acc31520c66a2e9d9306d64c028f161e3f0a77b123fca96b50ff56ece5e9434a924f0481d2bc34859d971b80225e
-  data.tar.gz: c8b49ba2fa7bb1f1b715fbbb04dea79ed361657a2483d7fc4f8a8fbe2a61694a28c6992ad0cffd7edc19e0fc81a6e03b09b1abd6ed8dd08300d91826a2cd63b2
+  metadata.gz: b43003dffdf9a21bb11877116e287e04b80a84618a3d2743b7a83a030025d7c0303d6842a5d478c048c49680e9a0955576de7497ae0275e08baa26540701abe5
+  data.tar.gz: 23de30730b6a02d26c1e71603966bb6811ef7ef4a7037c978f4dc69e75e6d3afe4abccafdd8d4eae6c9afd0d40eae674d3a47f470e52a8cffc56d14e1d9c07e9

data/VERSION

@@ -1 +1 @@
-0.2.4
+0.3.0

data/lib/bender/bot.rb

@@ -8,6 +8,8 @@ require 'robut/storage/yaml_store'
 require 'fuzzystringmatch'
 require 'queryparams'
 
+require_relative 'helpers'
+
 Bot = Robut # alias
 
 
@@ -29,59 +31,10 @@ end
 
 class BenderBot
   include Bot::Plugin
+  include Helpers
 
   JARO = FuzzyStringMatch::JaroWinkler.create :native
 
-  RESOLVED_TRANSITIONS = %w[ 51 ]
-
-  RESOLVED_STATE = /resolve/i
-
-  CLOSED_TRANSITIONS = %w[ 61 71 ]
-
-  CLOSED_STATE = /close/i
-
-  SEVERITIES = {
-    1 => '10480',
-    2 => '10481',
-    3 => '10482',
-    4 => '10483',
-    5 => '10484'
-  }
-
-  SHOW_FIELDS = {
-    'summary' => 'Summary',
-    'description' => 'Description',
-    'customfield_11250' => 'Severity',
-    'customfield_11251' => 'Impact Started',
-    'customfield_11252' => 'Impact Ended',
-    'customfield_11253' => 'Reported By',
-    'customfield_11254' => 'Services Affected',
-    'customfield_11255' => 'Cause',
-    'status' => 'Status',
-    'created' => 'Created',
-    'updated' => 'Updated'
-  }
-
-
-  QUOTES = [
-    'Bite my shiny metal ass!',
-    'This is the worst kind of discrimination there is: the kind against me!',
-    'I guess if you want children beaten, you have to do it yourself.',
-    "Hahahahaha! Oh wait you're serious. Let me laugh even harder.",
-    "You know what cheers me up? Other people's misfortune.",
-    'Anything less than immortality is a complete waste of time.',
-    "Blackmail is such an ugly word. I prefer extortion. The 'x' makes it sound cool.",
-    'Have you tried turning off the TV, sitting down with your children, and hitting them?',
-    "You're a pimple on society’s ass and you'll never amount to anything!",
-    'Shut up baby, I know it!',
-    "I'm so embarrassed. I wish everyone else was dead!",
-    "Afterlife? If I thought I had to live another life, I'd kill myself right now!",
-    "I'm back baby!",
-    "LET'S GO ALREADYYYYYY!",
-    "I don't have emotions and sometimes that makes me very sad",
-    "Life is hilariously cruel"
-  ]
-
   COMMANDS = {
     help: {
       desc: 'Display this help text',
@@ -134,7 +87,14 @@ class BenderBot
 
 
   def handle room, sender, message
-    @room_name = @@rooms.select { |r| r.xmpp_jid == room }.first.name
+    begin
+      @room_name = @@rooms.select { |r| r.xmpp_jid == room }.first.name
+      @@hipchat[@room_name].get_room # test
+    rescue
+      @@hipchat  = HipChat::Client.new(@@options.hipchat_token)
+      @@rooms    = @@hipchat.rooms
+      @room_name = @@rooms.select { |r| r.xmpp_jid == room }.first.name
+    end
     @room      = room
     @sender    = sender
     @message   = message
@@ -142,13 +102,6 @@ class BenderBot
     severity_field = SHOW_FIELDS.key 'Severity'
     severities = Hash.new { |h,k| h[k] = [] }
 
-    if message =~ /\/inc\s+(\w+)?\s*/
-      unless COMMANDS.include? $1.to_sym
-        reply_html 'Invalid usage', :red
-        reply_with_help
-        return
-      end
-    end
 
     case message
 
@@ -175,7 +128,7 @@ class BenderBot
 
       is = store['incidents'].reverse.map do |i|
         status = normalize_value i['fields']['status']
-        unless status =~ /done|complete|closed/i
+        unless status =~ /resolved|closed/i
           '%s (%s - %s) [%s]: %s' % [
             incident_link(i),
             short_severity(i['fields'][severity_field]['value']),
@@ -230,7 +183,7 @@ class BenderBot
       end
 
       if severities.empty?
-        reply_html 'No recent incidents! Woohoo!', :green
+        reply_html 'Good news everyone! No open incidents at the moment', :green
 
       else
         is = severities.keys.sort.map do |sev|
@@ -313,6 +266,11 @@ class BenderBot
       else
         reply_html 'Sorry, no such incident!', :red
       end
+
+
+    when /^\s*\/inc/i
+      reply_html 'Invalid usage', :red
+      reply_with_help
     end
 
 
@@ -346,31 +304,6 @@ private
 
 
 
-  def refresh_incidents
-    req_path = '/rest/api/2/search'
-    req_params = QueryParams.encode \
-      jql: "project = #{options.jira_project} ORDER BY created ASC, priority DESC",
-      fields: SHOW_FIELDS.keys.join(','),
-      startAt: 0,
-      maxResults: 1_000_000
-
-    uri = URI(options.jira_site + req_path + '?' + req_params)
-    http = Net::HTTP.new uri.hostname, uri.port
-
-    req = Net::HTTP::Get.new uri
-    req.basic_auth options.jira_user, options.jira_pass
-    req['Content-Type'] = 'application/json'
-    req['Accept'] = 'application/json'
-
-    resp = http.request req
-    issues = JSON.parse(resp.body)['issues']
-
-    store['incidents'] = issues.map! do |i|
-      i['num'] = i['key'].split('-', 2).last ; i
-    end
-  end
-
-
   def file_incident data
     req_path = '/rest/api/2/issue'
     uri = URI(options.jira_site + req_path)
@@ -499,62 +432,4 @@ private
     end
   end
 
-
-  def normalize_value val
-    case val
-    when Hash
-      val['name'] || val['value'] || val
-    when Array
-      val.map { |v| v['value'] }.join(', ')
-    when /^\d{4}\-\d{2}\-\d{2}/
-      '%s (%s)' % [ val, normalize_date(val) ]
-    else
-      val
-    end
-  end
-
-
-  def normalize_date val
-    Time.parse(val).utc.iso8601(0).sub(/Z$/, 'UTC')
-  end
-
-
-  def friendly_date val
-    Time.parse(val).strftime('%Y-%m-%d %H:%M %Z')
-  end
-
-
-  def recent_incident? i
-    it = Time.parse(i['fields']['created'])
-    Time.now - it < one_day
-  end
-
-
-  def one_day
-    24 * 60 * 60 # seconds/day
-  end
-
-
-  def select_incident num, refresh=true
-    refresh_incidents if refresh
-    store['incidents'].select { |i| i['num'] == num }.first
-  end
-
-
-  def short_severity s
-    s.split(' - ', 2).first
-  end
-
-
-  def incident_url incident
-    options.jira_site + '/browse/' + incident['key']
-  end
-
-  def incident_link incident
-    '<a href="%s">%s</a>' % [
-      incident_url(incident),
-      incident['key']
-    ]
-  end
-
 end

data/lib/bender/helpers.rb

@@ -0,0 +1,136 @@
+module Helpers
+
+  RESOLVED_TRANSITIONS = %w[ 51 ]
+
+  RESOLVED_STATE = /resolve/i
+
+  CLOSED_TRANSITIONS = %w[ 61 71 ]
+
+  CLOSED_STATE = /close/i
+
+  SEVERITIES = {
+    1 => '10480',
+    2 => '10481',
+    3 => '10482',
+    4 => '10483',
+    5 => '10484'
+  }
+
+  SHOW_FIELDS = {
+    'summary' => 'Summary',
+    'description' => 'Description',
+    'customfield_11250' => 'Severity',
+    'customfield_11251' => 'Impact Started',
+    'customfield_11252' => 'Impact Ended',
+    'customfield_11253' => 'Reported By',
+    'customfield_11254' => 'Services Affected',
+    'customfield_11255' => 'Cause',
+    'status' => 'Status',
+    'created' => 'Created',
+    'updated' => 'Updated'
+  }
+
+
+  QUOTES = [
+    'Bite my shiny metal ass!',
+    'This is the worst kind of discrimination there is: the kind against me!',
+    'I guess if you want children beaten, you have to do it yourself.',
+    "Hahahahaha! Oh wait you're serious. Let me laugh even harder.",
+    "You know what cheers me up? Other people's misfortune.",
+    'Anything less than immortality is a complete waste of time.',
+    "Blackmail is such an ugly word. I prefer extortion. The 'x' makes it sound cool.",
+    'Have you tried turning off the TV, sitting down with your children, and hitting them?',
+    "You're a pimple on society’s ass and you'll never amount to anything!",
+    'Shut up baby, I know it!',
+    "I'm so embarrassed. I wish everyone else was dead!",
+    "Afterlife? If I thought I had to live another life, I'd kill myself right now!",
+    "I'm back baby!",
+    "LET'S GO ALREADYYYYYY!",
+    "I don't have emotions and sometimes that makes me very sad",
+    "Life is hilariously cruel"
+  ]
+
+
+  def refresh_incidents bot=self
+    req_path = '/rest/api/2/search'
+    req_params = QueryParams.encode \
+      jql: "project = #{options.jira_project} ORDER BY created ASC, priority DESC",
+      fields: SHOW_FIELDS.keys.join(','),
+      startAt: 0,
+      maxResults: 1_000_000
+
+    uri = URI(options.jira_site + req_path + '?' + req_params)
+    http = Net::HTTP.new uri.hostname, uri.port
+
+    req = Net::HTTP::Get.new uri
+    req.basic_auth options.jira_user, options.jira_pass
+    req['Content-Type'] = 'application/json'
+    req['Accept'] = 'application/json'
+
+    resp = http.request req
+    issues = JSON.parse(resp.body)['issues']
+
+    bot.store['incidents'] = issues.map! do |i|
+      i['num'] = i['key'].split('-', 2).last ; i
+    end
+  end
+
+
+  def normalize_value val
+    case val
+    when Hash
+      val['name'] || val['value'] || val
+    when Array
+      val.map { |v| v['value'] }.join(', ')
+    when /^\d{4}\-\d{2}\-\d{2}/
+      '%s (%s)' % [ val, normalize_date(val) ]
+    else
+      val
+    end
+  end
+
+
+  def normalize_date val
+    Time.parse(val).utc.iso8601(0).sub(/Z$/, 'UTC')
+  end
+
+
+  def friendly_date val
+    Time.parse(val).strftime('%Y-%m-%d %H:%M %Z')
+  end
+
+
+  def recent_incident? i
+    it = Time.parse(i['fields']['created'])
+    Time.now - it < one_day
+  end
+
+
+  def one_day
+    24 * 60 * 60 # seconds/day
+  end
+
+
+  def select_incident num, refresh=true
+    refresh_incidents if refresh
+    store['incidents'].select { |i| i['num'] == num }.first
+  end
+
+
+  def short_severity s
+    s.split(' - ', 2).first
+  end
+
+
+  def incident_url incident
+    options.jira_site + '/browse/' + incident['key']
+  end
+
+  def incident_link incident
+    '<a href="%s">%s</a>' % [
+      incident_url(incident),
+      incident['key']
+    ]
+  end
+
+end

data/lib/bender/main.rb

@@ -6,12 +6,16 @@ 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
@@ -47,6 +51,16 @@ module Bender
       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 ],
@@ -77,11 +91,6 @@ module Bender
       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 ],
@@ -107,15 +116,21 @@ module Bender
       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
 
@@ -141,7 +156,78 @@ module Bender
     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,
@@ -169,7 +255,7 @@ module Bender
 
           bot.store['users'] = users
 
-          sleep options.refresh
+          sleep options.user_refresh
         end
       end
     end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: bender-bot
 version: !ruby/object:Gem::Version
-  version: 0.2.4
+  version: 0.3.0
 platform: ruby
 authors:
 - Sean Clemmer
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-08-26 00:00:00.000000000 Z
+date: 2015-08-27 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: thor
@@ -177,6 +177,7 @@ files:
 - bin/bender
 - lib/bender.rb
 - lib/bender/bot.rb
+- lib/bender/helpers.rb
 - lib/bender/main.rb
 - lib/bender/metadata.rb
 - lib/bender/mjolnir.rb