cloudservices/src/class/Server.ts

/* eslint-disable no-useless-return */
import express from 'express';
import bodyParser from 'body-parser';
import helmet from 'helmet';
import fs from 'fs-extra';
import { Client, Collection, Route } from '.';
import { Security } from '../api';


export default class Server {
  public routes: Collection<Route>

  public client: Client;

  public security: Security;

  public app: express.Express;

  public options: { port: number }

  constructor(client: Client, options?: { port: number }) {
    this.options = options;
    this.routes = new Collection();
    this.client = client;
    this.security = new Security(this.client);
    this.app = express();
    this.connect();
    this.loadRoutes();
  }

  private async loadRoutes(): Promise<void> {
    const routes = await fs.readdir(`${__dirname}/../api/routes`);
    routes.forEach(async (routeFile) => {
      if (routeFile === 'index.js') return;
      try {
        // eslint-disable-next-line new-cap
        const route: Route = new (require(`${__dirname}/../api/routes/${routeFile}`).default)(this);
        if (route.conf.deprecated === true) {
          route.deprecated();
        } else if (route.conf.maintenance === true) {
          route.maintenance();
        } else {
          route.bind();
        }
        this.routes.set(route.conf.path, route);
        this.app.use(route.conf.path, route.router);
        this.client.signale.success(`Successfully loaded route ${route.conf.path}`);
      } catch (error) {
        this.client.util.handleError(error);
      }
    });
  }

  private connect(): void {
    this.app.set('trust proxy', 'loopback');
    this.app.use(helmet({
      hsts: false,
      hidePoweredBy: false,
      contentSecurityPolicy: {
        directives: {
          defaultSrc: ["'self'"],
        },
      },
    }));
    this.app.use(bodyParser.json());
    this.app.listen(this.options.port, () => {
      this.client.signale.success(`API Server listening on port ${this.options.port}`);
    });
  }
}