certd/packages/ui/certd-server/src/controller/basic/file-controller.ts

import { Controller, Fields, Files, Get, Inject, Post, Provide, Query } from '@midwayjs/core';
import { BaseController, Constants, FileService, UploadFileItem, uploadTmpFileCacheKey } from '@certd/lib-server';
import send from 'koa-send';
import { nanoid } from 'nanoid';
import { cache } from '@certd/basic';
import { UploadFileInfo } from '@midwayjs/upload';

const imageExtSet = new Set(['.apng', '.avif', '.bmp', '.gif', '.ico', '.jpeg', '.jpg', '.png', '.svg', '.webp']);
const imageCacheSeconds = 3 * 24 * 60 * 60;

export function isImageFile(filePath: string) {
  return imageExtSet.has(filePath.substring(filePath.lastIndexOf('.')).toLowerCase());
}

/**
 */
@Provide()
@Controller('/api/basic/file')
export class FileController extends BaseController {
  @Inject()
  fileService: FileService;

  @Post('/upload', { description: Constants.per.authOnly })
  async upload(@Files() files: UploadFileInfo<string>[], @Fields() fields: any) {
    console.log('files', files, fields);
    const cacheKey = uploadTmpFileCacheKey + nanoid();
    const file = files[0];
    cache.set(
      cacheKey,
      {
        filename: file.filename,
        tmpFilePath: file.data,
      } as UploadFileItem,
      {
        ttl: 1000 * 60 * 60,
      }
    );
    return this.ok({
      key: cacheKey,
    });
  }

  @Get('/download', { description: Constants.per.guest })
  async download(@Query('key') key: string) {
    let userId: any = null;
    if (!key.startsWith('/public')) {
      userId = this.getUserId();
    }
    const filePath = this.fileService.getFile(key, userId);
    if (isImageFile(filePath)) {
      this.ctx.response.set('Cache-Control', `public,max-age=${imageCacheSeconds}`);
    } else {
      this.ctx.response.attachment(filePath);
    }
    await send(this.ctx, filePath);
  }
}