tinybase 5.0.0-beta.5 → 5.0.0-beta.6

package/lib/debug/persisters/persister-sync.js

@@ -0,0 +1,594 @@
+const getTypeOf = (thing) => typeof thing;
+const EMPTY_STRING = '';
+const STRING = getTypeOf(EMPTY_STRING);
+const UTF8 = 'utf8';
+const strCharCodeAt = (str, position) => str.charCodeAt(position);
+
+const promise = Promise;
+const mathMax = Math.max;
+const isInstanceOf = (thing, cls) => thing instanceof cls;
+const isUndefined = (thing) => thing == void 0;
+const ifNotUndefined = (value, then, otherwise) =>
+  isUndefined(value) ? otherwise?.() : then(value);
+const isString = (thing) => getTypeOf(thing) == STRING;
+const slice = (arrayOrString, start, end) => arrayOrString.slice(start, end);
+const size = (arrayOrString) => arrayOrString.length;
+const promiseNew = (resolver) => new promise(resolver);
+
+const arrayForEach = (array, cb) => array.forEach(cb);
+const arrayPush = (array, ...values) => array.push(...values);
+const arrayShift = (array) => array.shift();
+
+const object = Object;
+const getPrototypeOf = (obj) => object.getPrototypeOf(obj);
+const isObject = (obj) =>
+  !isUndefined(obj) &&
+  ifNotUndefined(
+    getPrototypeOf(obj),
+    (objPrototype) =>
+      objPrototype == object.prototype ||
+      isUndefined(getPrototypeOf(objPrototype)),
+    /* istanbul ignore next */
+    () => true,
+  );
+const objIds = object.keys;
+const objFreeze = object.freeze;
+const objNew = (entries = []) => object.fromEntries(entries);
+const objDel = (obj, id) => {
+  delete obj[id];
+  return obj;
+};
+const objSize = (obj) => size(objIds(obj));
+const objIsEmpty = (obj) => isObject(obj) && objSize(obj) == 0;
+
+const collHas = (coll, keyOrValue) => coll?.has(keyOrValue) ?? false;
+const collClear = (coll) => coll.clear();
+const collForEach = (coll, cb) => coll?.forEach(cb);
+const collDel = (coll, keyOrValue) => coll?.delete(keyOrValue);
+
+const mapNew = (entries) => new Map(entries);
+const mapGet = (map, key) => map?.get(key);
+const mapForEach = (map, cb) =>
+  collForEach(map, (value, key) => cb(key, value));
+const mapSet = (map, key, value) =>
+  isUndefined(value) ? (collDel(map, key), map) : map?.set(key, value);
+const mapEnsure = (map, key, getDefaultValue, hadExistingValue) => {
+  if (!collHas(map, key)) {
+    mapSet(map, key, getDefaultValue());
+  } else {
+    hadExistingValue?.(mapGet(map, key));
+  }
+  return mapGet(map, key);
+};
+
+const scheduleRunning = mapNew();
+const scheduleActions = mapNew();
+const getStoreFunctions = (supportsMergeableStore, store) =>
+  !supportsMergeableStore || isUndefined(store.getMergeableContent)
+    ? [
+        0,
+        store.getContent,
+        store.getTransactionChanges,
+        ([changedTables, changedValues]) =>
+          !objIsEmpty(changedTables) || !objIsEmpty(changedValues),
+      ]
+    : [
+        1,
+        store.getMergeableContent,
+        store.getTransactionMergeableChanges,
+        ([, [[, changedTables], [, changedValues]]]) =>
+          !objIsEmpty(changedTables) || !objIsEmpty(changedValues),
+      ];
+const createCustomPersister = (
+  store,
+  getPersisted,
+  setPersisted,
+  addPersisterListener,
+  delPersisterListener,
+  onIgnoredError,
+  supportsMergeableStore,
+  extra = {},
+  scheduleId = [],
+) => {
+  let loadSave = 0;
+  let loads = 0;
+  let saves = 0;
+  let action;
+  let autoLoadHandle;
+  let autoSaveListenerId;
+  mapEnsure(scheduleRunning, scheduleId, () => 0);
+  mapEnsure(scheduleActions, scheduleId, () => []);
+  const [isMergeableStore, getContent, getChanges, hasChanges] =
+    getStoreFunctions(supportsMergeableStore, store);
+  const run = async () => {
+    /* istanbul ignore else */
+    if (!mapGet(scheduleRunning, scheduleId)) {
+      mapSet(scheduleRunning, scheduleId, 1);
+      while (
+        !isUndefined((action = arrayShift(mapGet(scheduleActions, scheduleId))))
+      ) {
+        try {
+          await action();
+        } catch (error) {
+          /* istanbul ignore next */
+          onIgnoredError?.(error);
+        }
+      }
+      mapSet(scheduleRunning, scheduleId, 0);
+    }
+  };
+  const loadLock = async (actions) => {
+    /* istanbul ignore else */
+    if (loadSave != 2) {
+      loadSave = 1;
+      {
+        loads++;
+      }
+      await persister.schedule(async () => {
+        await actions();
+        loadSave = 0;
+      });
+    }
+    return persister;
+  };
+  const setContentOrChanges = (contentOrChanges) => {
+    (isMergeableStore && isString(contentOrChanges?.[0])
+      ? contentOrChanges?.[1][2] === 1
+        ? store.applyMergeableChanges
+        : store.setMergeableContent
+      : contentOrChanges?.[2] === 1
+        ? store.applyChanges
+        : store.setContent)(contentOrChanges);
+  };
+  const persister = {
+    load: async (initialTables, initialValues) =>
+      await loadLock(async () => {
+        try {
+          setContentOrChanges(await getPersisted());
+        } catch (error) {
+          onIgnoredError?.(error);
+          store.setContent([initialTables, initialValues]);
+        }
+      }),
+    startAutoLoad: async (initialTables = {}, initialValues = {}) => {
+      await persister.stopAutoLoad().load(initialTables, initialValues);
+      autoLoadHandle = addPersisterListener(
+        async (getContent2, getChanges2) => {
+          const changes = getChanges2?.();
+          await loadLock(async () => {
+            try {
+              setContentOrChanges(
+                changes ?? getContent2?.() ?? (await getPersisted()),
+              );
+            } catch (error) {
+              onIgnoredError?.(error);
+            }
+          });
+        },
+      );
+      return persister;
+    },
+    stopAutoLoad: () => {
+      if (autoLoadHandle) {
+        delPersisterListener(autoLoadHandle);
+        autoLoadHandle = void 0;
+      }
+      return persister;
+    },
+    isAutoLoading: () => !isUndefined(autoLoadHandle),
+    save: async (getChanges2) => {
+      /* istanbul ignore else */
+      if (loadSave != 1) {
+        loadSave = 2;
+        {
+          saves++;
+        }
+        await persister.schedule(async () => {
+          try {
+            await setPersisted(getContent, getChanges2);
+          } catch (error) {
+            /* istanbul ignore next */
+            onIgnoredError?.(error);
+          }
+          loadSave = 0;
+        });
+      }
+      return persister;
+    },
+    startAutoSave: async () => {
+      await persister.stopAutoSave().save();
+      autoSaveListenerId = store.addDidFinishTransactionListener(() => {
+        const changes = getChanges();
+        if (hasChanges(changes)) {
+          persister.save(() => changes);
+        }
+      });
+      return persister;
+    },
+    stopAutoSave: () => {
+      ifNotUndefined(autoSaveListenerId, store.delListener);
+      autoSaveListenerId = void 0;
+      return persister;
+    },
+    isAutoSaving: () => !isUndefined(autoSaveListenerId),
+    schedule: async (...actions) => {
+      arrayPush(mapGet(scheduleActions, scheduleId), ...actions);
+      await run();
+      return persister;
+    },
+    getStore: () => store,
+    destroy: () => persister.stopAutoLoad().stopAutoSave(),
+    getStats: () => ({loads, saves}),
+    ...extra,
+  };
+  return objFreeze(persister);
+};
+
+const textEncoder = new globalThis.TextEncoder();
+const getHash = (value) => {
+  let hash = 2166136261;
+  arrayForEach(textEncoder.encode(value), (char) => {
+    hash ^= char;
+    hash +=
+      (hash << 1) + (hash << 4) + (hash << 7) + (hash << 8) + (hash << 24);
+  });
+  return hash >>> 0;
+};
+
+const MASK6 = 63;
+const SHIFT36 = 2 ** 36;
+const SHIFT30 = 2 ** 30;
+const SHIFT24 = 2 ** 24;
+const SHIFT18 = 2 ** 18;
+const SHIFT12 = 2 ** 12;
+const SHIFT6 = 2 ** 6;
+const toB64 = (num) => String.fromCharCode(48 + (num & MASK6));
+const fromB64 = (str, pos) => strCharCodeAt(str, pos) - 48;
+const encodeHlc = (logicalTime42, counter24, clientHash30) =>
+  toB64(logicalTime42 / SHIFT36) +
+  toB64(logicalTime42 / SHIFT30) +
+  toB64(logicalTime42 / SHIFT24) +
+  toB64(logicalTime42 / SHIFT18) +
+  toB64(logicalTime42 / SHIFT12) +
+  toB64(logicalTime42 / SHIFT6) +
+  toB64(logicalTime42) +
+  toB64(counter24 / SHIFT18) +
+  toB64(counter24 / SHIFT12) +
+  toB64(counter24 / SHIFT6) +
+  toB64(counter24) +
+  toB64(clientHash30 / SHIFT24) +
+  toB64(clientHash30 / SHIFT18) +
+  toB64(clientHash30 / SHIFT12) +
+  toB64(clientHash30 / SHIFT6) +
+  toB64(clientHash30);
+const decodeHlc = (hlc16) => [
+  fromB64(hlc16, 0) * SHIFT36 +
+    fromB64(hlc16, 1) * SHIFT30 +
+    fromB64(hlc16, 2) * SHIFT24 +
+    fromB64(hlc16, 3) * SHIFT18 +
+    fromB64(hlc16, 4) * SHIFT12 +
+    fromB64(hlc16, 5) * SHIFT6 +
+    fromB64(hlc16, 6),
+  fromB64(hlc16, 7) * SHIFT18 +
+    fromB64(hlc16, 8) * SHIFT12 +
+    fromB64(hlc16, 9) * SHIFT6 +
+    fromB64(hlc16, 10),
+  fromB64(hlc16, 11) * SHIFT24 +
+    fromB64(hlc16, 12) * SHIFT18 +
+    fromB64(hlc16, 13) * SHIFT12 +
+    fromB64(hlc16, 14) * SHIFT6 +
+    fromB64(hlc16, 15),
+];
+const getHlcFunctions = (uniqueId) => {
+  let logicalTime = 0;
+  let lastCounter = -1;
+  const uniqueIdHash = getHash(uniqueId);
+  const getHlc = () => {
+    seenHlc();
+    return encodeHlc(logicalTime, ++lastCounter, uniqueIdHash);
+  };
+  const seenHlc = (hlc) => {
+    const previousLogicalTime = logicalTime;
+    const [remoteLogicalTime, remoteCounter] =
+      isUndefined(hlc) || hlc == '' ? [0, 0] : decodeHlc(hlc);
+    logicalTime = mathMax(
+      previousLogicalTime,
+      remoteLogicalTime,
+      globalThis.HLC_TIME ?? Date.now(),
+    );
+    lastCounter =
+      logicalTime == previousLogicalTime
+        ? logicalTime == remoteLogicalTime
+          ? mathMax(lastCounter, remoteCounter)
+          : lastCounter
+        : logicalTime == remoteLogicalTime
+          ? remoteCounter
+          : -1;
+  };
+  return [getHlc, seenHlc];
+};
+
+const clients = mapNew();
+const sends = objNew();
+const receives = objNew();
+const createLocalClient = () => {
+  const clientId = '' + Math.random();
+  const onReceive = (receive) => {
+    mapSet(clients, clientId, (...args) => {
+      receives[clientId] = (receives[clientId] ?? 0) + 1;
+      receive(...args);
+    });
+  };
+  const send = (toClientId, requestId, messageType, messageBody) => {
+    {
+      sends[clientId] = (sends[clientId] ?? 0) + 1;
+    }
+    isUndefined(toClientId)
+      ? mapForEach(clients, (otherClientId, receive) =>
+          otherClientId != clientId
+            ? receive(clientId, requestId, messageType, messageBody)
+            : 0,
+        )
+      : mapGet(clients, toClientId)?.(
+          clientId,
+          requestId,
+          messageType,
+          messageBody,
+        );
+  };
+  const destroy = () => {
+    collDel(clients, clientId);
+    {
+      objDel(sends, clientId);
+      objDel(receives, clientId);
+    }
+  };
+  const getStats = () => ({
+    sends: sends[clientId] ?? 0,
+    receives: receives[clientId] ?? 0,
+  });
+  return {send, onReceive, destroy, getStats};
+};
+
+const jsonString = (obj) =>
+  JSON.stringify(obj, (_key, value) =>
+    isInstanceOf(value, Map) ? object.fromEntries([...value]) : value,
+  );
+const jsonParse = JSON.parse;
+
+const MESSAGE_SEPARATOR = '\n';
+
+const createWsClient = async (webSocket) => {
+  let sends = 0;
+  let receives = 0;
+  let currentReceive;
+  webSocket.on('message', (data) => {
+    if (!isUndefined(currentReceive)) {
+      const payload = data.toString(UTF8);
+      const splitAt = payload.indexOf(MESSAGE_SEPARATOR);
+      if (splitAt !== -1) {
+        {
+          receives++;
+        }
+        currentReceive(
+          slice(payload, 0, splitAt),
+          ...jsonParse(slice(payload, splitAt + 1)),
+        );
+      }
+    }
+  });
+  const onReceive = (receive) => {
+    currentReceive = receive;
+  };
+  const send = (toClientId, ...args) => {
+    {
+      sends++;
+    }
+    webSocket.send(
+      (toClientId ?? EMPTY_STRING) + MESSAGE_SEPARATOR + jsonString(args),
+    );
+  };
+  const destroy = () => {
+    webSocket.close();
+  };
+  const getStats = () => ({sends, receives});
+  const client = {send, onReceive, destroy, getStats};
+  return promiseNew((resolve, reject) => {
+    if (webSocket.readyState != webSocket.OPEN) {
+      webSocket.on('error', reject);
+      webSocket.on('open', () => resolve(client));
+    } else {
+      resolve(client);
+    }
+  });
+};
+
+const createWsSimpleServer = (webSocketServer) => {
+  const clients = mapNew();
+  webSocketServer.on('connection', (webSocket, request) => {
+    const clientId = request.headers['sec-websocket-key'];
+    mapSet(clients, clientId, webSocket);
+    webSocket.on('message', (data) => {
+      const payload = data.toString(UTF8);
+      const splitAt = payload.indexOf(MESSAGE_SEPARATOR);
+      if (splitAt !== -1) {
+        const toClientId = slice(payload, 0, splitAt);
+        const message = slice(payload, splitAt + 1);
+        toClientId === EMPTY_STRING
+          ? mapForEach(clients, (otherClientId, otherWebSocket) =>
+              otherClientId != clientId
+                ? otherWebSocket.send(clientId + MESSAGE_SEPARATOR + message)
+                : 0,
+            )
+          : mapGet(clients, toClientId)?.send(
+              clientId + MESSAGE_SEPARATOR + message,
+            );
+      }
+    });
+    webSocket.on('close', () => collDel(clients, clientId));
+  });
+  const getWebSocketServer = () => webSocketServer;
+  const destroy = () => {
+    webSocketServer.close();
+    collClear(clients);
+  };
+  return {
+    getWebSocketServer,
+    destroy,
+  };
+};
+
+const RESPONSE = 0;
+const CONTENT_HASHES = 1;
+const GET_CONTENT_HASHES = 2;
+const GET_TABLE_IDS_DIFF = 3;
+const GET_ROW_IDS_DIFF = 4;
+const GET_TABLES_CHANGES = 5;
+const GET_VALUES_CHANGES = 6;
+const createSyncPersister = (
+  store,
+  client,
+  requestTimeoutSeconds = 1,
+  onIgnoredError,
+) => {
+  let persisterListener;
+  const {send, onReceive, destroy} = client;
+  const [getHlc] = getHlcFunctions(store.getId());
+  const pendingRequests = mapNew();
+  onReceive((fromClientId, requestId, messageType, messageBody) => {
+    if (messageType == RESPONSE) {
+      ifNotUndefined(
+        mapGet(pendingRequests, requestId),
+        ([toClientId, handleResponse]) =>
+          isUndefined(toClientId) || toClientId == fromClientId
+            ? handleResponse(messageBody, fromClientId)
+            : 0,
+      );
+    } else if (messageType == CONTENT_HASHES && persister.isAutoLoading()) {
+      getChangesFromOtherStore(fromClientId, messageBody).then((changes) =>
+        persisterListener?.(void 0, () => changes),
+      );
+    } else {
+      const response =
+        messageType == GET_CONTENT_HASHES && persister.isAutoSaving()
+          ? store.getMergeableContentHashes()
+          : messageType == GET_TABLE_IDS_DIFF
+            ? store.getMergeableTableIdsDiff(messageBody)
+            : messageType == GET_ROW_IDS_DIFF
+              ? store.getMergeableRowIdsDiff(messageBody)
+              : messageType == GET_TABLES_CHANGES
+                ? store.getMergeableTablesChanges(messageBody)
+                : messageType == GET_VALUES_CHANGES
+                  ? store.getMergeableValuesChanges(messageBody)
+                  : 0;
+      response === 0 ? 0 : send(fromClientId, requestId, RESPONSE, response);
+    }
+  });
+  const request = async (toClientId, messageType, messageBody = EMPTY_STRING) =>
+    promiseNew((resolve, reject) => {
+      const requestId = getHlc();
+      const timeout = setTimeout(() => {
+        collDel(pendingRequests, requestId);
+        reject(
+          `No response from ${toClientId ?? 'anyone'} to '${messageType}'`,
+        );
+      }, requestTimeoutSeconds * 1e3);
+      mapSet(pendingRequests, requestId, [
+        toClientId,
+        (response, fromClientId) => {
+          clearTimeout(timeout);
+          collDel(pendingRequests, requestId);
+          resolve([response, fromClientId]);
+        },
+      ]);
+      send(toClientId, requestId, messageType, messageBody);
+    });
+  const getChangesFromOtherStore = async (
+    otherClientId = null,
+    otherContentHashes,
+  ) => {
+    if (isUndefined(otherContentHashes)) {
+      [otherContentHashes, otherClientId] = await request(
+        otherClientId,
+        GET_CONTENT_HASHES,
+      );
+    }
+    const [otherContentTime, [otherTablesHash, otherValuesHash]] =
+      otherContentHashes;
+    const [, [tablesHash, valuesHash]] = store.getMergeableContentHashes();
+    const changes = [EMPTY_STRING, [[EMPTY_STRING, {}], [EMPTY_STRING, {}], 1]];
+    if (tablesHash != otherTablesHash) {
+      changes[0] = otherContentTime;
+      changes[1][0] = (
+        await request(
+          otherClientId,
+          GET_TABLES_CHANGES,
+          store.getMergeableCellHashes(
+            (
+              await request(
+                otherClientId,
+                GET_ROW_IDS_DIFF,
+                store.getMergeableRowHashes(
+                  (
+                    await request(
+                      otherClientId,
+                      GET_TABLE_IDS_DIFF,
+                      store.getMergeableTableHashes(),
+                    )
+                  )[0],
+                ),
+              )
+            )[0],
+          ),
+        )
+      )[0];
+    }
+    if (valuesHash != otherValuesHash) {
+      changes[0] = otherContentTime;
+      changes[1][1] = (
+        await request(
+          otherClientId,
+          GET_VALUES_CHANGES,
+          store.getMergeableValuesHashes(),
+        )
+      )[0];
+    }
+    return changes;
+  };
+  const getPersisted = async () => {
+    const changes = await getChangesFromOtherStore();
+    return changes[0] != EMPTY_STRING ? changes : void 0;
+  };
+  const setPersisted = async () => {
+    send(null, null, CONTENT_HASHES, store.getMergeableContentHashes());
+  };
+  const addPersisterListener = (listener) => (persisterListener = listener);
+  const delPersisterListener = () => (persisterListener = void 0);
+  const persister = createCustomPersister(
+    store,
+    getPersisted,
+    setPersisted,
+    addPersisterListener,
+    delPersisterListener,
+    onIgnoredError,
+    true,
+    {
+      getClient: () => client,
+      startSync: async () =>
+        await (await persister.startAutoLoad()).startAutoSave(),
+      stopSync: () => persister.stopAutoLoad().stopAutoSave(),
+      destroy: () => {
+        destroy();
+        return persister.stopSync();
+      },
+    },
+  );
+  return persister;
+};
+
+export {
+  createLocalClient,
+  createSyncPersister,
+  createWsClient,
+  createWsSimpleServer,
+};