david_bai
78 lines
2.4 KiB
TypeScript
78 lines
2.4 KiB
TypeScript
import { useEffect, useMemo } from "react";
|
|
import { webrtcService } from "@/lib/webrtcService";
|
|
import { useFileTransferStore } from "@/stores/fileTransferStore";
|
|
import type { Messages } from "@/types/messages";
|
|
|
|
// 保留类型定义以保持兼容性
|
|
export type PeerProgressDetails = { progress: number; speed: number };
|
|
export type FileProgressPeers = { [peerId: string]: PeerProgressDetails };
|
|
export type ProgressState = { [fileId: string]: FileProgressPeers };
|
|
|
|
interface UseWebRTCConnectionProps {
|
|
messages: Messages | null;
|
|
putMessageInMs: (
|
|
message: string,
|
|
isShareEnd?: boolean,
|
|
displayTimeMs?: number
|
|
) => void;
|
|
}
|
|
|
|
export function useWebRTCConnection({
|
|
messages,
|
|
putMessageInMs,
|
|
}: UseWebRTCConnectionProps) {
|
|
// 从 store 获取状态
|
|
const {
|
|
sharePeerCount,
|
|
retrievePeerCount,
|
|
senderDisconnected,
|
|
sendProgress,
|
|
receiveProgress,
|
|
setIsAnyFileTransferring,
|
|
} = useFileTransferStore();
|
|
|
|
// 计算是否有文件正在传输
|
|
const isAnyFileTransferring = useMemo(() => {
|
|
const allProgress = [
|
|
...Object.values(sendProgress),
|
|
...Object.values(receiveProgress),
|
|
];
|
|
return allProgress.some((fileProgress: any) => {
|
|
return Object.values(fileProgress).some((progress: any) => {
|
|
return progress.progress > 0 && progress.progress < 1;
|
|
});
|
|
});
|
|
}, [sendProgress, receiveProgress]);
|
|
|
|
useEffect(() => {
|
|
setIsAnyFileTransferring(isAnyFileTransferring);
|
|
}, [isAnyFileTransferring, setIsAnyFileTransferring]);
|
|
|
|
return {
|
|
// 状态从 store 获取
|
|
sharePeerCount,
|
|
retrievePeerCount,
|
|
senderDisconnected,
|
|
sendProgress,
|
|
receiveProgress,
|
|
|
|
// 方法直接从 service 暴露
|
|
broadcastDataToAllPeers:
|
|
webrtcService.broadcastDataToAllPeers.bind(webrtcService),
|
|
requestFile: webrtcService.requestFile.bind(webrtcService),
|
|
requestFolder: webrtcService.requestFolder.bind(webrtcService),
|
|
setReceiverDirectoryHandle:
|
|
webrtcService.setReceiverDirectoryHandle.bind(webrtcService),
|
|
getReceiverSaveType: webrtcService.getReceiverSaveType.bind(webrtcService),
|
|
manualSafeSave: webrtcService.manualSafeSave.bind(webrtcService),
|
|
|
|
// 重置连接方法
|
|
resetSenderConnection: () => webrtcService.leaveRoom(true),
|
|
resetReceiverConnection: () => webrtcService.leaveRoom(false),
|
|
|
|
// 为了兼容性，保留这些属性（但实际上不再需要）
|
|
sender: webrtcService.sender,
|
|
receiver: webrtcService.receiver,
|
|
};
|
|
}
|