refactor: ♻️ 重构合并拷贝和热辣通过网页数据获取图片列表的代码

Author: hymbz

rollup.config.ts

@@ -67,6 +67,7 @@ const packlist = [
   'components/Fab',
   'components/Toast',
   'userscript/dmzjApi',
+  'userscript/copyApi',
   'userscript/detectAd',
   'userscript/main',
   'worker/detectAd',

src/index.ts

@@ -21,6 +21,7 @@ import {
 } from 'helper';
 import { getInitLang } from 'helper/languages';
 import { request, toast, universal } from 'main';
+import { getImglistByHtml } from 'userscript/copyApi';
 import { otherSite } from 'userscript/otherSite';
 
 /** 站点配置 */
@@ -702,26 +703,16 @@ try {
     case 'www.relamanhua.org':
     case 'www.manga2024.com':
     case 'www.2024manga.com': {
-      if (
-        !location.pathname.includes('/chapter/') &&
-        !document.querySelector('.disData[contentkey]')
-      )
+      if (!location.pathname.includes('/chapter/')) break;
+
+      if (!document.querySelector('.disData[contentkey]')) {
+        toast.error(t('site.changed_load_failed'));
         break;
-      const getImgList = async () => {
-        const [, , comicName, , id] = location.pathname.split('/');
-        const res = await request<{
-          results: { chapter: { contents: { url: string }[] } };
-        }>(
-          `https://api.manga2025.com/api/v3/comic/${comicName}/chapter/${id}`,
-          { responseType: 'json' },
-        );
-        return res.response.results.chapter.contents.map(({ url }) =>
-          url.replace('.h800x.', '.h1500x.'),
-        );
-      };
+      }
+
       options = {
         name: 'relamanhua',
-        getImgList,
+        getImgList: () => getImglistByHtml(),
         onNext: querySelectorClick('.comicContent-next a:not(.prev-null)'),
         onPrev: querySelectorClick(
           '.comicContent-prev:not(.index,.list) a:not(.prev-null)',

src/site/copymanga.tsx

@@ -13,6 +13,7 @@ import {
   wait,
 } from 'helper';
 import { request, toast, useInit } from 'main';
+import { decryptData, getImglistByHtml } from 'userscript/copyApi';
 
 // API 参考：https://github.com/fumiama/copymanga/blob/279e08b06a70307bf20162900103ec1fdcb97751/app/src/main/res/values/strings.xml
 
@@ -51,62 +52,6 @@ const pcApi = new (class {
     );
 })();
 
-// by: https://github.com/MapoMagpie/comic-looms/blob/7799f87fdd5a8ac73c878f338b7ae6aa5c0b2d18/src/platform/matchers/mangacopy.ts#L96-L125
-const decryptData = async (
-  raw: string,
-  key: string = unsafeWindow.cct || 'oppzzivv.nzm.oip',
-) => {
-  const cipher = raw.slice(16);
-  const iv = raw.slice(0, 16);
-
-  const decryptedBuffer = await crypto.subtle.decrypt(
-    { name: 'AES-CBC', iv: new TextEncoder().encode(iv) },
-    await crypto.subtle.importKey(
-      'raw',
-      new TextEncoder().encode(key),
-      { name: 'AES-CBC' },
-      false,
-      ['decrypt'],
-    ),
-    new Uint8Array(
-      cipher.match(/.{1,2}/g)!.map((byte) => Number.parseInt(byte, 16)),
-    ).buffer,
-  );
-  return JSON.parse(new TextDecoder().decode(decryptedBuffer));
-};
-
-declare const contentKey: string;
-declare const cct: string;
-
-/** 通过解析网页变量获取图片列表 */
-const getImglistByHtml = async (comicName: string, id: string) => {
-  const getKeys = async (): Promise<[string, string]> => {
-    // 移动端没有 contentKey，就从 PC 端的网页获取
-    if (!unsafeWindow.contentKey) {
-      const html = await request(
-        `${location.origin}/comic/${comicName}/chapter/${id}`,
-        {
-          fetch: false,
-          headers: {
-            'User-Agent':
-              'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/104.0.5112.79 Safari/537.36',
-          },
-        },
-      );
-      const [script] = html.responseText.match(
-        /(?<=<script>\s+)(var .+?contentKey =.+?)(?=<\/script)/gs,
-      )!;
-      eval(script); // oxlint-disable-line no-eval
-    }
-
-    return [contentKey, cct];
-  };
-
-  const keys = await getKeys();
-  const res: { url: string }[] = await decryptData(...keys);
-  return res.map(({ url }) => url.replace(/(?<=(\/|\.))c800x/, 'c1500x'));
-};
-
 // 在目录页显示上次阅读记录
 const handleLastChapter = (comicName: string) => {
   let a: HTMLAnchorElement;
@@ -496,7 +441,9 @@ const buildChapters = async (comicName: string, hiddenType: HiddenType) => {
         if (titleDom) return getImglistByApi();
         // 其他普通漫画优先通过解析网页变量加载，避免触发 api 的限制
         try {
-          const imgList = await getImglistByHtml(comicName, id);
+          const imgList = await getImglistByHtml(
+            `${location.origin}/comic/${comicName}/chapter/${id}`,
+          );
           if (imgList.length === 0) throw new Error('解析网页变量失败');
           return imgList;
         } catch (error) {

src/userscript/copyApi.ts

@@ -0,0 +1,87 @@
+import { querySelector, t } from 'helper';
+import { toast } from 'main';
+import { request } from 'request';
+
+let contentKey = '';
+let key = '';
+
+const getKeys = async (url?: string): Promise<[string, string]> => {
+  if (contentKey && key) return [contentKey, key];
+
+  // 热辣漫画放在网页元素里
+  if (querySelector('.disData[contentkey]')) {
+    contentKey = querySelector('.disData[contentkey]')!.getAttribute(
+      'contentkey',
+    )!;
+    key = querySelector('.disPass[contentkey]')!.getAttribute('contentkey')!;
+    return [contentKey, key];
+  }
+
+  // 拷贝 PC 端直接放在网页变量里，不过另一个变量的名字会变
+  if (unsafeWindow.contentKey && unsafeWindow.cct) {
+    contentKey = unsafeWindow.contentKey; // oxlint-disable-line prefer-destructuring
+    key = unsafeWindow.cct;
+    return [contentKey, key];
+  }
+
+  // 如果另一个变量的名字变了，或者是在拷贝的移动端，就得从 PC 端的网页里解析获取了
+  if (url) {
+    const html = await request(url, {
+      fetch: false,
+      headers: {
+        'User-Agent':
+          'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/104.0.5112.79 Safari/537.36',
+      },
+    });
+    const [script] = html.responseText.match(
+      /(?<=<script>\s+)(var .+?contentKey =.+?)(?=<\/script)/gs,
+    )!;
+
+    const res: Record<string, string> = {};
+    for (const [, key, value] of script.matchAll(/var (\S+) = '(.+?)';\n/g))
+      res[key] = value;
+
+    contentKey = res.contentKey; // oxlint-disable-line prefer-destructuring
+
+    const passKey = Object.keys(res).find((key) => key !== 'contentKey');
+    if (!passKey) {
+      toast.error(t('site.changed_load_failed'));
+      throw new Error(t('site.changed_load_failed'));
+    }
+    key = res[passKey];
+    return [contentKey, key];
+  }
+
+  toast.error(t('site.changed_load_failed'));
+  throw new Error(t('site.changed_load_failed'));
+};
+
+// by: https://github.com/MapoMagpie/comic-looms/blob/7799f87fdd5a8ac73c878f338b7ae6aa5c0b2d18/src/platform/matchers/mangacopy.ts#L96-L125
+export const decryptData = async (raw: string, key?: string) => {
+  key ||= (await getKeys())[1]; // oxlint-disable-line no-await-expression-member
+
+  const cipher = raw.slice(16);
+  const iv = raw.slice(0, 16);
+
+  const decryptedBuffer = await crypto.subtle.decrypt(
+    { name: 'AES-CBC', iv: new TextEncoder().encode(iv) },
+    await crypto.subtle.importKey(
+      'raw',
+      new TextEncoder().encode(key),
+      { name: 'AES-CBC' },
+      false,
+      ['decrypt'],
+    ),
+    new Uint8Array(
+      cipher.match(/.{1,2}/g)!.map((byte) => Number.parseInt(byte, 16)),
+    ).buffer,
+  );
+  return JSON.parse(new TextDecoder().decode(decryptedBuffer));
+};
+
+/** 通过解析网页变量获取图片列表 */
+export const getImglistByHtml = async (url?: string) => {
+  const keys = await getKeys(url);
+  const res: { url: string }[] = await decryptData(...keys);
+  return res.map(({ url }) => url.replace(/(?<=(\/|\.))c800x/, 'c1500x'));
+};