myDrive/backend/services/chunk-service/utils/tempCreateVideoThumbnailFS.ts

import mongoose from "../../../db/connections/mongoose";
import crypto from "crypto";
import Thumbnail from "../../../models/thumbnail-model";
import sharp from "sharp";
import { FileInterface } from "../../../models/file-model";
import { UserInterface } from "../../../models/user-model";
import fs from "fs";
import uuid from "uuid";
import env from "../../../enviroment/env";
import { ObjectId } from "mongodb";
import File from "../../../models/file-model";
import ffmpeg from "fluent-ffmpeg";

const tempCreateVideoThumbnailFS = (
  file: FileInterface,
  filename: string,
  user: UserInterface
) => {
  return new Promise<FileInterface>(async (resolve, reject) => {
    const password = user.getEncryptionKey();

    let CIPHER_KEY = crypto.createHash("sha256").update(password!).digest();

    const thumbnailFilename = uuid.v4();

    const readStream = fs.createReadStream(file.metadata.filePath!);
    console.log("env", env);
    const writeStream = fs.createWriteStream(
      env.fsDirectory + thumbnailFilename
    );

    const tempDirectory = env.fsDirectory + "temp/" + thumbnailFilename;
    const tempWriteStream = fs.createWriteStream(tempDirectory);
    const decipher = crypto.createDecipheriv(
      "aes256",
      CIPHER_KEY,
      file.metadata.IV
    );

    const thumbnailIV = crypto.randomBytes(16);

    const thumbnailCipher = crypto.createCipheriv(
      "aes256",
      CIPHER_KEY,
      thumbnailIV
    );

    const decryptedReadStream = readStream.pipe(decipher);

    decryptedReadStream.pipe(tempWriteStream, { end: true });

    ffmpeg(tempDirectory, {
      timeout: 60,
    })
      .seek(1)
      .format("image2pipe")
      .outputOptions([
        "-f image2pipe",
        "-vframes 1",
        "-vf scale='if(gt(iw,ih),600,-1):if(gt(ih,iw),300,-1)'",
      ])
      .on("start", (command) => {
        /**
         * log
         */
      })
      .on("end", async () => {
        console.log("end");
        const thumbnailModel = new Thumbnail({
          name: filename,
          owner: user._id,
          IV: thumbnailIV,
          path: env.fsDirectory + thumbnailFilename,
        });

        await thumbnailModel.save();
        if (!file._id) {
          return reject();
        }
        const updateFileResponse = await File.updateOne(
          { _id: new ObjectId(file._id), "metadata.owner": user._id },
          {
            $set: {
              "metadata.hasThumbnail": true,
              "metadata.thumbnailID": thumbnailModel._id,
            },
          }
        );
        if (updateFileResponse.modifiedCount === 0) {
          return reject();
        }

        const updatedFile = await File.findById({
          _id: new ObjectId(file._id),
          "metadata.owner": user._id,
        });

        if (!updatedFile) return reject();

        fs.unlink(tempDirectory, (err) => {
          resolve(updatedFile?.toObject());
        });
      })
      .on("error", (err, _, stderr) => {
        console.log("error", err, stderr);
        resolve(file);
        /**
         * log
         */
      })
      .pipe(thumbnailCipher)
      .pipe(writeStream, { end: true });

    // ffmpeg()
    //   .input(readStream.pipe(decipher))
    //   .seekInput("00:00:01.00")
    //   .outputFormat("image2")
    //   .pipe(writeStream, { end: true })
    //   .on("end", async () => {
    //     const thumbnailModel = new Thumbnail({
    //       name: filename,
    //       owner: user._id,
    //       IV: thumbnailIV,
    //       path: env.fsDirectory + thumbnailFilename,
    //     });

    //     await thumbnailModel.save();
    //     resolve(file);
    //   })
    //   .on("error", (e) => {
    //     console.log("error", e);
    //     reject(e);
    //   });

    // ffmpeg()
    //   .input(readStream.pipe(decipher))
    //   .size("300x?")
    //   .outputOptions("-frames:v 1")
    //   .on("end", async () => {
    //     const thumbnailModel = new Thumbnail({
    //       name: filename,
    //       owner: user._id,
    //       IV: thumbnailIV,
    //       path: env.fsDirectory + thumbnailFilename,
    //     });

    //     await thumbnailModel.save();
    //     resolve(file);
    //   })
    //   .output(writeStream);
  });
};

export default tempCreateVideoThumbnailFS;