doctor-mail/apps/web/src/components/common/uploader/TusUploader.tsx

// TusUploader.tsx
import {
	useCallback,
	useState,
	useEffect,
	forwardRef,
	useImperativeHandle,
} from "react";
import {
	UploadOutlined,
	CheckCircleOutlined,
	DeleteOutlined,
} from "@ant-design/icons";
import { Upload, Progress, Button } from "antd";
import { useTusUpload } from "@web/src/hooks/useTusUpload";
import toast from "react-hot-toast";

export interface TusUploaderProps {
	value?: string[];
	onChange?: (value: string[]) => void;
}

export interface TusUploaderRef {
	reset: () => void;
}

interface UploadingFile {
	name: string;
	progress: number;
	status: "uploading" | "done" | "error";
	fileId?: string;
	fileKey?: string;
}

export const TusUploader = forwardRef<TusUploaderRef, TusUploaderProps>(
	({ value = [], onChange }, ref) => {
		const { handleFileUpload, uploadProgress } = useTusUpload();
		const [uploadingFiles, setUploadingFiles] = useState<UploadingFile[]>(
			[]
		);
		const [completedFiles, setCompletedFiles] = useState<UploadingFile[]>(
			[]
		);

		// 同步父组件value到completedFiles
		useEffect(() => {
			setCompletedFiles(
				value.map((fileId) => ({
					name: `文件 ${fileId}`,
					progress: 100,
					status: "done" as const,
					fileId,
				}))
			);
		}, [value]);

		// 暴露重置方法
		useImperativeHandle(ref, () => ({
			reset: () => {
				setCompletedFiles([]);
				setUploadingFiles([]);
			},
		}));

		const handleRemoveFile = useCallback(
			(fileId: string) => {
				setCompletedFiles((prev) =>
					prev.filter((f) => f.fileId !== fileId)
				);
				onChange?.(value.filter((id) => id !== fileId));
			},
			[onChange, value]
		);

		const handleRemoveUploadingFile = useCallback((fileKey: string) => {
			setUploadingFiles((prev) =>
				prev.filter((f) => f.fileKey !== fileKey)
			);
		}, []);

		const handleBeforeUpload = useCallback(
			(file: File) => {
				const fileKey = `${file.name}-${Date.now()}`;

				setUploadingFiles((prev) => [
					...prev,
					{
						name: file.name,
						progress: 0,
						status: "uploading",
						fileKey,
					},
				]);

				handleFileUpload(
					file,
					(result) => {
						const newValue = [...value, result.fileId];
						onChange?.(newValue);
						setUploadingFiles((prev) =>
							prev.filter((f) => f.fileKey !== fileKey)
						);
					},
					(error) => {
						console.error("上传错误:", error);
						toast.error(
							`上传失败: ${error instanceof Error ? error.message : "未知错误"}`
						);
						setUploadingFiles((prev) =>
							prev.map((f) =>
								f.fileKey === fileKey
									? { ...f, status: "error" }
									: f
							)
						);
					},
					fileKey
				);

				return false;
			},
			[handleFileUpload, onChange, value]
		);

		return (
			<div className="space-y-1">
				<Upload.Dragger
					name="files"
					multiple
					showUploadList={false}
					style={{ background: "transparent", borderStyle: "none" }}
					beforeUpload={handleBeforeUpload}>
					<p className="ant-upload-drag-icon">
						<UploadOutlined />
					</p>
					<p className="ant-upload-text">
						点击或拖拽文件到此区域进行上传
					</p>
					<p className="ant-upload-hint">支持单个或批量上传文件</p>

					<div className="px-2 py-0 rounded mt-1">
						{uploadingFiles.map((file) => (
							<div
								key={file.fileKey}
								className="flex flex-col gap-1 mb-2">
								<div className="flex items-center gap-2">
									<span className="text-sm">{file.name}</span>
								</div>
								<div className="flex items-center gap-2">
									<Progress
										percent={
											file.status === "done"
												? 100
												: Math.round(
														uploadProgress?.[
															file.fileKey!
														] || 0
													)
										}
										status={
											file.status === "error"
												? "exception"
												: "active"
										}
										className="flex-1"
									/>
									{file.status === "error" && (
										<Button
											type="text"
											danger
											icon={<DeleteOutlined />}
											onClick={(e) => {
												e.stopPropagation();
												if (file.fileKey)
													handleRemoveUploadingFile(
														file.fileKey
													);
											}}
										/>
									)}
								</div>
							</div>
						))}

						{completedFiles.map((file) => (
							<div
								key={file.fileId}
								className="flex items-center justify-between gap-2 mb-2">
								<div className="flex items-center gap-2">
									<CheckCircleOutlined className="text-green-500" />
									<span className="text-sm">{file.name}</span>
								</div>
								<Button
									type="text"
									danger
									icon={<DeleteOutlined />}
									onClick={(e) => {
										e.stopPropagation();
										if (file.fileId)
											handleRemoveFile(file.fileId);
									}}
								/>
							</div>
						))}
					</div>
				</Upload.Dragger>
			</div>
		);
	}
);