Fetch more sites config from IMDB Scout script

Moreasan · Aug 25, 2024 · d91a26e · d91a26e
1 parent 7774e21
commit d91a26e
Show file tree

Hide file tree

Showing 2 changed files with 1,651 additions and 827 deletions.
diff --git a/common/scripts/PrivateSitesGenerator.ts b/common/scripts/PrivateSitesGenerator.ts
@@ -24,6 +24,12 @@ let code: string = await new Promise((resolve, reject) => {
 const ast = esprima.parseScript(code, { tolerant: true });
 
 let privateSitesConfig: PrivateSiteConfig[] = [];
+const sites = [
+  "private_sites",
+  "chinese_sites",
+  "french_sites",
+  "german_sites",
+];
 
 estraverse.traverse(ast, {
   enter: function (node) {
@@ -36,10 +42,10 @@ estraverse.traverse(ast, {
           continue;
         }
         if (
-          declaration.id.name === "private_sites" &&
+          sites.includes(declaration.id.name) &&
           declaration.init.type === "ArrayExpression"
         ) {
-          privateSitesConfig = declaration.init.elements.map((element: any) => {
+          const sitesConfig = declaration.init.elements.map((element: any) => {
             if (element.type === "Literal") {
               return element.value;
             } else if (element.type === "ObjectExpression") {
@@ -64,6 +70,7 @@ estraverse.traverse(ast, {
               return obj;
             }
           });
+          privateSitesConfig = privateSitesConfig.concat(sitesConfig);
           return;
         }
       }
@@ -83,6 +90,12 @@ if (privateSitesConfig) {
   [key: string]: boolean | string | RegExp | undefined | number;
 };\n\n`;
 
+  privateSitesConfig.sort((a, b) => {
+    if (a.name < b.name) return -1;
+    if (a.name > b.name) return 1;
+    return 0;
+  });
+
   for (const item of privateSitesConfig) {
     const itemName = item.name;
     let itemProps = "";