qwtf_discord_bot 5.0.2 → 5.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 452714fe7a833e8c33d3825de55e52be1716f30fecd8c80df2705aa35df37bc7
-  data.tar.gz: bafc402721ac62475747ffe255b7f09f851672fff58ec05e89051d765282ce5b
+  metadata.gz: 9a23eb8820860898165bbdf94e448a7141449fa3850a8e66f079e93ca8590f0f
+  data.tar.gz: 33ea995e466b5379746ee57b12e24f3337581dd0c58e4e526b30eb2a9377778d
 SHA512:
-  metadata.gz: 17a594fdf8f231e4b067496cf59b9fdb777231ac9585c4df93df061948580100c5235bc4e7e31281a4606d22e6ca490e863aa13f2783b66292b73b599cca2bbc
-  data.tar.gz: 8fb14f30321620378acb31553bbfb89c0432b4ab58b4f69c7f27a294db94ff26686851e819b3cb74c2125ac6c2a8b5c509204eb90f70155c2af41e869960cae7
+  metadata.gz: 53393cfca043b1ff72cbc2a7f9e28d9a7f4c44381990ba23edf392f23afb84c66715e5ff274573d16587797fad5c9e76739edb67039eb72106f4767faf95e368
+  data.tar.gz: ae90377b7c6c4592e6811a91461ab13a3e8650d17c36a67f8c45717930e2956c738123c16db087c036508a0da1898fc0d0a1f5efac71a9149ce809da8d277067

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    qwtf_discord_bot (4.2.6)
+    qwtf_discord_bot (5.0.4)
       discordrb (~> 3.3)
       redis (~> 4.1)
       thor (~> 0.20)
@@ -9,14 +9,15 @@ PATH
 GEM
   remote: https://rubygems.org/
   specs:
-    activesupport (5.2.3)
+    activesupport (6.0.3.2)
       concurrent-ruby (~> 1.0, >= 1.0.2)
       i18n (>= 0.7, < 2)
       minitest (~> 5.1)
       tzinfo (~> 1.1)
-    coderay (1.1.2)
-    concurrent-ruby (1.1.5)
-    diff-lcs (1.3)
+      zeitwerk (~> 2.2, >= 2.2.2)
+    coderay (1.1.3)
+    concurrent-ruby (1.1.7)
+    diff-lcs (1.4.4)
     discordrb (3.3.0)
       discordrb-webhooks (~> 3.3.0)
       ffi (>= 1.9.24)
@@ -26,60 +27,61 @@ GEM
       websocket-client-simple (>= 0.3.0)
     discordrb-webhooks (3.3.0)
       rest-client (>= 2.1.0.rc1)
-    domain_name (0.5.20180417)
+    domain_name (0.5.20190701)
       unf (>= 0.0.5, < 1.0.0)
     event_emitter (0.2.6)
-    factory_bot (5.0.2)
-      activesupport (>= 4.2.0)
-    ffi (1.11.1)
+    factory_bot (6.1.0)
+      activesupport (>= 5.0.0)
+    ffi (1.13.1)
     http-accept (1.7.0)
     http-cookie (1.0.3)
       domain_name (~> 0.5)
-    i18n (1.6.0)
+    i18n (1.8.5)
       concurrent-ruby (~> 1.0)
-    method_source (0.9.2)
-    mime-types (3.2.2)
+    method_source (1.0.0)
+    mime-types (3.3.1)
       mime-types-data (~> 3.2015)
-    mime-types-data (3.2019.0331)
-    minitest (5.11.3)
+    mime-types-data (3.2020.0512)
+    minitest (5.14.1)
     netrc (0.11.0)
     opus-ruby (1.0.1)
       ffi
-    pry (0.12.2)
-      coderay (~> 1.1.0)
-      method_source (~> 0.9.0)
+    pry (0.13.1)
+      coderay (~> 1.1)
+      method_source (~> 1.0)
     rbnacl (3.4.0)
       ffi
-    redis (4.1.3)
-    rest-client (2.1.0.rc1)
+    redis (4.2.1)
+    rest-client (2.1.0)
       http-accept (>= 1.7.0, < 2.0)
       http-cookie (>= 1.0.2, < 2.0)
       mime-types (>= 1.16, < 4.0)
       netrc (~> 0.8)
-    rspec (3.8.0)
-      rspec-core (~> 3.8.0)
-      rspec-expectations (~> 3.8.0)
-      rspec-mocks (~> 3.8.0)
-    rspec-core (3.8.1)
-      rspec-support (~> 3.8.0)
-    rspec-expectations (3.8.4)
+    rspec (3.9.0)
+      rspec-core (~> 3.9.0)
+      rspec-expectations (~> 3.9.0)
+      rspec-mocks (~> 3.9.0)
+    rspec-core (3.9.2)
+      rspec-support (~> 3.9.3)
+    rspec-expectations (3.9.2)
       diff-lcs (>= 1.2.0, < 2.0)
-      rspec-support (~> 3.8.0)
-    rspec-mocks (3.8.1)
+      rspec-support (~> 3.9.0)
+    rspec-mocks (3.9.1)
       diff-lcs (>= 1.2.0, < 2.0)
-      rspec-support (~> 3.8.0)
-    rspec-support (3.8.2)
+      rspec-support (~> 3.9.0)
+    rspec-support (3.9.3)
     thor (0.20.3)
     thread_safe (0.3.6)
-    tzinfo (1.2.5)
+    tzinfo (1.2.7)
       thread_safe (~> 0.1)
     unf (0.1.4)
       unf_ext
-    unf_ext (0.0.7.6)
+    unf_ext (0.0.7.7)
     websocket (1.2.8)
     websocket-client-simple (0.3.0)
       event_emitter
       websocket
+    zeitwerk (2.4.0)
 
 PLATFORMS
   ruby

data/README.md

@@ -90,7 +90,8 @@ This responds to discord messages:
   - `!leave`
   - `!status`
   - `!maxplayers <no_of_players>`
-  - `!leave`
+  - `!notify <roles>`
+  - `!end`
 
 
 #### Watcher

data/VERSION

@@ -1 +1 @@
-5.0.2
+5.1.2

data/lib/event_decorator.rb

@@ -0,0 +1,54 @@
+class EventDecorator
+  def initialize(event)
+    @event = event
+  end
+
+  def channel
+    @event.channel
+  end
+
+  def channel_id
+    @event.channel.id
+  end
+
+  def username
+    user.username
+  end
+
+  def user_id
+    user.id
+  end
+
+  def users
+    server.users
+  end
+
+  def mentions_for(user_ids)
+    find_users(user_ids).map(&:mention)
+  end
+
+  def usernames_for(user_ids)
+    find_users(user_ids).map(&:username)
+  end
+
+  private
+
+  def find_users(user_ids)
+    user_ids.map do |user_id|
+      find_user(user_id)
+    end
+  end
+
+  def find_user(user_id)
+    users.find { |user| user.id == user_id }
+  end
+
+  def server
+    @event.server
+  end
+
+  def user
+    @event.user
+  end
+end
+

data/lib/pug.rb

@@ -0,0 +1,91 @@
+class Pug
+  DEFAULT_MAXPLAYERS = 8
+
+  def self.for(channel_id)
+    new(channel_id)
+  end
+
+  def initialize(channel_id)
+    @channel_id = channel_id
+  end
+
+  def join(user_id)
+    redis.setnx(pug_key, Time.now)
+    redis.sadd(players_key, user_id)
+  end
+
+  def joined_players
+    redis.smembers(players_key).map(&:to_i)
+  end
+
+  def full?
+    joined_player_count >= maxplayers
+  end
+
+  def joined_player_count
+    redis.scard(players_key).to_i
+  end
+
+  def player_slots
+    "#{joined_player_count}/#{maxplayers}"
+  end
+
+  def slots_left
+    maxplayers - joined_player_count
+  end
+
+  def notify_roles=(roles)
+    redis.set(notify_roles_key, roles)
+  end
+
+  def notify_roles
+    redis.get(notify_roles_key) || "@here"
+  end
+
+  def maxplayers=(maxplayers)
+    redis.set(maxplayers_key, maxplayers)
+  end
+
+  def maxplayers
+    (redis.get(maxplayers_key) || DEFAULT_MAXPLAYERS).to_i
+  end
+
+  def active?
+    redis.get(pug_key)
+  end
+
+  def leave(player_id)
+    redis.srem(players_key, player_id)
+  end
+
+  def empty?
+    joined_player_count == 0
+  end
+
+  def end_pug
+    redis.del(pug_key)
+    redis.del(players_key)
+  end
+
+  private
+
+  def maxplayers_key
+    [pug_key, "maxplayers"].join(":")
+  end
+
+  def players_key
+    [pug_key, "players"].join(":")
+  end
+
+  def pug_key
+    ["pug", "channel", @channel_id].join(":")
+  end
+
+  def notify_roles_key
+    [pug_key, "role"].join(":")
+  end
+
+  def redis
+    Redis.current
+  end
+end

data/lib/qwtf_discord_bot/qwtf_discord_bot_pug.rb

@@ -1,6 +1,9 @@
-require 'event_wrapper'
+# frozen_string_literal: true
 
-class QwtfDiscordBotPug
+require 'pug'
+require 'event_decorator'
+
+class QwtfDiscordBotPug # :nodoc:
   include QwtfDiscordBot
 
   FOUR_HOURS = 4 * 60 * 60
@@ -14,109 +17,127 @@ class QwtfDiscordBotPug
     )
 
     bot.command :join do |event, *args|
-      e = EventWrapper.new(event)
+      set_pug(event, args) do |e, pug|
+        pug.join(e.user_id)
+
+        message = if pug.joined_player_count == 1
+                    [
+                      "#{e.username} creates a PUG",
+                      pug.player_slots,
+                      pug.notify_roles
+                    ].join(' | ')
+                  elsif pug.slots_left.between?(1,3)
+                    [
+                      "#{e.username} joins the PUG",
+                      pug.player_slots,
+                      "#{pug.slots_left} more",
+                      pug.notify_roles
+                    ].join(' | ')
+                  else
+                    [
+                      "#{e.username} joins the PUG",
+                      pug.player_slots
+                    ].join(' | ')
+                  end
 
-      redis.setnx(e.pug_key, Time.now)
-      redis.sadd(e.players_key, e.user_id)
+        send_and_log_message(message, e.channel)
 
-      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)
+        if pug.full?
+          message = start_pug(
+            pug.player_slots,
+            e.mentions_for(pug.joined_players)
+          )
+
+          send_and_log_message(message, e.channel)
+        end
+      end
     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(" | ")
+      set_pug(event, args) do |e, pug|
+        message = if pug.active?
+                    [
+                      "#{e.usernames_for(pug.joined_players).join(', ')} joined",
+                      pug.player_slots
+                    ].join(' | ')
+                  else
+                    'No PUG has been started. `!join` to create'
+                  end
 
-      send_and_log_message(message, event)
+        send_and_log_message(message, e.channel)
+      end
     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
+      set_pug(event, args) do |e, pug|
+        new_maxplayers = args[0]
+
+        message = if new_maxplayers
+                    pug.maxplayers = new_maxplayers
+                    "Max number of players set to #{pug.maxplayers} | #{pug.player_slots} joined"
+                  else
+                    "Current max number of players is #{pug.maxplayers} | #{pug.player_slots} joined"
+                  end
 
-      send_and_log_message(message, event)
+        send_and_log_message(message, e.channel)
 
-      if e.joined_player_count >= e.maxplayers
-        mentions = joined_users(e).map do |user|
-          user.mention
-        end
+        if pug.full?
+          message = start_pug(
+            pug.player_slots,
+            e.mentions_for(pug.joined_players)
+          )
 
-        message = "Time to play! #{mentions.join(" ")}"
-        send_and_log_message(message, event)
+          send_and_log_message(message, e.channel)
+        end
       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)
+      set_pug(event, args) do |e, pug|
+        if !pug.active?
+          message = "There's no active PUG to leave"
+          send_and_log_message(message, e.channel)
+        elsif !pug.joined_players.include?(e.user_id)
+          message = "You're not in the PUG"
+          send_and_log_message(message, e.channel)
+        else
+          pug.leave(e.user_id)
+          message = "#{e.username} leaves the PUG | #{pug.player_slots} remain"
+          send_and_log_message(message, e.channel)
+
+          if pug.empty?
+            message = end_pug(pug)
+            send_and_log_message(message, e.channel)
+          end
+        end
       end
     end
 
     bot.command :end do |event, *args|
-      e = EventWrapper.new(event)
-
-      redis.del(e.pug_key)
-      redis.del(e.players_key)
+      set_pug(event, args) do |e, pug|
+        message = if !pug.active?
+                    "There's no active PUG to end"
+                  else
+                    end_pug(pug)
+                  end
 
-      message = "PUG ended"
-      send_and_log_message(message, event)
+        send_and_log_message(message, e.channel)
+      end
     end
 
-    bot.command :role do |event, *args|
-      e = EventWrapper.new(event)
-      role = args[0]
+    bot.command :notify do |event, *args|
+      set_pug(event, args) do |e, pug|
+        roles = args.join(' ')
+        pug.notify_roles = roles
 
-      redis.set(e.role_key, role)
+        message = if roles.empty?
+                    'Notification removed'
+                  else
+                    "Notification role set to #{roles}"
+                  end
 
-      message = "Notification role set to #{role}"
-      send_and_log_message(message, event)
+        send_and_log_message(message, e.channel)
+      end
     end
 
     bot.run
@@ -124,16 +145,26 @@ class QwtfDiscordBotPug
 
   private
 
-  def send_and_log_message(message, event)
-    event.channel.send_message(message)
-    puts message
+  def set_pug(event, *args)
+    e = EventDecorator.new(event)
+    pug = Pug.for(e.channel_id)
+    yield(e, pug)
   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
+  def start_pug(player_slots, mentions)
+    [
+      'Time to play!',
+      player_slots,
+      mentions.join(' ')
+    ].join(' | ')
+  end
+
+  def end_pug(pug)
+    pug.end_pug
+    'PUG ended'
+  end
+
+  def send_and_log_message(message, channel)
+    channel.send_message(message) && puts(message)
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: qwtf_discord_bot
 version: !ruby/object:Gem::Version
-  version: 5.0.2
+  version: 5.1.2
 platform: ruby
 authors:
 - Sheldon Johnson
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2020-08-23 00:00:00.000000000 Z
+date: 2020-08-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: discordrb
@@ -104,8 +104,9 @@ files:
 - exe/qwtf_discord_bot
 - lib/emoji.rb
 - lib/endpoint.rb
-- lib/event_wrapper.rb
+- lib/event_decorator.rb
 - lib/player.rb
+- lib/pug.rb
 - lib/qstat_request.rb
 - lib/qwtf_discord_bot.rb
 - lib/qwtf_discord_bot/config.rb

data/lib/event_wrapper.rb

@@ -1,58 +0,0 @@
-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