import {ID, IProgress, ItemType} from '#root/shared/interfaces/models.js';
import {Expose, Type} from 'class-transformer';
import {
  IsNotEmpty,
  IsString,
  IsMongoId,
  IsOptional,
  ValidateIf,
  IsBoolean,
  IsNumber,
  IsEnum,
  ValidateNested,
  IsEmail,
  Min,
  Max,
  IsArray,
} from 'class-validator';
import {JSONSchema} from 'class-validator-jsonschema';
import {WatchTime} from '../transformers/WatchTime.js';
import {UserQuizMetrics} from '#root/modules/quizzes/classes/index.js';

export class GetUserProgressParams {
  @JSONSchema({
    description: 'Course ID to retrieve progress for',
    type: 'string',
  })
  @IsNotEmpty()
  @IsString()
  @IsMongoId()
  courseId: string;

  @JSONSchema({
    description: 'Course version ID to retrieve progress for',
    type: 'string',
  })
  @IsNotEmpty()
  @IsString()
  @IsMongoId()
  versionId: string;
}

export class GetLeaderboardQuery {
  @JSONSchema({
    description: 'Page number (starts from 1)',
    default: 1,
    minimum: 1,
    type: 'number',
  })
  @IsOptional()
  page?: number = 1;

  @JSONSchema({
    description: 'Number of records per page',
    default: 10,
    minimum: 1,
    type: 'number',
  })
  @IsOptional()
  limit?: number = 10;

  @JSONSchema({
    description: 'Cohort ID to filter leaderboard by',
    type: 'string',
  })
  @IsOptional()
  @IsString()
  cohortId?: string;
}

export class LeaderboardNoAuthResponse {
  @JSONSchema({
    description: 'User ID',
    type: 'string',
  })
  @IsString()
  userId!: string;

  @JSONSchema({
    description: 'User full name',
    type: 'string',
  })
  @IsString()
  userName!: string;

  @JSONSchema({
    description: 'User email address',
    type: 'string',
    format: 'email',
  })
  @IsEmail()
  email!: string;

  @JSONSchema({
    description: 'Completion percentage of the course',
    type: 'number',
    minimum: 0,
    maximum: 100,
  })
  @IsNumber()
  @Min(0)
  @Max(100)
  completionPercentage!: number;

  @JSONSchema({
    description: 'Completion timestamp (null if not completed)',
    oneOf: [{type: 'string', format: 'date-time'}, {type: 'null'}],
  })
  @IsOptional()
  completedAt!: Date | string | null;

  @JSONSchema({
    description: 'Rank in leaderboard',
    type: 'number',
    minimum: 1,
  })
  @IsNumber()
  @Min(1)
  rank!: number;
}

export class GetLeaderboardResponse {
  @JSONSchema({
    description: 'Course name',
    type: 'string',
  })
  @IsString()
  course!: string;

  @JSONSchema({
    description: 'Course version',
    type: 'string',
  })
  @IsString()
  version!: string;

  @JSONSchema({
    description: 'Leaderboard data',
    type: 'array',
    items: {$ref: '#/components/schemas/LeaderboardNoAuthResponse'},
  })
  @IsArray()
  @ValidateNested({each: true})
  @Type(() => LeaderboardNoAuthResponse)
  data!: LeaderboardNoAuthResponse[];
}

export class StartItemBody {
  @JSONSchema({
    description: 'ID of the course item to start',
    example: '60d5ec49b3f1c8e4a8f8b8c4',
    type: 'string',
  })
  @IsNotEmpty()
  @IsString()
  @IsMongoId()
  itemId: string;

  @JSONSchema({
    description: 'ID of the module containing the item',
    example: '60d5ec49b3f1c8e4a8f8b8c5',
    type: 'string',
  })
  @IsNotEmpty()
  @IsString()
  @IsMongoId()
  moduleId: string;

  @JSONSchema({
    description: 'ID of the section containing the item',
    example: '60d5ec49b3f1c8e4a8f8b8c6',
    type: 'string',
  })
  @IsNotEmpty()
  @IsString()
  @IsMongoId()
  sectionId: string;
  
  @IsOptional()
  @IsString()
  cohortId?: string;
}

export class StartItemParams {
  @JSONSchema({
    description: 'Course ID to track progress for',
    type: 'string',
  })
  @IsNotEmpty()
  @IsString()
  @IsMongoId()
  courseId: string;

  @JSONSchema({
    description: 'Course version ID to track progress for',
    type: 'string',
  })
  @IsNotEmpty()
  @IsString()
  @IsMongoId()
  versionId: string;
}

export class StartItemResponse {
  @Expose()
  @JSONSchema({
    description: 'Watch item ID for tracking progress',
    example: '60d5ec49b3f1c8e4a8f8b8c7',
    type: 'string',
  })
  @IsNotEmpty()
  @IsString()
  @IsMongoId()
  watchItemId: string;

  constructor(data: Partial<StartItemResponse>) {
    Object.assign(this, data);
  }
}

export class StopItemParams {
  @JSONSchema({
    description: 'Course ID to stop tracking progress for',
    type: 'string',
  })
  @IsNotEmpty()
  @IsString()
  @IsMongoId()
  courseId: string;

  @JSONSchema({
    description: 'Course version ID to stop tracking progress for',
    type: 'string',
  })
  @IsNotEmpty()
  @IsString()
  @IsMongoId()
  versionId: string;
}

export class StopItemBody {
  @JSONSchema({
    description: 'Watch item ID used for tracking progress',
    example: '60d5ec49b3f1c8e4a8f8b8c7',
    type: 'string',
  })
  @IsNotEmpty()
  @IsString()
  @IsMongoId()
  watchItemId: string;

  @JSONSchema({
    description: 'ID of the course item to stop tracking',
    example: '60d5ec49b3f1c8e4a8f8b8c4',
    type: 'string',
  })
  @IsNotEmpty()
  @IsString()
  @IsMongoId()
  itemId: string;

  @JSONSchema({
    description: 'ID of the section containing the item',
    example: '60d5ec49b3f1c8e4a8f8b8c6',
    type: 'string',
  })
  @IsNotEmpty()
  @IsString()
  @IsMongoId()
  sectionId: string;

  @JSONSchema({
    description: 'ID of the module containing the item',
    example: '60d5ec49b3f1c8e4a8f8b8c5',
    type: 'string',
  })
  @IsNotEmpty()
  @IsString()
  @IsMongoId()
  moduleId: string;

  @JSONSchema({
    description: 'Attempt ID for quiz tracking',
    example: '60d5ec49b3f1c8e4a8f8b8c7',
    type: 'string',
  })
  @IsOptional()
  @IsString()
  @IsMongoId()
  attemptId?: string;

  @IsOptional()
  @IsBoolean()
  @JSONSchema({
    description: 'Whether this attempt is skipped',
    type: 'boolean',
    example: true,
  })
  isSkipped?: boolean;

  @IsOptional()
  @IsBoolean()
  @JSONSchema({
    description: 'Whether seek forward is allowed',
    type: 'boolean',
    example: true,
  })
  seekForwardEnabled?: boolean;

  @IsOptional()
  @JSONSchema({
    description: 'Next item ID to proceed to after stopping',
    type: 'string',
    example: "60d5ec49b3f1c8e4a8f8b8c8",
  })
  nextItemId?: string;

  @IsOptional()
  @IsString()
  @JSONSchema({
    description: 'Cohort to which the user belongs',
    example: 'Cohort A',
    type: 'string',
  })
  cohortId?: string;
}

export class ItemIdparams {
  @JSONSchema({
    description: 'Gives as ItemId',
  })
  @IsString()
  itemId: string;
}

export class UpdateProgressBody {
  @JSONSchema({
    description: 'ID of the module to update progress for',
    example: '60d5ec49b3f1c8e4a8f8b8c5',
    type: 'string',
  })
  @IsNotEmpty()
  @IsString()
  @IsMongoId()
  moduleId: string;

  @JSONSchema({
    description: 'ID of the section to update progress for',
    example: '60d5ec49b3f1c8e4a8f8b8c6',
    type: 'string',
  })
  @IsNotEmpty()
  @IsString()
  @IsMongoId()
  sectionId: string;

  @JSONSchema({
    description: 'ID of the item to update progress for',
    example: '60d5ec49b3f1c8e4a8f8b8c4',
    type: 'string',
  })
  @IsNotEmpty()
  @IsString()
  @IsMongoId()
  itemId: string;

  @JSONSchema({
    description: 'Watch item ID used for tracking progress',
    example: '60d5ec49b3f1c8e4a8f8b8c7',
    type: 'string',
  })
  @IsOptional()
  @IsString()
  @IsMongoId()
  watchItemId?: string;

  @JSONSchema({
    description: 'ID of the attempt for quiz',
    example: '60d5ec49b3f1c8e4a8f8b8c6',
    type: 'string',
  })
  @IsOptional()
  @IsString()
  @IsMongoId()
  attemptId?: string;
}

export class UpdateProgressParams {
  @JSONSchema({
    description: 'Course ID to update progress for',
    type: 'string',
  })
  @IsNotEmpty()
  @IsString()
  @IsMongoId()
  courseId: string;

  @JSONSchema({
    description: 'Course version ID to update progress for',
    type: 'string',
  })
  @IsNotEmpty()
  @IsString()
  @IsMongoId()
  versionId: string;
}

export class ResetCourseProgressParams {
  @JSONSchema({
    description: 'User ID to reset progress for',
    type: 'string',
  })
  @IsNotEmpty()
  @IsString()
  @IsMongoId()
  userId: string;

  @JSONSchema({
    description: 'Course ID to reset progress for',
    type: 'string',
  })
  @IsNotEmpty()
  @IsString()
  @IsMongoId()
  courseId: string;

  @JSONSchema({
    description: 'Course version ID to reset progress for',
    type: 'string',
  })
  @IsNotEmpty()
  @IsString()
  @IsMongoId()
  versionId: string;
}

export class ResetCourseProgressBody {
  @JSONSchema({
    description: 'Optional module ID to reset progress to',
    example: '60d5ec49b3f1c8e4a8f8b8c5',
    type: 'string',
    nullable: true,
  })
  @IsOptional()
  @IsString()
  @IsMongoId()
  moduleId?: string | null;

  @JSONSchema({
    description: 'Optional section ID to reset progress to',
    example: '60d5ec49b3f1c8e4a8f8b8c6',
    type: 'string',
    nullable: true,
  })
  @IsOptional()
  @IsString()
  @IsMongoId()
  sectionId?: string | null;

  @JSONSchema({
    description: 'Optional item ID to reset progress to',
    example: '60d5ec49b3f1c8e4a8f8b8c4',
    type: 'string',

    nullable: true,
  })
  @IsOptional()
  @IsString()
  @IsMongoId()
  itemId?: string | null;

  @Expose()
  @JSONSchema({
    description:
      'field to trigger validation error if moduleId is not provided',
    readOnly: true,
  })
  @IsOptional()
  @IsString()
  @IsMongoId()
  @ValidateIf(
    o => o.moduleId === null && (o.sectionId !== null || o.itemId !== null),
    {message: 'moduleId is required if sectionId or itemId is provided'},
  )
  invalidFieldsCheck?: any; // dummy field to trigger validation error

  @Expose()
  @JSONSchema({
    description:
      'field to trigger validation error if sectionId is not provided',
    readOnly: true,
  })
  @IsOptional()
  @IsString()
  @IsMongoId()
  @ValidateIf(o => o.sectionId === null && o.itemId !== null, {
    message: 'sectionId is required if itemId is provided',
  })
  invalidFieldsCheck2?: any; // dummy field to trigger validation error

  @Expose()
  @JSONSchema({
    description: 'Cohort to which the user belongs',
    example: 'Cohort A',
    type: 'string',
  })
  @IsOptional()
  @IsString()
  cohortId?: string;
}

export class ProgressDataResponse implements IProgress {
  @JSONSchema({
    description: 'Unique identifier for the progress record',
    example: '60d5ec49b3f1c8e4a8f8b8d1',
    type: 'string',
    readOnly: true,
  })
  @IsString()
  @IsMongoId()
  _id?: ID;

  @JSONSchema({
    description: 'User ID associated with this progress',
    example: '60d5ec49b3f1c8e4a8f8b8c1',
    type: 'string',
  })
  @IsNotEmpty()
  @IsString()
  @IsMongoId()
  userId: ID;

  @JSONSchema({
    description: 'Course ID associated with this progress',
    example: '60d5ec49b3f1c8e4a8f8b8c2',
    type: 'string',
  })
  @IsNotEmpty()
  @IsString()
  @IsMongoId()
  courseId: ID;

  @JSONSchema({
    description: 'Course version ID associated with this progress',
    example: '60d5ec49b3f1c8e4a8f8b8c3',
    type: 'string',
  })
  @IsNotEmpty()
  @IsString()
  @IsMongoId()
  courseVersionId: ID;

  @JSONSchema({
    description: 'ID of the current module in progress',
    example: '60d5ec49b3f1c8e4a8f8b8c5',
    type: 'string',
  })
  @IsNotEmpty()
  @IsString()
  @IsMongoId()
  currentModule: ID;

  @JSONSchema({
    description: 'ID of the current section in progress',
    example: '60d5ec49b3f1c8e4a8f8b8c6',
    type: 'string',
  })
  @IsNotEmpty()
  @IsString()
  @IsMongoId()
  currentSection: ID;

  @JSONSchema({
    description: 'ID of the current item in progress',
    example: '60d5ec49b3f1c8e4a8f8b8c4',
    type: 'string',
  })
  @IsNotEmpty()
  @IsString()
  @IsMongoId()
  currentItem: ID;

  @JSONSchema({
    description: 'Whether the course has been completed',
    example: false,
    type: 'boolean',
  })
  @IsNotEmpty()
  @IsBoolean()
  completed: boolean;
}

export class CompletedProgressResponse {
  @JSONSchema({
    description: 'Indicates whether the course has been completed',
    example: true,
    type: 'boolean',
    readOnly: true,
  })
  @IsNotEmpty()
  @IsBoolean()
  completed: boolean;

  @JSONSchema({
    description: 'Percentage of course completion',
    example: 75,
    type: 'number',
    readOnly: true,
  })
  @IsNotEmpty()
  @IsNumber()
  percentCompleted: number;

  @JSONSchema({
    description: 'Total number of items in the course',
    example: 20,
    type: 'number',
    readOnly: true,
  })
  @IsNotEmpty()
  @IsNumber()
  totalItems: number;

  @JSONSchema({
    description: 'Total number of completed items in the course',
    example: 15,
    type: 'number',
    readOnly: true,
  })
  @IsNotEmpty()
  @IsNumber()
  completedItems: number;
}

export class ProgressNotFoundErrorResponse {
  @JSONSchema({
    description: 'Error message indicating progress not found',
    example: 'Progress not found for the specified user and course version',
    type: 'string',
    readOnly: true,
  })
  @IsNotEmpty()
  message: string;
}

export class WatchTimeParams {
  @JSONSchema({
    description: 'user ID to get watch time for',
    type: 'string',
  })
  @IsNotEmpty()
  @IsString()
  @IsMongoId()
  userId: string;

  @JSONSchema({
    description: 'Course ID to get watch time for',
    type: 'string',
  })
  @IsNotEmpty()
  @IsString()
  @IsMongoId()
  courseId: string;

  @JSONSchema({
    description: 'Course version ID to get watch time for',
    type: 'string',
  })
  @IsNotEmpty()
  @IsString()
  @IsMongoId()
  versionId: string;

  @JSONSchema({
    description: 'Item ID to get watch time for',
    type: 'string',
  })
  @IsNotEmpty()
  @IsString()
  @IsMongoId()
  itemId: string;

  @JSONSchema({
    description: 'Type of the item (e.g., video, quiz)',
    example: 'VIDEO',
    type: 'string',
  })
  @IsNotEmpty()
  @IsString()
  @IsEnum(ItemType)
  type: ItemType;
}

export class WatchTimeResponse {
  @JSONSchema({
    description: 'Array of watch time records',
    type: 'array',
  })
  @IsNotEmpty()
  @ValidateNested()
  @Type(() => WatchTime)
  watchTime: WatchTime[];

  @JSONSchema({
    description: 'Quiz metrics if applicable',
    type: 'object',
    nullable: true,
  })
  @IsOptional()
  quizMetrics?: UserQuizMetrics;
}

export class TotalWatchTimeResponse {
  @JSONSchema({
    description: 'Total watch time of the user',
    example: 120,
    type: 'number',
  })
  @IsNumber()
  totalWatchTime: number;
}

export class UpsertWatchTimeBody {
  @JSONSchema({
    description: 'Watch item ID used for tracking progress',
    example: '60d5ec49b3f1c8e4a8f8b8c7',
    type: 'string',
  })
  @IsNotEmpty()
  @IsString()
  @IsMongoId()
  watchItemId: string;

  @JSONSchema({
    description: 'ID of the course item being watched',
    example: '60d5ec49b3f1c8e4a8f8b8c4',
    type: 'string',
  })
  @IsNotEmpty()
  @IsString()
  @IsMongoId()
  itemId: string;

  @JSONSchema({
    description: 'ID of the cohort if applicable',
    example: '60d5ec49b3f1c8e4a8f8b8c8',
    type: 'string',
  })
  @IsOptional()
  @IsString()
  @IsMongoId()
  cohortId?: string;
}

export class UpsertWatchTimeResponse {
  @JSONSchema({
    description: 'Watch item ID of the upserted record',
    example: '60d5ec49b3f1c8e4a8f8b8c7',
    type: 'string',
  })
  @IsString()
  watchItemId: string;
}

export const PROGRESS_VALIDATORS = [
  GetUserProgressParams,
  StartItemBody,
  StartItemParams,
  StartItemResponse,
  StopItemBody,
  StopItemParams,
  UpdateProgressBody,
  UpdateProgressParams,
  ResetCourseProgressBody,
  ResetCourseProgressParams,
  ProgressDataResponse,
  CompletedProgressResponse,
  ProgressNotFoundErrorResponse,
  WatchTimeParams,
  WatchTimeResponse,
  UpsertWatchTimeBody,
  UpsertWatchTimeResponse,
];