Create controller for status monitor

Author: ivanvs

examples/test-status-monitor/src/app.module.ts

@@ -1,9 +1,10 @@
 import { Module } from '@nestjs/common';
 import { AppController } from './app.controller';
 import { AppService } from './app.service';
+import { StatusMonitorModule } from '../../../dist/status.monitor.module';
 
 @Module({
-  imports: [],
+  imports: [StatusMonitorModule],
   controllers: [AppController],
   providers: [AppService],
 })

examples/test-status-monitor/src/main.ts

@@ -1,11 +1,8 @@
 import { NestFactory } from '@nestjs/core';
 import { AppModule } from './app.module';
-import { StatusMonitorModule } from '../../../dist/status.monitor.module';
 
 async function bootstrap() {
   const app = await NestFactory.create(AppModule);
-
-  StatusMonitorModule.setup('/monitor', app);
   await app.listen(3000);
 }
 bootstrap();

src/status.monitor.controller.ts

@@ -0,0 +1,29 @@
+import { Get, Controller, HttpCode } from '@nestjs/common';
+import * as fs from 'fs';
+import * as path from 'path';
+import Handlebars from 'handlebars';
+
+const controllerPath = 'monitor';
+
+@Controller(controllerPath)
+export class StatusMonitorController {
+  @Get()
+  @HttpCode(200)
+  root() {
+    const data = {
+      script: fs.readFileSync(
+        path.join(__dirname, '/public/javascripts/app.js'),
+      ),
+      style: fs.readFileSync(
+        path.join(__dirname, '/public/stylesheets/style.css'),
+      ),
+    };
+
+    const htmlTmpl = fs
+      .readFileSync(path.join(__dirname, '/public/index.html'))
+      .toString();
+
+    const render = Handlebars.compile(htmlTmpl);
+    return render(data);
+  }
+}

src/status.monitor.module.ts

@@ -1,45 +1,7 @@
-import { INestApplication } from '@nestjs/common';
-import { FastifyAdapter } from '@nestjs/core';
-import { loadPackage } from '@nestjs/common/utils/load-package.util';
-import * as fs from 'fs';
-import * as path from 'path';
-import Handlebars from 'handlebars';
+import { Module } from '@nestjs/common';
+import { StatusMonitorController } from './status.monitor.controller';
 
-export class StatusMonitorModule {
-  public static setup(endpointPath: string, app: INestApplication) {
-    const validatePath = (path): string =>
-      path.charAt(0) !== '/' ? '/' + path : path;
-
-    const httpServer = app.getHttpServer();
-    if (httpServer instanceof FastifyAdapter) {
-      return this.setupFastify(endpointPath, httpServer);
-    }
-
-    const finalPath = validatePath(endpointPath);
-
-    const data = {
-      script: fs.readFileSync(
-        path.join(__dirname, '/public/javascripts/app.js'),
-      ),
-      style: fs.readFileSync(
-        path.join(__dirname, '/public/stylesheets/style.css'),
-      ),
-    };
-
-    const htmlTmpl = fs
-      .readFileSync(path.join(__dirname, '/public/index.html'))
-      .toString();
-
-    const render = Handlebars.compile(htmlTmpl);
-
-    app.use(finalPath, (req, res) => res.send(render(data)));
-  }
-
-  private static setupFastify(path: string, httpServer: FastifyAdapter) {
-    httpServer.register(loadPackage('fastify-swagger', 'StatusMonitorModule'), {
-      exposeRoute: true,
-      routePrefix: path,
-      mode: 'static',
-    });
-  }
-}
+@Module({
+  controllers: [StatusMonitorController],
+})
+export class StatusMonitorModule {}