rslack 0.2.5

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: ed64057e1a6a7e9a5c3541d6ec477c551b480ce6
+  data.tar.gz: 9b0f306e0a6a311e4c862b58b48c92eee20e7f09
+SHA512:
+  metadata.gz: e3951ed8d9a74f734bdf9294009a24a88d8d1a4ed2c5e471ece1da4028623e6d93522f76bf820f0aee9ebbe92ddeefedf9d21d6119068577ae831225aaddd063
+  data.tar.gz: e904de9637818adf05ad7cf9593d106a5631ba1e88dcea2e98b8687e59745ff3fd91f8306df114c5071859b0d9fbce36a323001e410e812739560ddf12d3972b

data/.gitignore

@@ -0,0 +1,11 @@
+/.env
+/.bundle/
+/.yardoc
+/Gemfile.lock
+/_yardoc/
+/coverage/
+/doc/
+/pkg/
+/spec/reports/
+/tmp/
+/vendor/

data/Gemfile

@@ -0,0 +1,6 @@
+source "https://rubygems.org"
+
+git_source(:github) {|repo_name| "https://github.com/#{repo_name}" }
+
+# Specify your gem's dependencies in rslack.gemspec
+gemspec

data/README.md

@@ -0,0 +1,35 @@
+# RSlack
+
+Welcome to your new gem! In this directory, you'll find the files you need to be able to package up your Ruby library into a gem. Put your Ruby code in the file `lib/rslack`. To experiment with that code, run `bin/console` for an interactive prompt.
+
+TODO: Delete this and the text above, and describe your gem
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+```ruby
+gem 'rslack'
+```
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install rslack
+
+## Usage
+
+TODO: Write usage instructions here
+
+## Development
+
+After checking out the repo, run `bin/setup` to install dependencies. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
+
+To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and tags, and push the `.gem` file to [rubygems.org](https://rubygems.org).
+
+## Contributing
+
+Bug reports and pull requests are welcome on GitHub at https://github.com/[USERNAME]/rslack.

data/Rakefile

@@ -0,0 +1,2 @@
+require "bundler/gem_tasks"
+task :default => :spec

data/bin/console

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+
+require "bundler/setup"
+require "rslack"
+
+# You can add fixtures and/or initialization code here to make experimenting
+# with your gem easier. You can also use a different console, if you like.
+
+# (If you use this, don't forget to add pry to your Gemfile!)
+# require "pry"
+# Pry.start
+
+require "irb"
+IRB.start(__FILE__)

data/bin/rslack

@@ -0,0 +1,6 @@
+#! /usr/bin/env ruby
+
+$LOAD_PATH.unshift File.expand_path("../../lib", __FILE__)
+require 'rslack'
+
+RSlack.run

data/bin/setup

@@ -0,0 +1,8 @@
+#!/usr/bin/env bash
+set -euo pipefail
+IFS=$'\n\t'
+set -vx
+
+bundle install
+
+# Do any other automated setup that you need to do here

data/lib/rslack.rb

@@ -0,0 +1,46 @@
+require 'bundler'
+require 'dotenv'
+require 'optparse'
+require 'ostruct'
+
+module RSlack
+
+  class << self
+
+    def run(args = ARGV)
+      bot = build(args)
+
+      Signal.trap(:INT) do
+        bot.force_stop
+        exit 0
+      end
+
+      bot.start
+    end
+
+    private
+    def build(args)
+      Dotenv.load
+      Bundler.require
+
+      config = OpenStruct.new
+      option = OptionParser.new
+      option.on('--log FILE',    'Path to log file.') { |v| config.logfile = v }
+      option.on('--token TOKEN', 'Slack API token.')  { |v| config.token   = v }
+      option.parse!(args)
+
+      config.token ||= ENV['SLACK_API_TOKEN']
+      RSlack::Bot.new(config.token, log: config.logfile)
+    end
+
+  end
+
+end
+
+require 'rslack/bot'
+require 'rslack/action'
+require 'rslack/actions/cron'
+require 'rslack/actions/echo'
+require 'rslack/actions/ping'
+require 'rslack/actions/help'
+require 'rslack/version'

data/lib/rslack/action.rb

@@ -0,0 +1,37 @@
+require 'ostruct'
+require 'singleton'
+
+module RSlack
+
+  class Action
+
+    class << self
+
+      def route(regexp, **opts, &block)
+        case callback = opts[:callback]
+        when nil  then opts[:callback] = block
+        when Proc then nil
+        when String, Symbol
+          opts[:callback] = lambda { |msg| send(callback, msg) }
+        end
+
+        route = OpenStruct.new(**opts.merge(action: self, regexp: regexp))
+        RSlack::Bot.register(route)
+      end
+
+    end
+
+
+    attr_reader :bot
+
+    def initialize(bot)
+      @bot = bot
+    end
+
+    def memory
+      bot.memory[self.class.name] ||= {}
+    end
+
+  end
+
+end

data/lib/rslack/actions/cron.rb

@@ -0,0 +1,115 @@
+require 'stringio'
+require 'parse-cron'
+
+module RSlack
+
+  class CronAction < Action
+
+    route /cron add (?<syntax>\S+ \S+ \S+ \S+ \S+) (?<command>.+)$/,
+      helptext: 'Add cron schedule.',
+      callback: 'add'
+
+    route /cron del (?<id>\S+)/,
+      helptext: 'Delete cron schedule.',
+      callback: 'del'
+
+    route /cron list/,
+      helptext: 'List cron schedules.',
+      callback: 'list'
+
+    class << self
+      attr_accessor :timer
+    end
+
+    def initialize(bot)
+      super(bot)
+      @channel = {}
+    end
+
+    def add(msg)
+      start(msg.dup)
+
+      cron_id = unique_id
+      memory.update(cron_id => msg.dup)
+
+      bot.reply(msg, text: "_added schedule_ `[#{cron_id}] #{msgtostr(msg)}`")
+    end
+
+    def del(msg)
+      stop
+
+      cron_id = msg.var.id.to_i
+      deleted = memory.delete(cron_id)
+      raise "no such cron id: #{cron_id}" unless deleted
+
+      bot.reply(msg, text: "_deleted schedule_ `[#{cron_id}] #{msgtostr(deleted)}`")
+    end
+
+    def list(msg)
+      items = memory.select { |_, item| msg.channel == item.channel }
+
+      unless items.empty?
+        content = StringIO.new
+        content.puts "```"
+        content.puts items.map { |cron_id, item| "[#{cron_id}] #{msgtostr(item)}" }
+        content.puts "```"
+
+        bot.reply(msg, text: content.string)
+      else
+        bot.reply(msg, text: "_no schedules found_.")
+      end
+    end
+
+    def start(msg)
+      parser = CronParser.new(msg.var.syntax)
+      sleep  = parser.next(Time.now) - Time.now
+
+      CronAction.timer = EM::Timer.new(sleep) do
+        item = OpenStruct.new(msg.to_h.merge(text: msg.var.command))
+        item.delete_field(:var)
+        bot.react(item)
+        CronAction.timer = start(msg)
+      end
+    end
+
+    def stop
+      CronAction.timer.cancel
+    end
+
+    private
+    def msgtostr(msg)
+      @channel[msg.channel] ||= bot.client.channel(msg.channel).dig('channel', 'name')
+      "##{@channel[msg.channel]} #{msg.var.syntax} #{msg.var.command}"
+    end
+
+    def unique_id
+      loop do
+        cron_id = (1000..9999).to_a.sample
+        return cron_id unless memory.key?(cron_id)
+      end
+    end
+
+  end
+
+  class Bot
+
+    module CronApplicable
+
+      def wait_for_message
+        crons = memory['RSlack::CronAction']
+        unless crons.empty?
+          crons.each do |_, item|
+            action = RSlack::CronAction.new(self)
+            action.start(item)
+          end
+          logger.info("#{self.class} cron schedules loaded.")
+        end
+
+        super
+      end
+    end
+    prepend CronApplicable
+
+  end
+
+end

data/lib/rslack/actions/echo.rb

@@ -0,0 +1,15 @@
+module RSlack
+
+  class EchoAction < Action
+
+    route /echo (?<text>.+)$/,
+      helptext: 'Echo your message.',
+      callback: 'echo'
+
+    def echo(msg)
+      bot.reply(msg, text: msg.var.text)
+    end
+
+  end
+
+end

data/lib/rslack/actions/help.rb

@@ -0,0 +1,32 @@
+require 'stringio'
+
+module RSlack
+
+  class HelpAction < Action
+
+    route /help(?<options> --all)?/,
+      helptext: 'Show this message.',
+      callback: 'help'
+
+    def help(msg)
+      options = msg.var.options
+      routes  = bot.routes.select { |r| r.helptext || options.to_s.split.include?('--all') }
+
+      content = StringIO.new
+      content.puts "```"
+      routes.map do |route|
+        prefix = [(bot.name unless route.matchall), route.regexp.inspect].compact.join(' ') + ':'
+        indent = (prefix.size / 4 + 1) * 4
+
+        content.print "#{prefix.ljust(indent)}"
+        content.print "`#{route.helptext}`" if route.helptext
+        content.puts
+      end
+      content.puts "```"
+
+      bot.reply(msg, text: content.string)
+    end
+
+  end
+
+end

data/lib/rslack/actions/ping.rb

@@ -0,0 +1,15 @@
+module RSlack
+
+  class PingAction < Action
+
+    route /ping/,
+      callback: 'pong',
+      matchall: true
+
+    def pong(msg)
+      bot.reply(msg, text: "<@#{msg.user}> pong")
+    end
+
+  end
+
+end

data/lib/rslack/api.rb

@@ -0,0 +1,42 @@
+require 'http'
+require 'json'
+
+module RSlack
+
+  class APIClient
+
+    BASEURL = "https://slack.com/api/"
+
+    attr_accessor :token
+
+    def initialize(token)
+      @token = token
+    end
+
+    def rtm
+      call('rtm.start')
+    end
+
+    def auth
+      call('auth.test')
+    end
+
+    def post(text: '', **params)
+      call('chat.postMessage', params.merge(text: text))
+    end
+
+    def user(id)
+      call('users.info', user: id)
+    end
+
+    def channel(id)
+      call('channels.info', channel: id)
+    end
+
+    def call(endpoint, **params)
+      resp = HTTP.post(BASEURL + endpoint, params: { token: @token }.merge(params))
+      JSON.parse(resp.body)
+    end
+  end
+
+end

data/lib/rslack/bot.rb

@@ -0,0 +1,149 @@
+require 'eventmachine'
+require 'faye/websocket'
+require 'rslack/api'
+
+require 'json'
+require 'leveldb'
+require 'logger'
+require 'thread'
+require 'ostruct'
+
+module RSlack
+
+  class Bot
+
+    class << self
+
+      def register(route)
+        routes.push(route)
+      end
+
+      private
+      def routes
+        @routes ||= []
+      end
+
+    end
+
+    attr_reader :client
+    attr_reader :logger
+    attr_reader :memory
+    attr_reader :routes
+
+    STORAGE_KEYNAME  = 'rslack'
+    STORAGE_INTERVAL = 5
+
+    def initialize(token, log: nil)
+      @client  = APIClient.new(token)
+      @logger  = Logger.new(log || STDOUT)
+      @memory  = {}
+
+      @routes  = self.class.instance_variable_get(:@routes)
+      @storage = LevelDB::DB.new(File.join(Dir.pwd, "#{STORAGE_KEYNAME}.db"))
+
+      @auth = @client.auth
+      unless @auth['ok']
+        @logger.fatal("#{self.class} Failed to access slack web api.")
+        exit 1
+      end
+    end
+
+    def name
+      "@#{@auth['user']}"
+    end
+
+    def linked_name
+      "<@#{@auth['user_id']}>"
+    end
+
+    def reply(msg, **opts)
+      @client.post(**msg.to_h.merge(**opts))
+    end
+
+    def react(msg)
+      count = 0
+      @routes.each do |route|
+        matched = route.regexp.match(msg.text)
+        next unless matched
+        next unless route.matchall || msg.prefix == linked_name
+
+        count += 1
+        message = msg.dup
+        message.var = OpenStruct.new(matched.named_captures)
+        logger.info("#{self.class} passed msg to: '#{route.action}'.")
+
+        th = Thread.start do
+          begin
+            action = route.action.new(self)
+            action.instance_exec(message.freeze, &route.callback)
+          rescue => e
+            reply(msg, text: "_#{e}._")
+          end
+        end
+        th.abort_on_exception = true
+      end
+
+      reply(msg, text: 'I have no idea what to do...') if msg.prefix == linked_name && count.zero?
+    end
+
+    def start
+      load_storage
+      EM.run { wait_for_message }
+    end
+
+    def stop
+      EM.stop
+      dump_storage
+    end
+
+    def force_stop
+      EM.stop
+    end
+
+    def restart
+      stop
+      start
+    end
+
+    private
+    def wait_for_message
+      EM.add_periodic_timer(STORAGE_INTERVAL) { dump_storage }
+
+      url = @client.rtm['url']
+      rtm = Faye::WebSocket::Client.new(url)
+
+      rtm.on(:open) do |event|
+        logger.info("#{self.class} slack realtime api connection has opened.")
+      end
+
+      rtm.on(:close) do |event|
+        logger.info("#{self.class} slack realtime api connection has closed.")
+        restart if EM.reactor_running?
+      end
+
+      rtm.on(:message) do |event|
+        msg = OpenStruct.new(JSON.parse(event.data))
+        if msg.type == 'message' && msg.subtype.nil?
+          logger.info("#{self.class} recieved msg: '#{msg.text}'.")
+
+          if msg.text =~ /^\s*(#{linked_name})\s*(.*)$/
+            msg.prefix = $1.to_s
+            msg.text =   $2.to_s
+          end
+
+          react(msg)
+        end
+      end
+    end
+
+    def dump_storage
+      @storage[STORAGE_KEYNAME] = Marshal.dump(@memory)
+    end
+
+    def load_storage
+      @memory = Marshal.load(@storage[STORAGE_KEYNAME]) if @storage[STORAGE_KEYNAME]
+    end
+
+  end
+
+end

data/lib/rslack/version.rb

@@ -0,0 +1,3 @@
+module RSlack
+  VERSION = "0.2.5"
+end

data/rslack.gemspec

@@ -0,0 +1,34 @@
+# coding: utf-8
+lib = File.expand_path("../lib", __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require "rslack/version"
+
+Gem::Specification.new do |spec|
+  spec.name          = "rslack"
+  spec.version       = RSlack::VERSION
+  spec.authors       = ["haccht"]
+  spec.email         = ["haccht@users.noreply.github.com"]
+
+  spec.summary       = %q{super simple ruby slack bot framework.}
+  spec.description   = %q{super simple ruby slack bot framework.}
+  spec.homepage      = "http://github.com/haccht/rslack"
+
+  spec.files         = `git ls-files -z`.split("\x0").reject do |f|
+    f.match(%r{^(test|spec|features)/})
+  end
+  spec.bindir        = "bin"
+  spec.executables   = ["rslack"]
+  spec.require_paths = ["lib"]
+
+  spec.add_development_dependency "bundler", "~> 1.15"
+  spec.add_development_dependency "rake", "~> 10.0"
+
+  spec.add_dependency "bundler"
+  spec.add_dependency "dotenv"
+  spec.add_dependency "eventmachine"
+  spec.add_dependency "faye-websocket"
+  spec.add_dependency "http"
+  spec.add_dependency "parse-cron"
+  spec.add_dependency "leveldb"
+  spec.add_dependency "rake"
+end

metadata

@@ -0,0 +1,201 @@
+--- !ruby/object:Gem::Specification
+name: rslack
+version: !ruby/object:Gem::Version
+  version: 0.2.5
+platform: ruby
+authors:
+- haccht
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2018-03-05 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.15'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.15'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  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: dotenv
+  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: eventmachine
+  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: faye-websocket
+  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: http
+  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: parse-cron
+  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: leveldb
+  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: rake
+  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'
+description: super simple ruby slack bot framework.
+email:
+- haccht@users.noreply.github.com
+executables:
+- rslack
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- Gemfile
+- README.md
+- Rakefile
+- bin/console
+- bin/rslack
+- bin/setup
+- lib/rslack.rb
+- lib/rslack/action.rb
+- lib/rslack/actions/cron.rb
+- lib/rslack/actions/echo.rb
+- lib/rslack/actions/help.rb
+- lib/rslack/actions/ping.rb
+- lib/rslack/api.rb
+- lib/rslack/bot.rb
+- lib/rslack/version.rb
+- rslack.gemspec
+homepage: http://github.com/haccht/rslack
+licenses: []
+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.6.11
+signing_key: 
+specification_version: 4
+summary: super simple ruby slack bot framework.
+test_files: []