pydase/frontend/src/socket.ts

import { io } from "socket.io-client";
import { serializeDict, serializeList } from "./utils/serializationUtils";
import { SerializedObject } from "./types/SerializedObject";

export const hostname =
  process.env.NODE_ENV === "development" ? `localhost` : window.location.hostname;
export const port =
  process.env.NODE_ENV === "development" ? 8001 : window.location.port;
const URL = `ws://${hostname}:${port}/`;
console.debug("Websocket: ", URL);

export const socket = io(URL, { path: "/ws/socket.io", transports: ["websocket"] });

export const updateValue = (
  serializedObject: SerializedObject,
  callback?: (ack: unknown) => void,
) => {
  if (callback) {
    socket.emit(
      "update_value",
      { access_path: serializedObject["full_access_path"], value: serializedObject },
      callback,
    );
  } else {
    socket.emit("update_value", {
      access_path: serializedObject["full_access_path"],
      value: serializedObject,
    });
  }
};

export const runMethod = (
  accessPath: string,
  args: unknown[] = [],
  kwargs: Record<string, unknown> = {},
  callback?: (ack: unknown) => void,
) => {
  const serializedArgs = serializeList(args);
  const serializedKwargs = serializeDict(kwargs);

  if (callback) {
    socket.emit(
      "trigger_method",
      { access_path: accessPath, args: serializedArgs, kwargs: serializedKwargs },
      callback,
    );
  } else {
    socket.emit("trigger_method", {
      access_path: accessPath,
      args: serializedArgs,
      kwargs: serializedKwargs,
    });
  }
};