Initial Commit

mrepol742 · Oct 20, 2023 · d3989a3 · d3989a3
1 parent bfda75b
commit d3989a3
Show file tree

Hide file tree

Showing 3 changed files with 233 additions and 19 deletions.
diff --git a/index.ts b/index.ts
@@ -17,6 +17,7 @@
 
 import * as http from "http";
 import * as google from "googlethis";
+import { Innertube, UniversalCache, Utils } from "youtubei.js";
 
 const corsWhitelist = ["https://mrepol742.github.io"];
 
@@ -32,12 +33,44 @@ function getRoutes() {
         let url = ress.split("?")[0];
         console.log(req.method + " " + req.headers.origin + " " + url);
         if (req.method != "GET" || !(corsWhitelist.indexOf(req.headers.origin) !== -1)) {
-            res.writeHead(301, { Location: "https://mrepol742.github.io/unauthorized" });
+            res.writeHead(301, { Location: "https://mrepol742.github.io/unauthorized?utm_source=mrepol742.repl.co" });
             res.end();
         } else {
-            if (url == "/img" || url == "/img/index.html") {
-                let data = ress.split("?")[1];
-                let results = [];
+            let data = ress.split("?")[1];
+            let results = [];
+            if (url == "/vid" || url == "/vid/index.html") {
+                try {
+                    const yt = await Innertube.create({ cache: new UniversalCache(false), generate_session_locally: true });
+                    const search = await yt.search(data, { type: "video" });
+                    if (search.results) {
+                        res.setHeader("Access-Control-Allow-Origin", req.headers.origin);
+                        res.setHeader("Content-Type", "application/json");
+                        res.writeHead(200);
+                        res.end(JSON.stringify(search.results));
+                    }
+                } catch (err) {
+                    res.setHeader("Access-Control-Allow-Origin", req.headers.origin);
+                    res.setHeader("Content-Type", "application/json");
+                    res.writeHead(500);
+                    res.end('{error: "Internal Server Error", code: 500}');
+                }
+            } else if (url == "/mus" || url == "/mus/index.html") {
+                try {
+                    const yt = await Innertube.create({ cache: new UniversalCache(false), generate_session_locally: true });
+                    const search = await yt.music.search(data, { type: "song" });
+                    if (search.results) {
+                        res.setHeader("Access-Control-Allow-Origin", req.headers.origin);
+                        res.setHeader("Content-Type", "application/json");
+                        res.writeHead(200);
+                        res.end(JSON.stringify(search.results));
+                    }
+                } catch (err) {
+                    res.setHeader("Access-Control-Allow-Origin", req.headers.origin);
+                    res.setHeader("Content-Type", "application/json");
+                    res.writeHead(500);
+                    res.end('{error: "Internal Server Error", code: 500}');
+                }
+            } else if (url == "/img" || url == "/img/index.html") {
                 try {
                     const images = await google.image(data, { safe: true });
                     let i;
@@ -49,11 +82,12 @@ function getRoutes() {
                     res.writeHead(200);
                     res.end(JSON.stringify(results));
                 } catch (err) {
-                    console.log(err);
+                    res.setHeader("Access-Control-Allow-Origin", req.headers.origin);
+                    res.setHeader("Content-Type", "application/json");
+                    res.writeHead(500);
+                    res.end('{error: "Internal Server Error", code: 500}');
                 }
             } else if (url == "/" || url == "/index.html") {
-                let data = ress.split("?")[1];
-                let results = [];
                 try {
                     let response1 = await google.search(data, {
                         page: 0,
@@ -78,11 +112,23 @@ function getRoutes() {
                         res.setHeader("Content-Type", "application/json");
                         res.writeHead(200);
                         res.end(JSON.stringify(results));
-                    } catch (err) {}
-                } catch (err) {}
+                    } catch (err) {
+                        res.setHeader("Access-Control-Allow-Origin", req.headers.origin);
+                        res.setHeader("Content-Type", "application/json");
+                        res.writeHead(500);
+                        res.end('{error: "Internal Server Error", code: 500}');
+                    }
+                } catch (err) {
+                    res.setHeader("Access-Control-Allow-Origin", req.headers.origin);
+                    res.setHeader("Content-Type", "application/json");
+                    res.writeHead(500);
+                    res.end('{error: "Internal Server Error", code: 500}');
+                }
             } else {
-                res.writeHead(301, { Location: "https://mrepol742.github.io/404.html" });
-                res.end();
+                res.setHeader("Access-Control-Allow-Origin", req.headers.origin);
+                res.setHeader("Content-Type", "application/json");
+                res.writeHead(400);
+                res.end('{error: "Bad Request", code: 400}');
             }
         }
     };

diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -1,7 +1,7 @@
 {
     "name": "webvium-search",
     "version": "1.0.1",
-    "description": "Backend for Webvium Search Results using googlethis npm package.",
+    "description": "Backend for Webvium Search Results using googlethis & youtubei.js npm package.",
     "main": "index.js",
     "scripts": {
         "start": "clear && while [ true ]; do node index.js ; done",
@@ -20,7 +20,8 @@
     "homepage": "https://github.com/mrepol742/search#readme",
     "dependencies": {
         "@types/node": "^20.3.1",
-        "googlethis": "^1.8.0"
+        "googlethis": "^1.8.0",
+        "youtubei.js": "^5.1.0"
     },
     "devDependencies": {
         "prettier": "^2.8.8"