AugustBot 0.0.1

config.ru

require './slackbot'

app = App.new

run app

slackbot.rb

require 'rack'
require 'json'
require 'optparse'

class App

  def initialize(folder = 'commands/*.rb')
    @bot = Slack::Bot.new folder
  end

  def call(env)
    request = Rack::Request.new env
    
    # POST to "/"
    if request.path_info == '/' and request.post?
      status = 200
      begin
        response = @bot.handle request.params
      rescue JSON::ParserError
        response = {error: 'Malformed Request', status: 400}
        status = 400
      end
      [status, {'Content-Type' => 'application/json'}, [response]]
    else
      response = 'You are entering a robot-only facility.'
      [200, {'Content-Type' => 'application/plain'}, [response]]
    end
  end

end

module Slack
  class Bot

    def initialize(folder)
      @commands = {}

      # Eval commands in this instance
      Dir[folder].each do |file|
        puts "Loading command file:: #{file}"
        self.instance_eval File.open(file, 'r').read
      end
    end

    def handle(params)
      return {error: 'Missing parameters'}.to_json if !params['text'] or !params['user_name']
  
      response = {username: 'AugustBot', icon_emoji: ':diamond_shape_with_a_dot_inside:'}    
      args = params['text'][1, params['text'].length - 1].split ' '
      command_name = args.shift
      
      command = nil
      @commands.each_pair do |name, data|
        if name == command_name or data[:aliases].include? command_name
          command = data
        end
      end
      
      if command == nil
        response[:text] = 'Command not found.'
      else
        if command[:users]
          unless command[:users].include? params['user_name']
            response[:text] = "You don't have permission to use that command, #{params['user_name']}"
            return response.to_json
          end
        end
        puts "Running command #{command_name} for #{params['user_name']} with args: #{args}"
        time = Time.at params['timestamp'].to_i
        message = Message.new args, params['user_name'], time
        command[:proc].call message
        response[:text] = message.response.to_a
      end
      response.to_json
    end

    def on_command(name, options = {}, &block)
      command = options.merge proc: block
      # Ensure :aliases is an array
      unless command[:aliases].is_a? Array
        command[:aliases] = command[:aliases] ? [command[:aliases]] : []
      end
      # Ensure :help is a string
      command[:help] = command[:help].to_s if command[:help]
      unless command[:users].is_a? Array
        command[:users] = [command[:users]] if command[:users]
      end
      @commands[name] = command
    end

    alias_method :command, :on_command
  end

  class Message
    attr_reader :joined, :args, :user, :time

    attr_accessor :response

    alias_method :content, :joined
    alias_method :body, :joined
    alias_method :username, :user

    def initialize(args, username, time)
      @args, @user, @time = args, username, time
      @joined = args.join ' '
    end

    def reply(text = nil)
      if text
        (@response ||= []) << text
      end
      @response
    end
  end  
end