Merge pull request #3 from ombulabs/IIRR-13-admin-choice

[IIRR-13] Admin channel choice

Author: ABizzinotto

app/lib/discord/events/server_create.rb

@@ -21,6 +21,8 @@ def handle(event)
         else
           puts "No suitable channel found to add the bot to."
         end
+
+        create_server(server)
       end
 
       private
@@ -39,6 +41,23 @@ def welcome_message(server_name)
       end
 
       attr_reader :bot
+
+      def create_server(server)
+        # Only create a new record if it doesn't already exist
+        return if Server.exists?(server_id: server.id)
+
+        # Create the new Server record
+        discord_server = Server.create!(
+          server_id: server.id,
+          name: server.name
+        )
+        if server.system_channel
+          discord_server.create_channel!(
+            channel_id: server.system_channel.id,
+            name: server.system_channel.name
+          )
+        end
+      end
     end
   end
 end

app/lib/discord/events/set_channel.rb

@@ -0,0 +1,72 @@
+module Discord
+  module Events
+    class SetChannel
+      def initialize(bot)
+        @bot = bot
+      end
+
+      def listen
+        # Listen for the set_channel command
+        @bot.application_command(:set_channel) do |event|
+          handle(event)
+        end
+
+        # Listen for select menu event
+        listen_to_channel_select
+      end
+
+      def handle(event)
+        user = event.user
+        server = event.server
+
+        # Ensure the user has the necessary permissions to run the command
+        unless user.permission?(:manage_server)
+          event.respond("You do not have permission to run this command.")
+          return
+        end
+        display_settings(event)
+      end
+
+
+      private
+
+      def display_settings(event)
+        content = <<~MSG
+          Here you can configure the bot to suit the needs of the community!
+
+          What's the best channel to send Ruby or Rails questions to?
+        MSG
+        channel_select = Discordrb::Components::View.new do |view|
+          view.row do |row|
+            row.channel_select(custom_id: "set_channel_select", placeholder: "Discussion prompts channel")
+          end
+        end
+
+        event.respond(content: content, ephemeral: true, components: channel_select)
+      end
+
+
+
+      def listen_to_channel_select
+        # Listen for the channel select event
+        @bot.channel_select(custom_id: "set_channel_select") do |event|
+          channel = event.values.first  # Get the selected channel
+          server = Server.find_by!(server_id: event.server.id)
+
+          existing_channel = server.channel
+
+          if existing_channel
+            # If the channel exists, update it
+            existing_channel.update!(name: channel.name, channel_id: channel.id)
+          else
+            # If the channel doesn't exist, create a new one associated with the server
+            server.create_channel!(name: channel.name, channel_id: channel.id)
+          end
+
+          # Respond with a confirmation message
+          event.respond(content: "Thank you! Ruby or Rails questions will be sent to #{channel.name} from now on.", ephemeral: true)
+        end
+      end
+    end
+  end
+end

app/models/channel.rb

@@ -0,0 +1,3 @@
+class Channel < ApplicationRecord
+  belongs_to :server
+end

app/models/server.rb

@@ -0,0 +1,3 @@
+class Server < ApplicationRecord
+  has_one :channel
+end

db/migrate/20250319152948_create_servers.rb

@@ -0,0 +1,11 @@
+class CreateServers < ActiveRecord::Migration[8.0]
+  def change
+    create_table :servers do |t|
+      t.string :server_id, null: false
+      t.string :name
+
+      t.timestamps
+    end
+    add_index :servers, :server_id, unique: true
+  end
+end

db/migrate/20250319153043_create_channels.rb

@@ -0,0 +1,11 @@
+class CreateChannels < ActiveRecord::Migration[8.0]
+  def change
+    create_table :channels do |t|
+      t.references :server, null: false, foreign_key: true
+      t.string :channel_id, null: false
+      t.string :name
+
+      t.timestamps
+    end
+  end
+end

db/schema.rb

@@ -3,6 +3,22 @@ namespace :discord do
   task start_bot: :environment do
     bot.run
   end
+
+  desc "Clear all globally registered bot commands and re-register them"
+  task reset_commands: :environment do
+    # Fetch and delete all existing commands
+    commands = bot.get_application_commands
+    commands.each do |command|
+      bot.delete_application_command(command.id)
+    end
+
+    puts "✅ Cleared all global commands. Now re-registering..."
+
+    # Register the commands again
+    bot.register_application_command(:set_channel, "Set a channel for bot to be used in.")
+
+    puts "✅ Re-registered commands!"
+  end
 end
 
 def bot