import { NextRequest, NextResponse } from 'next/server';
import { mkdir, writeFile, appendFile, readFile, rm } from 'fs/promises';
import path from 'path';
import { generateHashedFileName, saveBuffer } from '@/lib/file-stroage';

export async function POST(request: NextRequest) {
  try {
    const formData = await request.formData();
    
    const chunk = formData.get('chunk') as File;
    const filename = formData.get('filename') as string;
    const chunkIndex = parseInt(formData.get('chunkIndex') as string);
    const totalChunks = parseInt(formData.get('totalChunks') as string);
    const fileId = formData.get('fileId') as string;
    
    if (!chunk || !filename || isNaN(chunkIndex) || isNaN(totalChunks) || !fileId) {
      return NextResponse.json({ 
        success: false, 
        error: '필수 매개변수가 누락되었습니다' 
      }, { status: 400 });
    }

    // 임시 디렉토리 생성
    const tempDir = path.join(process.cwd(), 'temp', fileId);
    await mkdir(tempDir, { recursive: true });

    // 청크 파일 저장
    const chunkPath = path.join(tempDir, `chunk-${chunkIndex}`);
    const buffer = Buffer.from(await chunk.arrayBuffer());
    await writeFile(chunkPath, buffer);

    console.log(`📦 청크 저장 완료: ${chunkIndex + 1}/${totalChunks} (${buffer.length} bytes)`);

    // 마지막 청크인 경우 모든 청크를 합쳐 최종 파일 생성
    if (chunkIndex === totalChunks - 1) {
      console.log(`🔄 파일 병합 시작: ${filename}`);
      
      try {
        // 모든 청크를 순서대로 읽어서 병합
        const chunks: Buffer[] = [];
        let totalSize = 0;
        
        for (let i = 0; i < totalChunks; i++) {
          const chunkData = await readFile(path.join(tempDir, `chunk-${i}`));
          chunks.push(chunkData);
          totalSize += chunkData.length;
        }
        
        // 모든 청크를 하나의 Buffer로 병합
        const mergedBuffer = Buffer.concat(chunks, totalSize);
        
        console.log(`📄 병합 완료: ${filename} (총 ${totalSize} bytes)`);

        // 공용 함수를 사용하여 파일 저장
        const saveResult = await saveBuffer({
          buffer: mergedBuffer,
          fileName: filename,
          directory: 'basicContract/template',
          originalName: filename
        });

        // 임시 파일 정리 (비동기로 처리)
        rm(tempDir, { recursive: true, force: true })
          .then(() => console.log(`🗑️ 임시 파일 정리 완료: ${fileId}`))
          .catch((e: unknown) => console.error('청크 정리 오류:', e));

        if (saveResult.success) {
          console.log(`✅ 최종 파일 저장 완료: ${saveResult.fileName}`);
          
          return NextResponse.json({
            success: true,
            fileName: filename,
            filePath: saveResult.publicPath,
            hashedFileName: saveResult.fileName,
            fileSize: totalSize
          });
        } else {
          console.error('파일 저장 실패:', saveResult.error);
          return NextResponse.json({ 
            success: false, 
            error: saveResult.error || '파일 저장에 실패했습니다' 
          }, { status: 500 });
        }
        
      } catch (mergeError) {
        console.error('파일 병합 오류:', mergeError);
        
        // 오류 발생 시 임시 파일 정리
        rm(tempDir, { recursive: true, force: true })
          .catch((e: unknown) => console.error('임시 파일 정리 오류:', e));
        
        return NextResponse.json({ 
          success: false, 
          error: '파일 병합 중 오류가 발생했습니다' 
        }, { status: 500 });
      }
    }

    return NextResponse.json({
      success: true,
      chunkIndex,
      message: `청크 ${chunkIndex + 1}/${totalChunks} 업로드 완료`
    });
    
  } catch (error) {
    console.error('청크 업로드 오류:', error);
    return NextResponse.json({ 
      success: false, 
      error: '서버 오류' 
    }, { status: 500 });
  }
}