Merge pull request #110 from waifuvault/landing-info

Add Files Served and Total Size to landing page

src/controllers/views/HomeView.ts

@@ -1,14 +1,23 @@
 import { Get, Hidden, View } from "@tsed/schema";
-import { Controller } from "@tsed/di";
+import { Controller, Inject } from "@tsed/di";
 import { Req, Res } from "@tsed/common";
+import { FileService } from "../../services/FileService.js";
+import { ObjectUtils } from "../../utils/Utils.js";
 
 @Controller("/")
 @Hidden()
 export class HomeView {
+    public constructor(@Inject() private fileService: FileService) {}
+
     @Get()
     @View("index.ejs")
-    public showRoot(): unknown {
-        return null;
+    public async showRoot(): Promise<unknown> {
+        const records = await this.fileService.getRecordCount();
+        const size = await this.fileService.getRecordSize();
+        return {
+            recordCount: records.toLocaleString(),
+            recordSize: ObjectUtils.sizeToHuman(size),
+        };
     }
 
     @Get("/login")

src/public/index.ejs

@@ -18,6 +18,8 @@
         <div class="p-5 mb-4 bg-body-tertiary rounded-3 border border-primary-subtle position-relative">
             <div class="container-fluid py-5 ">
                 <h1 class="display-5 fw-bold">No nonsense temporary file hosting</h1>
+                <h2 class="fw-bold">Serving <%- recordCount -%> public files totalling <%- recordSize -%>
+                </h2>
                 <p class="col-md-8 fs-4">Inspired by
                     <a href="https://0x0.st" target="_blank">https://0x0.st</a>, WaifuVault is a temporary file hosting service that allows for file uploads that are hosted for a set amount of time.
                 </p>

src/services/FileService.ts

@@ -228,6 +228,15 @@ export class FileService {
         return deleted;
     }
 
+    public getRecordCount(): Promise<number> {
+        return this.repo.getRecordCount();
+    }
+
+    public async getRecordSize(): Promise<number> {
+        const entries = await this.repo.getAllEntries();
+        return entries.reduce((acc, currentValue) => acc + currentValue.fileSize, 0);
+    }
+
     private async getFileHash(resourcePath: string): Promise<string> {
         const fileBuffer = await fs.readFile(resourcePath);
         const hashSum = crypto.createHash("md5");

src/utils/Utils.ts

@@ -13,6 +13,22 @@ export class ObjectUtils {
         return matches && matches[0] ? parseInt(matches[0]) : 0;
     }
 
+    public static sizeToHuman(value: number): string {
+        const sizeKB = Math.floor(value / 1024);
+        const sizeMB = Math.floor(sizeKB / 1024);
+        const sizeGB = Math.floor(sizeMB / 1024);
+        if (value < 1024) {
+            return `${value} B`;
+        }
+        if (sizeKB < 1024) {
+            return `${sizeKB} KB`;
+        }
+        if (sizeMB < 1024) {
+            return `${sizeMB} MB`;
+        }
+        return `${sizeGB} GB`;
+    }
+
     public static timeToHuman(value: number, timeUnit: TIME_UNIT = TIME_UNIT.milliseconds): string {
         let seconds: number;
         if (timeUnit === TIME_UNIT.milliseconds) {