/** * Use Case: RequestAvatarGenerationUseCase * * Handles the business logic for requesting avatar generation from a face photo. */ import type { Logger } from '@core/shared/domain/Logger'; import { Result } from '@core/shared/domain/Result'; import type { ApplicationErrorCode } from '@core/shared/errors/ApplicationErrorCode'; import { v4 as uuidv4 } from 'uuid'; import { AvatarGenerationRequest } from '../../domain/entities/AvatarGenerationRequest'; import { AvatarGenerationRepository } from '../../domain/repositories/AvatarGenerationRepository'; import type { RacingSuitColor } from '../../domain/types/AvatarGenerationRequest'; import type { AvatarGenerationPort } from '../ports/AvatarGenerationPort'; import type { FaceValidationPort } from '../ports/FaceValidationPort'; export interface RequestAvatarGenerationInput { userId: string; facePhotoData: string; suitColor: RacingSuitColor; style?: 'realistic' | 'cartoon' | 'pixel-art'; } export interface RequestAvatarGenerationResult { requestId: string; status: 'validating' | 'generating' | 'completed'; avatarUrls?: string[]; } export type RequestAvatarGenerationErrorCode = | 'FACE_VALIDATION_FAILED' | 'GENERATION_FAILED' | 'REPOSITORY_ERROR'; export type RequestAvatarGenerationApplicationError = ApplicationErrorCode< RequestAvatarGenerationErrorCode, { message: string } >; export class RequestAvatarGenerationUseCase { constructor( private readonly avatarRepo: AvatarGenerationRepository, private readonly faceValidation: FaceValidationPort, private readonly avatarGeneration: AvatarGenerationPort, private readonly logger: Logger, ) {} async execute( input: RequestAvatarGenerationInput, ): Promise> { this.logger.info('[RequestAvatarGenerationUseCase] Starting avatar generation request', { userId: input.userId, suitColor: input.suitColor, }); try { const requestId = uuidv4(); const requestProps: Parameters[0] = { id: requestId, userId: input.userId, facePhotoUrl: input.facePhotoData, suitColor: input.suitColor, }; if (input.style !== undefined) { requestProps.style = input.style; } const request = AvatarGenerationRequest.create(requestProps); await this.avatarRepo.save(request); request.markAsValidating(); await this.avatarRepo.save(request); const validationResult = await this.faceValidation.validateFacePhoto(input.facePhotoData); if (!validationResult.isValid || !validationResult.hasFace || validationResult.faceCount !== 1) { const errorMessage = validationResult.errorMessage || 'Invalid face photo: must contain exactly one face'; request.fail(errorMessage); await this.avatarRepo.save(request); return Result.err({ code: 'FACE_VALIDATION_FAILED', details: { message: errorMessage }, }); } request.markAsGenerating(); await this.avatarRepo.save(request); const generationOptions = { facePhotoUrl: input.facePhotoData, prompt: request.buildPrompt(), suitColor: input.suitColor, style: input.style || 'realistic', count: 3, }; const generationResult = await this.avatarGeneration.generateAvatars(generationOptions); if (!generationResult.success) { const errorMessage = generationResult.errorMessage || 'Failed to generate avatars'; request.fail(errorMessage); await this.avatarRepo.save(request); return Result.err({ code: 'GENERATION_FAILED', details: { message: errorMessage }, }); } const avatarUrls = generationResult.avatars.map(avatar => avatar.url); request.completeWithAvatars(avatarUrls); await this.avatarRepo.save(request); this.logger.info('[RequestAvatarGenerationUseCase] Avatar generation completed', { requestId, userId: input.userId, avatarCount: avatarUrls.length, }); return Result.ok({ requestId, status: 'completed', avatarUrls, }); } catch (error) { const err = error instanceof Error ? error : new Error(String(error)); this.logger.error('[RequestAvatarGenerationUseCase] Error during avatar generation', err, { userId: input.userId, }); return Result.err({ code: 'REPOSITORY_ERROR', details: { message: err.message ?? 'Internal error occurred during avatar generation' }, }); } } }