@trops/dash-core 0.1.159 → 0.1.160

package/dist/index.js

@@ -26713,7 +26713,7 @@ var useWidgetProviders = function useWidgetProviders() {
 var serverStates = new Map();
 // Map<serverName, { status, tools, resources, consumerCount }>
 
-var pendingConnects = new Map();
+var pendingConnects$1 = new Map();
 // Map<serverName, Promise<result>>
 
 /**
@@ -26928,13 +26928,13 @@ var useMcpProvider = function useMcpProvider(providerType) {
           setError(null);
 
           // 2. Another hook instance already connecting? Piggyback on its promise
-          if (!pendingConnects.has(selectedProviderName)) {
+          if (!pendingConnects$1.has(selectedProviderName)) {
             _context.next = 17;
             break;
           }
           _context.prev = 10;
           _context.next = 11;
-          return pendingConnects.get(selectedProviderName);
+          return pendingConnects$1.get(selectedProviderName);
         case 11:
           result = _context.sent;
           if (mountedRef.current) {
@@ -26976,7 +26976,7 @@ var useMcpProvider = function useMcpProvider(providerType) {
           // 3. First caller — fire the IPC call and share the promise
           connectPromise = new Promise(function (resolve, reject) {
             dashApi.mcpStartServer(selectedProviderName, provider.mcpConfig, provider.credentials, function (event, result) {
-              pendingConnects["delete"](selectedProviderName);
+              pendingConnects$1["delete"](selectedProviderName);
               if (result.error) {
                 serverStates.set(selectedProviderName, {
                   status: "error",
@@ -26997,7 +26997,7 @@ var useMcpProvider = function useMcpProvider(providerType) {
               });
               resolve(result);
             }, function (event, err) {
-              pendingConnects["delete"](selectedProviderName);
+              pendingConnects$1["delete"](selectedProviderName);
               serverStates.set(selectedProviderName, {
                 status: "error",
                 tools: [],
@@ -27007,7 +27007,7 @@ var useMcpProvider = function useMcpProvider(providerType) {
               reject(err);
             });
           });
-          pendingConnects.set(selectedProviderName, connectPromise);
+          pendingConnects$1.set(selectedProviderName, connectPromise);
           _context.prev = 18;
           _context.next = 19;
           return connectPromise;
@@ -27093,7 +27093,7 @@ var useMcpProvider = function useMcpProvider(providerType) {
           setResources([]);
           setStatus("disconnected");
           connectedRef.current = false;
-          pendingConnects["delete"](selectedProviderName);
+          pendingConnects$1["delete"](selectedProviderName);
           return _context2.abrupt("return", new Promise(function (resolve) {
             dashApi.mcpStopServer(selectedProviderName, function () {
               return resolve();
@@ -27535,6 +27535,528 @@ var useScheduler = function useScheduler() {
   };
 };
 
+/**
+ * Module-level shared state for WebSocket connections.
+ * Prevents multiple hook instances (e.g., 4 widgets using the same WS provider)
+ * from each firing their own IPC connect call.
+ *
+ * connectionStates: tracks connection status + consumer reference count per provider
+ * pendingConnects: deduplicates in-flight IPC calls so only 1 fires per provider
+ */
+var connectionStates = new Map();
+// Map<providerName, { status, consumerCount }>
+
+var pendingConnects = new Map();
+// Map<providerName, Promise<result>>
+
+var STATUS = {
+  DISCONNECTED: "disconnected",
+  CONNECTING: "connecting",
+  CONNECTED: "connected",
+  ERROR: "error"
+};
+
+/**
+ * useWebSocketProvider Hook
+ *
+ * Provides access to a WebSocket connection for a widget.
+ * Handles connection lifecycle, shared-connection ref counting,
+ * message buffering, and bidirectional communication.
+ *
+ * Mirrors useMcpProvider.js patterns exactly:
+ * - Module-level Maps (not component state) for connection sharing
+ * - consumerCount determines socket lifecycle
+ * - pendingConnects prevents duplicate connect calls during mount storms
+ *
+ * @param {string} providerType - The WebSocket provider type (e.g., "crypto-ws")
+ * @param {Object} options - Optional configuration
+ * @param {boolean} options.autoConnect - Whether to auto-connect on mount (default: true)
+ * @param {number} options.maxMessages - Max messages in buffer (default: 100)
+ *
+ * @returns {Object} WebSocket provider interface
+ */
+var useWebSocketProvider = function useWebSocketProvider(providerType) {
+  var _app$providers;
+  var options = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : {};
+  var _options$autoConnect = options.autoConnect,
+    autoConnect = _options$autoConnect === void 0 ? true : _options$autoConnect,
+    _options$maxMessages = options.maxMessages,
+    maxMessages = _options$maxMessages === void 0 ? 100 : _options$maxMessages;
+  var app = React.useContext(AppContext);
+  var workspace = React.useContext(WorkspaceContext);
+  var widgetContext = React.useContext(WidgetContext);
+  var _useState = React.useState(false),
+    _useState2 = _slicedToArray(_useState, 2),
+    isConnected = _useState2[0],
+    setIsConnected = _useState2[1];
+  var _useState3 = React.useState(false),
+    _useState4 = _slicedToArray(_useState3, 2),
+    isConnecting = _useState4[0],
+    setIsConnecting = _useState4[1];
+  var _useState5 = React.useState(null),
+    _useState6 = _slicedToArray(_useState5, 2),
+    error = _useState6[0],
+    setError = _useState6[1];
+  var _useState7 = React.useState(null),
+    _useState8 = _slicedToArray(_useState7, 2),
+    lastMessage = _useState8[0],
+    setLastMessage = _useState8[1];
+  var _useState9 = React.useState([]),
+    _useState0 = _slicedToArray(_useState9, 2),
+    messages = _useState0[0],
+    setMessages = _useState0[1];
+  var _useState1 = React.useState(STATUS.DISCONNECTED),
+    _useState10 = _slicedToArray(_useState1, 2),
+    status = _useState10[0],
+    setStatus = _useState10[1];
+  var connectedRef = React.useRef(false);
+  var mountedRef = React.useRef(true);
+  var messagesRef = React.useRef([]);
+  var maxMessagesRef = React.useRef(maxMessages);
+  maxMessagesRef.current = maxMessages;
+  var dashApi = app === null || app === void 0 ? void 0 : app.dashApi;
+
+  // Get the widget data
+  var widgetData = widgetContext === null || widgetContext === void 0 ? void 0 : widgetContext.widgetData;
+
+  // Get the selected WebSocket provider for this widget
+  // Same two-layer lookup as useMcpProvider:
+  // 1. Widget-level: stored directly on the layout item
+  // 2. Workspace-level: stored as workspace.selectedProviders[widgetId][providerType]
+  var widgetId = widgetData === null || widgetData === void 0 ? void 0 : widgetData.uuidString;
+  var selectedProviderName = function (_widgetData$selectedP, _workspace$workspaceD) {
+    if (widgetData !== null && widgetData !== void 0 && (_widgetData$selectedP = widgetData.selectedProviders) !== null && _widgetData$selectedP !== void 0 && _widgetData$selectedP[providerType]) {
+      return widgetData.selectedProviders[providerType];
+    }
+    if (widgetId && workspace !== null && workspace !== void 0 && (_workspace$workspaceD = workspace.workspaceData) !== null && _workspace$workspaceD !== void 0 && (_workspace$workspaceD = _workspace$workspaceD.selectedProviders) !== null && _workspace$workspaceD !== void 0 && (_workspace$workspaceD = _workspace$workspaceD[widgetId]) !== null && _workspace$workspaceD !== void 0 && _workspace$workspaceD[providerType]) {
+      return workspace.workspaceData.selectedProviders[widgetId][providerType];
+    }
+    return null;
+  }();
+
+  // Get the provider data (including credentials)
+  // Read from AppContext.providers (not DashboardContext)
+  var provider = selectedProviderName ? app === null || app === void 0 || (_app$providers = app.providers) === null || _app$providers === void 0 ? void 0 : _app$providers[selectedProviderName] : null;
+
+  /**
+   * Apply connection result to this hook instance's local state.
+   */
+  var applyConnected = React.useCallback(function () {
+    if (!mountedRef.current) return;
+    setIsConnected(true);
+    setIsConnecting(false);
+    setStatus(STATUS.CONNECTED);
+    connectedRef.current = true;
+  }, []);
+
+  /**
+   * Connect to the WebSocket server.
+   * Uses module-level deduplication so only one IPC call fires per provider,
+   * even when multiple hook instances call connect() simultaneously.
+   */
+  var connect = React.useCallback(/*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime.mark(function _callee2() {
+    var _provider$wsConfig;
+    var cached, statusResult, result, state, connectPromise, _result2, _t3, _t4;
+    return _regeneratorRuntime.wrap(function (_context2) {
+      while (1) switch (_context2.prev = _context2.next) {
+        case 0:
+          if (!connectedRef.current) {
+            _context2.next = 1;
+            break;
+          }
+          return _context2.abrupt("return");
+        case 1:
+          if (!(!(dashApi !== null && dashApi !== void 0 && dashApi.webSocket) || !provider)) {
+            _context2.next = 2;
+            break;
+          }
+          setError(!provider ? "No ".concat(providerType, " WebSocket provider selected for this widget") : "Dashboard API not available");
+          return _context2.abrupt("return");
+        case 2:
+          if (!(provider.providerClass !== "websocket")) {
+            _context2.next = 3;
+            break;
+          }
+          setError("Provider \"".concat(selectedProviderName, "\" is not a WebSocket provider"));
+          return _context2.abrupt("return");
+        case 3:
+          if ((_provider$wsConfig = provider.wsConfig) !== null && _provider$wsConfig !== void 0 && _provider$wsConfig.url) {
+            _context2.next = 4;
+            break;
+          }
+          setError("Provider \"".concat(selectedProviderName, "\" has no WebSocket URL configured"));
+          return _context2.abrupt("return");
+        case 4:
+          // 1. Already connected at module level? Verify with main process.
+          cached = connectionStates.get(selectedProviderName);
+          if (!(cached && cached.status === STATUS.CONNECTED)) {
+            _context2.next = 9;
+            break;
+          }
+          _context2.prev = 5;
+          _context2.next = 6;
+          return dashApi.webSocket.getStatus(selectedProviderName);
+        case 6:
+          statusResult = _context2.sent;
+          if (!((statusResult === null || statusResult === void 0 ? void 0 : statusResult.status) === "connected")) {
+            _context2.next = 7;
+            break;
+          }
+          cached.consumerCount++;
+          applyConnected();
+          return _context2.abrupt("return");
+        case 7:
+          // Server was stopped externally — clear stale cache and reconnect
+          connectionStates["delete"](selectedProviderName);
+          _context2.next = 9;
+          break;
+        case 8:
+          _context2.prev = 8;
+          _context2["catch"](5);
+          connectionStates["delete"](selectedProviderName);
+        case 9:
+          setIsConnecting(true);
+          setError(null);
+          setStatus(STATUS.CONNECTING);
+
+          // 2. Another hook instance already connecting? Piggyback on its promise
+          if (!pendingConnects.has(selectedProviderName)) {
+            _context2.next = 17;
+            break;
+          }
+          _context2.prev = 10;
+          _context2.next = 11;
+          return pendingConnects.get(selectedProviderName);
+        case 11:
+          result = _context2.sent;
+          if (mountedRef.current) {
+            _context2.next = 12;
+            break;
+          }
+          return _context2.abrupt("return");
+        case 12:
+          if (!result.error) {
+            _context2.next = 13;
+            break;
+          }
+          setError(result.message);
+          setIsConnecting(false);
+          setStatus(STATUS.ERROR);
+          return _context2.abrupt("return");
+        case 13:
+          state = connectionStates.get(selectedProviderName);
+          if (state) state.consumerCount++;
+          applyConnected();
+          _context2.next = 16;
+          break;
+        case 14:
+          _context2.prev = 14;
+          _t3 = _context2["catch"](10);
+          if (mountedRef.current) {
+            _context2.next = 15;
+            break;
+          }
+          return _context2.abrupt("return");
+        case 15:
+          setError((_t3 === null || _t3 === void 0 ? void 0 : _t3.message) || "Failed to connect to WebSocket server");
+          setIsConnecting(false);
+          setStatus(STATUS.ERROR);
+        case 16:
+          return _context2.abrupt("return");
+        case 17:
+          // 3. First caller — fire the IPC call and share the promise
+          connectPromise = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime.mark(function _callee() {
+            var _result, _t;
+            return _regeneratorRuntime.wrap(function (_context) {
+              while (1) switch (_context.prev = _context.next) {
+                case 0:
+                  _context.prev = 0;
+                  _context.next = 1;
+                  return dashApi.webSocket.connect(selectedProviderName, {
+                    url: provider.wsConfig.url,
+                    headers: provider.wsConfig.headers || null,
+                    subprotocols: provider.wsConfig.subprotocols || null,
+                    credentials: provider.credentials || null
+                  });
+                case 1:
+                  _result = _context.sent;
+                  pendingConnects["delete"](selectedProviderName);
+                  if (!_result.error) {
+                    _context.next = 2;
+                    break;
+                  }
+                  connectionStates.set(selectedProviderName, {
+                    status: STATUS.ERROR,
+                    consumerCount: 0
+                  });
+                  return _context.abrupt("return", _result);
+                case 2:
+                  connectionStates.set(selectedProviderName, {
+                    status: STATUS.CONNECTED,
+                    consumerCount: 1
+                  });
+                  return _context.abrupt("return", _result);
+                case 3:
+                  _context.prev = 3;
+                  _t = _context["catch"](0);
+                  pendingConnects["delete"](selectedProviderName);
+                  connectionStates.set(selectedProviderName, {
+                    status: STATUS.ERROR,
+                    consumerCount: 0
+                  });
+                  throw _t;
+                case 4:
+                case "end":
+                  return _context.stop();
+              }
+            }, _callee, null, [[0, 3]]);
+          }))();
+          pendingConnects.set(selectedProviderName, connectPromise);
+          _context2.prev = 18;
+          _context2.next = 19;
+          return connectPromise;
+        case 19:
+          _result2 = _context2.sent;
+          if (mountedRef.current) {
+            _context2.next = 20;
+            break;
+          }
+          return _context2.abrupt("return");
+        case 20:
+          if (!_result2.error) {
+            _context2.next = 21;
+            break;
+          }
+          setError(_result2.message);
+          setIsConnecting(false);
+          setStatus(STATUS.ERROR);
+          return _context2.abrupt("return");
+        case 21:
+          applyConnected();
+          _context2.next = 24;
+          break;
+        case 22:
+          _context2.prev = 22;
+          _t4 = _context2["catch"](18);
+          if (mountedRef.current) {
+            _context2.next = 23;
+            break;
+          }
+          return _context2.abrupt("return");
+        case 23:
+          setError((_t4 === null || _t4 === void 0 ? void 0 : _t4.message) || "Failed to connect to WebSocket server");
+          setIsConnecting(false);
+          setStatus(STATUS.ERROR);
+        case 24:
+        case "end":
+          return _context2.stop();
+      }
+    }, _callee2, null, [[5, 8], [10, 14], [18, 22]]);
+  })), [dashApi, provider, providerType, selectedProviderName, applyConnected]);
+
+  /**
+   * Disconnect from the WebSocket server.
+   * Only sends the IPC disconnect call when this is the last consumer.
+   */
+  var disconnect = React.useCallback(/*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime.mark(function _callee3() {
+    var state;
+    return _regeneratorRuntime.wrap(function (_context3) {
+      while (1) switch (_context3.prev = _context3.next) {
+        case 0:
+          if (!(!(dashApi !== null && dashApi !== void 0 && dashApi.webSocket) || !selectedProviderName)) {
+            _context3.next = 1;
+            break;
+          }
+          return _context3.abrupt("return");
+        case 1:
+          state = connectionStates.get(selectedProviderName);
+          if (!state) {
+            _context3.next = 3;
+            break;
+          }
+          state.consumerCount = Math.max(0, state.consumerCount - 1);
+          if (!(state.consumerCount > 0)) {
+            _context3.next = 2;
+            break;
+          }
+          // Other widgets still using this connection — just update local state
+          setIsConnected(false);
+          setLastMessage(null);
+          setMessages([]);
+          messagesRef.current = [];
+          setStatus(STATUS.DISCONNECTED);
+          connectedRef.current = false;
+          return _context3.abrupt("return");
+        case 2:
+          // Last consumer — actually disconnect
+          connectionStates["delete"](selectedProviderName);
+        case 3:
+          // Clear state synchronously before the IPC call
+          setIsConnected(false);
+          setLastMessage(null);
+          setMessages([]);
+          messagesRef.current = [];
+          setStatus(STATUS.DISCONNECTED);
+          connectedRef.current = false;
+          pendingConnects["delete"](selectedProviderName);
+          _context3.prev = 4;
+          _context3.next = 5;
+          return dashApi.webSocket.disconnect(selectedProviderName);
+        case 5:
+          _context3.next = 7;
+          break;
+        case 6:
+          _context3.prev = 6;
+          _context3["catch"](4);
+        case 7:
+        case "end":
+          return _context3.stop();
+      }
+    }, _callee3, null, [[4, 6]]);
+  })), [dashApi, selectedProviderName]);
+
+  /**
+   * Send data through the WebSocket connection
+   */
+  var send = React.useCallback(/*#__PURE__*/function () {
+    var _ref4 = _asyncToGenerator(/*#__PURE__*/_regeneratorRuntime.mark(function _callee4(data) {
+      var result;
+      return _regeneratorRuntime.wrap(function (_context4) {
+        while (1) switch (_context4.prev = _context4.next) {
+          case 0:
+            if (!(!(dashApi !== null && dashApi !== void 0 && dashApi.webSocket) || !selectedProviderName)) {
+              _context4.next = 1;
+              break;
+            }
+            throw new Error("WebSocket not connected");
+          case 1:
+            _context4.next = 2;
+            return dashApi.webSocket.send(selectedProviderName, data);
+          case 2:
+            result = _context4.sent;
+            if (!result.error) {
+              _context4.next = 3;
+              break;
+            }
+            throw new Error(result.message);
+          case 3:
+            return _context4.abrupt("return", result);
+          case 4:
+          case "end":
+            return _context4.stop();
+        }
+      }, _callee4);
+    }));
+    return function (_x) {
+      return _ref4.apply(this, arguments);
+    };
+  }(), [dashApi, selectedProviderName]);
+
+  // Keep a ref to connect so the auto-connect effect doesn't depend on it
+  var connectRef = React.useRef(connect);
+  connectRef.current = connect;
+
+  // Listen for incoming messages from main process
+  React.useEffect(function () {
+    if (!(dashApi !== null && dashApi !== void 0 && dashApi.webSocket) || !selectedProviderName) return;
+    var handleMessage = function handleMessage(_event, payload) {
+      if (payload.provider !== selectedProviderName) return;
+      if (!mountedRef.current) return;
+      var msg = payload.data;
+
+      // Update circular buffer
+      var next = [].concat(_toConsumableArray(messagesRef.current), [msg]);
+      if (next.length > maxMessagesRef.current) {
+        next.splice(0, next.length - maxMessagesRef.current);
+      }
+      messagesRef.current = next;
+      setMessages(next);
+      setLastMessage(msg);
+    };
+    dashApi.webSocket.onMessage(handleMessage);
+    return function () {
+      return dashApi.webSocket.offMessage(handleMessage);
+    };
+  }, [dashApi, selectedProviderName]);
+
+  // Listen for status changes from main process
+  React.useEffect(function () {
+    if (!(dashApi !== null && dashApi !== void 0 && dashApi.webSocket) || !selectedProviderName) return;
+    var handleStatusChange = function handleStatusChange(_event, payload) {
+      if (payload.provider !== selectedProviderName) return;
+      if (!mountedRef.current) return;
+      var newStatus = payload.status;
+      setStatus(newStatus);
+      if (newStatus === STATUS.CONNECTED) {
+        setIsConnected(true);
+        setIsConnecting(false);
+        setError(null);
+        connectedRef.current = true;
+      } else if (newStatus === STATUS.DISCONNECTED) {
+        setIsConnected(false);
+        setIsConnecting(false);
+        connectedRef.current = false;
+      } else if (newStatus === STATUS.ERROR) {
+        setIsConnected(false);
+        setIsConnecting(false);
+        setError(payload.error || "WebSocket error");
+        connectedRef.current = false;
+      } else if (newStatus === STATUS.CONNECTING) {
+        setIsConnecting(true);
+      }
+    };
+    dashApi.webSocket.onStatusChange(handleStatusChange);
+    return function () {
+      return dashApi.webSocket.offStatusChange(handleStatusChange);
+    };
+  }, [dashApi, selectedProviderName]);
+
+  // Auto-connect on mount or when provider selection changes
+  React.useEffect(function () {
+    if (autoConnect && selectedProviderName && !connectedRef.current) {
+      connectRef.current();
+    }
+    // eslint-disable-next-line react-hooks/exhaustive-deps
+  }, [autoConnect, selectedProviderName]);
+
+  // Track mounted state and cleanup on unmount
+  React.useEffect(function () {
+    mountedRef.current = true;
+    return function () {
+      mountedRef.current = false;
+
+      // Decrement consumer count; only disconnect if last consumer
+      if (connectedRef.current && dashApi !== null && dashApi !== void 0 && dashApi.webSocket && selectedProviderName) {
+        var state = connectionStates.get(selectedProviderName);
+        if (state) {
+          state.consumerCount = Math.max(0, state.consumerCount - 1);
+          if (state.consumerCount > 0) {
+            // Other widgets still using this connection — don't disconnect
+            return;
+          }
+
+          // Last consumer — disconnect
+          connectionStates["delete"](selectedProviderName);
+        }
+        dashApi.webSocket.disconnect(selectedProviderName)["catch"](function () {});
+      }
+    };
+  }, [dashApi, selectedProviderName]);
+  return {
+    isConnected: isConnected,
+    isConnecting: isConnecting,
+    error: error,
+    lastMessage: lastMessage,
+    messages: messages,
+    send: send,
+    connect: connect,
+    disconnect: disconnect,
+    status: status,
+    provider: provider,
+    serverName: selectedProviderName
+  };
+};
+
 function ownKeys$h(e, r) { var t = Object.keys(e); if (Object.getOwnPropertySymbols) { var o = Object.getOwnPropertySymbols(e); r && (o = o.filter(function (r) { return Object.getOwnPropertyDescriptor(e, r).enumerable; })), t.push.apply(t, o); } return t; }
 function _objectSpread$h(e) { for (var r = 1; r < arguments.length; r++) { var t = null != arguments[r] ? arguments[r] : {}; r % 2 ? ownKeys$h(Object(t), !0).forEach(function (r) { _defineProperty(e, r, t[r]); }) : Object.getOwnPropertyDescriptors ? Object.defineProperties(e, Object.getOwnPropertyDescriptors(t)) : ownKeys$h(Object(t)).forEach(function (r) { Object.defineProperty(e, r, Object.getOwnPropertyDescriptor(t, r)); }); } return e; }
 var PreviewComponentsPane = function PreviewComponentsPane(_ref) {
@@ -46555,6 +47077,7 @@ exports.useNotifications = useNotifications;
 exports.useProvider = useProvider;
 exports.useProviderClient = useProviderClient;
 exports.useScheduler = useScheduler;
+exports.useWebSocketProvider = useWebSocketProvider;
 exports.useWidgetEvents = useWidgetEvents;
 exports.useWidgetProviders = useWidgetProviders;
 exports.useWidgetSchedulerStatus = useWidgetSchedulerStatus;