diff --git a/d2m/actions/create-room.js b/d2m/actions/create-room.js index a207586..671940f 100644 --- a/d2m/actions/create-room.js +++ b/d2m/actions/create-room.js @@ -68,7 +68,7 @@ function convertNameAndTopic(channel, guild, customName) { * @param {DiscordTypes.APIGuild} guild */ async function channelToKState(channel, guild) { - const spaceID = await createSpace.ensureSpace(guild.id) + const spaceID = await createSpace.ensureSpace(guild) assert.ok(typeof spaceID === "string") const row = select("channel_room", ["nick", "custom_avatar"], "WHERE channel_id = ?").get(channel.id) diff --git a/d2m/actions/create-space.js b/d2m/actions/create-space.js index 96b074e..f90c1c3 100644 --- a/d2m/actions/create-space.js +++ b/d2m/actions/create-space.js @@ -74,30 +74,28 @@ async function guildToKState(guild) { } /** - * @param {string} guildID + * @param {DiscordTypes.APIGuild} guild * @param {boolean} shouldActuallySync false if just need to ensure nspace exists (which is a quick database check), * true if also want to efficiently sync space name, space avatar, and child room avatars * @returns {Promise} room ID */ -async function _syncSpace(guildID, shouldActuallySync) { - /** @ts-ignore @type {DiscordTypes.APIGuild} */ - const guild = discord.guilds.get(guildID) +async function _syncSpace(guild, shouldActuallySync) { assert.ok(guild) - if (inflightSpaceCreate.has(guildID)) { - await inflightSpaceCreate.get(guildID) // just waiting, and then doing a new db query afterwards, is the simplest way of doing it + if (inflightSpaceCreate.has(guild.id)) { + await inflightSpaceCreate.get(guild.id) // just waiting, and then doing a new db query afterwards, is the simplest way of doing it } - const spaceID = select("guild_space", "space_id", "WHERE guild_id = ?").pluck().get(guildID) + const spaceID = select("guild_space", "space_id", "WHERE guild_id = ?").pluck().get(guild.id) if (!spaceID) { const creation = (async () => { const guildKState = await guildToKState(guild) const spaceID = await createSpace(guild, guildKState) - inflightSpaceCreate.delete(guildID) + inflightSpaceCreate.delete(guild.id) return spaceID })() - inflightSpaceCreate.set(guildID, creation) + inflightSpaceCreate.set(guild.id, creation) return creation // Naturally, the newly created space is already up to date, so we can always skip syncing here. } @@ -136,14 +134,20 @@ async function _syncSpace(guildID, shouldActuallySync) { return spaceID } -/** Ensures the space exists. If it doesn't, creates the space with an accurate initial state. */ -function ensureSpace(guildID) { - return _syncSpace(guildID, false) +/** + * Ensures the space exists. If it doesn't, creates the space with an accurate initial state. + * @param {DiscordTypes.APIGuild} guild + */ +function ensureSpace(guild) { + return _syncSpace(guild, false) } -/** Actually syncs. Efficiently updates the space name, space avatar, and child room avatars. */ -function syncSpace(guildID) { - return _syncSpace(guildID, true) +/** + * Actually syncs. Efficiently updates the space name, space avatar, and child room avatars. + * @param {DiscordTypes.APIGuild} guild + */ +function syncSpace(guild) { + return _syncSpace(guild, true) } /** diff --git a/d2m/event-dispatcher.js b/d2m/event-dispatcher.js index 82c6adb..32b8f49 100644 --- a/d2m/event-dispatcher.js +++ b/d2m/event-dispatcher.js @@ -140,7 +140,7 @@ module.exports = { async onGuildUpdate(client, guild) { const spaceID = select("guild_space", "space_id", "WHERE guild_id = ?").pluck().get(guild.id) if (!spaceID) return - await createSpace.syncSpace(guild.id) + await createSpace.syncSpace(guild) }, /** diff --git a/scripts/migrate-from-old-bridge.js b/scripts/migrate-from-old-bridge.js index cd32729..84daad7 100644 --- a/scripts/migrate-from-old-bridge.js +++ b/scripts/migrate-from-old-bridge.js @@ -66,7 +66,7 @@ async function migrateGuild(guild) { console.log(`START MIGRATION of ${guild.name} (${guild.id})`) // Step 1: Create a new space for the guild (createSpace) - const spaceID = await createSpace.syncSpace(guild.id) + const spaceID = await createSpace.syncSpace(guild) let oldRooms = oldDB.prepare("SELECT matrix_id, discord_guild, discord_channel FROM room_entries INNER JOIN remote_room_data ON remote_id = room_id WHERE discord_guild = ?").all(guild.id) const migrated = db.prepare("SELECT discord_channel FROM migration WHERE migrated = 1").pluck().all() @@ -132,6 +132,6 @@ async function migrateGuild(guild) { } // Step 5: Call syncSpace to make sure everything is up to date - await createSpace.syncSpace(guild.id) + await createSpace.syncSpace(guild) console.log(`Finished migrating ${guild.name} to Out Of Your Element`) }