qwtf_discord_bot 5.2.5 → 5.3.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 27e4374d157a5f3cad0f5c1cc58ffb1e9fc0b9f6ca9b89391509635260bc460a
-  data.tar.gz: 2775f0952e46afb2af7dbff0a72b388e34f343ed896b3d909ca0b9d069f76ab4
+  metadata.gz: 601af47db687365971e97efd281732ed5effb977d1cd0d451aeda38be1ab27df
+  data.tar.gz: 2d06e9cb9f311d0b427445760a13729df7f8a59b582ed8c536e219137c930a25
 SHA512:
-  metadata.gz: 6cf2acfe24299674b69051e0d28ce628cea334840910b5bbd083d73e10d91cdf2d7c96312371ffed448424b769abb86c77a63de05b4a23136d05e92398bf2ff1
-  data.tar.gz: 1a5830ef4f29e3c331d5534a84837e8d3e29bccf016e478e439aa8727836fe957d20f4865265264aa69116110f9e7ba7cf322869f6738aff5ef365f4d860137d
+  metadata.gz: b67d9dcc35b75b0ea64dc482df3f0c2d9e180aa70288b9ebb98f2c58016e08d9eb504a542cfaf2e51e449a40e3dda33f9ed05da992c226f5a0f465dc1805f951
+  data.tar.gz: 3a75ce85122e1d3db1e2dc8dfa5dc7f3c614633f2506353d31cee66a7db2d09d29fe9130030f036885ed8d56fcbea776fdb5ad087251ae75698f8239ebf87e43

data/README.md

@@ -86,15 +86,20 @@ This responds to discord messages:
     qwtf-discord-bot pug
 
 This responds to discord messages:
+  - `!status`
   - `!join`
   - `!leave`
-  - `!status`
-  - `!kick <@player>`
   - `!teamsize <no_of_players>`
+  - `!kick <@player>`
   - `!team <team_no>`
   - `!unteam`
+  - `!addmap <map_name>`
+  - `!removemap <map_name>`
+  - `!maps`
+  - `!map <map_name>`
   - `!win <team_no>`
-  - `!notify <@role @role2>`
+  - `!draw`
+  - `!notify <@role>`
   - `!end`
 
 

data/VERSION

@@ -1 +1 @@
-5.2.5
+5.3.3

data/docker-compose.yml

@@ -34,6 +34,7 @@ services:
       - redis
     environment:
       - REDIS_URL=redis://redis
+      - RATINGS_API_URL
     volumes:
       - type: bind
         source: "/home/ubuntu/.config/qwtf_discord_bot/config.yaml"

data/lib/event_decorator.rb

@@ -19,6 +19,10 @@ class EventDecorator
     user.id
   end
 
+  def mention
+    user.mention
+  end
+
   def users
     server.users
   end

data/lib/pug.rb

@@ -27,6 +27,26 @@ class Pug
     end
   end
 
+  def add_maps(maps)
+    redis.sadd(maps_key, maps)
+  end
+
+  def remove_maps(maps)
+    redis.srem(maps_key, maps)
+  end
+
+  def maps
+    redis.smembers(maps_key)
+  end
+
+  def vote(player_id:, map:)
+    redis.sadd(votes_key(map), player_id)
+  end
+
+  def vote_count(map)
+    redis.scard(votes_key(map)).to_i
+  end
+
   def team(number)
     redis.smembers(team_key(number)).map(&:to_i)
   end
@@ -39,6 +59,10 @@ class Pug
     joined_player_count >= maxplayers
   end
 
+  def empty?
+    joined_player_count.zero?
+  end
+
   def joined_player_count
     joined_players.count
   end
@@ -55,6 +79,14 @@ class Pug
     maxplayers - joined_player_count
   end
 
+  def game_map=(map)
+    redis.set([pug_key, 'map'].join(':'), map)
+  end
+
+  def game_map
+    redis.get([pug_key, 'map'].join(':'))
+  end
+
   def notify_roles=(roles)
     redis.set(notify_roles_key, roles)
   end
@@ -73,7 +105,6 @@ class Pug
 
   def leave(player_id)
     leave_teams(player_id)
-    end_pug if empty?
   end
 
   def end_pug
@@ -90,10 +121,6 @@ class Pug
     teamsize * no_of_teams
   end
 
-  def won_by(team_no)
-    { teams: teams, winner: team_no }
-  end
-
   def teams
     teams_keys.inject({}) do |teams, team|
       teams.merge({ team.split(':').last => redis.smembers(team).map(&:to_i) })
@@ -116,10 +143,6 @@ class Pug
     redis.keys([pug_key, 'teams:*'].join(':'))
   end
 
-  def empty?
-    joined_player_count.zero?
-  end
-
   def team_key(team_no)
     [pug_key, 'teams', team_no].join(':')
   end
@@ -128,8 +151,8 @@ class Pug
     [channel_key, 'pug'].join(':')
   end
 
-  def channel_key
-    ['channel', @channel_id].join(':')
+  def maps_key
+    [channel_key, 'maps'].join(':')
   end
 
   def notify_roles_key
@@ -137,7 +160,19 @@ class Pug
   end
 
   def teamsize_key
-    [pug_key, 'teamsize'].join(':')
+    [channel_key, 'teamsize'].join(':')
+  end
+
+  def votes_keys
+    redis.keys([pug_key, 'votes:*'].join(':'))
+  end
+
+  def votes_key(map)
+    [channel_key, 'votes', map].join(':')
+  end
+
+  def channel_key
+    ['channel', @channel_id].join(':')
   end
 
   def redis

data/lib/qwtf_discord_bot/qwtf_discord_bot_pug.rb

@@ -32,6 +32,7 @@ class QwtfDiscordBotPug # :nodoc:
         send_msg(
           [
             "#{pug.player_slots} joined",
+            "Map: #{pug.game_map}",
             pug_teams_message(pug, e).join("\n")
           ].join("\n"),
           e.channel
@@ -41,6 +42,8 @@ class QwtfDiscordBotPug # :nodoc:
 
     bot.command :teamsize do |event, *args|
       setup_pug(event) do |e, pug|
+        return send_msg("Team size is #{pug.teamsize}", e.channel) unless args.any?
+
         new_teamsize = args[0].to_i
         return send_msg('Team size should be a number higher than 0', e.channel) unless new_teamsize > 0
 
@@ -87,12 +90,13 @@ class QwtfDiscordBotPug # :nodoc:
           e.channel
         )
 
-        send_msg(end_pug_message, e.channel) unless pug.active?
+        end_pug(pug, e.channel) if pug.empty?
       end
     end
 
     bot.command :kick do |event, *args|
       setup_pug(event) do |e, pug|
+        return send_msg("Kick who? e.g. `!kick @#{e.display_name}`", e.channel) unless args.any?
         return send_msg(no_active_pug_message, e.channel) unless pug.active?
 
         args.each do |arg|
@@ -101,7 +105,7 @@ class QwtfDiscordBotPug # :nodoc:
             next
           end
 
-          user_id = arg[3..-2].to_i
+          user_id = mention_to_user_id(arg)
           display_name = e.display_name_for(user_id) || arg
 
           unless pug.joined?(user_id)
@@ -123,30 +127,60 @@ class QwtfDiscordBotPug # :nodoc:
             e.channel
           )
 
-          break send_msg(end_pug_message, e.channel) unless pug.active?
+          break end_pug(pug, e.channel) if pug.empty?
         end
       end
     end
 
     bot.command :team do |event, *args|
       setup_pug(event) do |e, pug|
+        return send_msg("Which team? E.G. `!team 1`", e.channel) unless args.any?
+
         team_no = args[0].to_i
         return send_msg("Choose a team between 1 and 4", e.channel) unless team_no.between?(1, 4)
 
-        user_id = e.user_id
-        return send_msg("You're already in team #{team_no}", e.channel) if pug.team(team_no).include?(user_id)
+        pug_already_full = pug.full?
 
-        join_pug(e, pug) unless pug.joined?(user_id)
-        pug.join_team(team_no: team_no, player_id: user_id)
+        if args.count == 1
+          user_id = e.user_id
+          return send_msg("You're already in team #{team_no}", e.channel) if pug.team(team_no).include?(user_id)
 
-        send_msg(
-          [
-            "#{e.display_name} joins team #{team_no}",
-            "#{pug.team_player_count(team_no)}/#{pug.teamsize}"
-          ].join(MSG_SNIPPET_DELIMITER), e.channel
-        )
+          join_pug(e, pug) unless pug.joined?(user_id)
+          pug.join_team(team_no: team_no, player_id: user_id)
 
-        start_pug(pug, e) if pug.full?
+          send_msg(
+            [
+              "#{e.display_name} joins team #{team_no}",
+              "#{pug.team_player_count(team_no)}/#{pug.teamsize}"
+            ].join(MSG_SNIPPET_DELIMITER), e.channel
+          )
+        else
+          args[1..-1].each do |mention|
+            unless mention.match(/<@!\d+>/)
+              send_msg("#{arg} isn't a valid mention", e.channel)
+              next
+            end
+
+            user_id = mention_to_user_id(mention)
+            display_name = e.display_name_for(user_id) || arg
+
+            unless pug.joined?(user_id)
+              send_msg("#{display_name} isn't in the PUG", e.channel)
+              next
+            end
+
+            pug.join_team(team_no: team_no, player_id: user_id)
+
+            send_msg(
+              [
+                "#{display_name} joins team #{team_no}",
+                "#{pug.team_player_count(team_no)}/#{pug.teamsize}"
+              ].join(MSG_SNIPPET_DELIMITER), e.channel
+            )
+          end
+        end
+
+        start_pug(pug, e) if !pug_already_full && pug.full?
       end
     end
 
@@ -170,22 +204,60 @@ class QwtfDiscordBotPug # :nodoc:
 
         return send_msg("Not a valid team", e.channel) unless pug.team(winning_team_no).any?
 
-        pug.won_by(winning_team_no)
+        team_results = pug.teams.inject({}) do |teams, (name, player_ids)|
+          players = player_ids.inject({}) do |memo, id|
+            memo.merge({ id => e.display_name_for(id) })
+          end
 
-        winning_team = pug.team(winning_team_no).map do |player_id|
-          e.display_name_for(player_id)
+        result = winning_team_no.to_i == name.to_i ? 1 : -1
+        teams.merge({ name => { players: players, result: result } })
         end
 
-        non_winning_teams = pug.actual_teams.tap { |team| team.delete(winning_team_no) }
+        post_results(
+          {
+            match: {
+              map: pug.game_map,
+              teams: team_results
+            }
+          }.to_json
+        )
+
+        # winning_team = pug.team(winning_team_no).map do |player_id|
+        #   e.display_name_for(player_id)
+        # end
+
+        # non_winning_teams = pug.actual_teams.tap { |team| team.delete(winning_team_no) }
+
+        # losing_players = non_winning_teams.values.flatten.map do |player_id|
+        #   e.display_name_for(player_id)
+        # end
+
+        send_msg("Team #{winning_team_no} wins", e.channel)
+      end
+    end
 
-        losing_players = non_winning_teams.values.flatten.map do |player_id|
-          e.display_name_for(player_id)
+    bot.command :draw do |event, *args|
+      setup_pug(event) do |e, pug|
+        return send_msg(no_active_pug_message, e.channel) unless pug.active?
+
+        team_results = pug.teams.inject({}) do |teams, (name, player_ids)|
+          players = player_ids.inject({}) do |memo, id|
+            memo.merge({ id => e.display_name_for(id) })
+          end
+
+        teams.merge({ name => { players: players, result: 0 } })
         end
 
-        send_msg(
-          "#{winning_team.join(', ')} defeat #{losing_players.join(', ')}",
-          e.channel
+        post_results(
+          {
+            match: {
+              map: pug.game_map,
+              teams: team_results
+            }
+          }.to_json
         )
+
+        send_msg("Match drawn", e.channel)
       end
     end
 
@@ -193,9 +265,55 @@ class QwtfDiscordBotPug # :nodoc:
       setup_pug(event) do |e, pug|
         return send_msg(no_active_pug_message, e.channel) unless pug.active?
 
-        pug.end_pug
+        end_pug(pug, e.channel)
+      end
+    end
+
+    bot.command :addmap do |event, *args|
+      setup_pug(event) do |e, pug|
+        maps = args
+        return send_msg("What map? e.g. `!addmap 2fort5r`", e.channel) unless maps.any?
+
+        pug.add_maps(maps)
+        send_msg("#{maps.join(', ')} added to maps", e.channel)
+      end
+    end
+
+    bot.command :removemap do |event, *args|
+      setup_pug(event) do |e, pug|
+        maps = args
+        return send_msg("What map? e.g. `!removemap 2fort5r`", e.channel) unless maps.any?
+
+        pug.remove_maps(maps)
+        send_msg("#{maps.join(', ')} removed from maps", e.channel)
+      end
+    end
+
+    bot.command :maps do |event, *args|
+      setup_pug(event) do |e, pug|
+        maps = pug.maps
+        return send_msg('No maps have been added. `!addmap`', e.channel) unless maps.any?
 
-        send_msg(end_pug_message, e.channel)
+        send_msg(maps.join(', '), e.channel)
+      end
+    end
+
+    bot.command :map do |event, *args|
+      setup_pug(event) do |e, pug|
+        maps = pug.maps
+        return send_msg('No maps have been added. `!addmap`', e.channel) unless maps.any?
+        return send_msg(no_active_pug_message, e.channel) unless pug.active?
+
+        if args.empty?
+          return send_msg('No map has been set for the current PUG', e.channel) unless pug.game_map
+          send_msg("Current map is #{pug.game_map}", e.channel)
+        else
+          game_map = args.first
+          return send_msg("#{game_map} isn't in the map list. `!addmap` to add it.", e.channel) unless maps.include?(game_map)
+
+          pug.game_map = game_map
+          send_msg("Map set to #{game_map}", e.channel)
+        end
       end
     end
 
@@ -219,6 +337,10 @@ class QwtfDiscordBotPug # :nodoc:
 
   private
 
+  def mention_to_user_id(mention)
+    mention[3..-2].to_i
+  end
+
   def join_pug(e, pug)
     pug.join(e.user_id)
 
@@ -285,8 +407,9 @@ class QwtfDiscordBotPug # :nodoc:
     end
   end
 
-  def end_pug_message
-    'PUG ended'
+  def end_pug(pug, channel_id)
+    pug.end_pug
+    send_msg('PUG ended', channel_id)
   end
 
   def no_active_pug_message
@@ -296,4 +419,13 @@ class QwtfDiscordBotPug # :nodoc:
   def send_msg(message, channel)
     channel.send_message(message) && puts(message)
   end
+
+  def post_results(json)
+    uri = URI("#{ENV['RATINGS_API_URL']}matches/")
+    req = Net::HTTP::Post.new(uri, 'Content-Type' => 'application/json')
+    req.body = json
+    Net::HTTP.start(uri.hostname, uri.port) do |http|
+      http.request(req)
+    end
+  end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: qwtf_discord_bot
 version: !ruby/object:Gem::Version
-  version: 5.2.5
+  version: 5.3.3
 platform: ruby
 authors:
 - Sheldon Johnson
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2020-09-23 00:00:00.000000000 Z
+date: 2020-10-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: discordrb