ruboty-redmine 0.0.3 → 0.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 80d38692ef9bfa4c14e4641be38f067786234bb8
-  data.tar.gz: a6be7d580cf0b3ebcdf9780583e7f39181b176b1
+  metadata.gz: bcf104205740a7675b39dc610baaf02a1bd44dbf
+  data.tar.gz: e6904c6198c2673dc4c6e999ccb37c2e02f77527
 SHA512:
-  metadata.gz: c45f767065539bd4f8457a93e6853a7c7fdef682915b35b5981e403e640c1de0aaada64ce838e4d517a6751ccd85168337132a251db9d58d9816b5a5f6e286ba
-  data.tar.gz: 1d8f01c07cf5ba3317fc1345ee269fdfe1fb2eff811ebd46ab73050361ef05c91d480b6372b502b5fefa5bac73cad749e10b2a5646badde03216bba06370b100
+  metadata.gz: f28bc2644829ea3ad70ee3d8594a46ef779e08c5d75d16d0d4f82191b18f02d7e412937e6084a9fb2533abba48ccf3a6455fc7241bc9c5750ef90ba47a96f3a5
+  data.tar.gz: ed581a00f88e7932de5aa9f8ceb7d600b7519e06ac44af64685f2d7ae1494e7af04e840e7bd0b406fd48ebdd668edb279650910f7c86745f8b90cb295ec36d11

data/lib/ruboty/handlers/redmine.rb

@@ -1,10 +1,13 @@
 module Ruboty
   module Handlers
     class Redmine < Base
+      NAMESPACE = 'redmine'
+
       env :REDMINE_URL, 'Redmine url (e.g. http://your-redmine)', optional: false
       env :REDMINE_API_KEY, 'Redmine REST API key', optional: false
       env :REDMINE_BASIC_AUTH_USER, 'Basic Auth User', optional: true
       env :REDMINE_BASIC_AUTH_PASSWORD, 'Basic Auth Password', optional: true
+      env :REDMINE_CHECK_INTERVAL, 'Interval to check new issues', optional: true
 
       on(
         /create issue (?<rest>.+)/,
@@ -12,10 +15,52 @@ module Ruboty
         description: 'Create a new issue'
       )
 
+      on(
+        /register redmine alias "(?<name>[^"]+)" ("(?<expand_to>[^"]+)"|'(?<expand_to>[^']+)')/,
+        name: 'register_alias',
+        description: 'Register an alias'
+      )
+
+      on(
+        /watch redmine issues in "(?<tracker>[^"]+)" tracker of "(?<project>[^"]+)" project( and assign to (?<assignees>[\d,]+)|)/,
+        name: 'watch_issues',
+        description: 'Watch issues'
+      )
+
+      on(
+        /list watching redmine issues/,
+        name: 'list_watching',
+        description: 'List watching issues'
+      )
+
+      on(
+        /stop watching redmine issues (?<id>\d+)/,
+        name: 'stop_watching',
+        description: 'Stop watching issues',
+      )
+
+      on(
+        /associate redmine user (?<redmine_id>\d+) with "(?<chat_name>[^"]+)"/,
+        name: 'associate_user',
+        description: 'Associate redmine_id with chat_name',
+      )
+
+      def initialize(*args)
+        super
+
+        start_to_watch_issues
+      end
+
       def create_issue(message)
-        words = message[:rest].scan(/("([^"]+)"|([^ ]+))/).map {|v| v[1] || v[2] }
+        words = parse_arg(message[:rest])
         req = {}
         req[:subject] = words.shift
+
+        if words.size == 1
+          expand_to = alias_for(words.first)
+          words = parse_arg(expand_to) if expand_to
+        end
+
         words.each_with_index do |word, i|
           next if i == 0
 
@@ -23,13 +68,7 @@ module Ruboty
 
           case word
           when 'project'
-            project = redmine.projects.find do |project|
-              [
-                project.id.to_s,
-                project.name.downcase,
-                project.identifier.downcase,
-              ].include?(arg.downcase)
-            end
+            project = redmine.find_project(arg)
 
             unless project
               message.reply("Project '#{arg}' is not found.")
@@ -38,12 +77,7 @@ module Ruboty
 
             req[:project] = project
           when 'tracker'
-            tracker = redmine.trackers.find do |tracker|
-              [
-                tracker.id.to_s,
-                tracker.name.downcase,
-              ].include?(arg.downcase)
-            end
+            tracker = redmine.find_tracker(arg)
 
             unless tracker
               message.reply("Tracker '#{arg}' is not found.")
@@ -63,16 +97,160 @@ module Ruboty
         message.reply("Issue created: #{redmine.url_for_issue(issue)}")
       end
 
+      def register_alias(message)
+        aliases[message[:name]] = message[:expand_to]
+        message.reply("Registered.")
+      end
+
+      def watch_issues(message)
+        if message.match_data.names.include?('assignees')
+          assignees = message[:assignees].split(',').map(&:to_i)
+        else
+          assignees = []
+        end
+
+        watches << message.original.except(:robot).merge(
+          {id: watches.size + 1, project: message[:project], tracker: message[:tracker], assignees: assignees, assignee_index: 0}
+        ).stringify_keys
+        message.reply("Watching.")
+      end
+
+      def list_watching(message)
+        reply = watches.map do |watch|
+          s = "##{watch['id']} #{watch['tracker']} tracker in #{watch['project']} project"
+          if assignees = watch['assignees']
+            s << " and assign to #{assignees}"
+          end
+        end.join("\n")
+
+        message.reply(reply)
+      end
+
+      def stop_watching(message)
+        id = message[:id].to_i
+        watches.reject! do |watch|
+          watch['id'] == id
+        end
+
+        message.reply("Stopped.")
+      end
+
+      def associate_user(message)
+        users << {"redmine_id" => message[:redmine_id].to_i, "chat_name" => message[:chat_name]}
+
+        message.reply("Registered.")
+      end
+
       private
 
       def redmine
-        Ruboty::Redmine::Client.new(
+        @redmine ||= Ruboty::Redmine::Client.new(
           ENV['REDMINE_URL'],
           ENV['REDMINE_API_KEY'],
           basic_auth_user: ENV['REDMINE_BASIC_AUTH_USER'],
           basic_auth_password: ENV['REDMINE_BASIC_AUTH_PASSWORD'],
         )
       end
+
+      def alias_for(name)
+        aliases[name]
+      end
+
+      def aliases
+        robot.brain.data["#{NAMESPACE}_aliases"] ||= {}
+      end
+
+      def watches
+        robot.brain.data["#{NAMESPACE}_watches"] ||= []
+      end
+
+      def users
+        robot.brain.data["#{NAMESPACE}_users"] ||= []
+      end
+
+      def find_user_by_id(id)
+        users.find {|user| user['redmine_id'] == id }
+      end
+
+      def parse_arg(text)
+        text.scan(/("([^"]+)"|'([^']+)'|([^ ]+))/).map do |v|
+          v.shift
+          v.find {|itself| itself }
+        end
+      end
+
+      def start_to_watch_issues
+        thread = Thread.start do
+          last_issues_for_watch = {}
+
+          while true
+            sleep (ENV['REDMINE_CHECK_INTERVAL'] || 30).to_i
+            Ruboty::Redmine.log("Checking new issues...")
+            watches.each do |watch|
+              project = redmine.find_project(watch['project'])
+              tracker = redmine.find_tracker(watch['tracker'])
+
+              issues = redmine.issues(project: project, tracker: tracker, sort: 'id:desc')
+              if last_issues = last_issues_for_watch[watch]
+                new_issues = []
+                issues.each do |issue|
+                  found = last_issues.find do |last_issue|
+                    last_issue.id == issue.id
+                  end
+
+                  if found
+                    break
+                  else
+                    new_issues << issue
+                  end
+                end
+
+                new_issues.each do |new_issue|
+require 'pry'
+  Pry.config.input = STDIN
+  Pry.config.output = STDOUT
+binding.pry
+                  assignees = watch['assignees']
+                  assignee = nil
+                  unless assignees.empty?
+                    assignee = assignees[watch['assignee_index'] % assignees.size]
+                    watch['assignee_index'] += 1
+
+                    assignee = find_user_by_id(assignee)
+                  end
+
+                  if assignee
+                    redmine.update_issue(new_issue, assigned_to_id: assignee['redmine_id'])
+                  end
+
+                  msg = <<-EOC
+New Issue of #{tracker.name} in #{project.name} project
+-> #{new_issue.subject}
+                  EOC
+
+                  if assignee
+                    msg += <<-EOC
+-> Assigned to @#{assignee['chat_name']}
+                    EOC
+                  end
+
+                  msg += <<-EOC
+-> #{redmine.url_for_issue(new_issue)}
+                  EOC
+
+                  Message.new(
+                    watch.symbolize_keys.merge(robot: robot)
+                  ).reply(msg)
+                end
+              end
+
+              last_issues_for_watch[watch] = issues
+            end
+          end
+        end
+
+        thread.abort_on_exception = true
+      end
     end
   end
 end

data/lib/ruboty/redmine.rb

@@ -1,9 +1,16 @@
+require "active_support/core_ext/hash/keys"
+
 require "ruboty/redmine/version"
 require "ruboty/redmine/client"
 require "ruboty/handlers/redmine"
 
 module Ruboty
   module Redmine
-    # Your code goes here...
+    def self.log(msg)
+      Ruboty.logger.info "[redmine] #{msg}"
+
+      # FIX: thix dirty hack.
+      Ruboty.logger.instance_variable_get(:@logdev).dev.flush
+    end
   end
 end

data/lib/ruboty/redmine/client.rb

@@ -25,6 +25,37 @@ module Ruboty
         end
       end
 
+      def find_project(query)
+        projects.find do |project|
+          [
+            project.id.to_s,
+            project.name.downcase,
+            project.identifier.downcase,
+          ].include?(query.downcase)
+        end
+      end
+
+      def find_tracker(query)
+        trackers.find do |tracker|
+          [
+            tracker.id.to_s,
+            tracker.name.downcase,
+          ].include?(query.downcase)
+        end
+      end
+
+      def issues(opts)
+        params = {}
+        params[:project_id] = opts[:project].id if opts[:project]
+        params[:tracker_id] = opts[:tracker].id if opts[:tracker]
+        params[:sort] = opts[:sort] if opts[:sort]
+
+        res = JSON.parse(get('/issues.json', params).body)
+        res['issues'].map do |tracker|
+          OpenStruct.new(tracker)
+        end
+      end
+
       def create_issue(opts)
         req = {
           issue: {
@@ -39,6 +70,15 @@ module Ruboty
         )
       end
 
+      def update_issue(issue, opts)
+        req = {issue: opts}
+
+        res = put("/issues/#{issue.id}.json", req)
+        unless res.status == 200
+          raise "Updating issue failed. (#{res.body})"
+        end
+      end
+
       def url_for_issue(issue)
         URI.join(@url, "/issues/#{issue.id}")
       end
@@ -78,6 +118,15 @@ module Ruboty
           req.headers['Content-Type'] = 'application/json'
         end
       end
+
+      def put(path, params = {})
+        conn.put do |req|
+          req.url path
+          req.body = params.to_json
+          req.headers['X-Redmine-API-Key'] = @api_key
+          req.headers['Content-Type'] = 'application/json'
+        end
+      end
     end
   end
 end

data/lib/ruboty/redmine/version.rb

@@ -1,5 +1,5 @@
 module Ruboty
   module Redmine
-    VERSION = "0.0.3"
+    VERSION = "0.0.4"
   end
 end

data/ruboty-redmine.gemspec

@@ -19,7 +19,9 @@ Gem::Specification.new do |spec|
 
   spec.add_dependency "ruboty"
   spec.add_dependency "faraday"
+  spec.add_dependency "activesupport"
 
+  spec.add_development_dependency "pry-byebug"
   spec.add_development_dependency "bundler", "~> 1.7"
   spec.add_development_dependency "rake", "~> 10.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ruboty-redmine
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
 platform: ruby
 authors:
 - Ryota Arai
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-01-30 00:00:00.000000000 Z
+date: 2015-02-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ruboty
@@ -38,6 +38,34 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: activesupport
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: pry-byebug
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement