qwtf_discord_bot 5.4.11 → 5.5.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b7785ca7ed0ab47cdae89946225b60fe94c4e5ef21e8df14a1855ed70c262f9e
-  data.tar.gz: fbc91574a3557232b40236f750e4c6ad2e7e130f15e3176030ec18cfd519c776
+  metadata.gz: c0816f5d21f30d0692f0ad83adeff020e8898fc8c72cde120894dabcc14c133d
+  data.tar.gz: 69e9969b3e56e97351f5372a812f366ce0bda2d578b595c6e0e2567d5a7e40cc
 SHA512:
-  metadata.gz: 8ba026861ed819cbf511bf92f15781d88558788e4617c0a9e8e6af939907d3d1cde267ff1ad3e61f0f660e9e1c50b7f1d90b03299583c6a14037a23d5617df0f
-  data.tar.gz: 710d552cea8daf99b052e842db5120a5e399667d3a7af4391441ea410029d90ab3a316a5d67af93a9f9ffec6195ad05f59b3eb35c07b9a7dfad34eb7a73dba9b
+  metadata.gz: 18ab3a62f5eb12511dbea9f78889ad0de03b841b931aaa8eec1b3b746dd890e4c3cb9c13042976e4a3beae782dd568ba4d403c22c3caf86bcf8de70edc45555f
+  data.tar.gz: 72e3901ae00db397a0f4098adbdab40dabb7561cb677aa8159941ce1bd260ea53d95f077a4b8da3edaaa6c1d76971689b1020e879fa1456944db96822636b8c3

data/VERSION

@@ -1 +1 @@
-5.4.11
+5.5.2

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,44 @@
+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
+    )
+
+    puts JSON.parse(data)
+
+    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`",
@@ -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",

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: qwtf_discord_bot
 version: !ruby/object:Gem::Version
-  version: 5.4.11
+  version: 5.5.2
 platform: ruby
 authors:
 - Sheldon Johnson
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2020-11-01 00:00:00.000000000 Z
+date: 2020-11-16 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