feat(route/twitter): only apply third-party api supported routes

DIYgod · Dec 10, 2024 · 844e658 · 844e658
1 parent 17e2fc7
commit 844e658
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 4 deletions.
diff --git a/lib/routes/twitter/api/web-api/constants.ts b/lib/routes/twitter/api/web-api/constants.ts
@@ -15,6 +15,8 @@ const graphQLEndpointsPlain = [
 
 const gqlMap = Object.fromEntries(graphQLEndpointsPlain.map((endpoint) => [endpoint.split('/')[3].replace(/V2$|Query$|QueryV2$/, ''), endpoint]));
 
+const thirdPartySupportedAPI = ['UserByScreenName', 'UserByRestId', 'UserTweets', 'UserTweetsAndReplies'];
+
 const gqlFeatureUser = {
     hidden_profile_subscriptions_enabled: true,
     rweb_tipjar_consumption_enabled: true,
@@ -112,4 +114,4 @@ const timelineParams = {
 
 const bearerToken = 'Bearer AAAAAAAAAAAAAAAAAAAAANRILgAAAAAAnNwIzUejRCOuH5E6I8xnZz4puTs%3D1Zv7ttfk8LF81IUq16cHjhLTvJu4FA33AGWWjCpTnA';
 
-export { baseUrl, gqlMap, gqlFeatures, timelineParams, bearerToken };
+export { baseUrl, gqlMap, gqlFeatures, timelineParams, bearerToken, thirdPartySupportedAPI };
diff --git a/lib/routes/twitter/api/web-api/utils.ts b/lib/routes/twitter/api/web-api/utils.ts
@@ -1,5 +1,5 @@
 import ConfigNotFoundError from '@/errors/types/config-not-found';
-import { baseUrl, gqlFeatures, bearerToken, gqlMap } from './constants';
+import { baseUrl, gqlFeatures, bearerToken, gqlMap, thirdPartySupportedAPI } from './constants';
 import { config } from '@/config';
 import queryString from 'query-string';
 import { Cookie, CookieJar } from 'tough-cookie';
@@ -228,7 +228,7 @@ export const paginationTweets = async (endpoint: string, userId: number | undefi
     };
 
     const fetchData = async () => {
-        if (config.twitter.thirdPartyApi) {
+        if (config.twitter.thirdPartyApi && thirdPartySupportedAPI.includes(endpoint)) {
             const { data } = await ofetch(`${config.twitter.thirdPartyApi}${gqlMap[endpoint]}`, {
                 method: 'GET',
                 params,

diff --git a/lib/routes/twitter/keyword.ts b/lib/routes/twitter/keyword.ts
@@ -22,6 +22,10 @@ export const route: Route = {
                 name: 'TWITTER_AUTH_TOKEN',
                 description: 'Please see above for details.',
             },
+            {
+                name: 'TWITTER_THIRD_PARTY_API',
+                description: 'Please see above for details.',
+            },
         ],
         requirePuppeteer: false,
         antiCrawler: false,
@@ -30,7 +34,7 @@ export const route: Route = {
         supportScihub: false,
     },
     name: 'Keyword',
-    maintainers: ['DIYgod', 'yindaheng98', 'Rongronggg9'],
+    maintainers: ['DIYgod', 'yindaheng98', 'Rongronggg9', 'pseudoyu'],
     handler,
     radar: [
         {