embulk-input-slack_message 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 1f24a85713970b8586eaccf14732e5462b4618be
+  data.tar.gz: 19fc15c2631d750c37c488a63c57bcb3d60803cf
+SHA512:
+  metadata.gz: 593e59cc7e4d1a424d8c09c108300f4fa5d50973b45fa053e51506366217e8c359d9a8b4dda3f67bae3c92b810f355d634e22803cd7bfd37a5874cc04388bc36
+  data.tar.gz: 97d36c2cc8b58ba63ccb611139dd9f4cf847ac227d1285fa0df9178c321bff29d54451c3ae35600ddb4edc82fa71d07556ca45c1704ee1f72dba9325a61e0212

data/.gitignore

@@ -0,0 +1,6 @@
+*~
+/pkg/
+/tmp/
+/.bundle/
+/Gemfile.lock
+/vendor/bundle

data/.ruby-version

@@ -0,0 +1 @@
+jruby-9.0.4.0

data/Gemfile

@@ -0,0 +1,2 @@
+source 'https://rubygems.org/'
+gemspec

data/LICENSE.txt

@@ -0,0 +1,21 @@
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,74 @@
+# Slack Message input plugin for Embulk
+
+This plugin is embulk input from slack message.
+Inspire of the [embulk-input-slack_history](https://github.com/yaggytter/embulk-input-slack-history/blob/master/lib/embulk/input/slack_history.rb)
+
+## Overview
+
+* **Plugin type**: input
+* **Resume supported**: no
+* **Cleanup supported**: no
+* **Guess supported**: yes
+
+## Advanced
+
+### Complete use name
+
+User name be id come from slack in message.
+
+This plugin complete user name.
+
+### Guess supported
+
+This plugin guess supported.
+
+Calculate between time from latest and oldest.
+
+## Configuration
+
+channel settings is references slack description ([here](https://api.slack.com/methods/channels.history))
+
+- **channel**: channel settings (hash, required)
+  - name: channel name (string, required)
+  - type: channel type (string, required. this param is originated)
+    - channel: public channel
+    - group: private group
+    - direct: direct message
+    - multi_direct: multi direct message
+  - latest: End of time range of messages to include in results. (string, default: now date)
+  - oldest: Start of time range of messages to include in results. (string, default: 0)
+  - count: Number of messages to return, between 1 and 1000 (long, default: 100)
+  - inclusive: Include messages with latest or oldest timestamp in results (long, default: 0)
+  - unreads: Include `unread_count_display` in the output? (long, default: 0)
+- **token**: Slack API Token. Generate from [Slack Authentication](https://api.slack.com/web) (string, required)
+- **repeat**: Adjustment of time when next config generation (mill seconds). (long, default: 0)
+
+## Example
+
+```yaml
+in:
+  type: slack_message
+  channel:
+    name: general
+    type: channel
+    latest: '2016-02-16 00:00:00'
+    oldest: '2016-02-15 00:00:00'
+    count: 100
+    inclusive: 0
+    unreads: 0
+  token: 'aaaa-1111111111-2222222222-33333333333-bbbbbbbbbb'
+  repeat: 0
+  columns:
+  - {index: 0, name: datetime, type: timestamp, format: null}
+  - {index: 1, name: channel_id, type: string, format: null}
+  - {index: 2, name: channel_name, type: string, format: null}
+  - {index: 3, name: user_id, type: string, format: null}
+  - {index: 4, name: user_name, type: string, format: null}
+  - {index: 5, name: message, type: string, format: null}
+```
+
+## Build
+
+```
+$ rake
+```

data/Rakefile

@@ -0,0 +1,3 @@
+require "bundler/gem_tasks"
+
+task default: :build

data/embulk-input-slack_message.gemspec

@@ -0,0 +1,22 @@
+
+Gem::Specification.new do |spec|
+  spec.name          = "embulk-input-slack_message"
+  spec.version       = "0.1.0"
+  spec.authors       = ["yuemori"]
+  spec.summary       = "Slack Message input plugin for Embulk"
+  spec.description   = "Loads records from Slack Message."
+  spec.email         = ["yuemori@aiming-inc.com"]
+  spec.licenses      = ["MIT"]
+  spec.homepage      = "https://github.com/yuemori/embulk-input-slack_message"
+
+  spec.files         = `git ls-files`.split("\n") + Dir["classpath/*.jar"]
+  spec.test_files    = spec.files.grep(%r{^(test|spec)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency 'rest-client'
+  spec.add_dependency 'activesupport'
+  spec.add_development_dependency 'embulk', ['>= 0.8.3']
+  spec.add_development_dependency 'bundler', ['>= 1.10.6']
+  spec.add_development_dependency 'rake', ['>= 10.0']
+  spec.add_development_dependency 'pry'
+end

data/lib/embulk/input/slack_message.rb

@@ -0,0 +1,93 @@
+require 'slack'
+
+module Embulk
+  module Input
+    class SlackMessage < InputPlugin
+      Plugin.register_input("slack_message", self)
+
+      def self.transaction(config, &control)
+        task = {
+          'channel' => config.param('channel', :hash),
+          'token' => config.param('token', :string),
+          'repeat_at' => config.param('repeat_at', :long, default: 0)
+        }
+
+        yield(task, columns, 1)
+
+        channel = task['channel']
+        repeat_at = task['repeat_at']
+        latest = channel['latest'] ? Time.parse(channel['latest']).to_f : Time.now.to_f
+        oldest = channel['oldest'] ? Time.parse(channel['oldest']).to_f : 0
+
+        if repeat_at.zero?
+          time_diff = latest.to_i - oldest.to_i
+          channel = { oldest: Time.at(latest), latest: Time.at(latest + time_diff) }
+        else
+          channel = { oldest: oldest + repeat_at, latest: latest + repeat_at }
+        end
+
+        { channel: channel }
+      end
+
+      def self.columns
+        [
+          Column.new(0, "datetime", :timestamp),
+          Column.new(1, "channel_id", :string),
+          Column.new(2, "channel_name", :string),
+          Column.new(3, "user_id", :string),
+          Column.new(4, "user_name", :string),
+          Column.new(5, "message", :string)
+        ]
+      end
+
+      def self.guess(config)
+        channels = [
+          {
+            name: 'general',
+            type: 'channel',
+            latest: Time.now.strftime('%F %T'),
+            oldest: 0,
+            count: 100,
+            inclusive: 0,
+            unreads: 0
+          }
+        ]
+        { channel: channels.first, token: 'SLACK_API_TOKEN', repeat: 0, columns: Embulk::Schema.new(self.columns) }
+      end
+
+      def init
+        @channel = task["channel"]
+
+        token = task['token']
+        raise StandardError.new, 'slack token is not found' unless token
+
+        Slack.token = token
+      end
+
+      def run
+        latest = @channel['latest'] ? Time.parse(@channel['latest']).to_f : Time.now.to_f
+        oldest = @channel['oldest'] ? Time.parse(@channel['oldest']).to_f : 0
+        inclusive = @channel['inclusive'] || 0
+        count = @channel['count'] || 100
+        unreads = @channel['unreads'] || 0
+
+        options = {
+          latest: latest,
+          oldest: oldest,
+          inclusive: inclusive,
+          count: count,
+          unreads: unreads
+        }
+
+        Slack.messages(@channel['name'], @channel['type'], options).each do |message|
+          page_builder.add message
+        end
+
+        page_builder.finish
+
+        task_report = {}
+        return task_report
+      end
+    end
+  end
+end

data/lib/slack.rb

@@ -0,0 +1,22 @@
+require 'active_support'
+require 'active_support/core_ext'
+require 'rest-client'
+require 'json'
+require 'slack/client'
+
+module Slack
+  class << self
+    attr_accessor :token
+    delegate :channels, :messages, :users, :options=, to: :client
+
+    def configure(&block)
+      yield configuration
+    end
+
+    private
+
+    def client
+      @client = Slack::Client.new(token)
+    end
+  end
+end

data/lib/slack/client.rb

@@ -0,0 +1,73 @@
+module Slack
+  class Client
+    attr_reader :options, :token
+
+    def initialize(token)
+      @token = token
+    end
+
+    def messages(name, type, options)
+      channel = case type
+      when 'channel' then channels.find { |item| item['name'] == name }
+      when 'group' then groups.find { |item| item['name'] == name }
+      when 'direct' then im.find { |item| item['user'] == name }
+      when 'multi_direct' then mpim.find { |item| item['name'] == name }
+      end
+      messages = JSON.parse(get('channels.history', options.merge(channel: channel['id'])))['messages']
+      messages.select { |message| message['type'] == 'message' && message['subtype'].nil? }.map do |message|
+        [
+          Time.at(message['ts'].to_i),
+          channel['id'],
+          channel['name'],
+          message['user'],
+          user_by_id(message['user'])['name'],
+          filter_message(message['text'])
+        ]
+      end.reverse
+    end
+
+    def filter_message(text)
+      text = text.dup
+      scan = text.scan(/\<@[0-9A-Z]+\>/)
+      return text unless scan
+
+      scan.each do |pattern|
+        user_id = pattern.match(/<@(?<user_id>.*)>/)[:user_id]
+        text.gsub!(/#{pattern}/, "@#{user_by_id(user_id)['name']}")
+      end
+      text
+    end
+
+    def user_by_id(id)
+      users.find { |user| user['id'] == id }
+    end
+
+    def user_by_name(name)
+      users.find { |user| user['name'] == name }
+    end
+
+    def users
+      @users ||= JSON.parse(get('users.list', { presence: 0 }))['members']
+    end
+
+    def channels
+      @channels ||= JSON.parse(get('channels.list', { exclude_archived: 1 }))['channels']
+    end
+
+    def groups
+      @groups ||= JSON.parse(get('groups.list', { exclude_archived: 1 }))['groups']
+    end
+
+    def ims
+      @ims ||= JSON.parse(get('im.list'))['ims']
+    end
+
+    def mpim
+      @mpims ||= JSON.parse(get('mpim.list'))['groups']
+    end
+
+    def get(api, options = {})
+      RestClient.get("https://slack.com/api/#{api}", { params: options.merge(token: token) })
+    end
+  end
+end

metadata

@@ -0,0 +1,138 @@
+--- !ruby/object:Gem::Specification
+name: embulk-input-slack_message
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- yuemori
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2016-02-15 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rest-client
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  prerelease: false
+  type: :runtime
+- !ruby/object:Gem::Dependency
+  name: activesupport
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  prerelease: false
+  type: :runtime
+- !ruby/object:Gem::Dependency
+  name: embulk
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.8.3
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.8.3
+  prerelease: false
+  type: :development
+- !ruby/object:Gem::Dependency
+  name: bundler
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.10.6
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.10.6
+  prerelease: false
+  type: :development
+- !ruby/object:Gem::Dependency
+  name: rake
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  prerelease: false
+  type: :development
+- !ruby/object:Gem::Dependency
+  name: pry
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  prerelease: false
+  type: :development
+description: Loads records from Slack Message.
+email:
+- yuemori@aiming-inc.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".ruby-version"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- embulk-input-slack_message.gemspec
+- lib/embulk/input/slack_message.rb
+- lib/slack.rb
+- lib/slack/client.rb
+homepage: https://github.com/yuemori/embulk-input-slack_message
+licenses:
+- MIT
+metadata: {}
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project:
+rubygems_version: 2.4.8
+signing_key:
+specification_version: 4
+summary: Slack Message input plugin for Embulk
+test_files: []