ghostfolio/apps/api/src/app/frontend.middleware.ts

import * as fs from 'fs';
import * as path from 'path';

import { ConfigurationService } from '@ghostfolio/api/services/configuration.service';
import { DEFAULT_LANGUAGE_CODE } from '@ghostfolio/common/config';
import { Injectable, NestMiddleware } from '@nestjs/common';
import { ConfigService } from '@nestjs/config';
import { NextFunction, Request, Response } from 'express';

@Injectable()
export class FrontendMiddleware implements NestMiddleware {
  public indexHtmlDe = '';
  public indexHtmlEn = '';
  public indexHtmlEs = '';
  public indexHtmlIt = '';
  public indexHtmlNl = '';
  public isProduction: boolean;

  public constructor(
    private readonly configService: ConfigService,
    private readonly configurationService: ConfigurationService
  ) {
    const NODE_ENV =
      this.configService.get<'development' | 'production'>('NODE_ENV') ??
      'development';

    this.isProduction = NODE_ENV === 'production';

    try {
      this.indexHtmlDe = fs.readFileSync(
        this.getPathOfIndexHtmlFile('de'),
        'utf8'
      );
      this.indexHtmlEn = fs.readFileSync(
        this.getPathOfIndexHtmlFile(DEFAULT_LANGUAGE_CODE),
        'utf8'
      );
      this.indexHtmlEs = fs.readFileSync(
        this.getPathOfIndexHtmlFile('es'),
        'utf8'
      );
      this.indexHtmlIt = fs.readFileSync(
        this.getPathOfIndexHtmlFile('it'),
        'utf8'
      );
      this.indexHtmlNl = fs.readFileSync(
        this.getPathOfIndexHtmlFile('nl'),
        'utf8'
      );
    } catch {}
  }

  public use(req: Request, res: Response, next: NextFunction) {
    let featureGraphicPath = 'assets/cover.png';

    if (
      req.path === '/en/blog/2022/08/500-stars-on-github' ||
      req.path === '/en/blog/2022/08/500-stars-on-github/'
    ) {
      featureGraphicPath = 'assets/images/blog/500-stars-on-github.jpg';
    }

    if (
      req.path.startsWith('/api/') ||
      this.isFileRequest(req.url) ||
      !this.isProduction
    ) {
      // Skip
      next();
    } else if (req.path === '/de' || req.path.startsWith('/de/')) {
      res.send(
        this.interpolate(this.indexHtmlDe, {
          featureGraphicPath,
          languageCode: 'de',
          path: req.path,
          rootUrl: this.configurationService.get('ROOT_URL')
        })
      );
    } else if (req.path === '/es' || req.path.startsWith('/es/')) {
      res.send(
        this.interpolate(this.indexHtmlIt, {
          featureGraphicPath,
          languageCode: 'es',
          path: req.path,
          rootUrl: this.configurationService.get('ROOT_URL')
        })
      );
    } else if (req.path === '/it' || req.path.startsWith('/it/')) {
      res.send(
        this.interpolate(this.indexHtmlIt, {
          featureGraphicPath,
          languageCode: 'it',
          path: req.path,
          rootUrl: this.configurationService.get('ROOT_URL')
        })
      );
    } else if (req.path === '/nl' || req.path.startsWith('/nl/')) {
      res.send(
        this.interpolate(this.indexHtmlNl, {
          featureGraphicPath,
          languageCode: 'nl',
          path: req.path,
          rootUrl: this.configurationService.get('ROOT_URL')
        })
      );
    } else {
      res.send(
        this.interpolate(this.indexHtmlEn, {
          featureGraphicPath,
          languageCode: DEFAULT_LANGUAGE_CODE,
          path: req.path,
          rootUrl: this.configurationService.get('ROOT_URL')
        })
      );
    }
  }

  private getPathOfIndexHtmlFile(aLocale: string) {
    return path.join(__dirname, '..', 'client', aLocale, 'index.html');
  }

  private interpolate(template: string, context: any) {
    return template.replace(/[$]{([^}]+)}/g, (_, objectPath) => {
      const properties = objectPath.split('.');
      return properties.reduce(
        (previous, current) => previous?.[current],
        context
      );
    });
  }

  private isFileRequest(filename: string) {
    if (filename === '/assets/LICENSE') {
      return true;
    } else if (filename.includes('auth/ey')) {
      return false;
    }

    return filename.split('.').pop() !== filename;
  }
}