qwtf_discord_bot 5.4.13 → 5.5.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 256bc5dd2a321a2e9495764c946142a36bdc4cec6f4871a70c3d8b2b6e0d1357
-  data.tar.gz: 16d3964888b04bdb1580ef62b1541375876dfdd2a24847ff60bc2927871f8ab3
+  metadata.gz: 114b6c7c1ecc4b4cd957b359b58f42af35a29979830fbb397121ef37d26b2fdd
+  data.tar.gz: '0098c992179231b57e8b443b8f658354e5e05bc4e284de739c1e442c771d163f'
 SHA512:
-  metadata.gz: c153a53da144bd6af78ea5013dd080d1f7da909ec532da2fa3787aa872ef03d778aaf30857d63ed204934cdfe5be3ff23aab166f6a747291ccb67f94ea23f10e
-  data.tar.gz: 986cc092056e071b8dcd877a4869d98911e9366253842f48e75df0f333445bfab71da2a6bab20e8242127a38a4fa187c930279497a3b632bc81ad63327972b65
+  metadata.gz: c8da38ea19d0f3bd0ebcc0bf82468d8347ec7d343f0f67acafb1fa0bb567b50ad28ad2af268022cf4f1fd6e1f405203740318a6a9112ce9d4971a90d3fa66f68
+  data.tar.gz: c801b72f1614f818816a6ced4737d58cb4b59dfe2c5db779b22cf7d124ceb364bdc58d47307f842efe58992eee31c6c729e769f8b6a3f795b9ad462eb76b7e1c

data/VERSION

@@ -1 +1 @@
-5.4.13
+5.5.0

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,40 @@
+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?
+        next unless @messages[endpoint]
+
+        @messages[endpoint].delete
+        @messages.delete(endpoint)
+      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?

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

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: qwtf_discord_bot
 version: !ruby/object:Gem::Version
-  version: 5.4.13
+  version: 5.5.0
 platform: ruby
 authors:
 - Sheldon Johnson
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2020-11-04 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