extension/react-app/src/hooks/useWebsocket.ts


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42

import React, { useEffect, useState } from "react";
import { RootStore } from "../redux/store";
import { useSelector } from "react-redux";
import ContinueGUIClientProtocol from "./ContinueGUIClientProtocol";
import { postVscMessage } from "../vscode";

function useContinueGUIProtocol(useVscodeMessagePassing: boolean = true) {
  const sessionId = useSelector((state: RootStore) => state.config.sessionId);
  const serverHttpUrl = useSelector((state: RootStore) => state.config.apiUrl);
  const [client, setClient] = useState<ContinueGUIClientProtocol | undefined>(
    undefined
  );

  useEffect(() => {
    if (!sessionId || !serverHttpUrl) {
      if (useVscodeMessagePassing) {
        postVscMessage("onLoad", {});
      }
      setClient(undefined);
      return;
    }

    const serverUrlWithSessionId =
      serverHttpUrl.replace("http", "ws") +
      "/gui/ws?session_id=" +
      encodeURIComponent(sessionId);

    console.log(
      "Creating GUI websocket",
      serverUrlWithSessionId,
      useVscodeMessagePassing
    );
    const newClient = new ContinueGUIClientProtocol(
      serverUrlWithSessionId,
      useVscodeMessagePassing
    );
    setClient(newClient);
  }, [sessionId, serverHttpUrl]);

  return client;
}
export default useContinueGUIProtocol;