AFFiNE-Mirror/packages/backend/server/src/models/comment-attachment.ts

import { Injectable } from '@nestjs/common';
import { Prisma } from '@prisma/client';

import { BaseModel } from './base';

export type CreateCommentAttachmentInput =
  Prisma.CommentAttachmentUncheckedCreateInput;

/**
 * Comment Attachment Model
 */
@Injectable()
export class CommentAttachmentModel extends BaseModel {
  async upsert(input: CreateCommentAttachmentInput) {
    return await this.db.commentAttachment.upsert({
      where: {
        workspaceId_docId_key: {
          workspaceId: input.workspaceId,
          docId: input.docId,
          key: input.key,
        },
      },
      update: {
        name: input.name,
        mime: input.mime,
        size: input.size,
      },
      create: {
        workspaceId: input.workspaceId,
        docId: input.docId,
        key: input.key,
        name: input.name,
        mime: input.mime,
        size: input.size,
        createdBy: input.createdBy,
      },
    });
  }

  async delete(workspaceId: string, docId: string, key: string) {
    await this.db.commentAttachment.deleteMany({
      where: {
        workspaceId,
        docId,
        key,
      },
    });
    this.logger.log(`deleted comment attachment ${workspaceId}/${key}`);
  }

  async get(workspaceId: string, docId: string, key: string) {
    return await this.db.commentAttachment.findUnique({
      where: {
        workspaceId_docId_key: {
          workspaceId,
          docId,
          key,
        },
      },
    });
  }

  async list(workspaceId: string, docId?: string) {
    return await this.db.commentAttachment.findMany({
      where: {
        workspaceId,
        docId,
      },
    });
  }
}