mirror of
https://github.com/xCyanGrizzly/DragonsStash.git
synced 2026-05-10 22:01:16 +00:00
9adbdb2a77
- Add invokeWithTimeout wrapper for TDLib API calls (2min timeout per call) - Add stuck detection to getChannelMessages: break if from_message_id doesn't advance - Add stuck detection to getTopicMessages: same protection for topic scanning - Add stuck detection to getForumTopicList: break if pagination offsets don't advance - Add max page limit (5000) to all scanning loops to prevent infinite pagination - Add mutex wait timeout (30min) to prevent indefinite blocking when holder hangs - Add cycle timeout (4h default, configurable via WORKER_CYCLE_TIMEOUT_MINUTES) - Fix end-of-page detection to use actual limit value instead of hardcoded 100 Co-authored-by: xCyanGrizzly <53275238+xCyanGrizzly@users.noreply.github.com>
96 lines
3.7 KiB
JavaScript
96 lines
3.7 KiB
JavaScript
import tdl, { createClient } from "tdl";
|
|
import { getTdjson } from "prebuilt-tdlib";
|
|
import path from "path";
|
|
import { config } from "../util/config.js";
|
|
import { childLogger } from "../util/logger.js";
|
|
import { updateAccountAuthState, getAccountAuthCode, } from "../db/queries.js";
|
|
const log = childLogger("tdlib-client");
|
|
// Configure tdl to use the prebuilt tdjson shared library
|
|
tdl.configure({ tdjson: getTdjson() });
|
|
/**
|
|
* Create and authenticate a TDLib client for a Telegram account.
|
|
* Authentication flow communicates with the admin UI via the database:
|
|
* - Worker sets authState to AWAITING_CODE when TDLib asks for phone code
|
|
* - Admin enters the code via UI, which writes it to authCode field
|
|
* - Worker polls DB for the code and feeds it to TDLib
|
|
*/
|
|
export async function createTdlibClient(account) {
|
|
const dbPath = path.join(config.tdlibStateDir, account.id);
|
|
const client = createClient({
|
|
apiId: config.telegramApiId,
|
|
apiHash: config.telegramApiHash,
|
|
databaseDirectory: dbPath,
|
|
filesDirectory: path.join(dbPath, "files"),
|
|
});
|
|
client.on("error", (err) => {
|
|
log.error({ err, accountId: account.id }, "TDLib client error");
|
|
});
|
|
try {
|
|
await client.login(() => ({
|
|
getPhoneNumber: async () => {
|
|
log.info({ accountId: account.id }, "TDLib requesting phone number");
|
|
return account.phone;
|
|
},
|
|
getAuthCode: async () => {
|
|
log.info({ accountId: account.id }, "TDLib requesting auth code");
|
|
await updateAccountAuthState(account.id, "AWAITING_CODE");
|
|
// Poll database for the code entered via admin UI
|
|
const code = await pollForAuthCode(account.id);
|
|
if (!code) {
|
|
throw new Error("Auth code not provided within timeout");
|
|
}
|
|
// Clear the code after reading
|
|
await updateAccountAuthState(account.id, "AUTHENTICATED", null);
|
|
return code;
|
|
},
|
|
getPassword: async () => {
|
|
log.info({ accountId: account.id }, "TDLib requesting 2FA password");
|
|
await updateAccountAuthState(account.id, "AWAITING_PASSWORD");
|
|
// Poll database for the password entered via admin UI
|
|
const code = await pollForAuthCode(account.id);
|
|
if (!code) {
|
|
throw new Error("2FA password not provided within timeout");
|
|
}
|
|
await updateAccountAuthState(account.id, "AUTHENTICATED", null);
|
|
return code;
|
|
},
|
|
}));
|
|
await updateAccountAuthState(account.id, "AUTHENTICATED");
|
|
log.info({ accountId: account.id }, "TDLib client authenticated");
|
|
return client;
|
|
}
|
|
catch (err) {
|
|
log.error({ err, accountId: account.id }, "TDLib authentication failed");
|
|
await updateAccountAuthState(account.id, "EXPIRED");
|
|
throw err;
|
|
}
|
|
}
|
|
/**
|
|
* Poll the database every 5 seconds for an auth code, up to 5 minutes.
|
|
*/
|
|
async function pollForAuthCode(accountId, timeoutMs = 300_000) {
|
|
const start = Date.now();
|
|
while (Date.now() - start < timeoutMs) {
|
|
const result = await getAccountAuthCode(accountId);
|
|
if (result?.authCode) {
|
|
return result.authCode;
|
|
}
|
|
await sleep(5000);
|
|
}
|
|
return null;
|
|
}
|
|
function sleep(ms) {
|
|
return new Promise((resolve) => setTimeout(resolve, ms));
|
|
}
|
|
/**
|
|
* Close a TDLib client gracefully.
|
|
*/
|
|
export async function closeTdlibClient(client) {
|
|
try {
|
|
await client.close();
|
|
}
|
|
catch (err) {
|
|
log.warn({ err }, "Error closing TDLib client");
|
|
}
|
|
}
|
|
//# sourceMappingURL=client.js.map
|