qwtf_discord_bot 5.4.10 → 5.5.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c5bfd247ff3e811ab845a1cef86f043b93f54a6f30769885547d0ba1601c15ef
-  data.tar.gz: 72aebecdfaa233cd62e4d36b25c474a319dae5a1fa12d8ea8711536f23cf9549
+  metadata.gz: b1d76ac11d350eb38fbb60579e3ce26d561066f17e1bcda29029791afd98d36f
+  data.tar.gz: 1e75a7c3cecf17134bd41e2a9370bce29f82260319cd4d78bd9e9638e1c8e3af
 SHA512:
-  metadata.gz: 89c358f811c6c0d17b31fb26e1a8d6234bf170f82bd5a820da5d8474052a42594f1adeafcd6fa387488147559370533a10d73bf50526bdfedb5b948325f78f7d
-  data.tar.gz: e2eb01bcd75ff2ab9251452c6cfac97b381d1199d5fddc5a146ee451def5e56747b4528e2977fd15e2d3d732a63cac03a1dcaa38a37cecaa6a75758075bdf770
+  metadata.gz: d13de4098fde37b346b4a668befed0c59ba73cae7a7640b69356064d1446c081af3b0552ef651896bd4c08b2548882a1386508992f12022717e9519eb5317dd2
+  data.tar.gz: c0868734a288588818370904904a407789857c74164d4b4ef42e27bafc68724238dcd4235bd7c625d79f72949109eef08ead1f03e2dea5a9ab39472132d2e1be

data/VERSION

@@ -1 +1 @@
-5.4.10
+5.5.1

data/docker-compose.yml

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

data/exe/qwtf_discord_bot

@@ -26,6 +26,12 @@ class QwtfDiscordBotExe < Thor
     watcher_bot = QwtfDiscordBotWatcher.new
     watcher_bot.run
   end
+
+  desc 'dashboard', 'A live dashboard with active server information'
+  def dashboard
+    dashboard_bot = QwtfDiscordBotDashboard.new
+    dashboard_bot.run
+  end
 end
 
 QwtfDiscordBotExe.start

data/lib/dashboard.rb

@@ -0,0 +1,42 @@
+class Dashboard
+  def initialize(channel, bot)
+    @channel_id = channel["id"]
+    @endpoints = channel["endpoints"]
+    @bot = bot
+    @messages = {}
+  end
+
+  def update
+    @endpoints.each do |endpoint|
+      qstat_request = QstatRequest.new(endpoint)
+
+      if qstat_request.is_empty?
+        if @messages[endpoint]
+          @messages[endpoint].delete
+          @messages.delete(endpoint)
+        end
+
+        next
+      end
+
+      embed = qstat_request.to_full_embed
+
+      @messages[endpoint] = if @messages[endpoint]
+                              @messages[endpoint].edit(nil, embed)
+                            else
+                              channel.send_embed(nil, embed)
+                            end
+    end
+  end
+
+  private
+
+  def channel
+    data = Discordrb::API::Channel.resolve(
+      "Bot #{QwtfDiscordBot.config.token}",
+      @channel_id
+    )
+
+    Discordrb::Channel.new(JSON.parse(data), @bot)
+  end
+end

data/lib/qstat_request.rb

@@ -1,4 +1,6 @@
 class QstatRequest
+  MSG_SNIPPET_DELIMITER = ' · '
+
   attr_accessor :result
 
   def initialize(endpoint)
@@ -21,6 +23,19 @@ class QstatRequest
     embed
   end
 
+  def to_full_embed
+    Discordrb::Webhooks::Embed.new.tap do |embed|
+      embed.add_field(
+        name: name,
+        value: embed_summary,
+      )
+
+      teams.each do |team|
+        embed.add_field(team.to_embed_field)
+      end
+    end
+  end
+
   def to_message
     return server_summary if is_empty?
 
@@ -29,9 +44,40 @@ class QstatRequest
 
   def server_summary
     return "#{@endpoint} isn't responding" unless game_map
-    return "#{name} | #{@endpoint} | #{game_map} | #{numplayers}/#{maxplayers}" unless has_spectators?
 
-    "#{name} | #{@endpoint} | #{game_map} | #{numplayers}/#{maxplayers} players | #{numspectators}/#{maxspectators} spectators"
+    if !has_spectators?
+      return [
+        name,
+        @endpoint,
+        game_map,
+        "#{numplayers}/#{maxplayers}"
+      ].join(MSG_SNIPPET_DELIMITER)
+    end
+
+    [
+      name,
+      @endpoint,
+      game_map,
+      "#{numplayers}/#{maxplayers} players",
+      "#{numspectators}/#{maxspectators} spectators"
+    ].join(MSG_SNIPPET_DELIMITER)
+  end
+
+  def embed_summary
+    if !has_spectators?
+      return [
+        @endpoint,
+        game_map,
+        "#{numplayers}/#{maxplayers}"
+      ].join(MSG_SNIPPET_DELIMITER)
+    end
+
+    [
+      @endpoint,
+      game_map,
+      "#{numplayers}/#{maxplayers} players",
+      "#{numspectators}/#{maxspectators} spectators"
+    ].join(MSG_SNIPPET_DELIMITER)
   end
 
   def is_empty?
@@ -97,7 +143,7 @@ class QstatRequest
   end
 
   def build_roster
-    return nil if is_empty?
+    return [] if is_empty?
 
     roster = Roster.new
 

data/lib/qwtf_discord_bot.rb

@@ -1,11 +1,12 @@
+require 'discordrb'
+require 'yaml'
+require 'redis'
 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/qwtf_discord_bot_dashboard'
 require 'qwtf_discord_bot/config'
-require 'discordrb'
-require 'yaml'
-require 'redis'
 
 require 'qstat_request'
 require 'player'

data/lib/qwtf_discord_bot/config.rb

@@ -1,4 +1,5 @@
 require 'endpoint'
+require 'dashboard'
 
 class Config
   def initialize(config)
@@ -19,6 +20,10 @@ class Config
     end
   end
 
+  def dashboards
+    @dashboards ||= @config['dashboards']
+  end
+
   def emojis
     @emojis ||= @config['emojis']
   end

data/lib/qwtf_discord_bot/qwtf_discord_bot_dashboard.rb

@@ -0,0 +1,41 @@
+class QwtfDiscordBotDashboard
+  THIRTY_SECONDS = 30
+
+  def run
+    bot = Discordrb::Commands::CommandBot.new(
+      token: QwtfDiscordBot.config.token,
+      client_id: QwtfDiscordBot.config.client_id,
+      help_command: false,
+      prefix: proc do |message|
+        match = /^\!(\w+)(.*)/.match(message.content)
+        if match
+          first = match[1]
+          rest = match[2]
+          # Return the modified string with the first word lowercase:
+          "#{first.downcase}#{rest}"
+        end
+      end
+    )
+
+    @dashboards ||= QwtfDiscordBot.config.dashboards.map do |channel|
+      Dashboard.new(channel, bot)
+    end
+
+    every(THIRTY_SECONDS) do
+      @dashboards.each do |dashboard|
+        dashboard.update
+      end
+    end
+  end
+
+  private
+
+  def every(n_seconds)
+    loop do
+      before = Time.now
+      yield
+      interval = n_seconds - (Time.now - before)
+      sleep(interval) if interval > 0
+    end
+  end
+end

data/lib/qwtf_discord_bot/qwtf_discord_bot_pug.rb

@@ -288,26 +288,47 @@ class QwtfDiscordBotPug # :nodoc:
           )
         end
 
-        unless pug.joined?(user_id)
-          return send_embedded_message(
-            description: "You aren't in this PUG",
-            channel: e.channel
-          )
-        end
+        if args.empty?
+          unless pug.joined?(user_id)
+            return send_embedded_message(
+              description: "You aren't in this PUG",
+              channel: e.channel
+            )
+          end
 
-        if pug.team(0).include?(user_id)
-          return send_embedded_message(
-            description: "You aren't in a team",
+          if pug.team(0).include?(user_id)
+            return send_embedded_message(
+              description: "You aren't in a team",
+              channel: e.channel
+            )
+          end
+
+          pug.join_team(team_no: 0, player_id: user_id)
+
+          send_embedded_message(
+            description: "#{e.display_name} leaves team",
             channel: e.channel
           )
-        end
+        else
+          args.each do |mention|
+            unless mention.match(/<@!\d+>/)
+              send_embedded_message(
+                description: "#{arg} isn't a valid mention",
+                channel: e.channel
+              )
+              next
+            end
 
-        pug.join_team(team_no: 0, player_id: user_id)
+            user_id = mention_to_user_id(mention)
+            display_name = e.display_name_for(user_id) || arg
+            pug.join_team(team_no: 0, player_id: user_id)
 
-        send_embedded_message(
-          description: "#{e.display_name} leaves team",
-          channel: e.channel
-        )
+            send_embedded_message(
+              description: "#{display_name} leaves team",
+              channel: e.channel
+            )
+          end
+        end
       end
     end
 
@@ -320,6 +341,13 @@ class QwtfDiscordBotPug # :nodoc:
           )
         end
 
+        if !pug.full?
+          return send_embedded_message(
+            description: "Can't report unless PUG is full",
+            channel: event.channel
+          )
+        end
+
         unless args.any?
           return send_embedded_message(
             description: "Specify winning team; e.g. `!win 1`",
@@ -352,7 +380,7 @@ class QwtfDiscordBotPug # :nodoc:
           teams.merge({ name => { players: players, result: result } })
         end
 
-        post_results(
+        id = post_results(
           {
             match: {
               map: pug.game_map,
@@ -363,10 +391,10 @@ class QwtfDiscordBotPug # :nodoc:
               }
             }
           }.to_json
-        )
+        ).body
 
         send_embedded_message(
-          description: "#{TEAM_NAMES[winning_team_no]} wins. [Ratings](http://ratings.fortressone.org)",
+          description: "#{TEAM_NAMES[winning_team_no]} wins game ##{id}. [Ratings](http://ratings.fortressone.org)",
           channel: e.channel
         )
       end
@@ -381,6 +409,13 @@ class QwtfDiscordBotPug # :nodoc:
           )
         end
 
+        if !pug.full?
+          return send_embedded_message(
+            description: "Can't report unless PUG is full",
+            channel: event.channel
+          )
+        end
+
         if pug.actual_teams.count < 2
           return send_embedded_message(
             description: "There must be at least two teams with players to submit a result",
@@ -396,7 +431,7 @@ class QwtfDiscordBotPug # :nodoc:
         teams.merge({ name => { players: players, result: 0 } })
         end
 
-        post_results(
+        id = post_results(
           {
             match: {
               map: pug.game_map,
@@ -407,10 +442,10 @@ class QwtfDiscordBotPug # :nodoc:
               }
             }
           }.to_json
-        )
+        ).body
 
         send_embedded_message(
-          description: "Match drawn. [Ratings](http://ratings.fortressone.org)",
+          description: "Match ##{id} drawn. [Ratings](http://ratings.fortressone.org)",
           channel: e.channel
         )
       end
@@ -628,8 +663,8 @@ class QwtfDiscordBotPug # :nodoc:
 
   def status(pug:, event:, message_obj: nil)
     footer = [
-      pug.game_map,
-      "#{pug.player_slots} joined"
+      pug.game_map || "No map selected",
+      "#{pug.player_slots} joined",
     ].compact.join(MSG_SNIPPET_DELIMITER)
 
     send_embedded_message(

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: qwtf_discord_bot
 version: !ruby/object:Gem::Version
-  version: 5.4.10
+  version: 5.5.1
 platform: ruby
 authors:
 - Sheldon Johnson
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2020-10-31 00:00:00.000000000 Z
+date: 2020-11-15 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: discordrb
@@ -102,6 +102,7 @@ files:
 - emoji/red_soldier.png
 - emoji/red_spy.png
 - exe/qwtf_discord_bot
+- lib/dashboard.rb
 - lib/emoji.rb
 - lib/endpoint.rb
 - lib/event_decorator.rb
@@ -110,6 +111,7 @@ files:
 - lib/qstat_request.rb
 - lib/qwtf_discord_bot.rb
 - lib/qwtf_discord_bot/config.rb
+- lib/qwtf_discord_bot/qwtf_discord_bot_dashboard.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