@linkdlab/funcnodes_react_flow 0.1.0

package/src/funcnodes/workermanager.ts

@@ -0,0 +1,229 @@
+import { FuncNodesReactFlowZustandInterface, WorkersState } from "../state";
+import FuncNodesWorker from "./funcnodesworker";
+import WebSocketWorker from "./websocketworker";
+
+class WorkerManager {
+  private wsuri: string;
+  private workers: any;
+  private ws: WebSocket | null = null;
+  private reconnectAttempts: number = 0;
+  private maxReconnectAttempts: number = 999;
+  private initialTimeout: number = 200; // Initial reconnect delay in ms
+  private maxTimeout: number = 2000; // Maximum reconnect delay
+  private zustand: FuncNodesReactFlowZustandInterface;
+  private connectionTimeout?: NodeJS.Timeout;
+  on_setWorker: (worker: FuncNodesWorker | undefined) => void;
+  constructor(wsuri: string, zustand: FuncNodesReactFlowZustandInterface) {
+    this.wsuri = wsuri;
+    this.zustand = zustand;
+    this.workers = {};
+    this.on_setWorker = (worker: FuncNodesWorker | undefined) => {
+      console.log("WorkerManager: on_setWorker", worker);
+      this.zustand.worker = worker;
+    };
+    this.connect();
+  }
+  private connect(): void {
+    this.zustand.set_progress({
+      progress: 0,
+      message: "connecting to worker manager",
+      status: "info",
+      blocking: true,
+    });
+    console.log("Connecting to websocket");
+    this.ws = new WebSocket(this.wsuri);
+
+    this.ws.onopen = () => {
+      this.onopen();
+    };
+
+    this.ws.onclose = () => {
+      this.onclose();
+    };
+
+    this.ws.onerror = () => {
+      this.on_ws_error();
+    };
+
+    this.ws.onmessage = (event) => {
+      this.onmessage(event.data);
+    };
+
+    this.connectionTimeout = setTimeout(() => {
+      if (this.ws?.readyState !== WebSocket.OPEN) {
+        this.on_ws_error();
+      }
+    }, 5000);
+  }
+
+  on_ws_error() {
+    console.warn("Websocket error");
+    if (this.ws) {
+      this.ws.close(); // Ensure the connection is closed before attempting to reconnect
+    } else {
+      this.reconnect();
+    }
+  }
+
+  onopen() {
+    if (this.connectionTimeout) {
+      clearTimeout(this.connectionTimeout);
+      this.connectionTimeout = undefined;
+    }
+    this.zustand.auto_progress();
+    console.log("WorkerManager: onopen");
+    if (this.ws) {
+      this.ws.send("worker_status");
+    }
+
+    // Get active worker from window storage
+    const active_worker = window.localStorage.getItem(
+      "funcnodes__active_worker"
+    );
+    if (active_worker) {
+      this.set_active(active_worker);
+    }
+  }
+  onmessage(event: string) {
+    let msg = JSON.parse(event);
+    if (msg.type === "worker_status") {
+      console.debug("WorkerManager: worker_status", msg);
+      const new_state: WorkersState = {};
+      for (let worker of msg.active) {
+        worker.active = true;
+        new_state[worker.uuid] = worker;
+      }
+      for (let worker of msg.inactive) {
+        worker.active = false;
+        new_state[worker.uuid] = worker;
+      }
+      this.zustand.workers.setState(new_state);
+      return;
+    } else if (msg.type === "set_worker") {
+      if (msg.data.type === "WSWorker") {
+        let url =
+          "ws" +
+          (msg.data.ssl ? "s" : "") +
+          "://" +
+          msg.data.host +
+          ":" +
+          msg.data.port;
+        this.setWorker(
+          this.workers[msg.data.uuid] ||
+            new WebSocketWorker({
+              url,
+              zustand: this.zustand,
+              uuid: msg.data.uuid,
+            })
+        );
+      } else {
+        console.error("WorkerManager: unknown worker type", msg);
+      }
+
+      //store active worker in window storage
+
+      return;
+    } else if (msg.type === "progress") {
+      this.zustand.set_progress(msg as ProgressStateMessage);
+      return;
+    }
+    console.error("WorkerManager: unknown message", msg);
+  }
+
+  setWorker(worker: FuncNodesWorker | undefined) {
+    for (let w in this.workers) {
+      if (w !== worker?.uuid) {
+        this.workers[w].disconnect();
+      }
+    }
+    if (worker !== undefined) {
+      this.workers[worker.uuid] = worker;
+      worker.reconnect();
+    }
+    window.localStorage.setItem("funcnodes__active_worker", worker?.uuid || "");
+    if (this.zustand.worker !== undefined) {
+      this.zustand.clear_all();
+    }
+    this.zustand.worker = worker;
+    this.on_setWorker(worker);
+  }
+
+  async restart_worker(workerid: string) {
+    this.ws?.send(JSON.stringify({ type: "restart_worker", workerid }));
+  }
+
+  private calculateReconnectTimeout(): number {
+    // Increase timeout exponentially, capped at maxTimeout
+    let timeout = Math.min(
+      this.initialTimeout * Math.pow(2, this.reconnectAttempts),
+      this.maxTimeout
+    );
+    return timeout;
+  }
+
+  private reconnect(): void {
+    if (this.reconnectAttempts < this.maxReconnectAttempts) {
+      let timeout = this.calculateReconnectTimeout();
+      console.log(`Attempting to reconnect in ${timeout} ms`);
+
+      setTimeout(() => {
+        if (this.ws) {
+          if (this.ws.readyState === WebSocket.OPEN) {
+            return;
+          }
+        }
+        this.reconnectAttempts++;
+        this.connect();
+      }, timeout);
+    } else {
+      console.warn("Maximum reconnect attempts reached. Giving up.");
+    }
+  }
+
+  onclose() {
+    console.log("WorkerManager: onclose");
+    this.reconnect(); // Attempt to reconnect
+  }
+
+  set_active(workerid: string) {
+    if (!this.ws) return;
+    this.ws.send(JSON.stringify({ type: "set_active", workerid }));
+  }
+
+  new_worker({
+    name,
+    reference,
+    copyLib,
+    copyNS,
+  }: {
+    name?: string;
+    reference?: string;
+    copyLib?: boolean;
+    copyNS?: boolean;
+  }) {
+    if (!name) name = undefined;
+    if (!copyLib) copyLib = false;
+    if (!copyNS) copyNS = false;
+    if (!reference) {
+      reference = undefined;
+      copyLib = false;
+      copyNS = false;
+    }
+
+    if (this.ws) {
+      this.ws.send(
+        JSON.stringify({
+          type: "new_worker",
+          kwargs: {
+            name,
+            reference,
+            copyLib,
+            copyNS,
+          },
+        })
+      );
+    }
+  }
+}
+
+export default WorkerManager;

package/src/index.css

@@ -0,0 +1,13 @@
+body {
+  margin: 0;
+  font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', 'Roboto', 'Oxygen',
+    'Ubuntu', 'Cantarell', 'Fira Sans', 'Droid Sans', 'Helvetica Neue',
+    sans-serif;
+  -webkit-font-smoothing: antialiased;
+  -moz-osx-font-smoothing: grayscale;
+}
+
+code {
+  font-family: source-code-pro, Menlo, Monaco, Consolas, 'Courier New',
+    monospace;
+}

package/src/index.tsx

@@ -0,0 +1,19 @@
+import React from 'react';
+import ReactDOM from 'react-dom/client';
+import './index.css';
+import App from './App';
+import reportWebVitals from './reportWebVitals';
+
+const root = ReactDOM.createRoot(
+  document.getElementById('root') as HTMLElement
+);
+root.render(
+  <React.StrictMode>
+    <App />
+  </React.StrictMode>
+);
+
+// If you want to start measuring performance in your app, pass a function
+// to log results (for example: reportWebVitals(console.log))
+// or send to an analytics endpoint. Learn more: https://bit.ly/CRA-vitals
+reportWebVitals();

package/src/logo.svg

@@ -0,0 +1 @@
+<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 841.9 595.3"><g fill="#61DAFB"><path d="M666.3 296.5c0-32.5-40.7-63.3-103.1-82.4 14.4-63.6 8-114.2-20.2-130.4-6.5-3.8-14.1-5.6-22.4-5.6v22.3c4.6 0 8.3.9 11.4 2.6 13.6 7.8 19.5 37.5 14.9 75.7-1.1 9.4-2.9 19.3-5.1 29.4-19.6-4.8-41-8.5-63.5-10.9-13.5-18.5-27.5-35.3-41.6-50 32.6-30.3 63.2-46.9 84-46.9V78c-27.5 0-63.5 19.6-99.9 53.6-36.4-33.8-72.4-53.2-99.9-53.2v22.3c20.7 0 51.4 16.5 84 46.6-14 14.7-28 31.4-41.3 49.9-22.6 2.4-44 6.1-63.6 11-2.3-10-4-19.7-5.2-29-4.7-38.2 1.1-67.9 14.6-75.8 3-1.8 6.9-2.6 11.5-2.6V78.5c-8.4 0-16 1.8-22.6 5.6-28.1 16.2-34.4 66.7-19.9 130.1-62.2 19.2-102.7 49.9-102.7 82.3 0 32.5 40.7 63.3 103.1 82.4-14.4 63.6-8 114.2 20.2 130.4 6.5 3.8 14.1 5.6 22.5 5.6 27.5 0 63.5-19.6 99.9-53.6 36.4 33.8 72.4 53.2 99.9 53.2 8.4 0 16-1.8 22.6-5.6 28.1-16.2 34.4-66.7 19.9-130.1 62-19.1 102.5-49.9 102.5-82.3zm-130.2-66.7c-3.7 12.9-8.3 26.2-13.5 39.5-4.1-8-8.4-16-13.1-24-4.6-8-9.5-15.8-14.4-23.4 14.2 2.1 27.9 4.7 41 7.9zm-45.8 106.5c-7.8 13.5-15.8 26.3-24.1 38.2-14.9 1.3-30 2-45.2 2-15.1 0-30.2-.7-45-1.9-8.3-11.9-16.4-24.6-24.2-38-7.6-13.1-14.5-26.4-20.8-39.8 6.2-13.4 13.2-26.8 20.7-39.9 7.8-13.5 15.8-26.3 24.1-38.2 14.9-1.3 30-2 45.2-2 15.1 0 30.2.7 45 1.9 8.3 11.9 16.4 24.6 24.2 38 7.6 13.1 14.5 26.4 20.8 39.8-6.3 13.4-13.2 26.8-20.7 39.9zm32.3-13c5.4 13.4 10 26.8 13.8 39.8-13.1 3.2-26.9 5.9-41.2 8 4.9-7.7 9.8-15.6 14.4-23.7 4.6-8 8.9-16.1 13-24.1zM421.2 430c-9.3-9.6-18.6-20.3-27.8-32 9 .4 18.2.7 27.5.7 9.4 0 18.7-.2 27.8-.7-9 11.7-18.3 22.4-27.5 32zm-74.4-58.9c-14.2-2.1-27.9-4.7-41-7.9 3.7-12.9 8.3-26.2 13.5-39.5 4.1 8 8.4 16 13.1 24 4.7 8 9.5 15.8 14.4 23.4zM420.7 163c9.3 9.6 18.6 20.3 27.8 32-9-.4-18.2-.7-27.5-.7-9.4 0-18.7.2-27.8.7 9-11.7 18.3-22.4 27.5-32zm-74 58.9c-4.9 7.7-9.8 15.6-14.4 23.7-4.6 8-8.9 16-13 24-5.4-13.4-10-26.8-13.8-39.8 13.1-3.1 26.9-5.8 41.2-7.9zm-90.5 125.2c-35.4-15.1-58.3-34.9-58.3-50.6 0-15.7 22.9-35.6 58.3-50.6 8.6-3.7 18-7 27.7-10.1 5.7 19.6 13.2 40 22.5 60.9-9.2 20.8-16.6 41.1-22.2 60.6-9.9-3.1-19.3-6.5-28-10.2zM310 490c-13.6-7.8-19.5-37.5-14.9-75.7 1.1-9.4 2.9-19.3 5.1-29.4 19.6 4.8 41 8.5 63.5 10.9 13.5 18.5 27.5 35.3 41.6 50-32.6 30.3-63.2 46.9-84 46.9-4.5-.1-8.3-1-11.3-2.7zm237.2-76.2c4.7 38.2-1.1 67.9-14.6 75.8-3 1.8-6.9 2.6-11.5 2.6-20.7 0-51.4-16.5-84-46.6 14-14.7 28-31.4 41.3-49.9 22.6-2.4 44-6.1 63.6-11 2.3 10.1 4.1 19.8 5.2 29.1zm38.5-66.7c-8.6 3.7-18 7-27.7 10.1-5.7-19.6-13.2-40-22.5-60.9 9.2-20.8 16.6-41.1 22.2-60.6 9.9 3.1 19.3 6.5 28.1 10.2 35.4 15.1 58.3 34.9 58.3 50.6-.1 15.7-23 35.6-58.4 50.6zM320.8 78.4z"/><circle cx="420.9" cy="296.5" r="45.7"/><path d="M520.5 78.1z"/></g></svg>

package/src/react-app-env.d.ts

@@ -0,0 +1 @@
+/// <reference types="react-scripts" />

package/src/reportWebVitals.ts

@@ -0,0 +1,15 @@
+import { ReportHandler } from 'web-vitals';
+
+const reportWebVitals = (onPerfEntry?: ReportHandler) => {
+  if (onPerfEntry && onPerfEntry instanceof Function) {
+    import('web-vitals').then(({ getCLS, getFID, getFCP, getLCP, getTTFB }) => {
+      getCLS(onPerfEntry);
+      getFID(onPerfEntry);
+      getFCP(onPerfEntry);
+      getLCP(onPerfEntry);
+      getTTFB(onPerfEntry);
+    });
+  }
+};
+
+export default reportWebVitals;

package/src/setupTests.ts

@@ -0,0 +1,5 @@
+// jest-dom adds custom jest matchers for asserting on DOM nodes.
+// allows you to do things like:
+// expect(element).toHaveTextContent(/react/i)
+// learn more: https://github.com/testing-library/jest-dom
+import '@testing-library/jest-dom';

package/src/state/edge.ts

@@ -0,0 +1,35 @@
+interface BaseEdgeAction {
+  type: string;
+  src_nid: string;
+  src_ioid: string;
+  trg_nid: string;
+  trg_ioid: string;
+  from_remote: boolean;
+}
+
+interface EdgeActionAdd extends BaseEdgeAction {
+  type: "add";
+}
+
+interface EdgeActionDelete extends BaseEdgeAction {
+  type: "delete";
+}
+
+type EdgeAction = EdgeActionAdd | EdgeActionDelete;
+
+const generate_edge_id = ({
+  src_nid,
+  src_ioid,
+  trg_nid,
+  trg_ioid,
+}: {
+  src_nid: string;
+  src_ioid: string;
+  trg_nid: string;
+  trg_ioid: string;
+}) => {
+  return [`${src_nid}:${src_ioid}`, `${trg_nid}:${trg_ioid}`].sort().join("--");
+};
+
+export { generate_edge_id };
+export type { EdgeAction, EdgeActionAdd, EdgeActionDelete };