bender-bot 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 96464ae31bbfe17baeab335880504aa1900a363a
-  data.tar.gz: 6c140ca83b456be05cd33c7f7c24d27c28d00fe8
+  metadata.gz: 7c561e6f44af4f398b20ae38a1c8b506a3318a4b
+  data.tar.gz: 736e2de7ab537a0d070b9d65fc06f3a1831c1bc6
 SHA512:
-  metadata.gz: 7e29c40bb49d93c64bcc04c9292f557ba021a51a222a65d9a5e44ab2006fdc6dc5f2aff5dd3db5b6588bb1c0115655987fea8da34685be5fd77a25ac98636fbc
-  data.tar.gz: 1c5505dbe45432ace49b97196ca3f89a182622f19eb48b23f29fa72862891734f83f5948af7781e77691efbcd44e75b444f63f33f690e941804d12e8266fea50
+  metadata.gz: fc3a5c412677a9dbe0d3bdf61c7fecacff394636f7ecf9ca3c451964905c163ae2a7377a13a66215c8700dd4687a378c893c0b2e069826dc3d609c49c2dc9e96
+  data.tar.gz: 13229d4431b57107fbf104d598ff36d645db1d96086e9e5f9930eafd48eb6361f27ed1e18d938775fa2668949631ee35c367e2c7e4aabd7c50bb5fec9f489bd6

data/Readme.md

@@ -1,3 +1,3 @@
-# Bender ![Version](https://img.shields.io/gem/v/bender.svg?style=flat-square)
+# Bender ![Version](https://img.shields.io/gem/v/bender-bot.svg?style=flat-square)
 
 Yet another HipChat bot.

data/VERSION

@@ -1 +1 @@
-0.0.1
+0.0.2

data/lib/bender/bot.rb

@@ -2,33 +2,185 @@ require 'thread'
 
 require 'robut'
 require 'robut/storage/yaml_store'
+require 'fuzzystringmatch'
+require 'queryparams'
 
 Bot = Robut # alias
 
 
 
 module Bot
-  def self.run!
-    Bot::Plugin.plugins = [ Bot::Plugin::Bender ]
-    Bot::Web.set :connection, Bot::Connection.new.connect
+  def self.run! options
+    BenderBot.class_variable_set :@@options, options
+    Bot::Plugin.plugins = [ BenderBot ]
+    conn = Bot::Connection.new
+    conn.store['users'] ||= {}
+    Bot::Web.set :connection, conn.connect
+    return conn
   end
 end
 
 
-module Bot
-  module Plugin
-    class Bender
-      include Bot::Plugin
+class BenderBot
+  include Bot::Plugin
+
+  JARO = FuzzyStringMatch::JaroWinkler.create :native
+
+  SHOW_FIELDS = %w[
+    summary
+    description
+    priority
+    status
+    created
+    updated
+  ]
+
+
+  def handle time, sender, message
+    case message
+
+
+    when /^\s*\?opts\s*$/
+      reply options.inspect
+
+    when /^\s*\?whoami\s*$/
+      u = user_where name: sender
+      reply '%s: %s (%s)' % [ u[:nick], u[:name], u[:email] ]
+
+    when /^\s*\?lookup\s+(.+)\s*$/
+      u = user_where(name: $1) || user_where(nick: $1)
+      reply '%s: %s (%s)' % [ u[:nick], u[:name], u[:email] ]
+
+    when /^\s*\?incident\s*$/
+      reply [
+        '?icident - This help text',
+        '?incidents - List open incidents',
+        '?incident NUM - Show incident',
+        '?incident NUM FIELD - Show incident field',
+        '?incident SUMMARY - File a new incident'
+      ].join("\n")
+
+    when /^\s*\?incidents\s*$/
+      refresh_incidents
 
+      is = store['incidents'].map do |i|
+        '%s: %s' % [ i['num'], i['fields']['summary'] ]
+      end.join("\n")
 
-      def handle time, sender, message
-        if message =~ /bender/
-          reply 'What, %s?! (%s)' % [ sender, time ]
+      reply is
+
+    when /^\s*\?incident\s+(\d+)\s*$/
+      refresh_incidents
+      incident = store['incidents'].select { |i| i['num'] == $1 }.first
+
+      fields = SHOW_FIELDS - %w[ summary ]
+
+      i = fields.map do |f|
+        val = incident['fields'][f]
+        if val
+          val = val.is_a?(Hash) ? val['name'] : val
+          '%s: %s' % [ f, val ]
         end
+      end.compact
+
+      reply "%s: %s\n%s" % [
+        incident['key'],
+        incident['fields']['summary'],
+        i.join("\n")
+      ]
+
+    when /^\s*\?incident\s+(\d+)\s+(.*?)\s*$/
+      refresh_incidents
+      incident = store['incidents'].select { |i| i['num'] == $1 }.first
+      val = incident['fields'][$2]
+      val = val.is_a?(Hash) ? val['name'] : val
+      reply val
+
+    when /^\s*\?incident\s+(.*?)\s*$/
+      user = user_where name: sender
+      data = {
+        fields: {
+          project: { key: options.jira_project },
+          issuetype: { name: options.jira_type },
+          reporter: { name: user[:nick] },
+          summary: $1
+        }
+      }
+
+      reply file_incident(data)
+    end
+
+    return true
+  end
+
 
-        return true
-      end
 
+private
+
+  def options ; @@options end
+
+  def user_where fields, threshold=0.8
+    field, value = fields.to_a.shift
+    suggested_user = store['users'].values.sort_by do |u|
+      compare value, u[field]
+    end.last
+
+    distance = compare value, suggested_user[field]
+    return distance < threshold ? nil : suggested_user
+  end
+
+
+  def compare name1, name2
+    n1 = name1.gsub /\W/, ''
+    n2 = name2.gsub /\W/, ''
+    d1 = JARO.getDistance n1.downcase, n2.downcase
+    d2 = JARO.getDistance n1, n2
+    return d1 + d2 / 2.0
+  end
+
+
+
+  def refresh_incidents
+    req_path = '/rest/api/2/search'
+    req_params = QueryParams.encode \
+      jql: "project = #{options.jira_project} AND resolution = Unresolved ORDER BY created ASC, priority DESC",
+      fields: SHOW_FIELDS.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)
+    http = Net::HTTP.new uri.hostname, uri.port
+
+    req = Net::HTTP::Post.new uri
+    req.basic_auth options.jira_user, options.jira_pass
+    req['Content-Type'] = 'application/json'
+    req['Accept'] = 'application/json'
+    req.body = data.to_json
+
+    resp = http.request req
+    issue = JSON.parse(resp.body)
+
+    return options.jira_site + '/browse/' + issue['key']
+  end
+
+
 end

data/lib/bender/main.rb

@@ -1,6 +1,7 @@
 require 'logger'
 
 require 'tilt/erb'
+require 'queryparams'
 
 require_relative 'metadata'
 require_relative 'mjolnir'
@@ -70,8 +71,53 @@ 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 ],
+      desc: 'Set JIRA username',
+      required: true
+    option :jira_pass, \
+      type: :string,
+      aliases: %w[ -P ],
+      desc: 'Set JIRA password',
+      required: true
+    option :jira_site, \
+      type: :string,
+      aliases: %w[ -S ],
+      desc: 'Set JIRA site',
+      required: true
+    option :jira_project, \
+      type: :string,
+      aliases: %w[ -J ],
+      desc: 'Set JIRA project',
+      required: true
+    option :jira_type, \
+      type: :string,
+      aliases: %w[ -T ],
+      desc: 'Set JIRA issue type',
+      required: true
+    option :refresh, \
+      type: :numeric,
+      aliases: %w[ -R ],
+      desc: 'Set JIRA refresh rate',
+      default: 300
     include_common_options
     def start
+      bot = start_bot
+      refresh_users bot
+      serve_web bot
+    end
+
+
+
+  private
+
+    def start_bot
       Bot::Connection.configure do |config|
         config.jid = options.jid
         config.password = options.password
@@ -85,9 +131,46 @@ module Bender
         config.logger = log
       end
 
-      Bot.run!
+      Bot.run! options
+    end
+
+
+    def 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
+
+      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'
+
+      Thread.new do
+        loop do
+          resp = http.request req
+
+          users = JSON.parse(resp.body).inject({}) do |h, user|
+            h[user['key']] = {
+              nick: user['key'],
+              name: user['displayName'],
+              email: user['emailAddress']
+            } ; h
+          end
+
+          bot.store['users'] = users
+
+          sleep options.refresh
+        end
+      end
+    end
 
 
+    def serve_web bot
       Web.set :environment, options.environment
       Web.set :port, options.port
       Web.set :bind, options.bind
@@ -100,6 +183,7 @@ module Bender
         Web.set :logging, ::Logger::DEBUG
       end
 
+      Web.set :bot, bot
       Web.run!
     end
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: bender-bot
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - Sean Clemmer
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-04-23 00:00:00.000000000 Z
+date: 2015-04-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: thor
@@ -66,6 +66,48 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 0.5.2
+- !ruby/object:Gem::Dependency
+  name: hipchat
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1'
+- !ruby/object:Gem::Dependency
+  name: queryparams
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.0.3
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.0.3
+- !ruby/object:Gem::Dependency
+  name: fuzzy-string-match
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.9.7
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.9.7
 - !ruby/object:Gem::Dependency
   name: eventmachine
   requirement: !ruby/object:Gem::Requirement