async-matrix 1.0.0 → 1.1.1

data/lib/async/matrix/bridge/discord/db/file.rb

@@ -0,0 +1,120 @@
+# frozen_string_literal: true
+
+# Released under the Apache License, Version 2.0.
+# Copyright, 2026, by General Intelligence Systems.
+
+require "bundler/setup"
+require "async/matrix"
+
+module Async
+  module Matrix
+    module Bridge
+      module Discord
+        module DB
+          # Caches uploaded media to avoid re-uploading the same Discord
+          # attachment to the Matrix homeserver.
+          #
+          # Uses a composite primary key of (url, encrypted) because the same
+          # source URL may be uploaded both encrypted and unencrypted to
+          # different portal rooms.
+          #
+          #   cached = CachedFile.create(
+          #     url: "https://cdn.discordapp.com/...",
+          #     encrypted: false,
+          #     mxc: "mxc://example.com/abc"
+          #   )
+          #
+          class CachedFile < Sequel::Model
+            unrestrict_primary_key
+
+            def validate
+              super
+              errors.add(:url, "cannot be empty") if url.nil? || url.empty?
+              errors.add(:mxc, "cannot be empty") if mxc.nil? || mxc.empty?
+            end
+
+            # Look up a cached file by source URL and encryption status.
+            def self.by_url(url, encrypted: false)
+              first(url: url, encrypted: encrypted)
+            end
+          end
+        end
+      end
+    end
+  end
+end
+
+test do
+  describe "Async::Matrix::Bridge::Discord::DB::CachedFile" do
+    def setup_db
+      db = Sequel.sqlite
+      Sequel::Migrator.run(db, File.join(__dir__, "migrations"))
+      db
+    end
+
+    it "creates and retrieves a cached file" do
+      db = setup_db
+      Async::Matrix::Bridge::Discord::DB::CachedFile.dataset = db[:files]
+
+      cached = Async::Matrix::Bridge::Discord::DB::CachedFile.create(
+        url: "https://cdn.discordapp.com/attachments/123/456/image.png",
+        mxc: "mxc://example.com/abc",
+        mime_type: "image/png",
+        size: 102400,
+        width: 800,
+        height: 600
+      )
+      cached.url.should.include "discordapp.com"
+      cached.mxc.should == "mxc://example.com/abc"
+      cached.mime_type.should == "image/png"
+      cached.width.should == 800
+      cached.height.should == 600
+      db.disconnect
+    end
+
+    it "looks up by URL and encryption status" do
+      db = setup_db
+      Async::Matrix::Bridge::Discord::DB::CachedFile.dataset = db[:files]
+
+      Async::Matrix::Bridge::Discord::DB::CachedFile.create(
+        url: "https://example.com/file.png", encrypted: false,
+        mxc: "mxc://x/plain"
+      )
+      Async::Matrix::Bridge::Discord::DB::CachedFile.create(
+        url: "https://example.com/file.png", encrypted: true,
+        mxc: "mxc://x/encrypted",
+        decryption_info: '{"key":"abc"}'
+      )
+
+      plain = Async::Matrix::Bridge::Discord::DB::CachedFile.by_url("https://example.com/file.png")
+      plain.mxc.should == "mxc://x/plain"
+
+      enc = Async::Matrix::Bridge::Discord::DB::CachedFile.by_url("https://example.com/file.png", encrypted: true)
+      enc.mxc.should == "mxc://x/encrypted"
+      enc.decryption_info.should == '{"key":"abc"}'
+      db.disconnect
+    end
+
+    it "defaults encrypted to false" do
+      db = setup_db
+      Async::Matrix::Bridge::Discord::DB::CachedFile.dataset = db[:files]
+
+      cached = Async::Matrix::Bridge::Discord::DB::CachedFile.create(
+        url: "https://example.com/a.jpg", mxc: "mxc://x/y"
+      )
+      cached.encrypted.should == false
+      db.disconnect
+    end
+
+    it "validates required fields" do
+      db = setup_db
+      Async::Matrix::Bridge::Discord::DB::CachedFile.dataset = db[:files]
+
+      f = Async::Matrix::Bridge::Discord::DB::CachedFile.new(url: "", mxc: "")
+      f.valid?.should == false
+      f.errors[:url].should.not.be.empty
+      f.errors[:mxc].should.not.be.empty
+      db.disconnect
+    end
+  end
+end

data/lib/async/matrix/bridge/discord/db/guild.rb

@@ -0,0 +1,122 @@
+# frozen_string_literal: true
+
+# Released under the Apache License, Version 2.0.
+# Copyright, 2026, by General Intelligence Systems.
+
+require "bundler/setup"
+require "async/matrix"
+
+module Async
+  module Matrix
+    module Bridge
+      module Discord
+        module DB
+          # Maps a Discord guild (server) to a Matrix Space room.
+          #
+          # Bridging modes control how aggressively the bridge creates portals:
+          #   BRIDGE_NOTHING         = 0  — never bridge
+          #   BRIDGE_IF_PORTAL_EXISTS = 1  — only bridge existing portals
+          #   BRIDGE_CREATE_ON_MESSAGE = 2 — create portals on first message
+          #   BRIDGE_EVERYTHING      = 3  — proactively create all portals
+          #
+          #   guild = Guild.create(discord_id: "999", name: "My Server", bridging_mode: 3)
+          #   guild.portals  # => [Portal, ...]
+          #
+          class Guild < Sequel::Model
+            unrestrict_primary_key
+
+            BRIDGE_NOTHING           = 0
+            BRIDGE_IF_PORTAL_EXISTS  = 1
+            BRIDGE_CREATE_ON_MESSAGE = 2
+            BRIDGE_EVERYTHING        = 3
+
+            one_to_many :portals,
+              class: "Async::Matrix::Bridge::Discord::DB::Portal",
+              key: :discord_guild_id,
+              primary_key: :discord_id
+
+            def validate
+              super
+              errors.add(:discord_id, "cannot be empty") if discord_id.nil? || discord_id.empty?
+            end
+
+            def bridge_nothing?
+              bridging_mode == BRIDGE_NOTHING
+            end
+
+            def bridge_everything?
+              bridging_mode == BRIDGE_EVERYTHING
+            end
+          end
+        end
+      end
+    end
+  end
+end
+
+test do
+  describe "Async::Matrix::Bridge::Discord::DB::Guild" do
+    def setup_db
+      db = Sequel.sqlite
+      Sequel::Migrator.run(db, File.join(__dir__, "migrations"))
+      db
+    end
+
+    it "creates and retrieves a guild" do
+      db = setup_db
+      Async::Matrix::Bridge::Discord::DB::Guild.dataset = db[:guilds]
+
+      guild = Async::Matrix::Bridge::Discord::DB::Guild.create(
+        discord_id: "999",
+        name: "Test Guild",
+        mxid: "!space:example.com"
+      )
+      guild.discord_id.should == "999"
+      guild.name.should == "Test Guild"
+      guild.mxid.should == "!space:example.com"
+      db.disconnect
+    end
+
+    it "defaults bridging_mode to 0 (nothing)" do
+      db = setup_db
+      Async::Matrix::Bridge::Discord::DB::Guild.dataset = db[:guilds]
+
+      guild = Async::Matrix::Bridge::Discord::DB::Guild.create(discord_id: "100")
+      guild.bridging_mode.should == 0
+      guild.bridge_nothing?.should == true
+      guild.bridge_everything?.should == false
+      db.disconnect
+    end
+
+    it "supports all bridging modes" do
+      db = setup_db
+      Async::Matrix::Bridge::Discord::DB::Guild.dataset = db[:guilds]
+
+      guild = Async::Matrix::Bridge::Discord::DB::Guild.create(discord_id: "200", bridging_mode: 3)
+      guild.bridge_everything?.should == true
+      guild.bridge_nothing?.should == false
+      db.disconnect
+    end
+
+    it "validates discord_id is present" do
+      db = setup_db
+      Async::Matrix::Bridge::Discord::DB::Guild.dataset = db[:guilds]
+
+      guild = Async::Matrix::Bridge::Discord::DB::Guild.new(discord_id: "")
+      guild.valid?.should == false
+      guild.errors[:discord_id].should.not.be.empty
+      db.disconnect
+    end
+
+    it "enforces unique mxid" do
+      db = setup_db
+      Async::Matrix::Bridge::Discord::DB::Guild.dataset = db[:guilds]
+
+      Async::Matrix::Bridge::Discord::DB::Guild.create(discord_id: "300", mxid: "!a:x")
+      lambda {
+        Async::Matrix::Bridge::Discord::DB::Guild.create(discord_id: "301", mxid: "!a:x")
+      }.should.raise(Sequel::UniqueConstraintViolation)
+      db.disconnect
+    end
+  end
+end

data/lib/async/matrix/bridge/discord/db/message.rb

@@ -0,0 +1,162 @@
+# frozen_string_literal: true
+
+# Released under the Apache License, Version 2.0.
+# Copyright, 2026, by General Intelligence Systems.
+
+require "bundler/setup"
+require "async/matrix"
+
+module Async
+  module Matrix
+    module Bridge
+      module Discord
+        module DB
+          # Maps a Discord message to one or more Matrix events.
+          #
+          # A single Discord message may produce multiple Matrix events (one per
+          # attachment), distinguished by discord_attachment_id. The composite
+          # unique index on (discord_id, discord_attachment_id, discord_channel_id,
+          # discord_channel_receiver) ensures no duplicates.
+          #
+          #   msg = Message.create(
+          #     discord_id: "msg1", discord_channel_id: "ch1",
+          #     discord_sender: "user1", mxid: "$evt1", timestamp: 1234567890
+          #   )
+          #   msg.portal  # => Portal
+          #
+          class Message < Sequel::Model
+            many_to_one :portal,
+              class: "Async::Matrix::Bridge::Discord::DB::Portal",
+              key: [:discord_channel_id, :discord_channel_receiver],
+              primary_key: [:discord_id, :receiver]
+
+            def validate
+              super
+              errors.add(:discord_id, "cannot be empty") if discord_id.nil? || discord_id.empty?
+              errors.add(:mxid, "cannot be empty") if mxid.nil? || mxid.empty?
+              errors.add(:discord_sender, "cannot be empty") if discord_sender.nil? || discord_sender.empty?
+            end
+
+            # Find all parts of a Discord message (text + attachments).
+            def self.by_discord_id(discord_id, channel_id, receiver = "")
+              where(
+                discord_id: discord_id,
+                discord_channel_id: channel_id,
+                discord_channel_receiver: receiver
+              ).order(:discord_attachment_id).all
+            end
+
+            # Find a message by its Matrix event ID.
+            def self.by_mxid(mxid)
+              first(mxid: mxid)
+            end
+          end
+        end
+      end
+    end
+  end
+end
+
+test do
+  describe "Async::Matrix::Bridge::Discord::DB::Message" do
+    def setup_db
+      db = Sequel.sqlite
+      Sequel::Migrator.run(db, File.join(__dir__, "migrations"))
+      db
+    end
+
+    def set_datasets(db)
+      Async::Matrix::Bridge::Discord::DB::Message.dataset = db[:messages]
+      Async::Matrix::Bridge::Discord::DB::Portal.dataset = db[:portals]
+    end
+
+    it "creates and retrieves a message" do
+      db = setup_db
+      set_datasets(db)
+
+      msg = Async::Matrix::Bridge::Discord::DB::Message.create(
+        discord_id: "msg1",
+        discord_channel_id: "ch1",
+        discord_sender: "user1",
+        mxid: "$evt1",
+        timestamp: 1234567890
+      )
+      msg.discord_id.should == "msg1"
+      msg.mxid.should == "$evt1"
+      msg.discord_attachment_id.should == ""
+      db.disconnect
+    end
+
+    it "supports multi-part messages with attachments" do
+      db = setup_db
+      set_datasets(db)
+
+      Async::Matrix::Bridge::Discord::DB::Message.create(
+        discord_id: "msg2", discord_attachment_id: "",
+        discord_channel_id: "ch1", discord_sender: "u1",
+        mxid: "$text", timestamp: 100
+      )
+      Async::Matrix::Bridge::Discord::DB::Message.create(
+        discord_id: "msg2", discord_attachment_id: "att1",
+        discord_channel_id: "ch1", discord_sender: "u1",
+        mxid: "$img1", timestamp: 100
+      )
+      Async::Matrix::Bridge::Discord::DB::Message.create(
+        discord_id: "msg2", discord_attachment_id: "att2",
+        discord_channel_id: "ch1", discord_sender: "u1",
+        mxid: "$img2", timestamp: 100
+      )
+
+      parts = Async::Matrix::Bridge::Discord::DB::Message.by_discord_id("msg2", "ch1")
+      parts.length.should == 3
+      parts.map(&:mxid).should == ["$text", "$img1", "$img2"]
+      db.disconnect
+    end
+
+    it "finds by Matrix event ID" do
+      db = setup_db
+      set_datasets(db)
+
+      Async::Matrix::Bridge::Discord::DB::Message.create(
+        discord_id: "msg3", discord_channel_id: "ch1",
+        discord_sender: "u1", mxid: "$find_me", timestamp: 200
+      )
+
+      found = Async::Matrix::Bridge::Discord::DB::Message.by_mxid("$find_me")
+      found.should.not.be.nil
+      found.discord_id.should == "msg3"
+      db.disconnect
+    end
+
+    it "enforces unique composite key" do
+      db = setup_db
+      set_datasets(db)
+
+      Async::Matrix::Bridge::Discord::DB::Message.create(
+        discord_id: "msg4", discord_attachment_id: "",
+        discord_channel_id: "ch1", discord_channel_receiver: "",
+        discord_sender: "u1", mxid: "$a", timestamp: 300
+      )
+      lambda {
+        Async::Matrix::Bridge::Discord::DB::Message.create(
+          discord_id: "msg4", discord_attachment_id: "",
+          discord_channel_id: "ch1", discord_channel_receiver: "",
+          discord_sender: "u1", mxid: "$b", timestamp: 300
+        )
+      }.should.raise(Sequel::UniqueConstraintViolation)
+      db.disconnect
+    end
+
+    it "validates required fields" do
+      db = setup_db
+      set_datasets(db)
+
+      msg = Async::Matrix::Bridge::Discord::DB::Message.new(discord_id: "", mxid: "", discord_sender: "")
+      msg.valid?.should == false
+      msg.errors[:discord_id].should.not.be.empty
+      msg.errors[:mxid].should.not.be.empty
+      msg.errors[:discord_sender].should.not.be.empty
+      db.disconnect
+    end
+  end
+end

data/lib/async/matrix/bridge/discord/db/migrations/001_create_users.rb

@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+
+Sequel.migration do
+  change do
+    create_table(:users) do
+      String :mxid, primary_key: true
+      String :discord_id, unique: true
+      String :discord_token
+      String :management_room
+      String :space_room
+      String :dm_space_room
+    end
+  end
+end

data/lib/async/matrix/bridge/discord/db/migrations/002_create_guilds.rb

@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+
+Sequel.migration do
+  change do
+    create_table(:guilds) do
+      String :discord_id, primary_key: true
+      String :mxid, unique: true
+      String :name
+      String :avatar_hash
+      String :avatar_url
+      Integer :bridging_mode, default: 0, null: false
+    end
+  end
+end

data/lib/async/matrix/bridge/discord/db/migrations/003_create_portals.rb

@@ -0,0 +1,23 @@
+# frozen_string_literal: true
+
+Sequel.migration do
+  change do
+    create_table(:portals) do
+      String :discord_id, null: false
+      String :receiver, null: false, default: ""
+      String :mxid, unique: true
+      String :discord_guild_id
+      String :name
+      String :topic
+      String :avatar_hash
+      String :avatar_url
+      TrueClass :encrypted, default: false, null: false
+      Integer :channel_type, default: 0, null: false
+      String :relay_webhook_id
+      String :relay_webhook_secret
+
+      primary_key [:discord_id, :receiver]
+      foreign_key [:discord_guild_id], :guilds, key: :discord_id
+    end
+  end
+end

data/lib/async/matrix/bridge/discord/db/migrations/004_create_puppets.rb

@@ -0,0 +1,19 @@
+# frozen_string_literal: true
+
+Sequel.migration do
+  change do
+    create_table(:puppets) do
+      String :discord_id, primary_key: true
+      String :name
+      String :avatar_hash
+      String :avatar_url
+      String :username
+      String :global_name
+      TrueClass :is_bot, default: false, null: false
+      TrueClass :is_webhook, default: false, null: false
+      TrueClass :contact_info_set, default: false, null: false
+      String :custom_mxid
+      String :access_token
+    end
+  end
+end

data/lib/async/matrix/bridge/discord/db/migrations/005_create_messages.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+Sequel.migration do
+  change do
+    create_table(:messages) do
+      primary_key :id
+      String :discord_id, null: false
+      String :discord_attachment_id, null: false, default: ""
+      String :discord_channel_id, null: false
+      String :discord_channel_receiver, null: false, default: ""
+      String :discord_sender, null: false
+      String :mxid, null: false
+      Bignum :timestamp, null: false
+      String :discord_thread_id
+
+      unique [:discord_id, :discord_attachment_id, :discord_channel_id, :discord_channel_receiver]
+      index :mxid
+    end
+  end
+end

data/lib/async/matrix/bridge/discord/db/migrations/006_create_reactions.rb

@@ -0,0 +1,19 @@
+# frozen_string_literal: true
+
+Sequel.migration do
+  change do
+    create_table(:reactions) do
+      primary_key :id
+      String :discord_message_id, null: false
+      String :discord_sender, null: false
+      String :discord_emoji_name, null: false
+      String :mxid, null: false
+      String :discord_channel_id, null: false
+      String :discord_channel_receiver, null: false, default: ""
+      String :discord_thread_id
+
+      unique [:discord_message_id, :discord_sender, :discord_emoji_name]
+      index :mxid
+    end
+  end
+end

data/lib/async/matrix/bridge/discord/db/migrations/007_create_files.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+Sequel.migration do
+  change do
+    create_table(:files) do
+      String :url, null: false
+      TrueClass :encrypted, null: false, default: false
+      String :mxc, null: false
+      String :mime_type
+      Bignum :size
+      Integer :width
+      Integer :height
+      String :decryption_info
+
+      primary_key [:url, :encrypted]
+    end
+  end
+end

data/lib/async/matrix/bridge/discord/db/portal.rb

@@ -0,0 +1,152 @@
+# frozen_string_literal: true
+
+# Released under the Apache License, Version 2.0.
+# Copyright, 2026, by General Intelligence Systems.
+
+require "bundler/setup"
+require "async/matrix"
+
+module Async
+  module Matrix
+    module Bridge
+      module Discord
+        module DB
+          # Maps a Discord channel to a Matrix room.
+          #
+          # Uses a composite primary key of (discord_id, receiver). For guild
+          # channels, receiver is an empty string. For DMs, receiver is the
+          # Discord user ID of the bridge user who owns the portal.
+          #
+          #   portal = Portal.create(discord_id: "chan123", receiver: "", mxid: "!room:x")
+          #   portal.guild      # => Guild or nil
+          #   portal.messages   # => [Message, ...]
+          #
+          class Portal < Sequel::Model
+            unrestrict_primary_key
+
+            many_to_one :guild,
+              class: "Async::Matrix::Bridge::Discord::DB::Guild",
+              key: :discord_guild_id,
+              primary_key: :discord_id
+
+            one_to_many :messages,
+              class: "Async::Matrix::Bridge::Discord::DB::Message",
+              key: [:discord_channel_id, :discord_channel_receiver],
+              primary_key: [:discord_id, :receiver]
+
+            def validate
+              super
+              errors.add(:discord_id, "cannot be empty") if discord_id.nil? || discord_id.empty?
+            end
+
+            # Is this a DM portal? (has a receiver)
+            def dm?
+              receiver && !receiver.empty?
+            end
+
+            # Is this a guild channel portal?
+            def guild_channel?
+              !dm?
+            end
+          end
+        end
+      end
+    end
+  end
+end
+
+test do
+  describe "Async::Matrix::Bridge::Discord::DB::Portal" do
+    def setup_db
+      db = Sequel.sqlite
+      Sequel::Migrator.run(db, File.join(__dir__, "migrations"))
+      db
+    end
+
+    def set_datasets(db)
+      Async::Matrix::Bridge::Discord::DB::Portal.dataset = db[:portals]
+      Async::Matrix::Bridge::Discord::DB::Guild.dataset = db[:guilds]
+      Async::Matrix::Bridge::Discord::DB::Message.dataset = db[:messages]
+    end
+
+    it "creates a guild channel portal" do
+      db = setup_db
+      set_datasets(db)
+
+      portal = Async::Matrix::Bridge::Discord::DB::Portal.create(
+        discord_id: "chan1", receiver: "", mxid: "!room:x", name: "general"
+      )
+      portal.discord_id.should == "chan1"
+      portal.receiver.should == ""
+      portal.guild_channel?.should == true
+      portal.dm?.should == false
+      db.disconnect
+    end
+
+    it "creates a DM portal with receiver" do
+      db = setup_db
+      set_datasets(db)
+
+      portal = Async::Matrix::Bridge::Discord::DB::Portal.create(
+        discord_id: "dm1", receiver: "user123", mxid: "!dm:x"
+      )
+      portal.dm?.should == true
+      portal.guild_channel?.should == false
+      db.disconnect
+    end
+
+    it "finds by composite primary key" do
+      db = setup_db
+      set_datasets(db)
+
+      Async::Matrix::Bridge::Discord::DB::Portal.create(discord_id: "c1", receiver: "")
+      Async::Matrix::Bridge::Discord::DB::Portal.create(discord_id: "c1", receiver: "u1")
+
+      found = Async::Matrix::Bridge::Discord::DB::Portal["c1", ""]
+      found.should.not.be.nil
+      found.receiver.should == ""
+
+      found2 = Async::Matrix::Bridge::Discord::DB::Portal["c1", "u1"]
+      found2.receiver.should == "u1"
+      db.disconnect
+    end
+
+    it "associates with a guild" do
+      db = setup_db
+      set_datasets(db)
+
+      guild = Async::Matrix::Bridge::Discord::DB::Guild.create(discord_id: "g_assoc", name: "Guild")
+      portal = Async::Matrix::Bridge::Discord::DB::Portal.create(
+        discord_id: "c_assoc", receiver: "", discord_guild_id: "g_assoc"
+      )
+      # Query through the db directly to avoid races with parallel tests
+      # that rebind the class-level dataset on Sequel models.
+      portal.discord_guild_id.should == "g_assoc"
+      linked_guild = db[:guilds].where(discord_id: portal.discord_guild_id).first
+      linked_guild[:discord_id].should == "g_assoc"
+      linked_portals = db[:portals].where(discord_guild_id: guild.discord_id).all
+      linked_portals.length.should == 1
+      db.disconnect
+    end
+
+    it "enforces unique mxid" do
+      db = setup_db
+      set_datasets(db)
+
+      Async::Matrix::Bridge::Discord::DB::Portal.create(discord_id: "a", receiver: "", mxid: "!r:x")
+      lambda {
+        Async::Matrix::Bridge::Discord::DB::Portal.create(discord_id: "b", receiver: "", mxid: "!r:x")
+      }.should.raise(Sequel::UniqueConstraintViolation)
+      db.disconnect
+    end
+
+    it "defaults encrypted to false" do
+      db = setup_db
+      set_datasets(db)
+
+      portal = Async::Matrix::Bridge::Discord::DB::Portal.create(discord_id: "e1", receiver: "")
+      portal.encrypted.should == false
+      db.disconnect
+    end
+  end
+end