import { PrismaService } from "../common/prisma/prisma.service";
import { Injectable } from "@nestjs/common";
import { ComplianceEvidence } from "./entities/evidence.entity";

@Injectable()
export class EvidenceService {
  constructor(private readonly prismaService: PrismaService) {}

  /**
   * Create a new compliance evidence
   * @param {string} name evidence name
   * @param {string} description evidence description
   * @param {number} projectId project id
   * @param {string} fileName file name
   * @param {string} bucketName bucket name
   * @param {string} uploadedBy uploaded by
   * @returns {Promise<ComplianceEvidence>} created evidence
   */
  async createComplianceEvidence(
    name: string,
    description: string,
    projectId: number,
    fileName: string,
    bucketName: string,
    uploadedBy: string,
  ): Promise<ComplianceEvidence> {
    return new ComplianceEvidence(
      await this.prismaService.complianceEvidence.create({
        data: {
          name: name,
          description: description,
          fileName: fileName,
          bucketName: bucketName,
          uploadedBy: uploadedBy,
        },
      }),
    );
  }

  /**
   * Get compliance evidence by id
   * @param {number} id evidence id
   * @returns {Promise<ComplianceEvidence>} evidence
   */
  async findOne(id: number): Promise<ComplianceEvidence> {
    return new ComplianceEvidence(
      await this.prismaService.complianceEvidence.findUnique({
        where: { id },
        include: {
          riskSubmissions: {
            include: {
              risk: {
                include: {
                  project: true,
                },
              },
            },
          },
        },
      }),
    );
  }

  /**
   * Delete compliance evidence
   * @param {number} id evidence id
   * @returns {Promise<ComplianceEvidence>} deleted evidence
   */
  async deleteComplianceEvidence(id: number): Promise<ComplianceEvidence> {
    return new ComplianceEvidence(
      await this.prismaService.complianceEvidence.delete({
        where: { id },
      }),
    );
  }
}