fix: 替换imdb获取豆瓣id的方法豆瓣需要在已登陆状态

techmovie · Apr 26, 2023 · ec46ca1 · ec46ca1 · ningsibuqu · Apr 28, 2023
1 parent c1fddda
commit ec46ca1
Show file tree

Hide file tree

Showing 3 changed files with 34 additions and 4 deletions.
diff --git a/src/common.js b/src/common.js
@@ -1,6 +1,6 @@
 import {
   CURRENT_SITE_INFO, PIC_URLS,
-  DOUBAN_API_URL, DOUBAN_SEARCH_API, CURRENT_SITE_NAME,
+  DOUBAN_API_URL, DOUBAN_SEARCH_API, CURRENT_SITE_NAME, DOUBAN_SUGGEST_API,
 } from './const';
 const isChinese = (title) => {
   return /[\u4e00-\u9fa5]+/.test(title);
@@ -89,6 +89,33 @@ const getDoubanId = async (imdbId) => {
     console.log(error);
   }
 };
+const getDoubanIdByIMDB = async (imdbId) => {
+  try {
+    const url = DOUBAN_SUGGEST_API.replace('{query}', imdbId);
+    const options = {
+      cookie: '',
+      anonymous: false,
+      responseType: undefined,
+    };
+    const data = await fetch(url, options);
+    const doc = new DOMParser().parseFromString(data, 'text/html');
+    const linkDom = doc.querySelector('.result-list .result h3 a');
+    if (!linkDom) {
+      throw new Error('豆瓣ID获取失败');
+    } else {
+      const { href, textContent } = linkDom;
+      const season = textContent?.match(/第(.+?)季/)?.[1] ?? '';
+      const doubanId = decodeURIComponent(href).match(/subject\/(\d+)/)?.[1] ?? '';
+      return ({
+        id: doubanId,
+        season,
+        title: textContent || '',
+      });
+    }
+  } catch (error) {
+    throw new Error(error.message);
+  }
+};
 const getTvSeasonData = (data) => {
   const torrentTitle = getTorrentTitle();
   return new Promise((resolve, reject) => {
@@ -300,4 +327,5 @@ export {
   getDoubanId,
   createDoubanDom,
   getTvSeasonData,
+  getDoubanIdByIMDB,
 };
diff --git a/src/const.js b/src/const.js
@@ -13,6 +13,7 @@ const CURRENT_SITE_INFO = siteInfo;
 const CURRENT_SITE_NAME = CURRENT_SITE_INFO?.siteName ?? '';
 const DOUBAN_API_URL = 'https://movie.douban.com/subject/{doubanId}';
 const DOUBAN_SEARCH_API = 'https://movie.douban.com/j/subject_suggest?q={query}';
+const DOUBAN_SUGGEST_API = 'https://www.douban.com/search?cat=1002&q={query}';
 const PIC_URLS = {
   border: 'https://ptpimg.me/zz4632.png',
   icon2x: 'https://ptpimg.me/n74cjc.png',
@@ -25,4 +26,5 @@ export {
   DOUBAN_API_URL,
   DOUBAN_SEARCH_API,
   PIC_URLS,
+  DOUBAN_SUGGEST_API,
 };
diff --git a/src/index.js b/src/index.js
@@ -2,8 +2,8 @@ import {
   CURRENT_SITE_INFO, CURRENT_SITE_NAME,
 } from './const';
 import {
-  getImdbId, getDoubanId, getTvSeasonData, createDoubanDom,
-  getDoubanInfo, addToPtpPage,
+  getImdbId, getTvSeasonData, createDoubanDom,
+  getDoubanInfo, addToPtpPage, getDoubanIdByIMDB,
 } from './common';
 import './style.js';
 (async () => {
@@ -17,7 +17,7 @@ import './style.js';
       if (!savedIds[imdbId] ||
         (savedIds[imdbId] && savedIds[imdbId].updateTime && Date.now() - savedIds[imdbId].updateTime >= 30 * 24 * 60 * 60 * 1000)) {
         let doubanId = '';
-        const movieData = await getDoubanId(imdbId);
+        const movieData = await getDoubanIdByIMDB(imdbId);
         if (!movieData) {
           throw new Error('没有找到豆瓣条目');
         }