david_bai
77 lines
2.7 KiB
TypeScript
77 lines
2.7 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]);
|
|
|
|
// 确保服务在 React 生命周期内被初始化
|
|
useEffect(() => {
|
|
console.log('[useWebRTCConnection] WebRTC 服务已初始化');
|
|
|
|
return () => {
|
|
console.log('[useWebRTCConnection] Hook 清理');
|
|
};
|
|
}, []);
|
|
|
|
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,
|
|
};
|
|
} |