Contents

require 'qwtf_discord_bot/version'
require 'discordrb'

require 'qstat_request'
require 'player'
require 'team'
require 'emoji'

class QwtfDiscordBot
  TOKEN = ENV['QWTF_DISCORD_BOT_TOKEN'].strip
  CLIENT_ID = ENV['QWTF_DISCORD_BOT_CLIENT_ID'].strip
  CHANNEL_ID = ENV['QWTF_DISCORD_BOT_CHANNEL_ID'].strip

  def initialize(hostname:, port:)
    @hostname = hostname
    @port = port
  end

  def endpoint
    return @hostname if @port == 27500
    [@hostname, @port].join(':')
  end
end

class QwtfDiscordBotServer < QwtfDiscordBot
  def run
    bot = Discordrb::Commands::CommandBot.new(
      token: TOKEN,
      client_id: CLIENT_ID,
      prefix: '!'
    )

    bot.command :server do |_event|
      QstatRequest.new(endpoint).output
    end

    bot.run
  end
end

class QwtfDiscordBotWatcher < QwtfDiscordBot
  THIRTY_SECONDS = 30
  TEN_MINUTES = 10 * 60

  def run
    every(THIRTY_SECONDS) do
      request = QstatRequest.new(endpoint)
      numplayers = request.numplayers
      maxplayers = request.maxplayers
      map = request.map

      if request.players
        player_names = request.players.map(&:name)

        player_names.each do |name|
          unless seen_recently?(name)
            report_joined(name: name,
                          map: map,
                          numplayers: numplayers,
                          maxplayers: maxplayers)
          end

          history[name] = Time.now
        end
      end
    end
  end

  def every(n_seconds)
    loop do
      before = Time.now
      yield
      interval = n_seconds - (Time.now - before)
      sleep(interval) if interval > 0
    end
  end

  def seen_recently?(name)
    last_seen = history[name]
    last_seen && (Time.now - last_seen < TEN_MINUTES)
  end

  def report_joined(name:, map:, numplayers:, maxplayers:)
    Discordrb::API::Channel.create_message(
      "Bot #{TOKEN}",
      CHANNEL_ID,
      "**#{name}** has joined **#{endpoint} | #{map} | #{numplayers}/#{maxplayers}**"
    )
  end

  def history
    @history ||= {}
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
qwtf_discord_bot-0.4.1	lib/qwtf_discord_bot.rb