akeyless-client-commons 1.1.22 → 1.1.24

package/dist/hooks/index.js

@@ -1,4 +1,15 @@
 "use strict";
+function _array_like_to_array(arr, len) {
+    if (len == null || len > arr.length) len = arr.length;
+    for(var i = 0, arr2 = new Array(len); i < len; i++)arr2[i] = arr[i];
+    return arr2;
+}
+function _array_with_holes(arr) {
+    if (Array.isArray(arr)) return arr;
+}
+function _array_without_holes(arr) {
+    if (Array.isArray(arr)) return _array_like_to_array(arr);
+}
 function asyncGeneratorStep(gen, resolve, reject, _next, _throw, key, arg) {
     try {
         var info = gen[key](arg);
@@ -28,6 +39,25 @@ function _async_to_generator(fn) {
         });
     };
 }
+function _class_call_check(instance, Constructor) {
+    if (!(instance instanceof Constructor)) {
+        throw new TypeError("Cannot call a class as a function");
+    }
+}
+function _defineProperties(target, props) {
+    for(var i = 0; i < props.length; i++){
+        var descriptor = props[i];
+        descriptor.enumerable = descriptor.enumerable || false;
+        descriptor.configurable = true;
+        if ("value" in descriptor) descriptor.writable = true;
+        Object.defineProperty(target, descriptor.key, descriptor);
+    }
+}
+function _create_class(Constructor, protoProps, staticProps) {
+    if (protoProps) _defineProperties(Constructor.prototype, protoProps);
+    if (staticProps) _defineProperties(Constructor, staticProps);
+    return Constructor;
+}
 function _define_property(obj, key, value) {
     if (key in obj) {
         Object.defineProperty(obj, key, {
@@ -41,6 +71,39 @@ function _define_property(obj, key, value) {
     }
     return obj;
 }
+function _iterable_to_array(iter) {
+    if (typeof Symbol !== "undefined" && iter[Symbol.iterator] != null || iter["@@iterator"] != null) return Array.from(iter);
+}
+function _iterable_to_array_limit(arr, i) {
+    var _i = arr == null ? null : typeof Symbol !== "undefined" && arr[Symbol.iterator] || arr["@@iterator"];
+    if (_i == null) return;
+    var _arr = [];
+    var _n = true;
+    var _d = false;
+    var _s, _e;
+    try {
+        for(_i = _i.call(arr); !(_n = (_s = _i.next()).done); _n = true){
+            _arr.push(_s.value);
+            if (i && _arr.length === i) break;
+        }
+    } catch (err) {
+        _d = true;
+        _e = err;
+    } finally{
+        try {
+            if (!_n && _i["return"] != null) _i["return"]();
+        } finally{
+            if (_d) throw _e;
+        }
+    }
+    return _arr;
+}
+function _non_iterable_rest() {
+    throw new TypeError("Invalid attempt to destructure non-iterable instance.\\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method.");
+}
+function _non_iterable_spread() {
+    throw new TypeError("Invalid attempt to spread non-iterable instance.\\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method.");
+}
 function _object_spread(target) {
     for(var i = 1; i < arguments.length; i++){
         var source = arguments[i] != null ? arguments[i] : {};
@@ -80,10 +143,24 @@ function _object_spread_props(target, source) {
     }
     return target;
 }
+function _sliced_to_array(arr, i) {
+    return _array_with_holes(arr) || _iterable_to_array_limit(arr, i) || _unsupported_iterable_to_array(arr, i) || _non_iterable_rest();
+}
+function _to_consumable_array(arr) {
+    return _array_without_holes(arr) || _iterable_to_array(arr) || _unsupported_iterable_to_array(arr) || _non_iterable_spread();
+}
 function _type_of(obj) {
     "@swc/helpers - typeof";
     return obj && typeof Symbol !== "undefined" && obj.constructor === Symbol ? "symbol" : typeof obj;
 }
+function _unsupported_iterable_to_array(o, minLen) {
+    if (!o) return;
+    if (typeof o === "string") return _array_like_to_array(o, minLen);
+    var n = Object.prototype.toString.call(o).slice(8, -1);
+    if (n === "Object" && o.constructor) n = o.constructor.name;
+    if (n === "Map" || n === "Set") return Array.from(n);
+    if (n === "Arguments" || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(n)) return _array_like_to_array(o, minLen);
+}
 function _ts_generator(thisArg, body) {
     var f, y, t, g, _ = {
         label: 0,
@@ -240,6 +317,9 @@ var __toCommonJS = function(mod) {
 // src/hooks/index.ts
 var index_exports = {};
 __export(index_exports, {
+    useDbSnapshots: function() {
+        return useDbSnapshots;
+    },
     useDeepCompareEffect: function() {
         return useDeepCompareEffect;
     },
@@ -255,14 +335,17 @@ __export(index_exports, {
     useSetUserCountry: function() {
         return useSetUserCountry;
     },
-    useSnapshotBulk: function() {
-        return useSnapshotBulk;
+    useSmartSnapshots: function() {
+        return useSmartSnapshots;
+    },
+    useSocketSnapshots: function() {
+        return useSocketSnapshots;
     }
 });
 module.exports = __toCommonJS(index_exports);
 // src/hooks/global.ts
 var import_akeyless_types_commons2 = require("akeyless-types-commons");
-var import_react3 = require("react");
+var import_react2 = require("react");
 // src/helpers/firebase.ts
 var import_moment = __toESM(require("moment"));
 var import_app = require("firebase/app");
@@ -367,6 +450,50 @@ var simpleExtractData = function(doc2) {
         id: doc2.id
     });
 };
+var get_document_by_id = /*#__PURE__*/ function() {
+    var _ref = _async_to_generator(function(collection_path, doc_id) {
+        var doc_ref, doc_snap, error;
+        return _ts_generator(this, function(_state) {
+            switch(_state.label){
+                case 0:
+                    _state.trys.push([
+                        0,
+                        2,
+                        ,
+                        3
+                    ]);
+                    doc_ref = (0, import_firestore.doc)(db, collection_path, doc_id);
+                    return [
+                        4,
+                        (0, import_firestore.getDoc)(doc_ref)
+                    ];
+                case 1:
+                    doc_snap = _state.sent();
+                    if (!doc_snap.exists()) {
+                        throw new Error("Document not found, document id: " + doc_id);
+                    }
+                    return [
+                        2,
+                        simpleExtractData(doc_snap)
+                    ];
+                case 2:
+                    error = _state.sent();
+                    console.error("Error from get_document_by_id", error);
+                    return [
+                        2,
+                        null
+                    ];
+                case 3:
+                    return [
+                        2
+                    ];
+            }
+        });
+    });
+    return function get_document_by_id(collection_path, doc_id) {
+        return _ref.apply(this, arguments);
+    };
+}();
 var snapshot = function(config, snapshotsFirstTime, settings) {
     var resolvePromise;
     var isResolved = false;
@@ -392,11 +519,11 @@ var snapshot = function(config, snapshotsFirstTime, settings) {
             collectionRef = (0, import_firestore.query)(collectionRef, (0, import_firestore.orderBy)(order.fieldName, order.direction));
         });
     }
-    var unsubscribe = (0, import_firestore.onSnapshot)(collectionRef, function(snapshot2) {
+    var unsubscribe = (0, import_firestore.onSnapshot)(collectionRef, function(snapshot3) {
         if (!snapshotsFirstTime.includes(config.collectionName)) {
             var _config_onFirstTime, _config_extraParsers;
             snapshotsFirstTime.push(config.collectionName);
-            var documents = snapshot2.docs.map(function(doc2) {
+            var documents = snapshot3.docs.map(function(doc2) {
                 return simpleExtractData(doc2);
             });
             (_config_onFirstTime = config.onFirstTime) === null || _config_onFirstTime === void 0 ? void 0 : _config_onFirstTime.call(config, documents, config);
@@ -410,7 +537,7 @@ var snapshot = function(config, snapshotsFirstTime, settings) {
             var addedDocs = [];
             var modifiedDocs = [];
             var removedDocs = [];
-            snapshot2.docChanges().forEach(function(change) {
+            snapshot3.docChanges().forEach(function(change) {
                 if (change.type === "added") {
                     addedDocs.push(simpleExtractData(change.doc));
                 } else if (change.type === "modified") {
@@ -514,13 +641,363 @@ var biDomain = isLocal ? "http://localhost:9002/api/bi" : baseDomain + "/bi";
 var notificationsDomain = isLocal ? "http://localhost:9006/api/notifications" : baseDomain + "/notifications";
 var callCenterGeoDomain = isLocal ? "http://localhost:9007/api/call-center/geo" : baseDomain + "/call-center/geo";
 var callCenterEventsDomain = isLocal ? "http://localhost:9008/api/call-center/events" : baseDomain + "/call-center/events";
+var dataSocketDomain = isLocal ? "http://localhost:9009/api/data-socket" : baseDomain + "/data-socket";
+var dataSyncDomain = isLocal ? "http://localhost:9010/api/data-sync" : baseDomain + "/data-sync";
 // src/helpers/emails.ts
 var import_axios3 = __toESM(require("axios"));
+// src/helpers/socket.ts
+var import_socket = require("socket.io-client");
+var SESSION_STORAGE_KEY = "sessionId";
+var SocketService = /*#__PURE__*/ function() {
+    "use strict";
+    function _SocketService() {
+        _class_call_check(this, _SocketService);
+        this.socket = null;
+        this.connectCallbacks = [];
+        this.disconnectCallbacks = [];
+        this.authToken = null;
+    }
+    _create_class(_SocketService, [
+        {
+            /// Initialize the socket connection
+            key: "initSocket",
+            value: function initSocket() {
+                var _this = this;
+                if (!this.socket) {
+                    var socketUrl = isLocal ? "http://localhost:9009" : mode === "qa" ? "https://nx-api.xyz" : "https://nx-api.info";
+                    this.socket = (0, import_socket.io)(socketUrl, {
+                        path: "/api/data-socket/connect",
+                        auth: function(cb) {
+                            var sessionId = localStorage.getItem(SESSION_STORAGE_KEY) || void 0;
+                            var token = _this.authToken;
+                            var authPayload = {};
+                            if (token) authPayload.token = token;
+                            if (sessionId) authPayload.sessionId = sessionId;
+                            cb(authPayload);
+                        },
+                        transports: [
+                            "websocket"
+                        ],
+                        reconnection: true,
+                        reconnectionAttempts: 30,
+                        reconnectionDelay: 2 * 1e3
+                    });
+                    this.socket.on("connect", function() {
+                        var _this_socket, _this_socket1;
+                        console.log("\uD83D\uDFE2 Socket connected: ".concat((_this_socket = _this.socket) === null || _this_socket === void 0 ? void 0 : _this_socket.id, " (recovered - ").concat((_this_socket1 = _this.socket) === null || _this_socket1 === void 0 ? void 0 : _this_socket1.recovered, ")"));
+                        _this.connectCallbacks.forEach(function(cb) {
+                            return cb();
+                        });
+                    });
+                    this.socket.on("disconnect", function(reason) {
+                        console.log("Socket disconnected:", reason);
+                        _this.disconnectCallbacks.forEach(function(cb) {
+                            return cb();
+                        });
+                    });
+                    this.socket.on("session", function(param) {
+                        var session_id = param.session_id;
+                        if (session_id) {
+                            localStorage.setItem(SESSION_STORAGE_KEY, session_id);
+                        }
+                    });
+                    this.socket.on("connect_error", function(error) {
+                        console.error("Socket connection error:", error);
+                    });
+                }
+            }
+        },
+        {
+            /// get socket instance
+            key: "getSocketInstance",
+            value: function getSocketInstance() {
+                if (!this.socket) {
+                    this.initSocket();
+                }
+                if (!this.socket) {
+                    throw new Error("Socket not initialized");
+                }
+                if (!this.socket.connected) {
+                    this.socket.connect();
+                }
+                return this.socket;
+            }
+        },
+        {
+            /// connection management methods
+            key: "startSession",
+            value: function startSession(token) {
+                this.setAuthToken(token);
+                this.initSocket();
+            }
+        },
+        {
+            key: "onConnect",
+            value: function onConnect(callback) {
+                var _this = this;
+                var _this_socket;
+                if (!this.connectCallbacks.includes(callback)) {
+                    this.connectCallbacks.push(callback);
+                }
+                if ((_this_socket = this.socket) === null || _this_socket === void 0 ? void 0 : _this_socket.connected) {
+                    callback();
+                }
+                return function() {
+                    return _this.offConnect(callback);
+                };
+            }
+        },
+        {
+            key: "offConnect",
+            value: function offConnect(callback) {
+                this.connectCallbacks = this.connectCallbacks.filter(function(cb) {
+                    return cb !== callback;
+                });
+            }
+        },
+        {
+            key: "onDisconnect",
+            value: function onDisconnect(callback) {
+                var _this = this;
+                if (!this.disconnectCallbacks.includes(callback)) {
+                    this.disconnectCallbacks.push(callback);
+                }
+                if (this.socket && !this.socket.connected) {
+                    callback();
+                }
+                return function() {
+                    return _this.offDisconnect(callback);
+                };
+            }
+        },
+        {
+            key: "offDisconnect",
+            value: function offDisconnect(callback) {
+                this.disconnectCallbacks = this.disconnectCallbacks.filter(function(cb) {
+                    return cb !== callback;
+                });
+            }
+        },
+        {
+            key: "isConnected",
+            value: function isConnected() {
+                var _this_socket;
+                return ((_this_socket = this.socket) === null || _this_socket === void 0 ? void 0 : _this_socket.connected) || false;
+            }
+        },
+        {
+            key: "setAuthToken",
+            value: function setAuthToken(token) {
+                this.authToken = token;
+                if (this.socket) {
+                    this.socket.connect();
+                }
+            }
+        },
+        {
+            key: "disconnectSocket",
+            value: function disconnectSocket() {
+                if (this.socket) {
+                    this.socket.io.engine.close();
+                }
+            }
+        },
+        {
+            /// subscribe to collections
+            key: "subscribeToCollections",
+            value: function subscribeToCollections(config) {
+                var _this = this;
+                if (config.length === 0) {
+                    return function() {};
+                }
+                var s = this.getSocketInstance();
+                var collectionsNames = config.map(function(c) {
+                    return c.collectionName;
+                });
+                var eventHandlers = [];
+                config.forEach(function(configuration) {
+                    var collectionName = configuration.collectionName, onAdd = configuration.onAdd, onFirstTime = configuration.onFirstTime, onModify = configuration.onModify, onRemove = configuration.onRemove, extraParsers = configuration.extraParsers, conditions = configuration.conditions, orderBy2 = configuration.orderBy;
+                    var attach = function(eventName, handler) {
+                        _this.socket.off(eventName, handler);
+                        _this.socket.on(eventName, handler);
+                        eventHandlers.push({
+                            eventName: eventName,
+                            handler: handler
+                        });
+                    };
+                    attach("initial:".concat(collectionName), onFirstTime);
+                    attach("add:".concat(collectionName), onAdd);
+                    attach("update:".concat(collectionName), onModify);
+                    attach("delete:".concat(collectionName), onRemove);
+                    extraParsers === null || extraParsers === void 0 ? void 0 : extraParsers.forEach(function(parsers) {
+                        var extraOnAdd = parsers.onAdd, extraOnFirstTime = parsers.onFirstTime, extraOnModify = parsers.onModify, extraOnRemove = parsers.onRemove;
+                        attach("initial:".concat(collectionName), extraOnFirstTime);
+                        attach("add:".concat(collectionName), extraOnAdd);
+                        attach("update:".concat(collectionName), extraOnModify);
+                        attach("delete:".concat(collectionName), extraOnRemove);
+                    });
+                });
+                s.emit("subscribe_collections", collectionsNames, function(callback) {
+                    if (callback.success) {
+                        console.log("Successfully subscribed to: ".concat(collectionsNames.join(", ")));
+                    } else {
+                        console.error("Failed to subscribe to ".concat(config.join(", "), ": ").concat(callback.message));
+                    }
+                });
+                return function() {
+                    console.log("Cleaning up subscriptions for: ".concat(collectionsNames.join(", ")));
+                    s.emit("unsubscribe_collections", collectionsNames);
+                    eventHandlers.forEach(function(eh) {
+                        s.off(eh.eventName, eh.handler);
+                    });
+                };
+            }
+        },
+        {
+            /// set data
+            key: "setData",
+            value: function setData(payload) {
+                var s = this.getSocketInstance();
+                return new Promise(function(resolve, reject) {
+                    s.emit("set_data", payload, function(callback) {
+                        if (callback.success) {
+                            console.log("Data saved successfully:", payload);
+                            console.log("ack", callback);
+                            resolve(callback);
+                        } else {
+                            reject(new Error(callback.message || "Save operation failed"));
+                        }
+                    });
+                });
+            }
+        },
+        {
+            /// get data
+            key: "getCollectionData",
+            value: function getCollectionData(payload) {
+                var s = this.getSocketInstance();
+                s.emit("get_data", {
+                    collection_name: payload.collection_name
+                }, function(socketCallback) {
+                    if (socketCallback.success && socketCallback.data) {
+                        payload.callback(socketCallback.data);
+                    } else {
+                        payload.callback(payload.defaultValue);
+                    }
+                });
+            }
+        },
+        {
+            key: "getDocumentData",
+            value: function getDocumentData(payload) {
+                var s = this.getSocketInstance();
+                s.emit("get_data", {
+                    collection_name: payload.collection_name,
+                    key: payload.key
+                }, function(socketCallback) {
+                    if (socketCallback.success && socketCallback.data) {
+                        payload.callback(socketCallback.data);
+                    } else {
+                        payload.callback(payload.defaultValue);
+                    }
+                });
+            }
+        },
+        {
+            /// delete data
+            key: "deleteData",
+            value: function deleteData(payload) {
+                var s = this.getSocketInstance();
+                return new Promise(function(resolve, reject) {
+                    s.emit("delete_data", payload, function(callback) {
+                        if (callback.success) {
+                            console.log("Data deleted successfully:", payload);
+                            console.log("delete ack", callback);
+                            resolve(callback);
+                        } else {
+                            reject(new Error(callback.message || "Delete operation failed"));
+                        }
+                    });
+                });
+            }
+        },
+        {
+            key: "clearAllRedisData",
+            value: function clearAllRedisData() {
+                var s = this.getSocketInstance();
+                return new Promise(function(resolve, reject) {
+                    s.emit("clear_all_redis_data", function(ack) {
+                        if (ack.success) {
+                            resolve(ack);
+                        } else {
+                            reject(new Error(ack.message || "Clear all Redis data operation failed"));
+                        }
+                    });
+                });
+            }
+        }
+    ], [
+        {
+            key: "getInstance",
+            value: function getInstance() {
+                if (!_SocketService.instance) {
+                    _SocketService.instance = new _SocketService();
+                }
+                return _SocketService.instance;
+            }
+        }
+    ]);
+    return _SocketService;
+}();
+var socketServiceInstance = SocketService.getInstance();
+// src/hooks/global.ts
+var useDocumentTitle = function(title) {
+    (0, import_react2.useEffect)(function() {
+        document.title = title;
+    }, [
+        title
+    ]);
+    return null;
+};
+var useSetUserCountry = function(setUserCountry, changLang) {
+    (0, import_react2.useLayoutEffect)(function() {
+        var currentCountry = localStorage.getItem("userCountry");
+        if (!currentCountry) {
+            var updateCountry = /*#__PURE__*/ function() {
+                var _ref = _async_to_generator(function() {
+                    var country;
+                    return _ts_generator(this, function(_state) {
+                        switch(_state.label){
+                            case 0:
+                                return [
+                                    4,
+                                    getUserCountryByIp()
+                                ];
+                            case 1:
+                                country = _state.sent();
+                                changLang(country === import_akeyless_types_commons2.CountryOptions.IL ? "he" : "en");
+                                setUserCountry(country);
+                                localStorage.setItem("userCountry", country);
+                                return [
+                                    2
+                                ];
+                        }
+                    });
+                });
+                return function updateCountry() {
+                    return _ref.apply(this, arguments);
+                };
+            }();
+            updateCountry();
+        }
+    }, []);
+    return null;
+};
 // src/hooks/react.ts
 var import_lodash2 = require("lodash");
-var import_react2 = require("react");
+var import_react3 = require("react");
 function useSafeEffect(callback, dependencies, error_message) {
-    (0, import_react2.useEffect)(function() {
+    (0, import_react3.useEffect)(function() {
         try {
             callback();
         } catch (error) {
@@ -529,37 +1006,32 @@ function useSafeEffect(callback, dependencies, error_message) {
     }, dependencies);
 }
 var useDeepCompareMemo = function(factory, dependencies) {
-    var previousDepsRef = (0, import_react2.useRef)([]);
+    var previousDepsRef = (0, import_react3.useRef)([]);
     if (!(0, import_lodash2.isEqual)(dependencies, previousDepsRef.current)) {
         previousDepsRef.current = dependencies;
     }
-    return (0, import_react2.useMemo)(factory, previousDepsRef.current);
+    return (0, import_react3.useMemo)(factory, previousDepsRef.current);
 };
 function useDeepCompareEffect(effect, dependencies) {
-    var previousDepsRef = (0, import_react2.useRef)();
+    var previousDepsRef = (0, import_react3.useRef)();
     if (!(0, import_lodash2.isEqual)(previousDepsRef.current, dependencies)) {
         previousDepsRef.current = dependencies;
     }
-    (0, import_react2.useEffect)(effect, [
+    (0, import_react3.useEffect)(effect, [
         previousDepsRef.current
     ]);
 }
-// src/hooks/global.ts
-var useDocumentTitle = function(title) {
-    (0, import_react3.useEffect)(function() {
-        document.title = title;
-    }, [
-        title
-    ]);
-    return null;
-};
-var useSnapshotBulk = function(configs, label, settings) {
-    var snapshotsFirstTime = (0, import_react3.useRef)([]);
-    var unsubscribeFunctions = (0, import_react3.useRef)([]);
+// src/hooks/WebWorker.ts
+var import_react4 = require("react");
+// src/hooks/snapshots.ts
+var import_react5 = require("react");
+var useDbSnapshots = function(configs, label, settings) {
+    var snapshotsFirstTime = (0, import_react5.useRef)([]);
+    var unsubscribeFunctions = (0, import_react5.useRef)([]);
     useDeepCompareEffect(function() {
         var start = performance.now();
-        if (!(settings === null || settings === void 0 ? void 0 : settings.disableLogs)) {
-            console.log("==> ".concat(label || "Custom snapshots", " started... "));
+        if (!(settings === null || settings === void 0 ? void 0 : settings.disableLogs) && configs.length > 0) {
+            console.log("==> ".concat(label || "DB snapshots", " started from db... "));
         }
         var snapshotResults = configs.map(function(config) {
             return snapshot(config, snapshotsFirstTime.current, settings);
@@ -570,8 +1042,8 @@ var useSnapshotBulk = function(configs, label, settings) {
         Promise.all(snapshotResults.map(function(result) {
             return result.promise;
         })).then(function() {
-            if (!(settings === null || settings === void 0 ? void 0 : settings.disableLogs)) {
-                console.log("==> ".concat(label || "Custom snapshots", " ended. It took ").concat((performance.now() - start).toFixed(2), " ms"));
+            if (!(settings === null || settings === void 0 ? void 0 : settings.disableLogs) && configs.length > 0) {
+                console.log("==> ".concat(label || "DB snapshots", " ended from db. It took ").concat((performance.now() - start).toFixed(2), " ms"));
             }
         });
         if (settings === null || settings === void 0 ? void 0 : settings.cleanupForConfigChange) {
@@ -581,8 +1053,8 @@ var useSnapshotBulk = function(configs, label, settings) {
                         unsubscribe();
                     }
                 });
-                if (!(settings === null || settings === void 0 ? void 0 : settings.disableLogs)) {
-                    console.log("==> ".concat(label || "Custom snapshots", " unsubscribed"));
+                if (!(settings === null || settings === void 0 ? void 0 : settings.disableLogs) && configs.length > 0) {
+                    console.log("==> ".concat(label || "DB snapshots", " unsubscribed from db"));
                 }
             };
         }
@@ -591,7 +1063,7 @@ var useSnapshotBulk = function(configs, label, settings) {
         label,
         settings
     ]);
-    (0, import_react3.useEffect)(function() {
+    (0, import_react5.useEffect)(function() {
         return function() {
             unsubscribeFunctions.current.forEach(function(unsubscribe) {
                 if (unsubscribe) {
@@ -599,54 +1071,135 @@ var useSnapshotBulk = function(configs, label, settings) {
                 }
             });
             if (!(settings === null || settings === void 0 ? void 0 : settings.disableLogs)) {
-                console.log("==> ".concat(label || "Custom snapshots", " unsubscribed"));
+                console.log("==> ".concat(label || "DB snapshots", " unsubscribed"));
             }
         };
     }, []);
 };
-var useSetUserCountry = function(setUserCountry, changLang) {
-    (0, import_react3.useLayoutEffect)(function() {
-        var currentCountry = localStorage.getItem("userCountry");
-        if (!currentCountry) {
-            var updateCountry = /*#__PURE__*/ function() {
-                var _ref = _async_to_generator(function() {
-                    var country;
-                    return _ts_generator(this, function(_state) {
-                        switch(_state.label){
-                            case 0:
-                                return [
-                                    4,
-                                    getUserCountryByIp()
-                                ];
-                            case 1:
-                                country = _state.sent();
-                                changLang(country === import_akeyless_types_commons2.CountryOptions.IL ? "he" : "en");
-                                setUserCountry(country);
-                                localStorage.setItem("userCountry", country);
-                                return [
-                                    2
-                                ];
-                        }
-                    });
-                });
-                return function updateCountry() {
-                    return _ref.apply(this, arguments);
-                };
-            }();
-            updateCountry();
+var useSmartSnapshots = function(configs, label, settings) {
+    var _ref = _sliced_to_array((0, import_react5.useState)(null), 2), cacheCollectionsConfig = _ref[0], setCacheCollectionsConfig = _ref[1];
+    (0, import_react5.useEffect)(function() {
+        get_document_by_id("nx-settings", "cache_collections_config").then(function(res) {
+            return setCacheCollectionsConfig(res);
+        });
+        return function() {
+            return setCacheCollectionsConfig(null);
+        };
+    }, []);
+    var groupedConfig = (0, import_react5.useMemo)(function() {
+        if (!cacheCollectionsConfig) {
+            return {
+                configForDb: [],
+                configForCache: []
+            };
         }
+        var configForDb = [];
+        var configForCache = [];
+        configs.forEach(function(cfg) {
+            var collectionName = cfg.collectionName, _cfg_subscribeTo = cfg.subscribeTo, subscribeTo = _cfg_subscribeTo === void 0 ? "cache" : _cfg_subscribeTo;
+            if (subscribeTo === "cache" && cacheCollectionsConfig[collectionName]) {
+                configForCache.push(cfg);
+            } else {
+                configForDb.push(cfg);
+            }
+        });
+        return {
+            configForDb: configForDb,
+            configForCache: configForCache
+        };
+    }, [
+        configs,
+        cacheCollectionsConfig
+    ]);
+    useDbSnapshots(groupedConfig.configForDb, label, settings);
+    var socketConnected = useSocketSnapshots(groupedConfig.configForCache, label, settings).socketConnected;
+    return {
+        groupedConfig: groupedConfig,
+        socketConnected: socketConnected
+    };
+};
+var useSocketSnapshots = function(configs, label, settings) {
+    var _auth_currentUser;
+    var _ref = _sliced_to_array((0, import_react5.useState)(socketServiceInstance.isConnected()), 2), socketConnected = _ref[0], setSocketConnected = _ref[1];
+    var _ref1 = _sliced_to_array((0, import_react5.useState)([]), 2), cleanupSubscriptions = _ref1[0], setCleanupSubscriptions = _ref1[1];
+    var socketStarted = (0, import_react5.useRef)(false);
+    useDeepCompareEffect(function() {
+        if (!auth.currentUser) {
+            return;
+        }
+        var subscribe = function() {
+            if (configs.length === 0) {
+                return;
+            }
+            var disposer = socketServiceInstance.subscribeToCollections(configs);
+            setCleanupSubscriptions(function(prev) {
+                return _to_consumable_array(prev).concat([
+                    disposer
+                ]);
+            });
+            if (!(settings === null || settings === void 0 ? void 0 : settings.disableLogs)) {
+                console.log("==> ".concat(label || "Cache snapshots", " subscribed to ").concat(configs.map(function(c) {
+                    return c.collectionName;
+                }).join(", ")));
+            }
+        };
+        if (socketServiceInstance.isConnected()) {
+            setSocketConnected(true);
+            subscribe();
+        } else if (!socketStarted.current) {
+            socketStarted.current = true;
+            auth.currentUser.getIdToken().then(function(token) {
+                socketServiceInstance.startSession(token);
+                if (!(settings === null || settings === void 0 ? void 0 : settings.disableLogs)) {
+                    console.log("==> ".concat(label || "Cache snapshots", " started... "));
+                }
+            });
+        }
+        var offConnect = socketServiceInstance.onConnect(function() {
+            setSocketConnected(true);
+            socketStarted.current = false;
+            subscribe();
+        });
+        var offDisconnect = socketServiceInstance.onDisconnect(function() {
+            setSocketConnected(false);
+            cleanupSubscriptions.forEach(function(cleanup) {
+                return cleanup();
+            });
+        });
+        return function() {
+            cleanupSubscriptions.forEach(function(cleanup) {
+                return cleanup();
+            });
+            offConnect === null || offConnect === void 0 ? void 0 : offConnect();
+            offDisconnect === null || offDisconnect === void 0 ? void 0 : offDisconnect();
+            if (!(settings === null || settings === void 0 ? void 0 : settings.disableLogs) && configs.length > 0) {
+                console.log("==> ".concat(label || "Cache snapshots", " unsubscribed. "));
+            }
+        };
+    }, [
+        configs,
+        (_auth_currentUser = auth.currentUser) === null || _auth_currentUser === void 0 ? void 0 : _auth_currentUser.uid
+    ]);
+    (0, import_react5.useEffect)(function() {
+        return function() {
+            cleanupSubscriptions.forEach(function(cleanup) {
+                return cleanup();
+            });
+        };
     }, []);
-    return null;
+    return {
+        socketConnected: socketConnected
+    };
 };
-// src/hooks/WebWorker.ts
-var import_react5 = require("react");
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+    useDbSnapshots: useDbSnapshots,
     useDeepCompareEffect: useDeepCompareEffect,
     useDeepCompareMemo: useDeepCompareMemo,
     useDocumentTitle: useDocumentTitle,
     useSafeEffect: useSafeEffect,
     useSetUserCountry: useSetUserCountry,
-    useSnapshotBulk: useSnapshotBulk
+    useSmartSnapshots: useSmartSnapshots,
+    useSocketSnapshots: useSocketSnapshots
 });
 //# sourceMappingURL=index.js.map