d8dfun/claudecodeui
1
0
Fork 0
mirror of https://github.com/siteboon/claudecodeui.git synced 2026-01-31 13:57:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor: Extract WebSocket URL construction into a separate function

Browse Source
This commit is contained in:
Haileyesus
2026-01-31 12:03:40 +03:00
parent eca96c6973
commit 471892b2bd
1 changed files with 12 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
src/contexts/WebSocketContext.tsx
View File

@@ -17,7 +17,15 @@ export const useWebSocket = () => {
return context;
};
const useWebSocketProviderState = () => {
const buildWebSocketUrl = (token: string | null) => {
const isPlatform = import.meta.env.VITE_IS_PLATFORM === 'true';
const protocol = window.location.protocol === 'https:' ? 'wss:' : 'ws:';
if (isPlatform) return `${protocol}//${window.location.host}/ws`; // Platform mode: Use same domain as the page (goes through proxy)
if (!token) return null;
return `${protocol}//${window.location.host}/ws?token=${encodeURIComponent(token)}`; // OSS mode: Use same host:port that served the page
};
const useWebSocketProviderState = (): WebSocketContextType => {
const wsRef = useRef<WebSocket | null>(null);
const unmountedRef = useRef(false);
const [messages, setMessages] = useState<any[]>([]);
@@ -44,22 +52,10 @@ const useWebSocketProviderState = () => {
const isPlatform = import.meta.env.VITE_IS_PLATFORM === 'true';
// Construct WebSocket URL
let wsUrl: string;
const protocol = window.location.protocol === 'https:' ? 'wss:' : 'ws:';
if (isPlatform) {
// Platform mode: Use same domain as the page (goes through proxy)
wsUrl = `${protocol}//${window.location.host}/ws`;
} else {
// OSS mode: Connect to same host:port that served the page
const token = localStorage.getItem('auth-token');
if (!token) {
console.warn('No authentication token found for WebSocket connection');
return;
}
wsUrl = `${protocol}//${window.location.host}/ws?token=${encodeURIComponent(token)}`;
}
const wsUrl = buildWebSocketUrl(isPlatform ? null : localStorage.getItem('authToken'));
if (!wsUrl) return console.warn('No authentication token found for WebSocket connection');
const websocket = new WebSocket(wsUrl);
websocket.onopen = () => {