collect-system/apps/server/src/models/resource/processor/ImageProcessor.ts

import path from 'path';
import sharp from 'sharp';
import { FileMetadata, ImageMetadata, ResourceProcessor } from '../types';
import { Resource, ResourceStatus, db } from '@nice/common';
import { getUploadFilePath } from '@server/utils/file';
import { BaseProcessor } from './BaseProcessor';

export class ImageProcessor extends BaseProcessor {
  constructor() {
    super();
  }

  async process(resource: Resource): Promise<Resource> {
    const { url } = resource;
    const filepath = getUploadFilePath(url);
    const originMeta = resource.metadata as unknown as FileMetadata;
    if (!originMeta.mimeType?.startsWith('image/')) {
      this.logger.log(`Skipping non-image resource: ${resource.id}`);
      return resource;
    }
    try {
      const image = sharp(filepath);
      const metadata = await image.metadata();
      if (!metadata) {
        throw new Error(`Failed to get metadata for image: ${url}`);
      }
      // Create WebP compressed version
      const compressedDir = this.createOutputDir(filepath, 'compressed');
      const compressedPath = path.join(
        compressedDir,
        `${path.basename(filepath, path.extname(filepath))}.webp`,
      );
      await image
        .webp({
          quality: 80,
          lossless: false,
          effort: 5, // Range 0-6, higher means slower but better compression
        })
        .toFile(compressedPath);
      const imageMeta: ImageMetadata = {
        width: metadata.width || 0,
        height: metadata.height || 0,
        orientation: metadata.orientation,
        space: metadata.space,
        hasAlpha: metadata.hasAlpha,
      };
      const updatedResource = await db.resource.update({
        where: { id: resource.id },
        data: {
          metadata: {
            ...originMeta,
            ...imageMeta,
          },
        },
      });

      return updatedResource;
    } catch (error: any) {
      throw new Error(`Failed to process image: ${error.message}`);
    }
  }
}