qbert_bot 0.0.1

checksums.yaml

@@ -0,0 +1,15 @@
+---
+!binary "U0hBMQ==":
+  metadata.gz: !binary |-
+    ZDNiYWFhNTFhMDEwYzVkNGI4OTc0ODI0Zjc4N2MwNDhkNTQzODQ3Nw==
+  data.tar.gz: !binary |-
+    YTk5OWI0YTcyODFjNmYzZDVkMmI5OWI0N2E4NjM0YjFlNjdjNTY5NA==
+SHA512:
+  metadata.gz: !binary |-
+    MmE0NjY5YjMzMzYwYjkwZWJmMDUxNjRlNmEzZDdkZTgxYzUxYTllYjBjN2Y2
+    OTU2ZjU2MTFjNWI1NzZmOTE5MmUxYzU1OWViYzJlNWEyMzZiOWNkOGI3YmZl
+    MDAyYjk2OWVhOTk2MmM1ODkyZTg3NDg5M2VmZjA5MGFkM2QxN2Y=
+  data.tar.gz: !binary |-
+    NjdmY2E5ODgwYzliZjAxYmMyM2FmMmIzNmZmOGM5MDY2YzZjZmIzNWY0Yzgx
+    MjMzM2YxMzRiYmYzMGE5NjQwNmM5NWYzYTA4N2MzZGI5ODg2YTYwYzQwY2Zi
+    Nzc0ZDg4M2Q4MGYzODZhNTJhZmM0ZjI5MGY4Y2VmMDRhMmRkMzc=

data/bin/qbert_bot

@@ -0,0 +1,3 @@
+#!/usr/bin/env ruby
+require 'qbert_bot'
+QbertBot.run!

data/lib/qbert_bot/message.rb

@@ -0,0 +1,28 @@
+module QbertBot
+  class Message
+    attr_accessor :token, :team_id, :channel_id, :channel_name, :timestamp,
+      :user_id, :user_name, :raw_text, :trigger_word
+    attr_accessor :matches
+
+    def initialize(bot, post_data)
+      @bot = bot
+      [:token, :team_id, :channel_id, :channel_name, :timestamp,
+       :user_id, :user_name, :trigger_word].each do |k|
+        send("#{k.to_s}=", post_data[k])
+      end
+      self.raw_text = post_data[:text]
+    end
+
+    def say(*args)
+      @bot.say(*args)
+    end
+
+    def reply(text, opts = {})
+      say(channel_name, text, opts)
+    end
+
+    def text
+      raw_text.sub(trigger_word, '').strip
+    end
+  end
+end

data/lib/qbert_bot/plugin.rb

@@ -0,0 +1,17 @@
+module QbertBot
+  module Plugin
+    def self.included(klass)
+      @plugins ||= []
+      @plugins << klass
+      puts "Loading plugin #{klass}"
+    end
+
+    def self.plugins
+      @plugins
+    end
+
+    attr_accessor :bot, :slack, :router, :scheduler
+    def register
+    end
+  end
+end

data/lib/qbert_bot/robot.rb

@@ -0,0 +1,116 @@
+require 'singleton'
+require 'sinatra/base'
+require 'yaml'
+require 'rufus-scheduler'
+
+module QbertBot
+  Listener = Struct.new(:pattern, :proc)
+  BlankListener = Listener.new(//, proc {})
+
+  class Match
+    attr_reader :listener, :matches
+    def initialize(listener, matches)
+      @listener = listener
+      @matches = matches
+    end
+
+    def proc
+      @listener.proc
+    end
+  end
+
+  class Robot
+    include Singleton
+    attr_reader :config, :slack, :router, :scheduler
+
+    def initialize
+      @plugins = []
+      @listeners = []
+    end
+
+    def load_config
+      @config = YAML.load_file('qbert.yml')
+    end
+
+    def run!
+      prepare_workers
+      load_plugins
+      start_workers
+    end
+
+    def hear(pattern, &block)
+      puts("Adding #{pattern.source}")
+      @listeners << Listener.new(pattern, block)
+    end
+
+    def handle_web(method, request, params)
+      if method == :post && request.path_info == @router.endpoint
+        return handle_slack(params)
+      end
+
+      @router.exec_route(method, request, params)
+    end
+
+    def handle_slack(params)
+      msg = Message.new(self, params)
+      match = find_listener(msg)
+      msg.matches = match.matches
+      match.proc.call(msg)
+      ''
+    end
+
+    def say(*args)
+      @slack.say(*args)
+    end
+
+    private
+    def prepare_workers
+      @scheduler = Rufus::Scheduler.new
+      @router = Router.new(config['http'])
+      @slack = Slack.new(config['slack'])
+    end
+
+    def load_plugins
+      Plugin.plugins.each do |klass|
+        p = klass.new
+        p.bot = self
+        p.router = router
+        p.slack = slack
+        p.scheduler = scheduler
+        p.register
+        @plugins << p
+      end
+    end
+
+    def start_workers
+      Thread.new {
+        @scheduler.join
+      }
+      Web.run!
+    end
+
+    def find_listener(msg)
+      puts("Checking '#{msg.text}'")
+      @listeners.each do |l|
+        if match = msg.text.match(l.pattern)
+          puts("Found '#{l.pattern.source}'")
+          return Match.new(l, match)
+        end
+      end
+      return Match.new(BlankListener, [])
+    end
+  end
+
+  class Web < Sinatra::Base
+    set :port, proc { Robot.instance.router.port }
+    set :bind, proc { Robot.instance.router.bind }
+
+    get(//) do
+      Robot.instance.handle_web(:get, request, params)
+    end
+
+    post(//) do
+      Robot.instance.handle_web(:post, request, params)
+    end
+  end
+end

data/lib/qbert_bot/router.rb

@@ -0,0 +1,47 @@
+module QbertBot
+  Route = Struct.new(:method, :path, :proc)
+
+  class Router
+    attr_reader :routes, :port, :bind, :endpoint
+
+    def initialize(conf)
+      @routes = []
+      @bind = conf['bind']
+      @port = conf['port']
+      @endpoint = conf['endpoint']
+    end
+
+    def get(path, &block)
+      add_route(:get, path, &block)
+    end
+
+    def post(path, &block)
+      add_route(:post, path, &block)
+    end
+
+    def exec_route(method, request, params)
+      @routes.each do |route|
+        next unless route.method == method
+        next unless match = request.path_info.match(route.path)
+        return route.proc.call(request, params, match)
+      end
+      return 'no route'
+    end
+
+    private
+    def add_route(method, path, &block)
+      clean_path = nil
+
+      if path.is_a?(Regexp)
+        clean_path = path.source
+      elsif path.is_a?(String)
+        clean_path = Regexp.escape(path)
+        clean_path = "/#{clean_path}" unless clean_path[0] == '/'
+        clean_path = "#{clean_path}$" unless clean_path[-1, 1] == '$'
+      end
+
+      puts("Adding route for #{method} #{clean_path}")
+      @routes << Route.new(method, Regexp.new(clean_path), block)
+    end
+  end
+end

data/lib/qbert_bot/slack.rb

@@ -0,0 +1,37 @@
+require 'faraday'
+require 'json'
+
+module QbertBot
+  class Slack
+    attr_reader :hook, :default_name, :default_icon
+    def initialize(conf)
+      @hook = conf['hook']
+      @default_name = conf['name']
+      @default_icon = conf['icon']
+    end
+
+    def channel_or_dm(name)
+      return name if name[0] == '#'
+      return name if name[0] == '@'
+      "##{name}"
+    end
+
+
+    def say(to, text, opts = {})
+      payload = {
+        text: text,
+        channel: channel_or_dm(to),
+        username: default_name,
+        icon_emoji: default_icon,
+      }
+      payload[:icon_emoji] = opts[:icon] if opts.key?(:icon)
+      payload[:icon_url] = opts[:icon_url] if opts.key?(:icon_url)
+      payload[:username] = opts[:name] if opts.key?(:name)
+      payload[:attachments] = opts[:attachments] if opts.key?(:attachments)
+
+      puts("Saying: #{payload[:channel]} -> #{text}")
+      Faraday.post(@hook, payload: payload.to_json)
+    end
+
+  end
+end

data/lib/qbert_bot/version.rb

@@ -0,0 +1,3 @@
+module QbertBot
+  VERSION = "0.0.1"
+end

data/lib/qbert_bot.rb

@@ -0,0 +1,21 @@
+require "qbert_bot/version"
+require "qbert_bot/plugin"
+require "qbert_bot/message"
+require "qbert_bot/router"
+require "qbert_bot/slack"
+require "qbert_bot/robot"
+
+
+
+module QbertBot
+  def self.run!
+    bot = Robot.instance
+    bot.load_config
+
+    Dir['plugins/*.rb'].each do |f|
+      load f
+    end
+
+    bot.run!
+  end
+end

metadata

@@ -0,0 +1,123 @@
+--- !ruby/object:Gem::Specification
+name: qbert_bot
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- lxfontes
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-02-06 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rufus-scheduler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '3.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '3.0'
+- !ruby/object:Gem::Dependency
+  name: sinatra
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.4'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.4'
+- !ruby/object:Gem::Dependency
+  name: faraday
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0.9'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0.9'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.7'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.7'
+- !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'
+description: ''
+email:
+- lxfontes@gmail.com
+executables:
+- qbert_bot
+extensions: []
+extra_rdoc_files: []
+files:
+- bin/qbert_bot
+- lib/qbert_bot.rb
+- lib/qbert_bot/message.rb
+- lib/qbert_bot/plugin.rb
+- lib/qbert_bot/robot.rb
+- lib/qbert_bot/router.rb
+- lib/qbert_bot/slack.rb
+- lib/qbert_bot/version.rb
+homepage: http://github.com/lxfontes/qbert_bot
+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.4
+signing_key: 
+specification_version: 4
+summary: The no frills ruby bot for Slack
+test_files: []