feat：添加url的corpId替换脚本

scripts/replace-team-qrcode-url.js

@@ -5,8 +5,12 @@
  * 目的：
  * - 扫描 corp.team collection
  * - 将 team.qrcodes[].qrcode 中旧前缀替换为新前缀
+ * - 将 team.qrcodes[].qrcode 中 corpId 参数从旧值替换为新值
+ * - 将 team.groupWelcomeConfig 中的建档链接/附件链接同步替换
  * - 扫描 corp.staff-qrcode collection
  * - 将 staff-qrcode.archiveLink.url 中旧前缀替换为新前缀
+ * - 将 staff-qrcode.archiveLink.url 中 corpId 参数从旧值替换为新值
+ * - 将 staff-qrcode.fileList 中的附件链接同步替换
  *
  * 安全：
  * - 默认 dry-run（不落库）
@@ -90,6 +94,8 @@ function createDefaultOptions() {
     mongoPass: "",
     mongoAuthDb: "",
     corpId: "",
+    oldCorpId: "",
+    newCorpId: "",
     oldPrefix: "https://www.youcan365.com/h5/#/",
     newPrefix: "https://crm.gykqyy.com/gkPatientWeChat/#/",
   };
@@ -119,6 +125,8 @@ function parseArgs(argv) {
     else if (key === "mongopass") opts.mongoPass = value;
     else if (key === "mongoauthdb") opts.mongoAuthDb = value;
     else if (key === "corpid") opts.corpId = value;
+    else if (key === "oldcorpid" || key === "sourcecorpid" || key === "targetcorpid") opts.oldCorpId = value;
+    else if (key === "newcorpid" || key === "tokencorpid") opts.newCorpId = value;
     else if (key === "oldprefix") opts.oldPrefix = value;
     else if (key === "newprefix") opts.newPrefix = value;
     else if (key === "batchdocs") opts.batchDocs = Number(value);
@@ -175,6 +183,12 @@ function parseArgs(argv) {
     } else if (a === "--corpId") {
       opts.corpId = takeValue(i);
       i++;
+    } else if (a === "--oldCorpId" || a === "--sourceCorpId" || a === "--targetCorpId") {
+      opts.oldCorpId = takeValue(i);
+      i++;
+    } else if (a === "--newCorpId" || a === "--tokenCorpId") {
+      opts.newCorpId = takeValue(i);
+      i++;
     } else if (a === "--oldPrefix") {
       opts.oldPrefix = takeValue(i);
       i++;
@@ -202,8 +216,10 @@ function parseArgs(argv) {
 function applyEnvFallbacks(opts) {
   if (!opts.mongoUri) opts.mongoUri = process.env.MONGO_URI || process.env.GK_MONGO_URI || "";
   if (!opts.corpId) {
-    opts.corpId = process.env.TARGET_CORP_ID || process.env.TOKEN_CORP_ID || "";
+    opts.corpId = process.env.TOKEN_CORP_ID || process.env.TARGET_CORP_ID || "";
   }
+  if (!opts.oldCorpId) opts.oldCorpId = process.env.OLD_CORP_ID || process.env.SOURCE_CORP_ID || process.env.TARGET_CORP_ID || "";
+  if (!opts.newCorpId) opts.newCorpId = process.env.NEW_CORP_ID || process.env.TOKEN_CORP_ID || "";
   if (!opts.oldPrefix && process.env.OLD_QRCODE_PREFIX) opts.oldPrefix = process.env.OLD_QRCODE_PREFIX;
   if (!opts.newPrefix && process.env.NEW_QRCODE_PREFIX) opts.newPrefix = process.env.NEW_QRCODE_PREFIX;
 }
@@ -281,32 +297,24 @@ function replacePrefix(value, oldPrefix, newPrefix) {
   return `${newPrefix}${value.slice(oldPrefix.length)}`;
 }
 
-function updateTeamDoc(doc, { oldPrefix, newPrefix }) {
+function replaceUrlCorpId(value, oldCorpId, newCorpId) {
-  const qrcodes = Array.isArray(doc.qrcodes) ? doc.qrcodes : [];
+  if (!isNonEmptyString(value) || !isNonEmptyString(oldCorpId) || !isNonEmptyString(newCorpId)) return value;
-  let changed = 0;
+  return value.replace(
-  const nextQrcodes = qrcodes.map((item) => {
+    /([?&]corpId=)([^&#]*)/g,
-    if (!item || typeof item !== "object") return item;
+    (match, prefix, currentCorpId) => (currentCorpId === oldCorpId ? `${prefix}${newCorpId}` : match)
-    const current = item.qrcode;
+  );
-    const next = replacePrefix(current, oldPrefix, newPrefix);
-    if (next !== current) {
-      changed++;
-      return { ...item, qrcode: next };
-    }
-    return item;
-  });
-  return { changed, nextQrcodes };
 }
 
-function updateStaffQrcodeDoc(doc, { oldPrefix, newPrefix }) {
+function normalizeQrcodeUrl(value, { oldPrefix, newPrefix, oldCorpId, newCorpId }) {
-  const archiveLink =
+  return replaceUrlCorpId(replacePrefix(value, oldPrefix, newPrefix), oldCorpId, newCorpId);
-    doc && doc.archiveLink && typeof doc.archiveLink === "object"
+}
-      ? doc.archiveLink
+
-      : null;
+function normalizeArchiveLink(archiveLink, options) {
-  if (!archiveLink) {
+  if (!archiveLink || typeof archiveLink !== "object") {
     return { changed: 0, nextArchiveLink: archiveLink };
   }
   const current = archiveLink.url;
-  const next = replacePrefix(current, oldPrefix, newPrefix);
+  const next = normalizeQrcodeUrl(current, options);
   if (next === current) {
     return { changed: 0, nextArchiveLink: archiveLink };
   }
@@ -316,6 +324,120 @@ function updateStaffQrcodeDoc(doc, { oldPrefix, newPrefix }) {
   };
 }
 
+function normalizeFileList(fileList, options) {
+  if (!Array.isArray(fileList)) {
+    return { changed: 0, nextFileList: fileList };
+  }
+
+  let changed = 0;
+  const nextFileList = fileList.map((item) => {
+    if (!item || typeof item !== "object") return item;
+
+    let nextItem = item;
+
+    function setItemUrl(key) {
+      const current = item[key];
+      const next = normalizeQrcodeUrl(current, options);
+      if (next !== current) {
+        changed++;
+        nextItem = nextItem === item ? { ...item } : nextItem;
+        nextItem[key] = next;
+      }
+    }
+
+    setItemUrl("URL");
+    setItemUrl("url");
+    setItemUrl("href");
+
+    if (item.file && typeof item.file === "object") {
+      let nextFile = item.file;
+      for (const key of ["url", "URL", "href", "picurl", "imgUrl", "download_url"]) {
+        const current = item.file[key];
+        const next = normalizeQrcodeUrl(current, options);
+        if (next !== current) {
+          changed++;
+          nextFile = nextFile === item.file ? { ...item.file } : nextFile;
+          nextFile[key] = next;
+        }
+      }
+      if (nextFile !== item.file) {
+        nextItem = nextItem === item ? { ...item } : nextItem;
+        nextItem.file = nextFile;
+      }
+    }
+
+    return nextItem;
+  });
+
+  return { changed, nextFileList };
+}
+
+function updateTeamDoc(doc, { oldPrefix, newPrefix, oldCorpId, newCorpId }) {
+  const options = { oldPrefix, newPrefix, oldCorpId, newCorpId };
+  const qrcodes = Array.isArray(doc.qrcodes) ? doc.qrcodes : [];
+  let changed = 0;
+  const nextQrcodes = qrcodes.map((item) => {
+    if (!item || typeof item !== "object") return item;
+    const current = item.qrcode;
+    const next = normalizeQrcodeUrl(current, options);
+    if (next !== current) {
+      changed++;
+      return { ...item, qrcode: next };
+    }
+    return item;
+  });
+
+  const set = {};
+  if (changed > 0) {
+    set.qrcodes = nextQrcodes;
+  }
+
+  const groupWelcomeConfig =
+    doc && doc.groupWelcomeConfig && typeof doc.groupWelcomeConfig === "object"
+      ? doc.groupWelcomeConfig
+      : null;
+  if (groupWelcomeConfig) {
+    let nextGroupWelcomeConfig = groupWelcomeConfig;
+    const archiveResult = normalizeArchiveLink(groupWelcomeConfig.archiveLink, options);
+    if (archiveResult.changed > 0) {
+      changed += archiveResult.changed;
+      nextGroupWelcomeConfig = { ...nextGroupWelcomeConfig, archiveLink: archiveResult.nextArchiveLink };
+    }
+
+    const fileListResult = normalizeFileList(groupWelcomeConfig.fileList, options);
+    if (fileListResult.changed > 0) {
+      changed += fileListResult.changed;
+      nextGroupWelcomeConfig = { ...nextGroupWelcomeConfig, fileList: fileListResult.nextFileList };
+    }
+
+    if (nextGroupWelcomeConfig !== groupWelcomeConfig) {
+      set.groupWelcomeConfig = nextGroupWelcomeConfig;
+    }
+  }
+
+  return { changed, set };
+}
+
+function updateStaffQrcodeDoc(doc, { oldPrefix, newPrefix, oldCorpId, newCorpId }) {
+  const options = { oldPrefix, newPrefix, oldCorpId, newCorpId };
+  let changed = 0;
+  const set = {};
+
+  const archiveResult = normalizeArchiveLink(doc && doc.archiveLink, options);
+  if (archiveResult.changed > 0) {
+    changed += archiveResult.changed;
+    set.archiveLink = archiveResult.nextArchiveLink;
+  }
+
+  const fileListResult = normalizeFileList(doc && doc.fileList, options);
+  if (fileListResult.changed > 0) {
+    changed += fileListResult.changed;
+    set.fileList = fileListResult.nextFileList;
+  }
+
+  return { changed, set };
+}
+
 async function processCollectionDocs({
   collection,
   filter,
@@ -381,6 +503,8 @@ async function main() {
   --dbName <db>         数据库名，默认 corp
   --collection <name>   collection 名，默认 team
   --corpId <corpId>     仅处理指定机构
+  --oldCorpId <corpId>  URL 参数中的旧 corpId，默认取 OLD_CORP_ID / SOURCE_CORP_ID / TARGET_CORP_ID
+  --newCorpId <corpId>  URL 参数中的新 corpId，默认取 NEW_CORP_ID / TOKEN_CORP_ID
   --oldPrefix <url>     旧前缀，默认 https://www.youcan365.com/h5/#/
   --newPrefix <url>     新前缀，默认 https://crm.gykqyy.com/gkPatientWeChat/#/
   --apply --yes         真正写库；默认 dry-run
@@ -432,8 +556,10 @@ async function main() {
         updateTeamDoc(doc, {
           oldPrefix: opts.oldPrefix,
           newPrefix: opts.newPrefix,
+          oldCorpId: opts.oldCorpId,
+          newCorpId: opts.newCorpId,
         }),
-      buildUpdate: ({ nextQrcodes }) => ({ $set: { qrcodes: nextQrcodes } }),
+      buildUpdate: ({ set }) => ({ $set: set }),
       describeDoc: (doc) =>
         `_id=${doc._id} teamId=${doc.teamId || ""} corpId=${doc.corpId || ""}`,
     });
@@ -453,10 +579,10 @@ async function main() {
         updateStaffQrcodeDoc(doc, {
           oldPrefix: opts.oldPrefix,
           newPrefix: opts.newPrefix,
+          oldCorpId: opts.oldCorpId,
+          newCorpId: opts.newCorpId,
         }),
-      buildUpdate: ({ nextArchiveLink }) => ({
+      buildUpdate: ({ set }) => ({ $set: set }),
-        $set: { archiveLink: nextArchiveLink },
-      }),
       describeDoc: (doc) =>
         `_id=${doc._id} qrcodeId=${doc.qrcodeId || ""} corpId=${doc.corpId || ""}`,
     });