1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
|
import { NextRequest, NextResponse } from 'next/server';
import { mkdir, writeFile, appendFile, readFile, rm } from 'fs/promises';
import path from 'path';
import { generateHashedFileName, saveBuffer, saveDRMFile } from '@/lib/file-stroage';
import { decryptWithServerAction } from '@/components/drm/drmUtils';
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);
const mergedFile = new File([mergedBuffer], filename, {
type: chunk.type || 'application/octet-stream',
lastModified: Date.now(),
});
const saveResult = await saveDRMFile(
mergedFile,
decryptWithServerAction, // 복호화 함수
'basicContract/template', // 저장 디렉토리
// userId // 선택
);
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 });
}
}
|
