openclaw
Version:
Multi-channel AI gateway with extensible messaging integrations
590 lines (589 loc) • 23.9 kB
JavaScript
import { S as parseStrictInteger } from "./number-coercion-CkhMiRw8.mjs";
import { t as formatErrorMessage } from "./errors-BvOkmQM5.mjs";
import { M as takeMessageIdAfterStop, j as createFinalizableDraftStreamControlsForState } from "./channel-outbound-2iQdEUqa.mjs";
import "./error-runtime-oJAVOxQN.mjs";
import { n as isSingleUseReplyToMode } from "./reply-reference-CLjC4hNl.mjs";
import { b as isTelegramRateLimitError, d as isRetryableTelegramApiError, f as isSafeToRetrySendError, h as isTelegramClientRejection, l as TelegramRequestNotStartedError, v as isTelegramMessageNotModifiedError } from "./fetch-Dru0QTTk.mjs";
import "./reply-reference-D95Ecb-U.mjs";
import { N as telegramHtmlToPlainTextFallback, P as escapeTelegramHtml, m as warnTelegramRichBlocksDegradations, o as removeTelegramRichNativeQuoteParam, r as TELEGRAM_RICH_TEXT_LIMIT, t as TELEGRAM_TEXT_CHUNK_LIMIT } from "./text-chunk-limit-plWH6e89.mjs";
import { c as buildTelegramThreadParams } from "./helpers-ODclOX0q.mjs";
import "./outbound-adapter-BR_PxB2g.mjs";
import { E as planTelegramTextDeliveryPages, T as deliverTelegramTextPage, _t as runAuthorizedTelegramRequest, ft as buildTelegramThreadReplyParams, pt as withTelegramNativeQuoteFallback, vt as runReplaceableTelegramRequest } from "./send-CpiI0BKu.mjs";
//#region extensions/telegram/src/draft-stream-message.ts
function toDraftSnapshot(page) {
return {
text: page.plainText,
sourceText: page.sourceText,
sourceTextMode: page.sourceTextMode
};
}
async function deliverDraftPage(params) {
const { page } = params;
if (page.richMessage) warnTelegramRichBlocksDegradations({
context: params.context,
reasons: page.degradationReasons ?? [],
warn: (message) => params.warn?.(message)
});
const send = async () => {
const [delivery] = await deliverTelegramTextPage({
page: {
...page,
degradationReasons: void 0
},
html: page.sourceTextMode === "html",
context: params.context,
warn: (message) => params.warn?.(message),
fallbackLimit: Number.MAX_SAFE_INTEGER,
sender: {
sendPlain: (text) => params.send({ text }),
sendHtml: (text) => params.send({
text,
parseMode: "HTML"
}),
sendRich: (richMessage) => params.send({
text: page.plainText,
richMessage
})
}
});
return {
message: delivery.result,
snapshot: toDraftSnapshot(delivery.page)
};
};
return await (params.request ? params.request(send) : send());
}
async function editTelegramDraftMessage(params) {
return (await deliverDraftPage({
page: params.page,
context: "stream preview edit",
warn: params.warn,
request: params.request,
send: ({ text, parseMode, richMessage }) => {
if (richMessage) return params.api.raw.editMessageText({
chat_id: params.chatId,
message_id: params.messageId,
rich_message: richMessage
});
const other = parseMode ? {
parse_mode: parseMode,
...params.linkPreviewParams
} : params.linkPreviewParams;
return Object.keys(other).length > 0 ? params.api.editMessageText(params.chatId, params.messageId, text, other) : params.api.editMessageText(params.chatId, params.messageId, text);
}
})).snapshot;
}
async function sendTelegramDraftMessage(params) {
const { chatId, linkPreviewParams, page, assertCurrentSend } = params;
const sendPlannedMessage = (sendMessageParams) => deliverDraftPage({
page,
context: "stream preview",
warn: params.warn,
send: ({ text, parseMode, richMessage }) => {
assertCurrentSend();
return richMessage ? params.api.raw.sendRichMessage({
chat_id: chatId,
rich_message: richMessage,
...sendMessageParams
}) : params.api.sendMessage(chatId, text, {
...parseMode ? { parse_mode: parseMode } : {},
...sendMessageParams,
...linkPreviewParams
});
}
});
return (await withTelegramNativeQuoteFallback({
label: "stream-preview",
requestParams: params.sendMessageParams,
...params.page.richMessage ? { removeNativeQuoteParam: removeTelegramRichNativeQuoteParam } : {},
request: sendPlannedMessage
})).result;
}
//#endregion
//#region extensions/telegram/src/draft-stream.ts
const DEFAULT_THROTTLE_MS = 1e3;
const MAX_CONSECUTIVE_PREVIEW_FAILURES = 3;
const MIN_PREVIEW_DWELL_MS = 4e3;
function createTelegramDraftStream(params) {
const richMessages = params.richMessages === true;
const transportLimit = richMessages ? TELEGRAM_RICH_TEXT_LIMIT : TELEGRAM_TEXT_CHUNK_LIMIT;
const maxChars = Math.min(params.maxChars ?? transportLimit, transportLimit);
const throttleMs = Math.max(250, params.throttleMs ?? DEFAULT_THROTTLE_MS);
const minInitialChars = params.minInitialChars;
const chatId = params.chatId;
const threadParams = buildTelegramThreadParams(params.thread);
const replyToMessageId = parseStrictInteger(params.replyToMessageId);
const initialSendMessageParams = {
...threadParams,
...buildTelegramThreadReplyParams({
replyToMessageId,
replyQuoteMessageId: replyToMessageId,
replyQuoteText: params.replyQuote?.text,
replyQuotePosition: params.replyQuote?.position,
replyQuoteEntities: params.replyQuote?.entities,
nativeReply: true
})
};
const consumesReplyTarget = replyToMessageId != null && params.replyToMode !== void 0 && isSingleUseReplyToMode(params.replyToMode);
let replyTargetState = { kind: "available" };
const reserveReplyTargetForSend = (sendGeneration) => {
if (!consumesReplyTarget) return initialSendMessageParams;
if (replyTargetState.kind !== "available") return threadParams ?? {};
replyTargetState = {
kind: "pending",
generation: sendGeneration
};
return initialSendMessageParams;
};
const settlePendingReplyTarget = (sendGeneration, messageId) => {
if (replyTargetState.kind === "pending" && replyTargetState.generation === sendGeneration) replyTargetState = messageId === void 0 ? { kind: "available" } : {
kind: "retained",
messageId
};
};
const streamState = {
stopped: false,
final: false
};
let messageSendAttempted = false;
let consecutivePreviewFailures = 0;
let streamMessageId;
let streamMessageSnapshot;
let streamProviderMessage;
let terminalDeliveryError;
const pendingProviderObservations = /* @__PURE__ */ new Set();
let streamVisibleSinceMs;
let lastSentPreviewKey = "";
let lastDeliveredText = "";
let lastRequestedText = "";
let lastRequestedPreview;
let pendingPlatformSendDispatch;
let pendingPlatformSendAuthorization;
let requestedUpdates = 0;
let generation = 0;
let finalPagePlan;
const repositionedSendGenerations = /* @__PURE__ */ new Set();
let supersededPreviews = [];
const previewRequest = (assertCurrent, run) => runAuthorizedTelegramRequest(assertCurrent, () => streamState.final ? run() : runReplaceableTelegramRequest(run));
const scheduleProviderMessageObservation = (message) => {
if (!message) return;
const observation = (async () => {
try {
await params.onProviderMessage?.(message);
} catch (err) {
try {
params.warn?.(`telegram stream preview observation failed: ${formatErrorMessage(err)}`);
} catch {}
}
})();
pendingProviderObservations.add(observation);
observation.then(() => {
pendingProviderObservations.delete(observation);
});
};
const observeCurrentProviderMessage = () => {
const message = streamProviderMessage;
streamProviderMessage = void 0;
scheduleProviderMessageObservation(message);
};
const sendMessageTransportPreview = async (page, sendGeneration, disableLinkPreview) => {
const linkPreviewParams = disableLinkPreview ? { link_preview_options: { is_disabled: true } } : {};
if (pendingPlatformSendDispatch) {
await pendingPlatformSendDispatch();
pendingPlatformSendDispatch = void 0;
}
const assertPlatformSendAuthorized = pendingPlatformSendAuthorization;
assertPlatformSendAuthorized?.();
const targetMessageId = streamMessageId;
if (typeof targetMessageId === "number") {
const acceptedSnapshot = await editTelegramDraftMessage({
api: params.api,
chatId,
messageId: targetMessageId,
page,
linkPreviewParams,
warn: params.warn,
request: (send) => previewRequest(assertPlatformSendAuthorized, send)
});
if (sendGeneration === generation && streamMessageId === targetMessageId) streamMessageSnapshot = acceptedSnapshot;
return true;
}
messageSendAttempted = true;
const sendMessageParams = reserveReplyTargetForSend(sendGeneration);
let sent;
try {
sent = await previewRequest(assertPlatformSendAuthorized, () => sendTelegramDraftMessage({
api: params.api,
chatId,
page,
sendMessageParams,
linkPreviewParams,
warn: params.warn,
assertCurrentSend: () => {
if (sendGeneration !== generation || streamState.stopped) throw new TelegramRequestNotStartedError("Telegram preview generation retired");
assertPlatformSendAuthorized?.();
}
}));
} catch (err) {
const definitelyRejected = isSafeToRetrySendError(err) || isTelegramClientRejection(err);
if (sendGeneration === generation && definitelyRejected) messageSendAttempted = false;
if (definitelyRejected) settlePendingReplyTarget(sendGeneration);
throw err;
}
const sentMessageId = sent.message?.message_id;
const normalizedMessageId = typeof sentMessageId === "number" && Number.isFinite(sentMessageId) ? Math.trunc(sentMessageId) : void 0;
if (normalizedMessageId === void 0) {
if (sendGeneration === generation) {
streamState.stopped = true;
params.warn?.("telegram stream preview stopped (missing message id from sendMessage)");
return false;
}
return true;
}
settlePendingReplyTarget(sendGeneration, normalizedMessageId);
const adoptSentMessage = () => {
streamMessageId = normalizedMessageId;
streamMessageSnapshot = sent.snapshot;
streamProviderMessage = sent.message;
streamVisibleSinceMs = Date.now();
};
try {
if (params.validateProviderMessage) await params.validateProviderMessage(sent.message);
} catch (error) {
terminalDeliveryError ??= error instanceof Error ? error : new Error(formatErrorMessage(error));
streamState.stopped = true;
if (sendGeneration === generation) adoptSentMessage();
else if (repositionedSendGenerations.delete(sendGeneration)) retireWhenReplaced(normalizedMessageId, Date.now());
return false;
}
if (sendGeneration !== generation) {
const visibleSinceMs = Date.now();
if (repositionedSendGenerations.delete(sendGeneration)) {
retireWhenReplaced(normalizedMessageId, visibleSinceMs);
return true;
}
params.onRetainedPage?.({
messageId: normalizedMessageId,
textSnapshot: sent.snapshot.text,
visibleSinceMs
});
scheduleProviderMessageObservation(sent.message);
return true;
}
adoptSentMessage();
retireSupersededPreviews();
return true;
};
const sendOrEditPlannedPage = async (page, complete = false, disableLinkPreview = params.linkPreview === false) => {
const renderedPreviewKey = JSON.stringify([
page.sourceTextMode,
page.sourceText,
page.richMessage?.skip_entity_detection === true,
disableLinkPreview
]);
if (renderedPreviewKey === lastSentPreviewKey) return true;
const sendGeneration = generation;
if (typeof streamMessageId !== "number" && minInitialChars != null && !streamState.final && !complete && page.plainText.length < minInitialChars) return false;
const previousSentPreviewKey = lastSentPreviewKey;
lastSentPreviewKey = renderedPreviewKey;
const updateAtSend = requestedUpdates;
const settleAuthorization = () => {
if (requestedUpdates === updateAtSend) pendingPlatformSendAuthorization = void 0;
};
try {
const sent = await sendMessageTransportPreview(page, sendGeneration, disableLinkPreview);
if (sent) settleAuthorization();
if (sendGeneration !== generation) return true;
if (sent) consecutivePreviewFailures = 0;
return sent;
} catch (err) {
if (sendGeneration !== generation) return true;
const isEdit = typeof streamMessageId === "number";
if (isEdit && isTelegramMessageNotModifiedError(err)) {
settleAuthorization();
consecutivePreviewFailures = 0;
streamMessageSnapshot = toDraftSnapshot(page);
return true;
}
lastSentPreviewKey = previousSentPreviewKey;
if (!streamState.final && isTelegramRateLimitError(err)) return false;
const retryable = isTelegramRateLimitError(err) || (isEdit ? isRetryableTelegramApiError(err) : isSafeToRetrySendError(err));
consecutivePreviewFailures += 1;
if (retryable && consecutivePreviewFailures <= MAX_CONSECUTIVE_PREVIEW_FAILURES) {
params.warn?.(`telegram stream preview ${isEdit ? "edit" : "send"} failed (retrying): ${formatErrorMessage(err)}`);
return false;
}
streamState.stopped = true;
params.warn?.(`telegram stream preview failed: ${formatErrorMessage(err)}`);
return false;
}
};
const retainCurrentPage = () => {
if (typeof streamMessageId !== "number" || !streamMessageSnapshot?.text) return;
params.onRetainedPage?.({
messageId: streamMessageId,
textSnapshot: streamMessageSnapshot.text,
visibleSinceMs: streamVisibleSinceMs
});
observeCurrentProviderMessage();
};
const resolveExactRemainingPage = (plan) => {
if (plan.nextPageIndex <= 0 || plan.nextPageIndex >= plan.pages.length) return;
const fullSourceText = plan.pages[0]?.fullSourceText;
if (fullSourceText === void 0) return;
const acceptedSourceText = plan.pages.slice(0, plan.nextPageIndex).map((page) => page.sourceText).join("");
if (!fullSourceText?.startsWith(acceptedSourceText)) return;
const sourceText = fullSourceText.slice(acceptedSourceText.length);
const plainText = telegramHtmlToPlainTextFallback(sourceText);
return plainText.length <= maxChars ? {
plainText,
sourceText,
sourceTextMode: "html",
fullSourceText,
htmlText: sourceText
} : void 0;
};
const sendOrEditStreamMessage = async (update) => {
const isLazy = typeof update !== "string";
const text = isLazy ? update.resolveText() : update;
if (text === void 0) return true;
if (isLazy) {
lastRequestedPreview = void 0;
lastRequestedText = text;
}
if (streamState.stopped && !streamState.final) return false;
const trimmed = text.trimEnd();
if (!trimmed) return false;
const fullPreview = lastRequestedPreview?.text === trimmed ? lastRequestedPreview : params.renderText?.(trimmed) ?? { text: trimmed };
const pages = streamState.final && finalPagePlan ? finalPagePlan.pages : planTelegramTextDeliveryPages({
text: fullPreview.markdownSource?.text ?? fullPreview.text,
maxChars,
richMessages,
richMessage: fullPreview.richMessage,
tableMode: fullPreview.markdownSource?.tableMode,
...richMessages || fullPreview.markdownSource ? {} : { textMode: fullPreview.parseMode === "HTML" ? "html" : "plain" }
});
const firstPage = pages[0];
if (!firstPage) return false;
const disableLinkPreview = fullPreview.linkPreview === false || params.linkPreview === false;
if (!streamState.final) {
finalPagePlan = void 0;
const updateGeneration = generation;
const sent = await sendOrEditPlannedPage(firstPage, fullPreview.complete, disableLinkPreview);
if (updateGeneration !== generation) return true;
if (sent) lastDeliveredText = pages.length === 1 ? trimmed : firstPage.plainText.trimEnd();
return sent;
}
const activePlan = finalPagePlan ??= {
pages,
nextPageIndex: 0
};
for (let index = activePlan.nextPageIndex; index < pages.length; index += 1) {
const exactRemainingPage = resolveExactRemainingPage(activePlan);
const page = exactRemainingPage ?? pages[index];
if (index > 0 && typeof streamMessageId === "number") {
retainCurrentPage();
resetStreamToNewMessage("page");
}
if (!await sendOrEditPlannedPage(page, false, disableLinkPreview)) return false;
if (finalPagePlan !== activePlan) return true;
activePlan.nextPageIndex = exactRemainingPage ? pages.length : index + 1;
if (exactRemainingPage) break;
}
finalPagePlan = void 0;
lastDeliveredText = trimmed;
return true;
};
const { loop, update: updateDraft, stopForClear } = createFinalizableDraftStreamControlsForState({
throttleMs,
state: streamState,
sendOrEditStreamMessage
});
const throwTerminalDeliveryError = () => {
if (terminalDeliveryError !== void 0) throw terminalDeliveryError;
};
const waitForInFlight = async () => {
await loop.waitForInFlight();
throwTerminalDeliveryError();
};
const flush = async () => {
await waitForInFlight();
if (!streamState.stopped) await loop.flush();
throwTerminalDeliveryError();
};
const requestDraftUpdate = (update, preview, onPlatformSendDispatch, assertPlatformSendAuthorized) => {
if (streamState.stopped || streamState.final) return;
if (typeof update === "string") {
lastRequestedPreview = preview;
lastRequestedText = update;
pendingPlatformSendDispatch = onPlatformSendDispatch;
}
pendingPlatformSendAuthorization = assertPlatformSendAuthorized;
requestedUpdates += 1;
updateDraft(update);
};
const stop = async () => {
const stopGeneration = generation;
streamState.final = true;
loop.resetThrottleWindow();
await loop.waitForInFlight();
throwTerminalDeliveryError();
if (generation !== stopGeneration || streamState.stopped) return;
await flush();
if (generation !== stopGeneration || streamState.stopped) return;
const finalText = lastRequestedText.trimEnd();
if (finalText && finalText !== lastDeliveredText.trimEnd()) for (let attempt = 0; attempt < 2; attempt += 1) {
const sent = await sendOrEditStreamMessage(finalText);
throwTerminalDeliveryError();
if (generation !== stopGeneration) return;
if (sent) {
loop.resetPending();
break;
}
if (!finalPagePlan || streamState.stopped) break;
}
streamState.final = true;
observeCurrentProviderMessage();
await Promise.all(pendingProviderObservations);
pendingPlatformSendDispatch = void 0;
pendingPlatformSendAuthorization = void 0;
};
/** Prepared final content not yet accepted after retained pagination pages. */
const remainingFinalContent = () => {
const plan = finalPagePlan;
if (!plan || plan.nextPageIndex <= 0 || plan.nextPageIndex >= plan.pages.length) return;
const pages = plan.pages.slice(plan.nextPageIndex);
const exactRemainingPage = resolveExactRemainingPage(plan);
const sourceText = exactRemainingPage?.sourceText || pages.map((page) => page.sourceTextMode === "html" ? page.sourceText : escapeTelegramHtml(page.plainText)).join("");
return {
text: exactRemainingPage?.plainText ?? pages.map((page) => page.plainText).join(""),
sourceText,
sourceTextMode: "html"
};
};
const resetStreamToNewMessage = (mode) => {
const continueFinalPagination = mode === "page";
if (mode === "retain") observeCurrentProviderMessage();
streamState.stopped = false;
streamState.final = continueFinalPagination;
if (!continueFinalPagination) generation += 1;
messageSendAttempted = false;
streamMessageId = void 0;
streamMessageSnapshot = void 0;
streamProviderMessage = void 0;
streamVisibleSinceMs = void 0;
lastSentPreviewKey = "";
if (!continueFinalPagination) {
finalPagePlan = void 0;
lastRequestedText = "";
lastDeliveredText = "";
loop.resetPending();
lastRequestedPreview = void 0;
pendingPlatformSendAuthorization = void 0;
requestedUpdates += 1;
}
loop.resetThrottleWindow();
};
const scheduleDetachedDelete = (messageId, visibleSince, minDelayMs = 0) => {
const runDelete = async () => {
try {
if (!await params.api.deleteMessage(chatId, messageId)) {
params.warn?.(`telegram stream preview cleanup was not confirmed (chat=${chatId}, message=${messageId})`);
return;
}
if (replyTargetState.kind === "retained" && replyTargetState.messageId === messageId) replyTargetState = { kind: "available" };
params.log?.(`telegram stream preview deleted (chat=${chatId}, message=${messageId})`);
} catch (err) {
params.warn?.(`telegram stream preview cleanup failed: ${formatErrorMessage(err)}`);
}
};
const elapsedMs = typeof visibleSince === "number" ? Date.now() - visibleSince : MIN_PREVIEW_DWELL_MS;
const remainingDwellMs = Math.max(0, MIN_PREVIEW_DWELL_MS - elapsedMs);
const delayMs = Math.max(remainingDwellMs, minDelayMs);
if (delayMs <= 0) runDelete();
else setTimeout(() => {
runDelete();
}, delayMs);
};
const retireSupersededPreviews = () => {
const previews = supersededPreviews;
supersededPreviews = [];
for (const { messageId, visibleSinceMs } of previews) scheduleDetachedDelete(messageId, visibleSinceMs, REPOSITION_DELETE_DELAY_MS);
};
const retireWhenReplaced = (messageId, visibleSinceMs) => {
supersededPreviews.push({
messageId,
visibleSinceMs
});
if (typeof streamMessageId === "number") retireSupersededPreviews();
};
const clear = async () => {
retireSupersededPreviews();
const visibleSince = streamVisibleSinceMs;
const messageId = await takeMessageIdAfterStop({
stopForClear,
readMessageId: () => streamMessageId,
clearMessageId: () => {
streamMessageId = void 0;
streamMessageSnapshot = void 0;
streamProviderMessage = void 0;
}
});
retireSupersededPreviews();
if (typeof messageId === "number" && Number.isFinite(messageId)) scheduleDetachedDelete(messageId, visibleSince);
await Promise.all(pendingProviderObservations);
};
const REPOSITION_DELETE_DELAY_MS = 1500;
/** Rewind and delete the superseded preview only after its replacement lands. */
const rotateToNewMessageDeferringDelete = () => {
const supersededMessageId = streamMessageId;
const supersededVisibleSince = streamVisibleSinceMs;
if (messageSendAttempted && streamMessageId === void 0) repositionedSendGenerations.add(generation);
resetStreamToNewMessage("replace");
if (typeof supersededMessageId === "number" && Number.isFinite(supersededMessageId)) retireWhenReplaced(supersededMessageId, supersededVisibleSince);
};
params.log?.(`telegram stream preview ready (maxChars=${maxChars}, throttleMs=${throttleMs})`);
return {
update: (text, options) => requestDraftUpdate(text, void 0, options?.onPlatformSendDispatch, options?.assertPlatformSendAuthorized),
updateLazy: (resolveText) => requestDraftUpdate({ resolveText }),
updatePreview: (preview, assertPlatformSendAuthorized) => {
const text = preview.text.trimEnd();
if (!text) return;
requestDraftUpdate(text, {
...preview,
text
}, void 0, assertPlatformSendAuthorized);
},
flush,
waitForInFlight,
messageId: () => streamMessageId,
lastDeliveredText: () => lastDeliveredText,
/** A failed preview stops editing for good; it can no longer carry live progress. */
isStopped: () => streamState.stopped,
currentMessageSnapshot: () => {
const ownsReplyTarget = !consumesReplyTarget || replyTargetState.kind === "retained" && replyTargetState.messageId === streamMessageId;
return streamMessageSnapshot && ownsReplyTarget && replyToMessageId !== void 0 ? {
...streamMessageSnapshot,
replyToMessageId
} : streamMessageSnapshot;
},
clear,
stop,
/** Stop without a final flush or delete. */
discard: async () => {
await stopForClear();
observeCurrentProviderMessage();
await Promise.all(pendingProviderObservations);
},
remainingFinalContent,
/** True while a pending or visible draft owns a first/batched reply target. */
hasConsumedReplyTarget: () => replyTargetState.kind !== "available",
/** Reset internal state so the next update creates a new message instead of editing. */
forceNewMessage: () => resetStreamToNewMessage("retain"),
rotateToNewMessageDeferringDelete,
/** True when a preview sendMessage was attempted but the response was lost. */
sendMayHaveLanded: () => messageSendAttempted && typeof streamMessageId !== "number"
};
}
//#endregion
export { createTelegramDraftStream as t };