qwtf_discord_bot 4.2.6 → 5.0.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 724d0971cb9732c8095fa84f03c9be9ab7e215515ee9f65dcbe117fa8dc409c5
-  data.tar.gz: 2c81f16c2ed703ec9d3671eb0f0731ffaa34f7590bf1a3241e729aa7bd13bcb5
+  metadata.gz: 1952a9bb33cf18bcda01f202ed5900cb5168f0c3e1ad8aa083b1ee0ae54b6ed0
+  data.tar.gz: 31ab6af67990f6479640402bf20ec90fdcfad9943b5cc84697ea2c18776e9676
 SHA512:
-  metadata.gz: d314677b730b1fdce5866120c97eb7448bcbab24ea46d2263f5dee35f8dacaec1ec433b4a5303942d80f2429a626673a7d1306956740f1579b3767993fb2181f
-  data.tar.gz: 9499263c0c76de7590a2aacb0956fb7083d04ce378a17497b259b90ef3adbe6eb2c505a1e7077568c48391547e09e493d10c514d14c3d1b8e888619b1d629f0a
+  metadata.gz: 3ae59b76eda7fe87baa54c6b79d5d0d32de7f2cb947d2631eed2b2d2a3769376727c855a8ba1293c5e00fd23376357a5153cd3bdfa00e328868ee2f1b669f3e3
+  data.tar.gz: c0fd9ef59318cde38734e3f28d9f8a41a9c6ad0a829aae70ee48a07aeb5f509b6a22a58e6e6148a6c4324a1733f6d9d3c7e0104994ccbfd6d4ef6305cda803fb

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    qwtf_discord_bot (4.2.4)
+    qwtf_discord_bot (4.2.6)
       discordrb (~> 3.3)
       redis (~> 4.1)
       thor (~> 0.20)
@@ -95,4 +95,4 @@ DEPENDENCIES
   thor
 
 BUNDLED WITH
-   1.17.1
+   1.17.2

data/README.md

@@ -66,7 +66,7 @@ A Discord bot for checking the status of QuakeWorld Team Fortress servers
 
 ### Commands
 
-There are two modules:
+There are three modules:
 
 
 #### Server
@@ -81,6 +81,18 @@ This responds to discord messages:
 ![screenshot of bot responding to !server command](server_screenshot.png)
 
 
+#### Pug
+
+    qwtf-discord-bot pug
+
+This responds to discord messages:
+  - `!join`
+  - `!leave`
+  - `!status`
+  - `!maxplayers <no_of_players>`
+  - `!leave`
+
+
 #### Watcher
 
     qwtf-discord-bot watcher
@@ -99,6 +111,12 @@ Build:
     gem build qwtf_discrd_bot.gemspec
 
 
+Install:
+
+ 
+    gem install --local qwtf_discord_bot-$(cat VERSION).gem
+
+
 Push:
 
     gem push qwtf_discord_bot-$(cat VERSION).gem
@@ -129,6 +147,14 @@ Only discord-bot watcher:
       discord-bot watcher
 
 
+Only discord-bot pug:
+
+    docker run -it \
+      --env QWTF_DISCORD_BOT_CONFIG_FILE=config.yaml \
+      --mount type=bind,source="$(pwd)"/config.yaml,target=/discord-bot/config.yaml \
+      discord-bot pug
+
+
 Build:
 
     docker build --tag=discord-bot .
@@ -140,6 +166,19 @@ Push:
     docker push fortressone/discord-bot:latest
 
 
+Create AWS instance:
+
+```
+docker-machine create \
+  --driver amazonec2 \
+  --amazonec2-access-key <AWS_ACCESS_KEY> \
+  --amazonec2-secret-key <AWS_SECRET_KEY> \
+  --amazonec2-root-size 30 \
+  --amazonec2-region ap-southeast-2 \
+  discord-bot
+```
+
+
 ## License
 
 The gem is available as open source under the terms of the [MIT License](https://opensource.org/licenses/MIT).

data/VERSION

@@ -1 +1 @@
-4.2.6
+5.0.1

data/docker-compose.yml

@@ -26,3 +26,15 @@ services:
       - type: bind
         source: "/home/ubuntu/.config/qwtf_discord_bot/config.yaml"
         target: /discord-bot/config.yaml
+  discord-pug-bot:
+    image: fortressone/discord-bot:latest
+    command: pug
+    restart: always
+    depends_on:
+      - redis
+    environment:
+      - REDIS_URL=redis://redis
+    volumes:
+      - type: bind
+        source: "/home/ubuntu/.config/qwtf_discord_bot/config.yaml"
+        target: /discord-bot/config.yaml

data/exe/qwtf_discord_bot

@@ -2,6 +2,7 @@
 
 require 'qwtf_discord_bot'
 require 'thor'
+require 'pry'
 
 class QwtfDiscordBotExe < Thor
   def self.exit_on_failure?
@@ -14,6 +15,12 @@ class QwtfDiscordBotExe < Thor
     server_bot.run
   end
 
+  desc 'pug', 'Start, join, leave, record pick-up games.'
+  def pug
+    pug_bot = QwtfDiscordBotPug.new
+    pug_bot.run
+  end
+
   desc 'watcher', 'Watches servers and accounces when a player joins'
   def watcher
     watcher_bot = QwtfDiscordBotWatcher.new

data/lib/event_wrapper.rb

@@ -0,0 +1,58 @@
+class EventWrapper
+  include QwtfDiscordBot
+
+  DEFAULT_MAXPLAYERS = 8
+
+  def initialize(event)
+    @event = event
+  end
+
+  def user_id
+    @event.user.id
+  end
+
+  def username
+    @event.user.username
+  end
+
+  def maxplayers_key
+    [pug_key, "maxplayers"].join(":")
+  end
+
+  def maxplayers
+    redis.setnx(maxplayers_key, DEFAULT_MAXPLAYERS)
+    redis.get(maxplayers_key).to_i
+  end
+
+  def joined_player_count
+    redis.scard(players_key).to_i
+  end
+
+  def slots_left
+    maxplayers - joined_player_count
+  end
+
+  def pug_key
+    ["pug", "channel", @event.channel.id].join(":")
+  end
+
+  def players_key
+    [pug_key, "players"].join(":")
+  end
+
+  def player_slots
+    "#{joined_player_count}/#{maxplayers}"
+  end
+
+  def users
+    @event.server.users
+  end
+
+  def role_key
+    [pug_key, "role"].join(":")
+  end
+
+  def role
+    redis.get(role_key) || "@here"
+  end
+end

data/lib/qwtf_discord_bot.rb

@@ -1,9 +1,11 @@
 require 'qwtf_discord_bot/version'
 require 'qwtf_discord_bot/qwtf_discord_bot_server'
+require 'qwtf_discord_bot/qwtf_discord_bot_pug'
 require 'qwtf_discord_bot/qwtf_discord_bot_watcher'
 require 'qwtf_discord_bot/config'
 require 'discordrb'
 require 'yaml'
+require 'redis'
 
 require 'qstat_request'
 require 'player'
@@ -12,9 +14,17 @@ require 'emoji'
 require 'roster'
 
 module QwtfDiscordBot # :nodoc:
-  CONFIG_FILE = ENV['QWTF_DISCORD_BOT_CONFIG_FILE'] || "#{Dir.pwd}/config.yaml"
-
   def self.config
-    @config ||= Config.new(CONFIG_FILE)
+    @config ||= Config.new(config_file)
+  end
+
+  def self.config_file
+    return ENV['QWTF_DISCORD_BOT_CONFIG_FILE'] if ENV['QWTF_DISCORD_BOT_CONFIG_FILE']
+    return "#{Dir.pwd}/config.yaml" if FileTest.exist?("#{Dir.pwd}/config.yaml")
+    "#{Dir.home}/.config/qwtf_discord_bot/config.yaml"
+  end
+
+  def redis
+    Redis.current
   end
 end

data/lib/qwtf_discord_bot/qwtf_discord_bot_pug.rb

@@ -0,0 +1,138 @@
+require 'event_wrapper'
+
+class QwtfDiscordBotPug
+  include QwtfDiscordBot
+
+  FOUR_HOURS = 4 * 60 * 60
+
+  def run
+    bot = Discordrb::Commands::CommandBot.new(
+      token: QwtfDiscordBot.config.token,
+      client_id: QwtfDiscordBot.config.client_id,
+      prefix: '!'
+    )
+
+    bot.command :join do |event, *args|
+      e = EventWrapper.new(event)
+
+      redis.setnx(e.pug_key, Time.now)
+      redis.sadd(e.players_key, e.user_id)
+
+      message = if e.joined_player_count == e.maxplayers
+                  mentions = joined_users(e).map do |user|
+                    user.mention
+                  end
+                  "Time to play! #{mentions.join(" ")}"
+                elsif (e.joined_player_count == 1)
+                  [
+                    "#{e.username} creates a PUG",
+                    e.player_slots,
+                    e.role,
+                  ].join(" | ")
+                elsif e.slots_left <= 3
+                  [
+                    "#{e.username} joins the PUG",
+                    e.player_slots,
+                    "#{e.slots_left} more",
+                    e.role,
+                  ].join(" | ")
+                else
+                  [
+                    "#{e.username} joins the PUG",
+                    e.player_slots,
+                  ].join(" | ")
+                end
+
+      send_and_log_message(message, event)
+    end
+
+    bot.command :status do |event, *args|
+      e = EventWrapper.new(event)
+      usernames = joined_users(e).map(&:username)
+
+      message = [
+        "#{usernames.join(", ")} joined",
+        e.player_slots
+      ].join(" | ")
+
+      send_and_log_message(message, event)
+    end
+
+    bot.command :maxplayers do |event, *args|
+      e = EventWrapper.new(event)
+      new_maxplayers = args[0]
+
+      message = if new_maxplayers
+        redis.set(e.maxplayers_key, new_maxplayers)
+        "Max number of players set to #{e.maxplayers} | #{e.player_slots}"
+      else
+        "Current max number of players is #{e.maxplayers} | #{e.player_slots}"
+      end
+
+      send_and_log_message(message, event)
+
+      if e.joined_player_count >= e.maxplayers
+        mentions = joined_users(e).map do |user|
+          user.mention
+        end
+
+        message = "Time to play! #{mentions.join(" ")}"
+        send_and_log_message(message, event)
+      end
+    end
+
+    bot.command :leave do |event, *args|
+      e = EventWrapper.new(event)
+
+      redis.srem(e.players_key, e.user_id)
+
+      message = "#{e.username} leaves the PUG | #{e.player_slots}"
+
+      send_and_log_message(message, event)
+
+      if e.joined_player_count == 0
+        redis.del(e.pug_key)
+
+        message = "PUG ended"
+        send_and_log_message(message, event)
+      end
+    end
+
+    bot.command :end do |event, *args|
+      e = EventWrapper.new(event)
+
+      redis.del(e.pug_key)
+      redis.del(e.players_key)
+
+      message = "PUG ended"
+      send_and_log_message(message, event)
+    end
+
+    bot.command :role do |event, *args|
+      e = EventWrapper.new(event)
+      role = args[0]
+
+      redis.set(e.role_key, role)
+
+      message = "Notification role set to #{role}"
+      send_and_log_message(message, event)
+    end
+
+    bot.run
+  end
+
+  private
+
+  def send_and_log_message(message, event)
+    event.channel.send_message(message)
+    puts message
+  end
+
+  def joined_users(event)
+    redis.smembers(event.players_key).map do |user_id|
+      event.users.find do |user|
+        user.id.to_s == user_id
+      end
+    end
+  end
+end

data/lib/qwtf_discord_bot/qwtf_discord_bot_watcher.rb

@@ -1,5 +1,3 @@
-require "redis"
-
 class QwtfDiscordBotWatcher
   include QwtfDiscordBot
 
@@ -14,7 +12,7 @@ class QwtfDiscordBotWatcher
         next if request.is_empty?
 
         request.player_names.each do |name|
-          redis_key = "#{address}:#{name}"
+          redis_key = ["watcher", address, name].join(":")
 
           unless seen_recently?(redis_key)
             endpoint.channel_ids.each do |channel_id|
@@ -51,7 +49,7 @@ class QwtfDiscordBotWatcher
   end
 
   def report_joined(name:, channel_id:, server_summary:)
-    message = "#{name} has joined #{server_summary}"
+    message = "#{name} joins #{server_summary}"
 
     Discordrb::API::Channel.create_message(
       "Bot #{QwtfDiscordBot.config.token}",
@@ -61,10 +59,4 @@ class QwtfDiscordBotWatcher
 
     puts message
   end
-
-  private
-
-  def redis
-    @redis ||= Redis.new
-  end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: qwtf_discord_bot
 version: !ruby/object:Gem::Version
-  version: 4.2.6
+  version: 5.0.1
 platform: ruby
 authors:
 - Sheldon Johnson
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2020-01-19 00:00:00.000000000 Z
+date: 2020-08-23 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: discordrb
@@ -104,10 +104,12 @@ files:
 - exe/qwtf_discord_bot
 - lib/emoji.rb
 - lib/endpoint.rb
+- lib/event_wrapper.rb
 - lib/player.rb
 - lib/qstat_request.rb
 - lib/qwtf_discord_bot.rb
 - lib/qwtf_discord_bot/config.rb
+- lib/qwtf_discord_bot/qwtf_discord_bot_pug.rb
 - lib/qwtf_discord_bot/qwtf_discord_bot_server.rb
 - lib/qwtf_discord_bot/qwtf_discord_bot_watcher.rb
 - lib/qwtf_discord_bot/version.rb
@@ -135,8 +137,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubyforge_project: 
-rubygems_version: 2.7.6.2
+rubygems_version: 3.1.2
 signing_key: 
 specification_version: 4
 summary: Works by wrapping the excellent CLI server query tool qstat. Accepts !server,