Telegram: harden network retries and config

Co-authored-by: techboss <techboss@users.noreply.github.com>

src/infra/retry-policy.test.ts

@@ -0,0 +1,27 @@
+import { afterEach, describe, expect, it, vi } from "vitest";
+
+import { createTelegramRetryRunner } from "./retry-policy.js";
+
+describe("createTelegramRetryRunner", () => {
+  afterEach(() => {
+    vi.useRealTimers();
+  });
+
+  it("retries when custom shouldRetry matches non-telegram error", async () => {
+    vi.useFakeTimers();
+    const runner = createTelegramRetryRunner({
+      retry: { attempts: 2, minDelayMs: 0, maxDelayMs: 0, jitter: 0 },
+      shouldRetry: (err) => err instanceof Error && err.message === "boom",
+    });
+    const fn = vi
+      .fn<[], Promise<string>>()
+      .mockRejectedValueOnce(new Error("boom"))
+      .mockResolvedValue("ok");
+
+    const promise = runner(fn, "request");
+    await vi.runAllTimersAsync();
+
+    await expect(promise).resolves.toBe("ok");
+    expect(fn).toHaveBeenCalledTimes(2);
+  });
+});

src/infra/retry-policy.ts

@@ -72,16 +72,21 @@ export function createTelegramRetryRunner(params: {
   retry?: RetryConfig;
   configRetry?: RetryConfig;
   verbose?: boolean;
+  shouldRetry?: (err: unknown) => boolean;
 }): RetryRunner {
   const retryConfig = resolveRetryConfig(TELEGRAM_RETRY_DEFAULTS, {
     ...params.configRetry,
     ...params.retry,
   });
+  const shouldRetry = params.shouldRetry
+    ? (err: unknown) => params.shouldRetry?.(err) || TELEGRAM_RETRY_RE.test(formatErrorMessage(err))
+    : (err: unknown) => TELEGRAM_RETRY_RE.test(formatErrorMessage(err));
+
   return <T>(fn: () => Promise<T>, label?: string) =>
     retryAsync(fn, {
       ...retryConfig,
       label,
-      shouldRetry: (err) => TELEGRAM_RETRY_RE.test(formatErrorMessage(err)),
+      shouldRetry,
       retryAfterMs: getTelegramRetryAfterMs,
       onRetry: params.verbose
         ? (info) => {