fixed: fetch cover Referer

zhifengle · Jul 15, 2021 · b75d3d1 · b75d3d1
1 parent 26ea419
commit b75d3d1
Show file tree

Hide file tree

Showing 7 changed files with 62 additions and 1 deletion.
diff --git a/extension/manifest.json b/extension/manifest.json
@@ -53,6 +53,7 @@
     "tabs",
     "storage",
     "webRequest",
+    "webRequestBlocking",
     "<all_urls>"
   ]
 }
diff --git a/src/bg/index.ts b/src/bg/index.ts
@@ -21,14 +21,19 @@ interface Config {
 
 let E_USER_CONFIG: Config = {};
 
+// ref: header editor
+const IS_CHROME = /Chrome\/(\d+)\.(\d+)/.test(navigator.userAgent);
+
 async function sendMsgToCurrentTab(
   payload: LogMsg & Record<string, string | number>
 ) {
   const tabs = await browser.tabs.query({
     active: true,
     currentWindow: true,
   });
-  browser.tabs.sendMessage(tabs[0].id, payload);
+  if (tabs && Array.isArray(tabs) && tabs[0]) {
+    browser.tabs.sendMessage(tabs[0].id, payload);
+  }
 }
 
 async function handleMessage(request: ExtMsg) {
@@ -137,7 +142,9 @@ async function updateAuxData(payload: {
       duration: 0,
     });
     console.info('the start of updating aux data');
+    window._fetch_url_bg = auxSite;
     const auxData = await getWikiDataByURL(auxSite, auxSiteOpts);
+    window._fetch_url_bg = null;
     const obj = await browser.storage.local.get(['wikiData']);
     console.info('current wikiData: ', obj.wikiData);
     if (!auxData || (auxData && auxData.length === 0)) {
@@ -186,6 +193,18 @@ async function updateAuxData(payload: {
   }
 }
 
+function createHeaderListener(type: string): any {
+  const result = ['blocking'];
+  result.push(type);
+  if (
+    IS_CHROME &&
+    // @ts-ignore
+    chrome.webRequest.OnBeforeSendHeadersOptions.hasOwnProperty('EXTRA_HEADERS')
+  ) {
+    result.push('extraHeaders');
+  }
+  return result;
+}
 async function init() {
   // 初始化设置
   const obj = await browser.storage.local.get(['version', 'config']);
@@ -213,6 +232,23 @@ async function init() {
       console.log('E_CONFIG: ', E_USER_CONFIG);
     }
   });
+  browser.webRequest.onBeforeSendHeaders.addListener(
+    (obj) => {
+      let m = (obj?.url ?? '').match(/brandnew\/(\d+)/);
+      if (m) {
+        obj.requestHeaders.push({
+          name: 'Referer',
+          value: `http://www.getchu.com/soft.phtml?id=${m[1]}`,
+        });
+      }
+      return { requestHeaders: obj.requestHeaders };
+    },
+    {
+      // urls: ['http://*/brandnew/*'],
+      urls: ['http://www.getchu.com/brandnew/*'],
+    },
+    createHeaderListener('requestHeaders')
+  );
 }
 
 init();
diff --git a/src/global.d.ts b/src/global.d.ts
@@ -10,4 +10,6 @@ declare var GM_getResourceText: any;
 // @TODO avoid use global variable
 interface Window {
   _parsedEl: Element | Document;
+  // 后台的 url
+  _fetch_url_bg?: string;
 }
diff --git a/src/sites/erogamescape.ts b/src/sites/erogamescape.ts
@@ -155,6 +155,9 @@ export const erogamescapeTools: SiteTools = {
                 cookie: 'getchu_adalt_flag=getchu.com',
                 decode: 'EUC-JP',
               },
+              prefs: {
+                targetNames: ['cover'],
+              },
             },
           },
         };

diff --git a/src/sites/lib.ts b/src/sites/lib.ts
@@ -11,13 +11,22 @@ export async function getCover($d: Element, site: ModelKey) {
   }
   if (!url) return;
   try {
+    // 在其它网站上获取的相对路径的链接
+    // @TODO 这里临时使用的全局变量来处理
+    if (window._fetch_url_bg && !/^https?:/.test(url)) {
+      const urlObj = new URL(window._fetch_url_bg);
+      url = `${urlObj.origin}/${url.replace(/^\.?\/?/, '')}`;
+    }
     // 跨域的图片不能用这种方式
     // dataUrl = convertImgToBase64($d as any);
     let opts: any = {};
     if (site.includes('getchu')) {
       opts.headers = {
         Referer: location.href,
       };
+      if (!location.href.includes('getchu.com') && window._fetch_url_bg) {
+        opts.headers.Referer = window._fetch_url_bg;
+      }
     }
     dataUrl = await getImageDataByURL(url, opts);
     if (dataUrl) {

diff --git a/src/user-script/index.ts b/src/user-script/index.ts
@@ -42,7 +42,9 @@ async function updateAuxData(payload: {
       duration: 0,
     });
     console.info('the start of updating aux data');
+    window._fetch_url_bg = auxSite;
     const auxData = await getWikiDataByURL(auxSite, auxSiteOpts);
+    window._fetch_url_bg = null;
     if (!auxData || (auxData && auxData.length === 0)) {
       logMessage({
         type: 'error',

diff --git a/src/utils/domUtils.ts b/src/utils/domUtils.ts
@@ -305,3 +305,11 @@ export function genAnonymousLinkText(url: string, text: string): string {
       ${text}</a>
     `;
 }
+
+export function addHTMLBase(html: string, url: string): string {
+  if (html.match(/<base.+>/)) {
+    return html;
+  }
+  const obj = new URL(url);
+  return html.replace('</head>', `<base href="${obj.origin}"></head>`);
+}