lita-kintai 0.2.0 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 0e5ec578207f077b2d7cb7d6f9c0bd5bec0de09c
-  data.tar.gz: 31762b9c377b44c6629431824ffc2ff4b7e15271
+  metadata.gz: 5d058f92110471659693ea8060d40a4a02b85a1f
+  data.tar.gz: 5463d339eaa7b46964df52ca7ab34661f7e9936f
 SHA512:
-  metadata.gz: 6d133cc94a15c8bc24570eba382a0b65af58ff3534528a674b6212b70847fd1110410e8892931cb24509be5ff9391cef9525e4011160f756d562a16673f93bae
-  data.tar.gz: 3e8ec3f773dd2e5619591cfdcf84081083c94fc7aedbfabc5389608be9c76df8e585b32cd9c810de6ac4658ff19a50292b51c887741a9de78ed544141e07c8f2
+  metadata.gz: eac814178e90adb59f1c643c261c05403713936c046012cd07e4446bbfd3c83ef2b3965b165f837287b2d41068d72f64f21611536b3764648ead142ddf5a5f63
+  data.tar.gz: afc014be3319b2c57fb72393161be0acc2d12c92c01ef2e5669a7999a1f293b58dbd390424804f5f1d105df72381633fcb4dd2f0c80a3818931405a0106d47b8

data/lib/gmail.rb

@@ -0,0 +1,86 @@
+require 'google/apis/gmail_v1'
+require 'googleauth'
+require 'googleauth/stores/file_token_store'
+
+require 'fileutils'
+
+class Gmail
+  OOB_URI = 'urn:ietf:wg:oauth:2.0:oob'
+  APPLICATION_NAME = 'Lita Kintai'
+  CLIENT_SECRETS_PATH = 'client_secret.json'
+  CREDENTIALS_PATH = File.join(Dir.home, '.credentials',
+                               "lita-kintai.yaml")
+  SCOPE = [Google::Apis::GmailV1::AUTH_GMAIL_READONLY, Google::Apis::GmailV1::AUTH_GMAIL_SEND]
+  USER_ID = 'default'
+
+  def self.authorized?
+    !service.nil? && !service.authorization.nil?
+  end
+
+  def self.credentials_from_code(code)
+    authorizer.get_and_store_credentials_from_code(
+    user_id: USER_ID, code: code, base_url: OOB_URI)
+  end
+
+  def self.authorization_url
+    authorizer.get_authorization_url(base_url: OOB_URI)
+  end
+
+  def self.find_mail(query)
+    ids = service.list_user_messages('me', q: query)
+
+    return [] unless ids.messages
+    ids.messages.map do |message|
+      find_mail_by_id(message.id)
+    end
+  end
+
+  def self.send_message(mail)
+    message = Google::Apis::GmailV1::Message.new(raw: mail.to_s )
+    result = service.send_user_message('me', message)
+  end
+
+private
+  def self.service
+    return @service if !@service.nil? && !@service.authorization.nil?
+
+    credentials = authorizer.get_credentials(USER_ID)
+    if credentials
+      @service.authorization = credentials
+      return @service
+    end
+
+    return @service
+  end
+
+  def self.authorizer
+    return @authorizer unless @authorizer.nil?
+
+    @service = Google::Apis::GmailV1::GmailService.new
+    @service.client_options.application_name = APPLICATION_NAME
+
+    FileUtils.mkdir_p(File.dirname(CREDENTIALS_PATH))
+
+    client_id = Google::Auth::ClientId.from_file(CLIENT_SECRETS_PATH)
+    token_store = Google::Auth::Stores::FileTokenStore.new(
+      file: CREDENTIALS_PATH)
+    @authorizer = Google::Auth::UserAuthorizer.new(
+      client_id, SCOPE, token_store)
+  end
+
+  def self.find_mail_by_id(id)
+    results = @service.get_user_message('me', id)
+
+    body = results.payload.parts ?
+      results.payload.parts.first.body.data :
+      results.payload.body.data
+    headers = results.payload.headers
+
+    {
+      subject: headers.select { |e| e.name == 'Subject'}.first.value,
+      from: headers.select { |e| e.name == 'From'}.first.value,
+      date: Time.parse(headers.select { |e| e.name == 'Date'}.first.value),
+      body: body.force_encoding('utf-8'),
+    }
+  end
+end

data/lib/lita-kintai.rb

@@ -4,6 +4,7 @@ Lita.load_locales Dir[File.expand_path(
   File.join("..", "..", "locales", "*.yml"), __FILE__
 )]
 
+require "gmail"
 require "lita/handlers/kintai"
 
 Lita::Handlers::Kintai.template_root File.expand_path(

data/lib/lita/handlers/kintai.rb

@@ -1,41 +1,52 @@
-require 'google/apis/gmail_v1'
-require 'googleauth'
-require 'googleauth/stores/file_token_store'
-
-require 'fileutils'
 require 'date'
-
 require 'rufus-scheduler'
+require 'mail'
 
 module Lita
   module Handlers
     class Kintai < Handler
       config :query, type: String
+      config :mail_to, type: String, default: ''
+      config :mail_cc, type: String, default: ''
+      config :template_subject, type: String, default: ''
       config :template_header, type: String, default: ''
       config :template_footer, type: String, default: ''
+      config :template_info, type: String, default: ''
       config :schedule_cron, type: String, default: nil
       config :schedule_room, type: String, default: nil
 
       route /kintai/i, :kintai, command: true
+      route /^draft\s+(.+)/im, :draft, command: true
       route /^code\s+(.+)/, :code, command: true
 
       on :loaded, :load_on_start
-
-      OOB_URI = 'urn:ietf:wg:oauth:2.0:oob'
-      APPLICATION_NAME = 'Lita Kintai'
-      CLIENT_SECRETS_PATH = 'client_secret.json'
-      CREDENTIALS_PATH = File.join(Dir.home, '.credentials',
-                                   "lita-kintai.yaml")
-      SCOPE = Google::Apis::GmailV1::AUTH_GMAIL_READONLY
-      USER_ID = 'default'
+      on :slack_reaction_added, :reaction_added
 
       def kintai(response)
-        response.reply(current_kintai)
+        if Gmail.authorized?
+          register_draft(response, kintai_info)
+        else
+          response.reply(authenticate_info)
+        end
       end
 
-      def send_kintai(user: user, room: room)
-        target = Source.new(user: user, room: room)
-        robot.send_message(target, current_kintai)
+      def draft(response)
+        info = response.matches[0][0]
+        register_draft(response, info)
+      end
+
+      def register_draft(response, info)
+        mail = create_kintai_mail(info)
+        reply = response.reply(mail_to_message(mail))
+        @@draft = { channel: reply["channel"], ts: reply["ts"], mail: mail }
+        reply
+      end
+
+      def code(response)
+        code = response.matches[0][0]
+        Gmail.credentials_from_code(code)
+
+        response.reply("Confirmed")
       end
 
       def load_on_start(_payload)
@@ -51,119 +62,144 @@ module Lita
         end
       end
 
-      def code(response)
-        code = response.matches[0][0]
-        authorizer.get_and_store_credentials_from_code(
-          user_id: USER_ID, code: code, base_url: OOB_URI)
+      def send_kintai(user: user, room: room)
+        if Gmail.authorized?
+          mail = create_kintai_mail(kintai_info)
+          reply = send_message(user: user, room: room, message: mail)
+          @@draft = { channel: reply["channel"], ts: reply["ts"], mail: mail }
+          reply
+        else
+          send_message(user: user, room: room, message: authenticate_info)
+        end
+      end
 
-        response.reply("Confirmed")
+      def reaction_added(_payload)
+        case _payload[:name]
+        when '+1' then
+          if !@@draft.nil? &&
+            _payload[:item]["type"] == "message" &&
+            _payload[:item]["channel"] == @@draft[:channel] &&
+            _payload[:item]["ts"] == @@draft[:ts]
+            if Gmail.authorized?
+              # TODO: 成功失敗
+              send_mail(@@draft[:mail])
+              @@draft = nil
+              send_message(room: _payload[:item]["channel"],
+                message: 'Sent email.')
+            else
+              send_message(room: _payload[:item]["channel"],
+                message: authenticate_info)
+            end
+          end
+        end
       end
 
-      def current_kintai
-        if authorize.nil?
-          auth_url = authorizer.get_authorization_url(base_url: OOB_URI)
-          return <<-EOS
-Authenticate your Google account.
-Then tell me the code as follows: `code \#{your_code}`
+      def send_message(user: user, room: room, message: message)
+        target = Source.new(user: user, room: room)
+        robot.send_message(target, message)
+      end
+
+      def create_kintai_mail(info)
+        create_mail(
+          to: config.mail_to,
+          cc: config.mail_cc,
+          subject: kintai_subject,
+          body: kintai_body(info)
+        )
+      end
 
-#{auth_url}
-          EOS
+      def mail_to_message(mail)
+        <<-EOS
+To: #{mail.to}
+Cc: #{mail.cc}
+Subject: #{mail.subject}
+#{mail.body.to_s}
+        EOS
+      end
+
+      def create_mail(to: to, cc: cc, subject: subject, body: body)
+        Mail.new do
+          to to
+          cc cc
+          subject subject
+          body body
         end
+      end
 
-        kintai_info
+      def send_mail(mail)
+        return Gmail.send_message(mail)
       end
 
       def kintai_info
         texts = ""
-        texts << "#{Date.today.strftime("%m/%d")} (#{%w(日 月 火 水 木 金 土)[Date.today.wday]})#{config.template_header}"
 
-        mails = find_mail(config.query)
+        mails = Gmail.find_mail(config.query)
         # query の `newer:#{Date.today.strftime("%Y/%m/%d")}` 昨日のも一部返ってくる
         # `newer_than:1d` だと24h以内になるので、ここで今日のだけにする
         mails.select{ |m| m[:date] > Date.today.to_time }.each do |m|
           name = m[:from].split("\"")[1]
 
           text = m[:subject] + m[:body]
+          info = kintai_from_text(text)
 
-          reason = "私用のため、"
-          if text.match(/電車|列車/)
-            reason = "電車遅延のため、"
-          end
-          if text.match(/体調|痛/)
-            reason = "体調不良のため、"
-          end
-          if text.match(/健康診断|検診|健診/)
-            reason = "健康診断のため、"
-          end
-
-          at = "出社時刻未定です。"
-          if hm = text.match(/([0-1][0-9]|[2][0-3]):[0-5][0-9]/)
-            at = "#{hm}頃出社予定です。"
-          elsif min = text.match(/([0-5][0-9])分/)
-            at = "10:#{min[1]}頃出社予定です。"
-          end
-
-          if text.match(/おやすみ|休み|有給|休暇/)
-            reason = "本日お休みです。"
-            at = ""
-          end
-
-          texts << "#{name}さん: #{reason}#{at}\n"
+          texts << "#{name}さん: #{info}\n"
         end
-
-        texts << config.template_footer
+        texts << config.template_info
       end
 
-      def authorize
-        return @service if !@servise.nil? && !@servise.authorization.nil?
-
-        credentials = authorizer.get_credentials(USER_ID)
-        if credentials
-          @service.authorization = credentials
-          return @service
-        end
-
-        return nil
+      def self.kintai_from_text(text)
+        reason = kintai_reason(text)
+        time = kintai_time(text)
+        "#{reason}のため、#{time}です。"
       end
 
-      def authorizer
-        return @authorizer unless @authorizer.nil?
+      def kintai_subject
+        "#{Date.today.strftime("%m/%d")} (#{%w(日 月 火 水 木 金 土)[Date.today.wday]})#{config.template_subject}"
+      end
 
-        @service = Google::Apis::GmailV1::GmailService.new
-        @service.client_options.application_name = APPLICATION_NAME
+      def kintai_body(info)
+        <<-EOS
+#{config.template_header}
 
-        FileUtils.mkdir_p(File.dirname(CREDENTIALS_PATH))
+#{info}
 
-        client_id = Google::Auth::ClientId.from_file(CLIENT_SECRETS_PATH)
-        token_store = Google::Auth::Stores::FileTokenStore.new(
-          file: CREDENTIALS_PATH)
-        @authorizer = Google::Auth::UserAuthorizer.new(
-          client_id, SCOPE, token_store)
+#{config.template_footer}
+        EOS
       end
 
-      def find_mail(query)
-        ids = @service.list_user_messages('me', q: query)
-
-        return [] unless ids.messages
-        ids.messages.map do |message|
-          find_mail_by_id(message.id)
+      def self.kintai_reason(text)
+        if text.match(/電車|列車/)
+          return "電車遅延"
+        elsif text.match(/体調|痛/)
+          return "体調不良"
+        elsif text.match(/健康診断|検診|健診/)
+          return "健康診断"
         end
+        return  "私用"
       end
 
-      def find_mail_by_id(id)
-        results = @service.get_user_message('me', id)
+      def self.kintai_time(text)
+        if hm = text.match(/([0-1][0-9]|[2][0-3]):[0-5][0-9]/)
+          return "#{hm}頃出社予定"
+        elsif min = text.match(/(([0-5])*[0-9])分/)
+          return "10:#{min[1].rjust(2, "0")}頃出社予定"
+        elsif half = text.match(/([0-1][0-9]|[2][0-3])時半/)
+          return "#{half[1]}:30頃出社予定"
+        elsif half = text.match(/(\d)時間半/)
+          return "#{10+half[1].to_i}:30頃出社予定"
+        elsif text.match(/おやすみ|休み|有給|休暇/)
+          return "本日お休み"
+        end
+        return "出社時刻未定"
+      end
 
-        body = results.payload.parts ?
-          results.payload.parts.first.body.data :
-          results.payload.body.data
-        headers = results.payload.headers
+      def authenticate_info
+        <<-EOS
+Authenticate your Google account.
+Then tell me the code as follows: `code \#{your_code}`
 
-        {
-          subject: headers.select { |e| e.name == 'Subject'}.first.value,
-          from: headers.select { |e| e.name == 'From'}.first.value,
-          date: Time.parse(headers.select { |e| e.name == 'Date'}.first.value),
-          body: body.force_encoding('utf-8'),
-        }
+#{Gmail.authorization_url}
+        EOS
       end
 
       Lita.register_handler(self)

data/lita-kintai.gemspec

@@ -1,6 +1,6 @@
 Gem::Specification.new do |spec|
   spec.name          = "lita-kintai"
-  spec.version       = "0.2.0"
+  spec.version       = "0.3.0"
   spec.authors       = ["Shoma SATO"]
   spec.email         = ["noir.neo.04@gmail.com"]
   spec.description   = "A lita handler for summarize attendance emails. Fuckin' legacy attendance management method by email!"
@@ -17,6 +17,7 @@ Gem::Specification.new do |spec|
   spec.add_runtime_dependency "lita", ">= 4.7"
   spec.add_runtime_dependency "google-api-client"
   spec.add_runtime_dependency "rufus-scheduler"
+  spec.add_runtime_dependency "mail"
 
   spec.add_development_dependency "bundler", "~> 1.3"
   spec.add_development_dependency "pry-byebug"

data/spec/lita/handlers/kintai_spec.rb

@@ -9,8 +9,8 @@ describe Lita::Handlers::Kintai, lita_handler: true do
   describe '#kintai' do
     context 'when authenticated' do
       before do
-        allow_any_instance_of(Lita::Handlers::Kintai).to receive(:authorize).and_return(Google::Apis::GmailV1::GmailService.new)
-        allow_any_instance_of(Lita::Handlers::Kintai).to receive(:find_mail).and_return(
+        allow(Gmail).to receive(:authorized?).and_return(true)
+        allow(Gmail).to receive(:find_mail).and_return(
           [
             {
               subject: '',
@@ -22,10 +22,73 @@ describe Lita::Handlers::Kintai, lita_handler: true do
         )
       end
 
-      it 'returns kintai list' do
+      xit 'returns kintai list' do
         send_command('kintai')
         expect(replies.last).not_to be_nil
       end
     end
   end
+
+  describe '#kintai_reason(text)' do
+    subject { Lita::Handlers::Kintai.kintai_reason(text) }
+
+    context '「電車」が含まれる時' do
+      let(:text) { "電車遅延のため" }
+      it { is_expected.to eq "電車遅延" }
+    end
+
+    context '「体調不良」が含まれる時' do
+      let(:text) { "体調不良のため" }
+      it { is_expected.to eq "体調不良" }
+    end
+
+    context '「健康診断」が含まれる時' do
+      let(:text) { "健康診断に行くので" }
+      it { is_expected.to eq "健康診断" }
+    end
+
+    context 'いずれにもマッチしない時' do
+      let(:text) { "寝坊したので" }
+      it { is_expected.to eq "私用" }
+    end
+  end
+
+  describe '#kintai_time(text)' do
+    subject { Lita::Handlers::Kintai.kintai_time(text) }
+
+    context '「HH:mm」が含まれる時' do
+      let(:text) { "12:00頃出社します" }
+      it { is_expected.to eq "12:00頃出社予定" }
+    end
+
+    context '「mm分」が含まれる時' do
+      let(:text) { "10分ほど遅れます" }
+      it { is_expected.to eq "10:10頃出社予定" }
+    end
+
+    context '「m分」が含まれる時' do
+      let(:text) { "5分ほど遅れます" }
+      it { is_expected.to eq "10:05頃出社予定" }
+    end
+
+    context '「HH時半」が含まれる時' do
+      let(:text) { "10時半ごろに出社します" }
+      it { is_expected.to eq "10:30頃出社予定" }
+    end
+
+    context '「n時間半」が含まれる時' do
+      let(:text) { "1時間半ほど遅れます" }
+      it { is_expected.to eq "11:30頃出社予定" }
+    end
+
+    context '「休み」が含まれる時' do
+      let(:text) { "お休みをいただいてます" }
+      it { is_expected.to eq "本日お休み" }
+    end
+
+    context 'いずれにもマッチしない時' do
+      let(:text) { "出社時刻わかり次第また連絡します" }
+      it { is_expected.to eq "出社時刻未定" }
+    end
+  end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: lita-kintai
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.3.0
 platform: ruby
 authors:
 - Shoma SATO
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-11-27 00:00:00.000000000 Z
+date: 2017-04-16 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: lita
@@ -52,6 +52,20 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: mail
+  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: bundler
   requirement: !ruby/object:Gem::Requirement
@@ -134,6 +148,7 @@ files:
 - Gemfile
 - README.md
 - Rakefile
+- lib/gmail.rb
 - lib/lita-kintai.rb
 - lib/lita/handlers/kintai.rb
 - lita-kintai.gemspec