fix: proper error handler (#23)

* re use legacy errors

* 8.3.6-alpha

.gitignore

@@ -10,4 +10,5 @@ p.ts
 .idea
 CHANGELOG.md
 theme.zip
-AGENTS.md
+AGENTS.md
+lustpress-express

package.json

@@ -1,6 +1,6 @@
 {
   "name": "lustpress",
-  "version": "8.2.4-alpha",
+  "version": "8.3.6-alpha",
   "description": "RESTful and experimental API for PH and other R18 websites",
   "main": "src/index.ts",
   "scripts": {

src/LustPress.ts

@@ -25,12 +25,14 @@ function cachedLocationOrUnknown(): string {
 class LustPress {
   url: string;
   useragent: string;
-  private browserUA = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/121.0.0.0 Safari/537.36";
+  private browserUA = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/116.0.0.0 Safari/537.36";
   private cookieCache: { [domain: string]: string } = {};
 
   constructor() {
     this.url = "";
-    this.useragent = `${pkg.name}/${pkg.version} Bun/${Bun.version}`;
+    this.browserUA =
+      "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/116.0.0.0 Safari/537.36";
+    this.useragent = process.env.USER_AGENT || `${pkg.name}/${pkg.version} Bun/${Bun.version}`;
   }
 
   async getPornhubCookies(): Promise<string> {
@@ -86,7 +88,7 @@ class LustPress {
     const domain = new URL(url).hostname;
 
     const headers: Record<string, string> = {
-      "User-Agent": isPornhub ? this.browserUA : this.useragent,
+      "User-Agent": isPornhub ? this.browserUA : this.useragent
     };
 
     if (isPornhub) {
@@ -101,20 +103,23 @@ class LustPress {
 
     console.log(`[FETCH] ${url}`);
     let res = await fetch(url, { headers, redirect: "follow" });
-    let text = await res.text();
+    let arrayBuf = await res.arrayBuffer();
+    let body = Buffer.from(arrayBuf);
 
-    if (isPornhub && text.includes("leastFactor")) {
+    if (isPornhub && body.toString("utf8").includes("leastFactor")) {
       console.log("Challenge detected at target URL. Re-authenticating...");
       this.cookieCache[domain] = await this.getPornhubCookies();
       res = await fetch(url, { 
         headers: { ...headers, Cookie: this.cookieCache[domain] },
         redirect: "follow" 
       });
-      text = await res.text();
+      arrayBuf = await res.arrayBuffer();
+      body = Buffer.from(arrayBuf);
     }
 
-    const body = Buffer.from(text);
+    if (res.ok && !url.includes("/random")) {
-    if (!url.includes("/random")) await keyv.set(url, body, ttl);
+      await keyv.set(url, body, ttl);
+    }
     return body;
   }
 

src/index.ts

@@ -1,56 +1,69 @@
-import { Elysia } from "elysia";
+import { Elysia } from "elysia";
-import { cors } from "@elysiajs/cors";
+import { cors } from "@elysiajs/cors";
-import { swagger } from "@elysiajs/swagger";
+import { swagger } from "@elysiajs/swagger";
-import { lust } from "./LustPress";
+import { lust } from "./LustPress";
-import { scrapeRoutes } from "./router/endpoint";
+import { scrapeRoutes } from "./router/endpoint";
-import pkg from "../package.json";
+import pkg from "../package.json";
-
+
-const app = new Elysia()
+const app = new Elysia()
-  .use(cors())
+  .use(cors())
-  .use(
+  .use(
-    swagger({
+    swagger({
-      path: "/docs",
+      path: "/docs",
-      documentation: {
+      documentation: {
-        info: {
+        info: {
-          title: "Lustpress API",
+          title: "Lustpress API",
-          version: pkg.version,
+          version: pkg.version,
-          description: pkg.description,
+          description: pkg.description,
-        },
+        },
-      },
+      },
-    })
+    })
-  )
+  )
-  .get("/", async () => ({
+  .get("/", async () => ({
-    success: true,
+    success: true,
-    playground: "https://sinkaroid.github.io/lustpress",
+    playground: "https://sinkaroid.github.io/lustpress",
-    endpoint:
+    endpoint:
-      "https://github.com/sinkaroid/lustpress/blob/master/README.md#routing",
+      "https://github.com/sinkaroid/lustpress/blob/master/README.md#routing",
-    date: new Date().toLocaleString(),
+    date: new Date().toLocaleString(),
-    rss: lust.currentProccess().rss,
+    rss: lust.currentProccess().rss,
-    heap: lust.currentProccess().heap,
+    heap: lust.currentProccess().heap,
-    server: await lust.getServer(),
+    server: await lust.getServer(),
-    version: pkg.version,
+    version: pkg.version,
-  }))
+  }))
-  .use(scrapeRoutes)
+  .onError(({ code, error, set }) => {
-  .onError(({ code, error, set }) => {
+    if (code === "NOT_FOUND") {
-    console.log("Error occurred:", error);
+      set.status = 404;
-    if (code === "NOT_FOUND") {
+      return {
-      set.status = 404;
+        success: false,
-      return {
+        message: (error as Error).message || "Not Found",
-        success: false,
+      };
-        message: (error as Error).message || "Not Found",
+    }
-      };
+    if (code === "VALIDATION") {
-    }
+      set.status = 400;
-    set.status = 500;
+      return {
-    return {
+        success: false,
-      success: false,
+        message: JSON.parse((error as Error).message)[0]?.message || "Validation Error",
-      message: (error as Error).message || "Internal Server Error",
+      };
-      stack: (error as Error).stack,
+    }
-    };
+    if (code === "UNKNOWN") {
-  })
+      set.status = 400;
-  .listen(process.env.PORT || 3000);
+      return {
-
+        success: false,
-console.log(
+        message: (error as Error).message,
-  `Lustpress is running at ${app.server?.hostname}:${app.server?.port}`
+      };
-);
+    }
-
+    set.status = 500;
-export type App = typeof app;
+    return {
+      success: false,
+      message: (error as Error).message || "Internal Server Error",
+      stack: (error as Error).stack,
+    };
+  })
+  .use(scrapeRoutes) // reuse legacy error handling
+  .listen(process.env.PORT || 3000);
+
+console.log(
+  `Lustpress is running at ${app.server?.hostname}:${app.server?.port}`
+);
+
+export type App = typeof app;