Static module: extract logic from controller to service

2026-07-09 04:47:41 +00:00 · 2022-12-05 20:38:29 +02:00
parent b4ad786d62
commit 9f85c11d2e
5 changed files with 99 additions and 110 deletions
@@ -0,0 +1,88 @@
+import {
+  Injectable,
+  NotFoundException,
+  UnauthorizedException,
+} from '@nestjs/common';
+import { readdir, stat } from 'fs/promises';
+import { Stats } from 'fs';
+import { join } from 'path';
+
+@Injectable()
+export class StaticService {
+  async provideDirIndex(relPath: string): Promise<string> {
+    if (relPath.indexOf('..') > -1) {
+      throw new UnauthorizedException(
+        'You are not authorized to visit ' + relPath,
+      );
+    }
+
+    const path = join(__dirname, process.env.STATIC_SERVE_FOLDER, relPath);
+
+    const stats: Stats | false = await stat(path).catch(() => false);
+
+    if (stats === false) {
+      throw new NotFoundException('Path not found: ' + relPath);
+    }
+
+    if (!stats.isDirectory()) {
+      return null;
+    }
+
+    return this.readDirectory(path, relPath);
+  }
+
+  private async readDirectory(path: string, relPath: string) {
+    let files = await readdir(path);
+    let longestFileLength = 0;
+
+    files = files.filter((fileName: string) => {
+      const fileIncluded = fileName[0] !== '.';
+      longestFileLength =
+        fileIncluded && longestFileLength < fileName.length
+          ? fileName.length
+          : longestFileLength;
+
+      return fileIncluded;
+    });
+
+    longestFileLength += 10;
+
+    let result: string;
+    const resultHtmlStart = `
+    <html lang="en">
+        <head><title>Index of ${relPath} </title></head>
+        <body>
+          <h1>Index of ${relPath}</h1>
+          <hr><pre>`;
+
+    result = `<a href="../">../</a>`;
+
+    for (const file of files) {
+      const stats: Stats | false = await stat(join(path, file)).catch(
+        () => false,
+      );
+
+      if (stats === false) {
+        continue;
+      }
+      const arrDate: Array<string> = stats.birthtime.toUTCString().split(' ');
+      const strDate: string =
+        ' '.repeat(longestFileLength - file.length) +
+        `${arrDate[1]}-${arrDate[2]}-${arrDate[3]} ${arrDate[4]}`;
+
+      if (stats.isDirectory()) {
+        const strSize = ' '.repeat(19) + '-';
+        result += '\n' + `<a href="${file}/">${file}/</a>${strDate + strSize}`;
+        continue;
+      }
+      let strSize = stats.size.toString();
+      strSize = ' '.repeat(20 - strSize.length) + strSize;
+
+      result += '\n' + `<a href="${file}">${file}</a> ${strDate + strSize}`;
+    }
+
+    const resultHtmlEnd = `</pre><hr></body></html>`;
+
+    return resultHtmlStart + result + resultHtmlEnd;
+  }
+}