feat: ✨ pixiv 支持切换加载原图/非原图

#313

Author: hymbz

src/helper/other.ts

@@ -653,8 +653,8 @@ export const onUrlChange = (
 /** wait，但是只在 url 变化时判断 */
 export const waitUrlChange = (isValidUrl: () => unknown) =>
   new Promise<void>((resolve) => {
-    const abort = onUrlChange(() => {
-      if (!isValidUrl()) return;
+    const abort = onUrlChange(async () => {
+      if (!(await isValidUrl())) return;
       resolve();
       abort();
     });

src/index.ts

@@ -933,29 +933,7 @@ try {
     // #其他[Pixiv](https://www.pixiv.net)
     // test: https://www.pixiv.net/artworks/128841242
     case 'www.pixiv.net': {
-      let imgList: string[] = [];
-
-      options = {
-        name: 'pixiv',
-        getImgList: () => imgList,
-        SPA: {
-          async isMangaPage() {
-            const id = Number(location.pathname.split('/')[2]);
-            if (!id || !location.pathname.startsWith('/artworks/')) {
-              imgList.length = 0;
-              return false;
-            }
-
-            type resData = { body: { urls: { original: string } }[] };
-            const res = await request<resData>(`/ajax/illust/${id}/pages`, {
-              responseType: 'json',
-            });
-            imgList = res.response.body.map((e) => e.urls.original);
-            return imgList.length > 1;
-          },
-        },
-        initOptions: { autoShow: false, defaultOption: { pageNum: 1 } },
-      };
+      inject('site/pixiv');
       break;
     }
 

src/site/pixiv.tsx

@@ -0,0 +1,78 @@
+import { createEffectOn, onUrlChange, waitUrlChange } from 'helper';
+import { request, useInit } from 'main';
+
+(async () => {
+  let imgs: {
+    urls: { original: string; regular: string };
+    height: number;
+    width: number;
+  }[] = [];
+
+  const isMangaPage = async () => {
+    const id = Number(location.pathname.split('/')[2]);
+    if (!id || !location.pathname.startsWith('/artworks/')) {
+      imgs.length = 0;
+      return false;
+    }
+
+    const res = await request<{ body: typeof imgs }>(
+      `/ajax/illust/${id}/pages`,
+      { responseType: 'json' },
+    );
+    imgs = res.response.body;
+    return imgs.length > 1;
+  };
+  await waitUrlChange(isMangaPage);
+
+  const { store, setState, showComic, init } = await useInit('pixiv', {
+    autoShow: false,
+    defaultOption: { pageNum: 1 },
+    /** 加载原图 */
+    load_original_image: true,
+  });
+
+  // 在切换时重新获取图片
+  createEffectOn(
+    () => store.options.load_original_image,
+    (isOriginal, prev) => {
+      setState('nowComic', isOriginal ? 'original' : 'regular');
+      if (prev) showComic();
+    },
+  );
+
+  const getImgList = (isOriginal: boolean) => () =>
+    imgs.map((img) => {
+      const src = isOriginal ? img.urls.original : img.urls.regular;
+      return { src, height: img.height, width: img.width };
+    });
+
+  setState((state) => {
+    state.comicMap.original = {
+      getImgList: getImgList(true),
+      imgList: getImgList(true)(),
+    };
+    state.comicMap.regular = {
+      getImgList: getImgList(false),
+      imgList: getImgList(false)(),
+    };
+  });
+
+  init();
+
+  onUrlChange(async (lastUrl) => {
+    if (!lastUrl) return;
+
+    if (!isMangaPage())
+      return setState((state) => {
+        state.fab.show = false;
+        state.manga.show = false;
+      });
+
+    setState((state) => {
+      state.fab.show = undefined;
+      state.manga.show = false;
+    });
+
+    if (store.options.autoShow) await showComic();
+  });
+})();