import {
  ObjectIdToString,
  StringToObjectId,
} from '#root/shared/constants/transformerConstants.js';
import {
  EnrollmentRole,
  EnrollmentStatus,
  ID,
  IEnrollment,
} from '#root/shared/interfaces/models.js';
import { Expose, Transform, Type } from 'class-transformer';
import { ObjectId } from 'mongodb';
import { Progress } from './Progress.js';
import { IsArray, IsOptional, IsString, ValidateNested } from 'class-validator';
import { JSONSchema } from 'class-validator-jsonschema';

@Expose()
export class Enrollment implements IEnrollment {
  @Expose({ toClassOnly: true })
  @Transform(ObjectIdToString.transformer, { toPlainOnly: true })
  @Transform(StringToObjectId.transformer, { toClassOnly: true })
  _id?: ID;

  @Expose()
  @Transform(ObjectIdToString.transformer, { toPlainOnly: true })
  @Transform(StringToObjectId.transformer, { toClassOnly: true })
  userId: ID;

  @Expose()
  @Transform(ObjectIdToString.transformer, { toPlainOnly: true })
  @Transform(StringToObjectId.transformer, { toClassOnly: true })
  courseId: ID;

  @Expose()
  @Transform(ObjectIdToString.transformer, { toPlainOnly: true })
  @Transform(StringToObjectId.transformer, { toClassOnly: true })
  courseVersionId: ID;

  @Expose()
  role: EnrollmentRole;

  @Expose()
  status: EnrollmentStatus;

  @Expose()
  @Type(() => Date)
  enrollmentDate: Date;

  @IsString()
  @Expose()
  percentCompleted: number;

  @IsString()
  @Expose()
  @IsOptional()
  completedItemsCount?: number;

  @IsOptional()
  cohortId?: ID;

  @Expose()
  @IsOptional()
  @Type(() => Date)
  ethicsConsentSignedAt?: Date;

  @IsString()
  @Expose()
  @IsOptional()
  ethicsConsentSignature?: string;

  @IsString()
  @Expose()
  @IsOptional()
  ethicsConsentVersion?: string;

  @Expose()
  @IsOptional()
  ethicsAdditionalImageConsent?: boolean;

  constructor(userId?: string, courseId?: string, courseVersionId?: string, cohortId?: ID) {
    if (userId && courseId && courseVersionId) {
      this.userId = new ObjectId(userId);
      this.courseId = new ObjectId(courseId);
      this.courseVersionId = new ObjectId(courseVersionId);
      this.status = 'ACTIVE';
      this.enrollmentDate = new Date();
      this.percentCompleted = 0;
      this.completedItemsCount = 0;
    }
    if(cohortId){
      this.cohortId = cohortId;
    }
  }
}

@Expose({ toPlainOnly: true })
export class EnrollUserResponse {
  @ValidateNested()
  @Expose()
  @JSONSchema({
    type: 'object',
    $ref: '#/components/schemas/Enrollment'
  })
  @Type(() => Enrollment)
  enrollment: Enrollment;

  @Expose()
  @Type(() => Progress)
  progress: Progress;

  @Expose()
  @Type(() => String)
  role: EnrollmentRole;

  constructor(
    enrollment: Enrollment,
    progress: Progress,
    role: EnrollmentRole,
  ) {
    this.enrollment = enrollment;
    this.progress = progress;
    this.role = role;
  }
}
export class EnrolledUserResponse {
  @ValidateNested()
  @Expose()
  @Type(() => String)
  role: EnrollmentRole;

  @ValidateNested()
  @Expose()
  @Type(() => String)
  status: EnrollmentStatus;

  @Expose()
  @Type(() => Date)
  enrollmentDate: Date;

  constructor(
    role: EnrollmentRole,
    status: EnrollmentStatus,
    enrollmentDate: Date,
  ) {
    this.role = role;
    this.status = status;
    this.enrollmentDate = enrollmentDate;
  }
}