goddonebianu
605c57f8bb
All checks were successful
continuous-integration/drone/push Build is passing
85 lines
2.2 KiB
TypeScript
85 lines
2.2 KiB
TypeScript
"use client";
|
||
|
||
import { useCallback } from "react";
|
||
|
||
export function useFileUpload() {
|
||
const uploadFile = useCallback((
|
||
file: File,
|
||
onSuccess: (url: string) => void,
|
||
onError?: (error: Error) => void
|
||
) => {
|
||
try {
|
||
// 验证文件大小（限制为100MB）
|
||
const maxSize = 100 * 1024 * 1024; // 100MB
|
||
if (file.size > maxSize) {
|
||
throw new Error(`文件大小超过限制 (${(file.size / 1024 / 1024).toFixed(2)}MB > 100MB)`);
|
||
}
|
||
|
||
const url = URL.createObjectURL(file);
|
||
onSuccess(url);
|
||
} catch (error) {
|
||
const errorMessage = error instanceof Error ? error.message : '文件上传失败';
|
||
onError?.(new Error(errorMessage));
|
||
}
|
||
}, []);
|
||
|
||
const uploadVideo = useCallback((
|
||
onVideoUpload: (url: string) => void,
|
||
onError?: (error: Error) => void
|
||
) => {
|
||
const input = document.createElement('input');
|
||
input.type = 'file';
|
||
input.accept = 'video/*';
|
||
|
||
input.onchange = (e) => {
|
||
const file = (e.target as HTMLInputElement).files?.[0];
|
||
if (file) {
|
||
// 验证文件类型
|
||
if (!file.type.startsWith('video/')) {
|
||
onError?.(new Error('请选择有效的视频文件'));
|
||
return;
|
||
}
|
||
uploadFile(file, onVideoUpload, onError);
|
||
}
|
||
};
|
||
|
||
input.onerror = () => {
|
||
onError?.(new Error('文件选择失败'));
|
||
};
|
||
|
||
input.click();
|
||
}, [uploadFile]);
|
||
|
||
const uploadSubtitle = useCallback((
|
||
onSubtitleUpload: (url: string) => void,
|
||
onError?: (error: Error) => void
|
||
) => {
|
||
const input = document.createElement('input');
|
||
input.type = 'file';
|
||
input.accept = '.srt';
|
||
|
||
input.onchange = (e) => {
|
||
const file = (e.target as HTMLInputElement).files?.[0];
|
||
if (file) {
|
||
// 验证文件扩展名
|
||
if (!file.name.toLowerCase().endsWith('.srt')) {
|
||
onError?.(new Error('请选择.srt格式的字幕文件'));
|
||
return;
|
||
}
|
||
uploadFile(file, onSubtitleUpload, onError);
|
||
}
|
||
};
|
||
|
||
input.onerror = () => {
|
||
onError?.(new Error('文件选择失败'));
|
||
};
|
||
|
||
input.click();
|
||
}, [uploadFile]);
|
||
|
||
return {
|
||
uploadVideo,
|
||
uploadSubtitle,
|
||
uploadFile,
|
||
};
|
||
} |