slack-ruby-bot-server 0.1.0

data/lib/slack-ruby-bot-server/api/presenters/root_presenter.rb

@@ -0,0 +1,44 @@
+module SlackRubyBotServer
+  module Api
+    module Presenters
+      module RootPresenter
+        include Roar::JSON::HAL
+        include Roar::Hypermedia
+        include Grape::Roar::Representer
+
+        link :self do |opts|
+          "#{base_url(opts)}/api"
+        end
+
+        link :status do |opts|
+          "#{base_url(opts)}/api/status"
+        end
+
+        link :teams do |opts|
+          {
+            href: "#{base_url(opts)}/api/teams/#{link_params(Helpers::PaginationParameters::ALL, :active)}",
+            templated: true
+          }
+        end
+
+        link :team do |opts|
+          {
+            href: "#{base_url(opts)}/api/teams/{id}",
+            templated: true
+          }
+        end
+
+        private
+
+        def base_url(opts)
+          request = Grape::Request.new(opts[:env])
+          request.base_url
+        end
+
+        def link_params(*args)
+          "{?#{args.join(',')}}"
+        end
+      end
+    end
+  end
+end

data/lib/slack-ruby-bot-server/api/presenters/status_presenter.rb

@@ -0,0 +1,40 @@
+module SlackRubyBotServer
+  module Api
+    module Presenters
+      module StatusPresenter
+        include Roar::JSON::HAL
+        include Roar::Hypermedia
+        include Grape::Roar::Representer
+
+        link :self do |opts|
+          "#{base_url(opts)}/status"
+        end
+
+        property :teams_count
+        property :active_teams_count
+        property :ping
+
+        private
+
+        def ping
+          team = Team.asc(:_id).first
+          return unless team
+          team.ping!
+        end
+
+        def teams_count
+          Team.count
+        end
+
+        def active_teams_count
+          Team.active.count
+        end
+
+        def base_url(opts)
+          request = Grape::Request.new(opts[:env])
+          request.base_url
+        end
+      end
+    end
+  end
+end

data/lib/slack-ruby-bot-server/api/presenters/team_presenter.rb

@@ -0,0 +1,24 @@
+module SlackRubyBotServer
+  module Api
+    module Presenters
+      module TeamPresenter
+        include Roar::JSON::HAL
+        include Roar::Hypermedia
+        include Grape::Roar::Representer
+
+        property :id, type: String, desc: 'Team ID.'
+        property :team_id, type: String, desc: 'Slack team ID.'
+        property :name, type: String, desc: 'Team name.'
+        property :domain, type: String, desc: 'Team domain.'
+        property :active, type: Boolean, desc: 'Team is active.'
+        property :created_at, type: DateTime, desc: 'Date/time when the team was created.'
+        property :updated_at, type: DateTime, desc: 'Date/time when the team was accepted, declined or canceled.'
+
+        link :self do |opts|
+          request = Grape::Request.new(opts[:env])
+          "#{request.base_url}/api/teams/#{id}"
+        end
+      end
+    end
+  end
+end

data/lib/slack-ruby-bot-server/api/presenters/teams_presenter.rb

@@ -0,0 +1,14 @@
+module SlackRubyBotServer
+  module Api
+    module Presenters
+      module TeamsPresenter
+        include Roar::JSON::HAL
+        include Roar::Hypermedia
+        include Grape::Roar::Representer
+        include PaginatedPresenter
+
+        collection :results, extend: TeamPresenter, as: :teams, embedded: true
+      end
+    end
+  end
+end

data/lib/slack-ruby-bot-server/app.rb

@@ -0,0 +1,85 @@
+module SlackRubyBotServer
+  class App
+    def prepare!
+      silence_loggers!
+      check_mongodb_provider!
+      check_database!
+      create_indexes!
+      mark_teams_active!
+      migrate_from_single_team!
+      update_team_name_and_id!
+      purge_inactive_teams!
+      configure_global_aliases!
+    end
+
+    def self.instance
+      @instance ||= SlackRubyBotServer::App.new
+    end
+
+    private
+
+    def logger
+      @logger ||= begin
+        $stdout.sync = true
+        Logger.new(STDOUT)
+      end
+    end
+
+    def silence_loggers!
+      Mongoid.logger.level = Logger::INFO
+      Mongo::Logger.logger.level = Logger::INFO
+    end
+
+    def check_mongodb_provider!
+      return unless ENV['RACK_ENV'] == 'production'
+      fail "Missing ENV['MONGO_URL'], ENV['MONGOHQ_URI'] or ENV['MONGOLAB_URI']." unless ENV['MONGO_URL'] || ENV['MONGOHQ_URI'] || ENV['MONGOLAB_URI']
+    end
+
+    def check_database!
+      rc = Mongoid.default_client.command(ping: 1)
+      return if rc && rc.ok?
+      fail rc.documents.first['error'] || 'Unexpected error.'
+    rescue Exception => e
+      warn "Error connecting to MongoDB: #{e.message}"
+      raise e
+    end
+
+    def create_indexes!
+      ::Mongoid::Tasks::Database.create_indexes
+    end
+
+    def mark_teams_active!
+      Team.where(active: nil).update_all(active: true)
+    end
+
+    def update_team_name_and_id!
+      Team.active.where(team_id: nil).each do |team|
+        begin
+          auth = team.ping![:auth]
+          team.update_attributes!(team_id: auth['team_id'], name: auth['team'])
+        rescue StandardError => e
+          logger.warn "Error pinging team #{team.id}: #{e.message}."
+          team.set(active: false)
+        end
+      end
+    end
+
+    def migrate_from_single_team!
+      return unless ENV.key?('SLACK_API_TOKEN')
+      logger.info 'Migrating from env SLACK_API_TOKEN ...'
+      team = Team.find_or_create_from_env!
+      logger.info "Automatically migrated team: #{team}."
+      logger.warn "You should unset ENV['SLACK_API_TOKEN']."
+    end
+
+    def purge_inactive_teams!
+      Team.purge!
+    end
+
+    def configure_global_aliases!
+      SlackRubyBot.configure do |config|
+        config.aliases = ENV['SLACK_RUBY_BOT_ALIASES'].split(' ') if ENV['SLACK_RUBY_BOT_ALIASES']
+      end
+    end
+  end
+end

data/lib/slack-ruby-bot-server/info.rb

@@ -0,0 +1,11 @@
+module SlackRubyBotServer
+  INFO = <<-EOS
+BotServer #{SlackRubyBotServer::VERSION}
+
+© 2016 Daniel Doubrovkine & Contributors, MIT License
+https://twitter.com/dblockdotorg
+
+Free Service at http://slack-ruby-bot-server.herokuapp.com
+Open-Source at https://github.com/dblock/slack-ruby-bot-server
+  EOS
+end

data/lib/slack-ruby-bot-server/models.rb

@@ -0,0 +1 @@
+require 'slack-ruby-bot-server/models/team'

data/lib/slack-ruby-bot-server/models/team.rb

@@ -0,0 +1,66 @@
+class Team
+  include Mongoid::Document
+  include Mongoid::Timestamps
+
+  SORT_ORDERS = ['created_at', '-created_at', 'updated_at', '-updated_at']
+
+  field :team_id, type: String
+  field :name, type: String
+  field :domain, type: String
+  field :token, type: String
+  field :active, type: Boolean, default: true
+
+  scope :active, -> { where(active: true) }
+
+  validates_uniqueness_of :token, message: 'has already been used'
+  validates_presence_of :token
+  validates_presence_of :team_id
+
+  def deactivate!
+    update_attributes!(active: false)
+  end
+
+  def activate!(token)
+    update_attributes!(active: true, token: token)
+  end
+
+  def to_s
+    {
+      name: name,
+      domain: domain,
+      id: team_id
+    }.map do |k, v|
+      "#{k}=#{v}" if v
+    end.compact.join(', ')
+  end
+
+  def ping!
+    client = Slack::Web::Client.new(token: token)
+    auth = client.auth_test
+    {
+      auth: auth,
+      presence: client.users_getPresence(user: auth['user_id'])
+    }
+  end
+
+  def self.find_or_create_from_env!
+    token = ENV['SLACK_API_TOKEN']
+    return unless token
+    team = Team.where(token: token).first
+    team ||= Team.new(token: token)
+    info = Slack::Web::Client.new(token: token).team_info
+    team.team_id = info['team']['id']
+    team.name = info['team']['name']
+    team.domain = info['team']['domain']
+    team.save!
+    team
+  end
+
+  def self.purge!
+    # destroy teams inactive for two weeks
+    Team.where(active: false, :updated_at.lte => 2.weeks.ago).each do |team|
+      Mongoid.logger.info "Destroying #{team}, inactive since #{team.updated_at}, over two weeks ago."
+      team.destroy
+    end
+  end
+end

data/lib/slack-ruby-bot-server/rspec.rb

@@ -0,0 +1,5 @@
+require 'slack-ruby-bot/rspec'
+require 'slack-ruby-bot-server'
+require 'fabrication'
+require 'faker'
+require_relative 'rspec/fabricators/team'

data/lib/slack-ruby-bot-server/rspec/fabricators/team.rb

@@ -0,0 +1,5 @@
+Fabricator(:team) do
+  token { Fabricate.sequence(:team_token) { |i| "abc-#{i}" } }
+  team_id { Fabricate.sequence(:team_id) { |i| "T#{i}" } }
+  name { Faker::Lorem.word }
+end

data/lib/slack-ruby-bot-server/server.rb

@@ -0,0 +1,21 @@
+module SlackRubyBotServer
+  class Server < SlackRubyBot::Server
+    attr_accessor :team
+
+    def initialize(attrs = {})
+      @team = attrs[:team]
+      fail 'Missing team' unless @team
+      options = { token: @team.token }
+      super(options)
+      client.owner = @team
+    end
+
+    def restart!(wait = 1)
+      # when an integration is disabled, a live socket is closed, which causes the default behavior of the client to restart
+      # it would keep retrying without checking for account_inactive or such, we want to restart via service which will disable an inactive team
+      logger.info "#{team.name}: socket closed, restarting ..."
+      SlackRubyBotServer::Service.instance.restart! team, self, wait
+      client.owner = team
+    end
+  end
+end

data/lib/slack-ruby-bot-server/service.rb

@@ -0,0 +1,81 @@
+module SlackRubyBotServer
+  class Service
+    include SlackRubyBot::Loggable
+
+    def self.start!
+      Thread.new do
+        Thread.current.abort_on_exception = true
+        instance.start_from_database!
+      end
+    end
+
+    def self.instance
+      @instance ||= new
+    end
+
+    def initialize
+      @lock = Mutex.new
+      @services = {}
+    end
+
+    def start!(team)
+      fail 'Token already known.' if @services.key?(team.token)
+      logger.info "Starting team #{team}."
+      server = SlackRubyBotServer::Server.new(team: team)
+      @lock.synchronize do
+        @services[team.token] = server
+      end
+      restart!(team, server)
+    rescue StandardError => e
+      logger.error e
+    end
+
+    def stop!(team)
+      @lock.synchronize do
+        fail 'Token unknown.' unless @services.key?(team.token)
+        logger.info "Stopping team #{team}."
+        @services[team.token].stop!
+        @services.delete(team.token)
+      end
+    rescue StandardError => e
+      logger.error e
+    end
+
+    def start_from_database!
+      Team.active.each do |team|
+        start!(team)
+      end
+    end
+
+    def restart!(team, server, wait = 1)
+      server.start_async
+    rescue StandardError => e
+      case e.message
+      when 'account_inactive', 'invalid_auth' then
+        logger.error "#{team.name}: #{e.message}, team will be deactivated."
+        deactivate!(team)
+      else
+        logger.error "#{team.name}: #{e.message}, restarting in #{wait} second(s)."
+        sleep(wait)
+        restart! team, server, [wait * 2, 60].min
+      end
+    end
+
+    def deactivate!(team)
+      team.deactivate!
+      @lock.synchronize do
+        @services.delete(team.token)
+      end
+    rescue Mongoid::Errors::Validations => e
+      logger.error "#{team.name}: #{e.message}, error - #{e.document.class}, #{e.document.errors.to_hash}, ignored."
+    rescue StandardError => e
+      logger.error "#{team.name}: #{e.class}, #{e.message}, ignored."
+    end
+
+    def reset!
+      @services.values.to_a.each do |server|
+        stop!(server.team)
+      end
+    end
+  end
+end

data/lib/slack-ruby-bot-server/version.rb

@@ -0,0 +1,3 @@
+module SlackRubyBotServer
+  VERSION = '0.1.0'
+end

data/public/index.html.erb

@@ -0,0 +1,28 @@
+<html>
+ <head>
+  <title>Slack Ruby Bot Server</title>
+  <script src="/scripts/jquery-1.7.1.min.js"></script>
+  <script src="/scripts/url.min.js"></script>
+  <script src="/scripts/register.js"></script>
+  <script src="/scripts/stats.js"></script>
+</head>
+ <body style="text-align: center">
+  <p>
+    <img src='img/slack.png' width="120px">
+  </p>
+  <p>
+    <h3>An opinionated boilerplate for a Slack bot service.</h3>
+  </p>
+  <p id='messages' />
+  <p id='register'>
+    <a href="https://slack.com/oauth/authorize?scope=bot&client_id=<%= ENV['SLACK_CLIENT_ID'] %>"><img alt="Add to Slack" height="40" width="139" src="https://platform.slack-edge.com/img/add_to_slack.png" srcset="https://platform.slack-edge.com/img/add_to_slack.png 1x, https://platform.slack-edge.com/img/add_to_slack@2x.png 2x"></a>
+  </p>
+  <p id='active_teams_count'>&nbsp;</p>
+  <p>
+    <small>
+      made by <a href="https://twitter.com/dblockdotorg" target="_blank">@dblockdotorg</a>, <a href="https://github.com/dblock/slack-ruby-bot-server" target="_blank">fork me on github</a>
+    </small>
+  </p>
+ </body>
+</html>
+