From 08a89190ed64457c004535dabf642407a83a238c Mon Sep 17 00:00:00 2001
From: Artemy <artegoser@gmail.com>
Date: Fri, 18 Aug 2023 11:38:05 +0300
Subject: [PATCH] fix: proxy url

---
 src/app.ts                   | 10 +++++++---
 src/config/config.service.ts |  5 ++++-
 src/handlers/main.ts         |  3 ++-
 src/routes/get.ts            | 11 ++++++++++-
 src/routes/parse.ts          | 11 ++++++++++-
 src/routes/raw-html.ts       | 12 +++++++++++-
 src/utils/generate.ts        |  8 ++++++--
 7 files changed, 50 insertions(+), 10 deletions(-)

diff --git a/src/app.ts b/src/app.ts
index f92fc18..eb3df5c 100644
--- a/src/app.ts
+++ b/src/app.ts
@@ -27,6 +27,7 @@ class App {
   async init() {
     const fastify = Fastify({
       logger: true,
+      trustProxy: this.config.reverse_proxy_enabled,
     });
 
     fastify.register(fastifyStatic, {
@@ -58,9 +59,12 @@ class App {
 
     fastify.setErrorHandler(errorHandler);
 
-    fastify.listen({ host: this.config.host, port: this.config.port }, (err) => {
-      err && console.log(err);
-    });
+    fastify.listen(
+      { host: this.config.host, port: this.config.port },
+      (err) => {
+        err && console.log(err);
+      }
+    );
   }
 }
 
diff --git a/src/config/config.service.ts b/src/config/config.service.ts
index f88c60c..10b8bfc 100644
--- a/src/config/config.service.ts
+++ b/src/config/config.service.ts
@@ -3,6 +3,7 @@ import { config } from "dotenv";
 export class ConfigService {
   public readonly host: string;
   public readonly port: number;
+  public readonly reverse_proxy_enabled: boolean;
 
   constructor() {
     const parsed = config().parsed;
@@ -11,7 +12,9 @@ export class ConfigService {
       throw new Error("Invalid .env file");
     }
 
-    this.host = process.env.HOST || 'localhost';
+    this.host = process.env.HOST || "localhost";
     this.port = Number(process.env.PORT) || 8080;
+    this.reverse_proxy_enabled =
+      Boolean(process.env.REVERSE_PROXY_ENABLED) || false;
   }
 }
diff --git a/src/handlers/main.ts b/src/handlers/main.ts
index 01d8983..6769283 100644
--- a/src/handlers/main.ts
+++ b/src/handlers/main.ts
@@ -18,6 +18,7 @@ import {
 
 export default async function handlePage(
   url: string, // remote URL
+  requestUrl: URL, // proxy URL
   engine?: string
 ): Promise<IHandlerOutput> {
   const urlObj = new URL(url);
@@ -40,7 +41,7 @@ export default async function handlePage(
   const window = new JSDOM(response.data, { url }).window;
 
   [...window.document.getElementsByTagName("a")].forEach((link) => {
-    link.href = generateProxyUrl(link.href, engine);
+    link.href = generateProxyUrl(requestUrl, link.href, engine);
   });
 
   if (engine) {
diff --git a/src/routes/get.ts b/src/routes/get.ts
index e6afe6b..dbd1e59 100644
--- a/src/routes/get.ts
+++ b/src/routes/get.ts
@@ -2,6 +2,7 @@ import { FastifyInstance } from "fastify";
 
 import { GetSchema, IGetSchema } from "../types/requests";
 import handlePage from "../handlers/main";
+import { generateRequestUrl } from "../utils/generate";
 
 export default async function getRoute(fastify: FastifyInstance) {
   fastify.get<IGetSchema>(
@@ -11,7 +12,15 @@ export default async function getRoute(fastify: FastifyInstance) {
       const remoteUrl = request.query.url;
       const engine = request.query.engine;
 
-      const parsed = await handlePage(remoteUrl, engine);
+      const parsed = await handlePage(
+        remoteUrl,
+        generateRequestUrl(
+          request.protocol,
+          request.hostname,
+          request.originalUrl
+        ),
+        engine
+      );
 
       if (request.query.format === "text") {
         reply.type("text/plain; charset=utf-8");
diff --git a/src/routes/parse.ts b/src/routes/parse.ts
index 09bb06a..bca6363 100644
--- a/src/routes/parse.ts
+++ b/src/routes/parse.ts
@@ -1,13 +1,22 @@
 import { EngineRequest, IParseSchema, parseSchema } from "../types/requests";
 import { FastifyInstance } from "fastify";
 import handlePage from "../handlers/main";
+import { generateRequestUrl } from "../utils/generate";
 
 export default async function parseRoute(fastify: FastifyInstance) {
   fastify.get<IParseSchema>(
     "/parse",
     { schema: parseSchema },
     async (request: EngineRequest) => {
-      return await handlePage(request.query.url, request.query.engine);
+      return await handlePage(
+        request.query.url,
+        generateRequestUrl(
+          request.protocol,
+          request.hostname,
+          request.originalUrl
+        ),
+        request.query.engine
+      );
     }
   );
 }
diff --git a/src/routes/raw-html.ts b/src/routes/raw-html.ts
index 20bcae5..30c520f 100644
--- a/src/routes/raw-html.ts
+++ b/src/routes/raw-html.ts
@@ -2,13 +2,23 @@ import { FastifyInstance } from "fastify";
 
 import { GetRequest, IParseSchema, rawHtmlSchema } from "../types/requests";
 import handlePage from "../handlers/main";
+import { generateRequestUrl } from "../utils/generate";
 
 export default async function rawHtml(fastify: FastifyInstance) {
   fastify.get<IParseSchema>(
     "/raw-html",
     { schema: rawHtmlSchema },
     async (request: GetRequest) => {
-      return (await handlePage(request.query.url)).content;
+      return (
+        await handlePage(
+          request.query.url,
+          generateRequestUrl(
+            request.protocol,
+            request.hostname,
+            request.originalUrl
+          )
+        )
+      ).content;
     }
   );
 }
diff --git a/src/utils/generate.ts b/src/utils/generate.ts
index 4a0095a..08f5c90 100644
--- a/src/utils/generate.ts
+++ b/src/utils/generate.ts
@@ -6,8 +6,12 @@ export function generateRequestUrl(
   return new URL(`${protocol}://${host}${originalUrl}`);
 }
 
-export function generateProxyUrl(href: string, engine?: string): string {
+export function generateProxyUrl(
+  requestUrl: URL,
+  href: string,
+  engine?: string
+): string {
   const urlParam = `?url=${encodeURIComponent(href)}`;
   const engineParam = engine ? `&engine=${engine}` : "";
-  return `/get${urlParam}${engineParam}`;
+  return `${requestUrl.origin}/get${urlParam}${engineParam}`;
 }