initial

2025-01-01 07:33:33 +01:00
commit 396cfe70de
25 changed files with 10518 additions and 0 deletions
@@ -0,0 +1,90 @@
+import * as plugins from './rendertron.plugins.js';
+import * as paths from './rendertron.paths.js';
+import { logger } from './rendertron.logging.js';
+import { PrerenderResult } from './rendertron.classes.prerenderresult.js';
+import { PrerenderManager } from './rendertron.classes.prerendermanager.js';
+import { TaskManager } from './rendertron.taskmanager.js';
+
+export class Rendertron {
+  public projectinfo: plugins.projectinfo.ProjectInfo;
+  public serviceServerInstance: plugins.typedserver.utilityservers.UtilityServiceServer;
+  public prerenderManager: PrerenderManager;
+  public taskManager: TaskManager;
+
+  /**
+   * starts the financeflow instance
+   */
+  public async start() {
+    this.projectinfo = new plugins.projectinfo.ProjectInfo(paths.packageDir);
+    this.prerenderManager = new PrerenderManager();
+    this.taskManager = new TaskManager(this);
+    await this.prerenderManager.start();
+    await this.taskManager.start();
+    this.serviceServerInstance = new plugins.typedserver.utilityservers.UtilityServiceServer({
+      serviceDomain: 'rendertron.lossless.one',
+      serviceName: 'rendertron',
+      serviceVersion: this.projectinfo.npm.version,
+      addCustomRoutes: async (serverArg) => {
+        serverArg.addRoute(
+          '/render/*',
+          new plugins.typedserver.servertools.Handler('GET', async (req, res) => {
+            const requestedUrl = req.url.replace('/render/', '');
+            logger.log('info', `Got SSR request for ${requestedUrl}`);
+            if (requestedUrl.startsWith('https://url(')) {
+              logger.log('warn', `relative url error for ${requestedUrl}`);
+              res.status(500);
+              res.write('error due to relative protocol');
+              res.end();
+              return;
+            }
+            const originResponse = await plugins.smartrequest
+              .request(
+                requestedUrl,
+                {
+                  method: 'GET',
+                  keepAlive: false,
+                  // headers: req.headers,
+                },
+                true
+              )
+              .catch((error) => {
+                logger.log('warn', `the origin request errored for ${requestedUrl}`);
+                res.write(`rendertron encountered an error for ${requestedUrl}`);
+                res.end();
+              });
+            if (!originResponse) {
+              return;
+            }
+            for (const header of Object.keys(originResponse.headers)) {
+              res.setHeader(header, originResponse.headers[header]);
+            }
+            if (originResponse.headers['content-type']?.includes('text/html')) {
+              logger.log('info', `Piping ${requestedUrl} through smartssr.`);
+              res.write(await this.prerenderManager.getPrerenderResultForUrl(requestedUrl));
+              res.end();
+            } else {
+              logger.log('info', `Serving ${requestedUrl} directly.`);
+              for (const headerKey of Object.keys(originResponse.headers)) {
+                console.log(`${headerKey}: ${originResponse.headers[headerKey]}`);
+                res.set(headerKey, originResponse.headers[headerKey]);
+              }
+              originResponse.on('data', (data) => {
+                res.write(data);
+              });
+              originResponse.on('end', () => {
+                res.end();
+              });
+            }
+          })
+        );
+      },
+    });
+    await this.serviceServerInstance.start();
+  }
+
+  public async stop() {
+    this.serviceServerInstance ? await this.serviceServerInstance.stop() : null;
+    this.prerenderManager ? await this.prerenderManager.stop() : null;
+    this.taskManager ? await this.taskManager.stop() : null;
+  }
+}