bliss/api/src/modules/root/root.service.ts

import {
  Injectable,
  InternalServerErrorException,
  Logger,
  NotFoundException,
} from "@nestjs/common";
import { Request, Response } from "express";
import fastFolderSize from "fast-folder-size";
import { createReadStream, existsSync } from "fs";
import { stat } from "fs/promises";
import { thumbnailDir, uploadDir } from "lib/constants";
import { CustomSession } from "lib/types";
import {
  formatBytes,
  formatDate,
  generateRandomHexColor,
  lookUp,
} from "lib/utils";
import { PrismaService } from "modules/prisma/prisma.service";
import { join } from "path";
import { promisify } from "util";

@Injectable()
export class RootService {
  private logger = new Logger(RootService.name);
  constructor(private readonly prismaService: PrismaService) {}

  downloadFile(filename: string, res: Response) {
    if (!filename) {
      throw new NotFoundException();
    }
    const fn = decodeURIComponent(filename);
    return stat(join(uploadDir, fn))
      .then((stats) => {
        res.setHeader("Accept-Ranges", "bytes");
        res.setHeader("Content-Length", stats.size);
        res.setHeader("Content-Disposition", `'attachment'; filename=${fn}`);
        createReadStream(join(uploadDir, fn)).pipe(res);
      })
      .catch((err) => {
        if (err === "ENOENT") {
          throw new NotFoundException();
        } else {
          this.logger.error(err);
          throw new InternalServerErrorException();
        }
      });
  }

  async getFile(slug: string, req: Request) {
    if (!slug) {
      throw new NotFoundException();
    }
    const file = await this.prismaService.file.findUnique({
      where: { slug: slug.split(".").shift() },
      include: { user: { include: { embed_settings: true } } },
    });

    if (!file) {
      throw new NotFoundException();
    }

    const protocol = req.headers["x-forwarded-proto"] || req.protocol;
    const ext = file.filename.split(".").pop();

    let oembed: string;
    let url: string;
    let baseUrl: string;
    let vw = file.views;

    if (
      !existsSync(join(uploadDir, `${file.slug}.${ext}`)) &&
      process.env.UPLOADER === "local"
    ) {
      throw new NotFoundException();
    }

    if ((req.session as CustomSession).userId !== file.userId) {
      const { views } = await this.prismaService.file.update({
        where: { slug },
        data: { views: file.views + 1 },
      });
      vw = views;
    }

    const isVideo = lookUp(file.filename).includes("video");
    const isImage = lookUp(file.filename).includes("image");
    const isAudio = lookUp(file.filename).includes("audio");
    const cannotDisplay = !isImage && !isVideo && !isAudio;
    const timezone = new Date().getTimezoneOffset() / 60;

    if (process.env.UPLOADER === "s3") {
      baseUrl =
        process.env.CDN_URL ??
        "https://" + process.env.BUCKET_NAME + process.env.S3_ENDPOINT;
      oembed = `${baseUrl}/${slug}.json`;
      url = `${baseUrl}/${slug}.${ext}`;
    } else {
      baseUrl = `${protocol}://${req.headers.host}`;
      oembed = `${baseUrl}/${slug}.json`;
      url = `${baseUrl}/${slug}.${ext}`;
    }

    const {
      user: { embed_settings },
    } = file;

    return {
      oembed,
      url,
      title: embed_settings.enabled ? embed_settings?.title : null,
      description: embed_settings.enabled ? embed_settings?.description : null,
      color: embed_settings?.color ?? generateRandomHexColor(),
      ogType: isVideo ? "video.other" : isImage ? "image" : "website",
      urlType: isVideo ? "video" : isAudio ? "audio" : "image",
      mimetype: lookUp(file.filename),
      filename: file.filename,
      slug: file.slug + "." + file.filename.split(".").pop(),
      size: formatBytes(file.size),
      username: file.user.username,
      embed_enabled: embed_settings?.enabled,
      views: vw,
      timestamp: formatDate(file.createdAt) + ` (UTC${timezone})`,
      isVideo,
      isImage,
      isAudio,
      cannotDisplay,
      id: file.id,
    };
  }

  async getStatistics() {
    try {
      const fastSize = promisify(fastFolderSize);
      const [files, uploads] = await Promise.all([
        this.prismaService.file.count(),
        fastSize(uploadDir),
      ]);

      return {
        files,
        size: formatBytes(uploads),
      };
    } catch (error) {
      this.logger.error(error.message);
      return {
        files: "N/A",
        size: "N/A",
      };
    }
  }
}