@gobing-ai/knowledge-kit 0.1.10 → 0.1.12

package/plugins/publishings/wechatmp-pub/dist/profile-check.js

@@ -20026,9 +20026,11 @@ import { isAbsolute, join as join3, relative, resolve as resolve3, sep } from "p
 // ../../packages/publish-harness/src/bot-fs.ts
 import {
   chmodSync,
+  closeSync,
   copyFileSync,
   lstatSync,
   mkdirSync as mkdirSync2,
+  openSync,
   readdirSync as readdirSync2,
   realpathSync as realpathSync2,
   rmdirSync
@@ -20069,6 +20071,9 @@ function rmdir(path) {
 function chmod(path, mode) {
   chmodSync(path, mode);
 }
+function createExclusiveFile(path) {
+  closeSync(openSync(path, "wx", 384));
+}
 
 // ../../packages/publish-harness/src/bot-transaction.ts
 import { homedir } from "os";
@@ -20273,7 +20278,6 @@ class BotError extends Error {
     this.code = code;
   }
 }
-
 // ../../packages/publish-harness/src/index.ts
 import { homedir as homedir4 } from "os";
 import { join as join4, resolve as resolve4 } from "path";
@@ -20409,6 +20413,12 @@ async function waitUntil(predicate, timeoutMs, pollMs = DEFAULT_POLL_MS) {
   }
   return predicate();
 }
+var DEFAULT_SIGNAL_CONTAINERS = [
+  '[class*="toast" i]',
+  '[class*="notification" i]',
+  '[role="alert"]',
+  '[class*="message" i]'
+];
 function errorMessage2(error) {
   return error instanceof Error ? error.message : String(error);
 }
@@ -20416,6 +20426,9 @@ function errorMessage2(error) {
 // ../../plugins/publishings/wechatmp-pub/src/profile-check.ts
 init_dist();
 
+// ../../plugins/publishings/wechatmp-pub/src/index.ts
+import { randomUUID } from "crypto";
+
 // ../../packages/kk-core/src/content-limits.ts
 var PLATFORM_CONTENT_LIMITS = Object.freeze({
   schemaVersion: 5,
@@ -20686,6 +20699,9 @@ if (false) {}
 var CJK_TERMINATORS = new Set(["\u3002", "\uFF01", "\uFF1F"]);
 var LATIN_TERMINATORS = new Set([".", "!", "?"]);
 var ATTACHED_CLOSERS = new Set(["\u201D", "\u2019", "\u300D", "\u300F", "\uFF09", ")", '"', "'"]);
+// ../../plugins/publishings/wechatmp-pub/src/index.ts
+init_dist();
+
 // ../../plugins/publishings/wechatmp-pub/src/cgi.ts
 var MP_ORIGIN = "https://mp.weixin.qq.com";
 var FILETRANSFER_URL = `${MP_ORIGIN}/cgi-bin/filetransfer`;
@@ -20747,6 +20763,13 @@ var URL_CANCEL = `${URL_POPOVER} a.jsPopoverBt:text-is("\u53D6\u6D88")`;
 var ORIGINAL_DECLARATION = `${DIALOG} label:has(.js_original_type_radio):has-text("\u6587\u5B57\u539F\u521B")`;
 var ORIGINAL_AGREEMENT = `${DIALOG} label.original_agreement:has(input[type="checkbox"]:not(:checked))`;
 
+// ../../plugins/publishings/wechatmp-pub/src/flow.ts
+var PUBLIC_SAVE_REFUSAL_CONTAINERS = [
+  ...DEFAULT_SIGNAL_CONTAINERS,
+  "div.version_fail_dialog:visible",
+  "div.weui-desktop-dialog:visible"
+];
+
 // ../../plugins/publishings/wechatmp-pub/src/index.ts
 var CHANNEL = TARGET;
 var CONSOLE_READY_TIMEOUT_MS = 30000;
@@ -20756,6 +20779,25 @@ function isConsoleReady(page) {
 function storageStatePath(profileDir) {
   return `${profileDir}/storage-state.json`;
 }
+async function persistWechatSessionState(context, profileDir) {
+  const target = storageStatePath(profileDir);
+  const temporary = `${target}.${process.pid}.${randomUUID()}.tmp`;
+  const fs = createNodeFileSystem();
+  let reserved = false;
+  try {
+    createExclusiveFile(temporary);
+    reserved = true;
+    await context.storageState({ path: temporary });
+    chmod(temporary, 384);
+    await fs.rename(temporary, target);
+  } finally {
+    if (reserved) {
+      await Promise.resolve().then(() => fs.deleteFile(temporary)).catch(() => {
+        return;
+      });
+    }
+  }
+}
 async function checkWechatSession(profileDir, opts = {}) {
   const { page, context } = await openWithState(profileDir, opts.headless ?? true);
   try {
@@ -20778,6 +20820,8 @@ async function openWechatLogin(opts) {
     headless: false,
     launcher: kkBrowserLauncher
   });
+  let failed = false;
+  let failure;
   try {
     const page = context.pages()[0] ?? await context.newPage();
     await page.goto(CONSOLE_URL);
@@ -20785,10 +20829,23 @@ async function openWechatLogin(opts) {
     if (!ready)
       throw new Error(`QR scan not completed within ${timeoutMs / 1000}s`);
     await page.waitForTimeout(1e4);
-    await context.storageState({ path: storageStatePath(opts.profileDir) });
-  } finally {
+    try {
+      await persistWechatSessionState(context, opts.profileDir);
+    } catch {
+      throw new Error("wechatmp authenticated session snapshot failed");
+    }
+  } catch (error) {
+    failed = true;
+    failure = error;
+  }
+  try {
     await context.close();
+  } catch {
+    if (!failed)
+      echoError("wechatmp browser context close failed after authenticated session snapshot; saved state preserved");
   }
+  if (failed)
+    throw failure;
 }
 async function openWechatSession(profileDir, opts = {}) {
   const { context } = await openWithState(profileDir, opts.headless ?? false, opts.url ?? CONSOLE_URL);