refactor(web): split provider module

2025-11-26 01:16:54 +01:00
parent e5f677803f
commit 4dd2f3b7f7
10 changed files with 1079 additions and 940 deletions
--- a/src/web/login.ts
+++ b/src/web/login.ts
@@ -0,0 +1,80 @@
+import fs from "node:fs/promises";
+
+import { DisconnectReason } from "@whiskeysockets/baileys";
+
+import { danger, info, success } from "../globals.js";
+import { logInfo } from "../logger.js";
+import { defaultRuntime, type RuntimeEnv } from "../runtime.js";
+import {
+  createWaSocket,
+  formatError,
+  WA_WEB_AUTH_DIR,
+  waitForWaConnection,
+} from "./session.js";
+
+export async function loginWeb(
+  verbose: boolean,
+  waitForConnection: typeof waitForWaConnection = waitForWaConnection,
+  runtime: RuntimeEnv = defaultRuntime,
+) {
+  const sock = await createWaSocket(true, verbose);
+  logInfo("Waiting for WhatsApp connection...", runtime);
+  try {
+    await waitForConnection(sock);
+    console.log(success("✅ Linked! Credentials saved for future sends."));
+  } catch (err) {
+    const code =
+      (err as { error?: { output?: { statusCode?: number } } })?.error?.output
+        ?.statusCode ??
+      (err as { output?: { statusCode?: number } })?.output?.statusCode;
+    if (code === 515) {
+      console.log(
+        info(
+          "WhatsApp asked for a restart after pairing (code 515); creds are saved. Restarting connection once…",
+        ),
+      );
+      try {
+        sock.ws?.close();
+      } catch {
+        // ignore
+      }
+      const retry = await createWaSocket(false, verbose);
+      try {
+        await waitForConnection(retry);
+        console.log(
+          success(
+            "✅ Linked after restart; web session ready. You can now send with provider=web.",
+          ),
+        );
+        return;
+      } finally {
+        setTimeout(() => retry.ws?.close(), 500);
+      }
+    }
+    if (code === DisconnectReason.loggedOut) {
+      await fs.rm(WA_WEB_AUTH_DIR, { recursive: true, force: true });
+      console.error(
+        danger(
+          "WhatsApp reported the session is logged out. Cleared cached web session; please rerun warelay login and scan the QR again.",
+        ),
+      );
+      throw new Error("Session logged out; cache cleared. Re-run login.");
+    }
+    const formatted = formatError(err);
+    console.error(
+      danger(
+        `WhatsApp Web connection ended before fully opening. ${formatted}`,
+      ),
+    );
+    throw new Error(formatted);
+  } finally {
+    // Let Baileys flush any final events before closing the socket.
+    setTimeout(() => {
+      try {
+        sock.ws?.close();
+      } catch {
+        // ignore
+      }
+    }, 500);
+  }
+}