FastGPT/packages/service/common/file/upload/multer.ts

import type { NextApiRequest, NextApiResponse } from 'next';
import { customAlphabet } from 'nanoid';
import multer from 'multer';
import path from 'path';
import { BucketNameEnum } from '@fastgpt/global/common/file/constants';
import fs from 'fs';

const nanoid = customAlphabet('1234567890abcdef', 12);

type FileType = {
  fieldname: string;
  originalname: string;
  encoding: string;
  mimetype: string;
  filename: string;
  path: string;
  size: number;
};

export function getUploadModel({ maxSize = 500 }: { maxSize?: number }) {
  maxSize *= 1024 * 1024;
  class UploadModel {
    uploader = multer({
      limits: {
        fieldSize: maxSize
      },
      preservePath: true,
      storage: multer.diskStorage({
        filename: (_req, file, cb) => {
          const { ext } = path.parse(decodeURIComponent(file.originalname));
          cb(null, nanoid() + ext);
        }
      })
    }).any();

    async doUpload<T = Record<string, any>>(req: NextApiRequest, res: NextApiResponse) {
      return new Promise<{
        files: FileType[];
        metadata: T;
        bucketName?: `${BucketNameEnum}`;
      }>((resolve, reject) => {
        // @ts-ignore
        this.uploader(req, res, (error) => {
          if (error) {
            return reject(error);
          }

          resolve({
            ...req.body,
            files:
              // @ts-ignore
              req.files?.map((file) => ({
                ...file,
                originalname: decodeURIComponent(file.originalname)
              })) || [],
            metadata: (() => {
              if (!req.body?.metadata) return {};
              try {
                return JSON.parse(req.body.metadata);
              } catch (error) {
                console.log(error);
                return {};
              }
            })()
          });
        });
      });
    }
  }

  return new UploadModel();
}

export const removeFilesByPaths = (paths: string[]) => {
  paths.forEach((path) => {
    fs.unlink(path, (err) => {
      if (err) {
        console.error(err);
      }
    });
  });
};