brainiac-discord 0.0.22 → 0.0.24

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 0c4e465a488b1eb8ff21dc46d539271cd85a35086fbed35a32e49bad609ee949
-  data.tar.gz: 5095805d24f0422c6ad9fe0ea7bb363965732b6d2b6df047d00295ce14b05194
+  metadata.gz: 7111ffc72dd19a6df6442ba8634e288c5085f06d696d51941a9fe64f9223a7db
+  data.tar.gz: 6748908008c71f05cdf84eff403bfdeee5ef30bc950acb327de7e5556bbdfc3d
 SHA512:
-  metadata.gz: e8c98fba862167e6c12096161dfb5fd964b12c6cb938c767279b00c2002a0770993c0f3a68b6b9f35fd391fbecba83d19f6fb21b1d1972a05aad4adcd89c78ff
-  data.tar.gz: 37d9072132b6c961f15c7237f9b5bae98db6ddf0933f39ea9bfee4c9b983a85eb8b007e6cf2b11d8fc19fe75c261f733429cbad1827d124c4c47cbdaa811147e
+  metadata.gz: a01f1a444b3714dce15727b9c6572295f20b77d1891abaac462da1ee49ebcc3cefedcf44a7e18d00e2eeb339c0a3c70d3513a481a4c8524751aa91d3d7fc59f1
+  data.tar.gz: 72e2b469f446a22464134230c5d16a0c7d2d4ea3fee06adb450e795441b4061f87e9f0346ef8be50be573e9f355f6a15857e944ccc5469f3a40d337644f8fa71

data/lib/brainiac/plugins/discord/agent_setup.rb

@@ -0,0 +1,204 @@
+# frozen_string_literal: true
+
+require "base64"
+require "fileutils"
+require "json"
+require "net/http"
+require "uri"
+
+module Brainiac
+  module Plugins
+    module Discord
+      # Provisions a Discord bot for a new brainiac agent (`brainiac agent create`
+      # / `brainiac agent setup`).
+      #
+      # Discord has no API for creating applications, so the one manual step is
+      # "New Application → Bot → Reset Token" in the developer portal. Given that
+      # token, everything else is automated: bot username + avatar, app icon +
+      # description, the Message Content intent, the token in agents.json, the
+      # user_mappings entry, and the server invite link.
+      module AgentSetup
+        API_BASE = "https://discord.com/api/v10"
+        PORTAL_URL = "https://discord.com/developers/applications"
+
+        # Application flags that PATCH /applications/@me is allowed to change.
+        GATEWAY_PRESENCE_LIMITED = 1 << 13
+        GATEWAY_GUILD_MEMBERS_LIMITED = 1 << 15
+        GATEWAY_MESSAGE_CONTENT_LIMITED = 1 << 19
+        EDITABLE_FLAGS = GATEWAY_PRESENCE_LIMITED | GATEWAY_GUILD_MEMBERS_LIMITED | GATEWAY_MESSAGE_CONTENT_LIMITED
+
+        # View channels, send/read messages, embeds, files, reactions, threads.
+        BOT_PERMISSIONS = [10, 11, 14, 15, 16, 6, 34, 35, 38].sum { |bit| 1 << bit }
+
+        APP_DESCRIPTION_LIMIT = 400
+        IMAGE_TYPES = { ".png" => "image/png", ".jpg" => "image/jpeg", ".jpeg" => "image/jpeg",
+                        ".gif" => "image/gif", ".webp" => "image/webp" }.freeze
+
+        class << self
+          def run(agent_key, entry)
+            display_name = entry["display_name"] || agent_key.capitalize
+            token = registry.dig(agent_key, "env", "DISCORD_BOT_TOKEN")
+
+            if token && discord_config.dig("user_mappings", display_name)
+              puts "  [Discord] ✓ #{display_name} already has a bot (skipping)"
+              return
+            end
+
+            token ||= prompt_for_token(display_name, entry["open_browser"])
+            return unless token
+
+            me = request(:get, "/users/@me", token)
+            unless me&.dig("id")
+              puts "  [Discord] ✗ Discord rejected that token — re-run: brainiac agent setup #{agent_key}"
+              return
+            end
+
+            avatar = image_data_uri(entry["avatar_path"])
+            update_bot_user(me, display_name, avatar, token)
+            app = request(:get, "/applications/@me", token) || {}
+            update_application(app, entry["description"], avatar, token)
+
+            save_token(agent_key, token)
+            save_user_mapping(display_name, me["id"])
+            puts "  [Discord] ✓ Saved DISCORD_BOT_TOKEN and user_mappings[#{display_name}] = #{me["id"]}"
+            puts "  [Discord] Invite #{display_name} to your server:"
+            puts "    #{invite_url(app["id"] || me["id"])}"
+            open_url(invite_url(app["id"] || me["id"])) if entry["open_browser"]
+          end
+
+          private
+
+          def prompt_for_token(display_name, browser)
+            return nil unless $stdin.tty?
+
+            puts ""
+            puts "  [Discord] Opening the developer portal in your browser…" if browser && open_url(PORTAL_URL)
+            puts "  [Discord] Create a bot for #{display_name} (the only manual step):"
+            puts "    1. #{PORTAL_URL} → New Application → name it \"#{display_name}\""
+            puts "    2. Bot tab → Reset Token → copy it"
+            puts "  Name, avatar, description and Message Content intent are set automatically."
+            print "  Bot token (blank to skip): "
+            token = read_secret
+            puts ""
+            token.nil? || token.empty? ? nil : token
+          end
+
+          # Best-effort: launch the default browser without blocking on it.
+          # Returns false when no launcher is available (headless, SSH, ...).
+          def open_url(url)
+            launcher = RUBY_PLATFORM.include?("darwin") ? "open" : "xdg-open"
+            pid = Process.spawn(launcher, url, %i[out err] => File::NULL)
+            Process.detach(pid)
+            true
+          rescue SystemCallError
+            false
+          end
+
+          def read_secret
+            require "io/console"
+            $stdin.noecho(&:gets)&.strip
+          rescue Errno::ENOTTY, NoMethodError
+            $stdin.gets&.strip
+          end
+
+          def update_bot_user(bot_user, display_name, avatar, token)
+            body = {}
+            body[:username] = display_name unless bot_user["username"] == display_name
+            body[:avatar] = avatar if avatar
+            return if body.empty?
+
+            if request(:patch, "/users/@me", token, body)
+              puts "  [Discord] ✓ Bot user: #{body.keys.join(" + ")} updated"
+            else
+              puts "  [Discord] ⚠ Couldn't update bot #{body.keys.join("/")} (username taken or rate limited?) — set it in the portal"
+            end
+          end
+
+          def update_application(app, description, avatar, token)
+            body = {}
+            body[:description] = truncate(description, APP_DESCRIPTION_LIMIT) if description
+            body[:icon] = avatar if avatar
+            if body.any? && !request(:patch, "/applications/@me", token, body)
+              puts "  [Discord] ⚠ Couldn't update app #{body.keys.join("/")} — set it in the portal"
+            end
+
+            # Separate request so a flags rejection doesn't lose the icon/description.
+            flags = ((app["flags"] || 0) & EDITABLE_FLAGS) | GATEWAY_MESSAGE_CONTENT_LIMITED
+            if request(:patch, "/applications/@me", token, { flags: flags })
+              puts "  [Discord] ✓ App #{(body.keys + [:message_content_intent]).join(" + ")} set"
+            else
+              puts "  [Discord] ⚠ Couldn't enable Message Content intent — Bot tab → Privileged Gateway Intents"
+            end
+          end
+
+          def invite_url(client_id)
+            "https://discord.com/oauth2/authorize?client_id=#{client_id}&scope=bot&permissions=#{BOT_PERMISSIONS}"
+          end
+
+          def image_data_uri(path)
+            return nil unless path && File.file?(path)
+
+            mime = IMAGE_TYPES[File.extname(path).downcase]
+            return nil unless mime
+
+            "data:#{mime};base64,#{Base64.strict_encode64(File.binread(path))}"
+          end
+
+          def truncate(text, limit)
+            text = text.strip
+            text.length > limit ? "#{text[0, limit - 1].rstrip}…" : text
+          end
+
+          # Returns the parsed body on 2xx, nil otherwise.
+          def request(method, path, token, body = nil)
+            uri = URI("#{API_BASE}#{path}")
+            klass = { get: Net::HTTP::Get, patch: Net::HTTP::Patch }.fetch(method)
+            req = klass.new(uri)
+            req["Authorization"] = "Bot #{token}"
+            req["Content-Type"] = "application/json"
+            req["User-Agent"] = "DiscordBot (https://github.com/stowzilla/brainiac-discord, 1.0)"
+            req.body = body.to_json if body
+
+            response = Net::HTTP.start(uri.host, uri.port, use_ssl: true) { |http| http.request(req) }
+            return nil unless response.is_a?(Net::HTTPSuccess)
+
+            response.body.to_s.empty? ? {} : JSON.parse(response.body)
+          rescue StandardError
+            nil
+          end
+
+          def brainiac_dir
+            ENV.fetch("BRAINIAC_DIR", File.join(Dir.home, ".brainiac"))
+          end
+
+          def registry_file = File.join(brainiac_dir, "agents.json")
+          def config_file = File.join(brainiac_dir, "discord.json")
+
+          def registry
+            File.exist?(registry_file) ? JSON.parse(File.read(registry_file)) : {}
+          end
+
+          def discord_config
+            File.exist?(config_file) ? JSON.parse(File.read(config_file)) : {}
+          end
+
+          def save_token(agent_key, token)
+            reg = registry
+            reg[agent_key] ||= {}
+            reg[agent_key]["env"] ||= {}
+            reg[agent_key]["env"]["DISCORD_BOT_TOKEN"] = token
+            FileUtils.mkdir_p(brainiac_dir)
+            File.write(registry_file, JSON.pretty_generate(reg))
+          end
+
+          def save_user_mapping(display_name, user_id)
+            config = discord_config
+            config["user_mappings"] ||= {}
+            config["user_mappings"][display_name] = user_id
+            File.write(config_file, JSON.pretty_generate(config))
+          end
+        end
+      end
+    end
+  end
+end

data/lib/brainiac/plugins/discord/cli.rb

@@ -3,6 +3,7 @@
 require "json"
 require "net/http"
 require "uri"
+require_relative "agent_setup"
 
 module Brainiac
   module Plugins
@@ -270,25 +271,12 @@ module Brainiac
         %w[setup config default map owner token status]
       end
 
-      # Called by brainiac CLI after `agent create` — prompts for Discord user ID.
+      # Called by brainiac CLI after `agent create` / `agent setup` — provisions
+      # the agent's Discord bot (see AgentSetup). Safe to re-run.
       def self.on_agent_created(agent_key, entry)
-        return unless $stdin.tty?
-
-        display_name = entry["display_name"] || agent_key.capitalize
-        puts ""
-        puts "  [Discord] Configure Discord for #{display_name}?"
-        print "  Discord user ID (or blank to skip): "
-        user_id = $stdin.gets&.chomp
-        return if user_id.nil? || user_id.empty?
-
-        config_file = File.join(ENV.fetch("BRAINIAC_DIR", File.join(Dir.home, ".brainiac")), "discord.json")
-        config = File.exist?(config_file) ? JSON.parse(File.read(config_file)) : {}
-        config["user_mappings"] ||= {}
-        config["user_mappings"][display_name] = user_id
-        File.write(config_file, JSON.pretty_generate(config))
-        puts "  ✓ Added #{display_name} → #{user_id} to discord.json user_mappings"
+        AgentSetup.run(agent_key, entry)
       rescue JSON::ParserError => e
-        puts "  ⚠ Could not update discord.json: #{e.message}"
+        puts "  ⚠ Could not update Discord config: #{e.message}"
       end
 
       # Called by brainiac CLI after `agent remove` — removes from Discord user_mappings.

data/lib/brainiac/plugins/discord/version.rb

@@ -3,7 +3,7 @@
 module Brainiac
   module Plugins
     module Discord
-      VERSION = "0.0.22"
+      VERSION = "0.0.24"
     end
   end
 end

data/lib/brainiac/plugins/discord.rb

@@ -65,6 +65,21 @@ module Brainiac
             end
           end
 
+          # `agent create` writes the agent before the Discord step saves its token,
+          # so the server can see the agent first and only get the token here.
+          Brainiac.on(:agent_updated) do |ctx|
+            agent_key = ctx[:agent_key]
+            token = ((ctx[:entry].is_a?(Hash) && ctx[:entry]["env"]) || {})["DISCORD_BOT_TOKEN"]
+            next if token == Gateway.bot_token(agent_key)
+
+            Config.reload! # pick up the user_mappings entry the setup step wrote
+            Gateway.stop_bot!(agent_key)
+            if token
+              Gateway.start_bot!(agent_key, token)
+              LOG.info "[Discord] Started bot for #{ctx[:display_name]} (token added/changed)" if defined?(LOG)
+            end
+          end
+
           Brainiac.on(:agent_removed) do |ctx|
             agent_key = ctx[:agent_key]
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: brainiac-discord
 version: !ruby/object:Gem::Version
-  version: 0.0.22
+  version: 0.0.24
 platform: ruby
 authors:
 - Andy Davis
@@ -103,6 +103,7 @@ extra_rdoc_files: []
 files:
 - README.md
 - lib/brainiac/plugins/discord.rb
+- lib/brainiac/plugins/discord/agent_setup.rb
 - lib/brainiac/plugins/discord/api.rb
 - lib/brainiac/plugins/discord/branching.rb
 - lib/brainiac/plugins/discord/cli.rb