Function startDevTools

src/commands/devTools.ts:25–191

(statusBarItem: vscode.StatusBarItem)

const WebSocketServer = Server;

export const startDevTools = async (statusBarItem: vscode.StatusBarItem) => {
  if (wss) {
    wss.clients.forEach((client) => {
      client.close();
    });
    wss.close();
    wss = undefined;
    return setStatusBarToStopped(statusBarItem);
  }
  const config = getConfig();
  const port = config.get<number>("devToolsPort");
  wss = new WebSocketServer({ port: port || 3003 });
  setStatusBarToRunning(statusBarItem);

  const workspaceUri = getWorkspaceUri();
  if (!workspaceUri) {
    return;
  }
  let rootDir = workspaceUri;

  if (!(await isRemixDir(rootDir))) {
    const currentFile = vscode.window.activeTextEditor?.document.uri;

    if (!currentFile) {
      return;
    }
    const remixDir = await getRemixRootFromFileUri(currentFile);
    if (!remixDir) {
      return;
    }
    rootDir = remixDir;
  }

  const isTS = await isTSConfigExists(rootDir);

  // Handle WebSocket connection
  wss.on("connection", (socket: WebSocket) => {
    // Handle messages received from the React app
    socket.onmessage = async (event: MessageEvent) => {
      const message = JSON.parse(event.data.toString());

      if (message.type === "plugin") {
        if (message.subtype === "read_file") {
          const file = await tryReadFile(joinPath(rootDir, message.path).path);
          if (file) {
            return socket.send(JSON.stringify({ type: "plugin", subtype: "read_file", error: false, data: file }));
          }
          socket.send(JSON.stringify({ type: "plugin", subtype: "read_file", error: true, data: "File not found" }));
        }
        if (message.subtype === "open_file") {
          try {
            await openFileInEditor(joinPath(rootDir, message.path).path);
            socket.send(JSON.stringify({ type: "plugin", subtype: "open_file", error: false, data: "success" }));
          } catch (err) {
            socket.send(
              JSON.stringify({ type: "plugin", subtype: "open_file", error: true, data: (err as any)?.message }),
            );
          }