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

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

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

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

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

    console.log("Creating websocket", serverUrlWithSessionId);
    console.log("Using vscode message passing", useVscodeMessagePassing);
    const newClient = new ContinueNotebookClientProtocol(
      serverUrlWithSessionId,
      useVscodeMessagePassing
    );
    setClient(newClient);
  }, [sessionId, serverHttpUrl]);

  return client;
}
export default useContinueNotebookProtocol;