tracer: Avoid hardcoding the WebSocket port

frida · Sep 19, 2024 · ab9ed88 · ab9ed88
1 parent 662d2d5
commit ab9ed88
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/apps/tracer/src/App.tsx b/apps/tracer/src/App.tsx
@@ -33,7 +33,10 @@ export default function App() {
     const [addingTargets, setAddingTargets] = useState(false);
     const [events, setEvents] = useState<Event[]>([]);
     const [stagedItems, setStagedItems] = useState<StagedItem[]>([]);
-    const { sendJsonMessage, lastJsonMessage, readyState } = useWebSocket<TracerMessage>("ws://localhost:1337" /* window.location.origin */);
+    const { sendJsonMessage, lastJsonMessage, readyState } = useWebSocket<TracerMessage>(
+        (import.meta.env.MODE === "development")
+            ? "ws://localhost:1337"
+            : `ws://${window.location.host}`);
 
     function handleHandlerSelection(id: HandlerId) {
         setSelectedHandler(id);