diff --git a/scripts/backfill.js b/scripts/backfill.js index 941e803..c0c440e 100644 --- a/scripts/backfill.js +++ b/scripts/backfill.js @@ -10,6 +10,7 @@ if (!channelID) { process.exit(1) } +const assert = require("assert/strict") const sqlite = require("better-sqlite3") const backfill = new sqlite("scripts/backfill.db") backfill.prepare("CREATE TABLE IF NOT EXISTS backfill (channel_id TEXT NOT NULL, message_id INTEGER NOT NULL, PRIMARY KEY (channel_id, message_id))").run() @@ -55,29 +56,35 @@ async function event(event) { if (!channel) return const guild_id = event.d.id - try { - await createRoom.syncRoom(channelID) - let last = backfill.prepare("SELECT cast(max(message_id) as TEXT) FROM backfill WHERE channel_id = ?").pluck().get(channelID) || "0" - console.log(`OK, processing messages for #${channel.name}, continuing from ${last}`) - - while (last) { - const messages = await discord.snow.channel.getChannelMessages(channelID, {limit: 50, after: String(last)}) - messages.reverse() // More recent messages come first -> More recent messages come last - for (const message of messages) { - const simulatedGatewayDispatchData = { - guild_id, - backfill: true, - ...message - } - await eventDispatcher.MESSAGE_CREATE(discord, simulatedGatewayDispatchData) - preparedInsert.run(channelID, message.id) - } - last = messages.at(-1)?.id + let roomID = passthrough.select("channel_room", "room_id", {channel_id: channelID}).pluck().get() + if (!roomID) { + console.log(`Channel #${channel.name} is not bridged yet. Attempting to auto-create...`) + try { + roomID = await createRoom.syncRoom(channelID) + console.log(`Successfully bridged to new room: ${roomID}`) + } catch (e) { + console.error(`Failed to auto-create room: ${e.message}`) + process.exit(1) } - - process.exit() - } catch (e) { - console.error(e) - process.exit(1) // won't exit automatically on thrown error due to living discord connection, so manual exit is necessary } + + let last = backfill.prepare("SELECT cast(max(message_id) as TEXT) FROM backfill WHERE channel_id = ?").pluck().get(channelID) || "0" + console.log(`OK, processing messages for #${channel.name}, continuing from ${last}`) + + while (last) { + const messages = await discord.snow.channel.getChannelMessages(channelID, {limit: 50, after: String(last)}) + messages.reverse() // More recent messages come first -> More recent messages come last + for (const message of messages) { + const simulatedGatewayDispatchData = { + guild_id, + backfill: true, + ...message + } + await eventDispatcher.MESSAGE_CREATE(discord, simulatedGatewayDispatchData) + preparedInsert.run(channelID, message.id) + } + last = messages.at(-1)?.id + } + + process.exit() }