import { ApiProperty, OmitType, PartialType } from "@nestjs/swagger";
import {
  IsEnum,
  IsNotEmpty,
  IsNumber,
  IsOptional,
  IsString,
} from "class-validator";
import { ComplianceRiskStatus } from "../entities/compliance.types";

export class ComplianceRiskDto {
  @IsString()
  @IsOptional()
  @ApiProperty({ required: false })
  assignedTo?: string;

  @IsNotEmpty()
  @ApiProperty({ required: true })
  @IsNumber()
  projectId: number;

  @IsNotEmpty()
  @ApiProperty({ required: true })
  @IsNumber()
  complianceSubmissionId: number;

  @IsNotEmpty()
  @ApiProperty({ required: true })
  @IsNumber()
  ruleId: number;
}

export class UpdateComplianceRiskDto extends PartialType(
  OmitType(ComplianceRiskDto, ["projectId", "complianceSubmissionId"] as const),
) {
  @IsOptional()
  @ApiProperty({ required: false })
  @IsEnum(ComplianceRiskStatus, { each: true })
  status?: ComplianceRiskStatus;
}

export class ComplianceRiskCreateInput extends ComplianceRiskDto {
  @IsNotEmpty()
  @IsEnum(ComplianceRiskStatus, { each: true })
  @ApiProperty({ required: true })
  status: ComplianceRiskStatus;
}