tinybase 5.3.2 → 5.3.3

package/es6/persisters/persister-powersync/with-schemas/index.js

@@ -49,8 +49,6 @@ const errorNew = (message) => {
   throw new Error(message);
 };
 
-const arrayNew = (size2, cb) =>
-  arrayMap(new Array(size2).fill(0), (_, index) => cb(index));
 const arrayForEach = (array, cb) => array.forEach(cb);
 const arrayJoin = (array, sep = EMPTY_STRING) => array.join(sep);
 const arrayMap = (array, cb) => array.map(cb);
@@ -60,6 +58,45 @@ const arrayClear = (array, to) => array.splice(0, to);
 const arrayPush = (array, ...values) => array.push(...values);
 const arrayShift = (array) => array.shift();
 
+const object = Object;
+const getPrototypeOf = (obj) => object.getPrototypeOf(obj);
+const objEntries = object.entries;
+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 objMerge = (...objs) => object.assign({}, ...objs);
+const objHas = (obj, id) => id in obj;
+const objDel = (obj, id) => {
+  delete obj[id];
+  return obj;
+};
+const objToArray = (obj, cb) =>
+  arrayMap(objEntries(obj), ([id, value]) => cb(value, id));
+const objMap = (obj, cb) =>
+  objNew(objToArray(obj, (value, id) => [id, cb(value, id)]));
+const objValues = (obj) => object.values(obj);
+const objSize = (obj) => size(objIds(obj));
+const objIsEmpty = (obj) => isObject(obj) && objSize(obj) == 0;
+
+const setNew = (entryOrEntries) =>
+  new Set(
+    isArray(entryOrEntries) || isUndefined(entryOrEntries)
+      ? entryOrEntries
+      : [entryOrEntries],
+  );
+const setAdd = (set, value) => (set == null ? void 0 : set.add(value));
+
 var __async$6 = (__this, __arguments, generator) => {
   return new Promise((resolve, reject) => {
     var fulfilled = (value) => {
@@ -105,9 +142,11 @@ const getWrappedCommand = (executeCommand, onSqlCommand) =>
         })
     : executeCommand;
 const escapeId = (str) => `"${str.replace(/"/g, '""')}"`;
-const getPlaceholders = (array) =>
+const escapeColumnNames = (...columnNames) =>
+  arrayJoin(arrayMap(columnNames, escapeId), COMMA);
+const getPlaceholders = (array, offset = [1]) =>
   arrayJoin(
-    arrayMap(array, (_, index) => '$' + (index + 1)),
+    arrayMap(array, () => '$' + offset[0]++),
     COMMA,
   );
 
@@ -131,37 +170,6 @@ const collForEach = (coll, cb) => (coll == null ? void 0 : coll.forEach(cb));
 const collDel = (coll, keyOrValue) =>
   coll == null ? void 0 : coll.delete(keyOrValue);
 
-const object = Object;
-const getPrototypeOf = (obj) => object.getPrototypeOf(obj);
-const objEntries = object.entries;
-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 objMerge = (...objs) => object.assign({}, ...objs);
-const objHas = (obj, id) => id in obj;
-const objDel = (obj, id) => {
-  delete obj[id];
-  return obj;
-};
-const objToArray = (obj, cb) =>
-  arrayMap(objEntries(obj), ([id, value]) => cb(value, id));
-const objMap = (obj, cb) =>
-  objNew(objToArray(obj, (value, id) => [id, cb(value, id)]));
-const objValues = (obj) => object.values(obj);
-const objSize = (obj) => size(objIds(obj));
-const objIsEmpty = (obj) => isObject(obj) && objSize(obj) == 0;
-
 const jsonString = JSON.stringify;
 const jsonParse = JSON.parse;
 const jsonStringWithUndefined = (obj) =>
@@ -214,14 +222,6 @@ const visitTree = (node, path, ensureLeaf, pruneLeaf, p = 0) =>
     },
   );
 
-const setNew = (entryOrEntries) =>
-  new Set(
-    isArray(entryOrEntries) || isUndefined(entryOrEntries)
-      ? entryOrEntries
-      : [entryOrEntries],
-  );
-const setAdd = (set, value) => (set == null ? void 0 : set.add(value));
-
 const INTEGER = /^\d+$/;
 const getPoolFunctions = () => {
   const pool = [];
@@ -633,7 +633,7 @@ const getCommandFunctions = (
   querySchema,
   onIgnoredError,
   columnType,
-  orReplace,
+  upsert = defaultUpsert,
   encode,
   decode,
 ) => {
@@ -678,27 +678,27 @@ const getCommandFunctions = (
     partial = false,
   ) =>
     __async$4(void 0, null, function* () {
-      const tableCellOrValueIds = setNew();
+      const settingColumnNameSet = setNew();
       objToArray(content != null ? content : {}, (contentRow) =>
         arrayMap(
           objIds(contentRow != null ? contentRow : {}),
-          (cellOrValueId) => setAdd(tableCellOrValueIds, cellOrValueId),
+          (cellOrValueId) => setAdd(settingColumnNameSet, cellOrValueId),
         ),
       );
-      const tableColumnNames = collValues(tableCellOrValueIds);
+      const settingColumnNames = collValues(settingColumnNameSet);
       if (
         !partial &&
         deleteEmptyTable &&
-        arrayIsEmpty(tableColumnNames) &&
+        arrayIsEmpty(settingColumnNames) &&
         collHas(schemaMap, tableName)
       ) {
         yield databaseExecuteCommand('DROP ' + TABLE + escapeId(tableName));
         mapSet(schemaMap, tableName);
         return;
       }
-      const tableSchemaColumns = mapGet(schemaMap, tableName);
-      const currentColumnNames = setNew(collValues(tableSchemaColumns));
-      if (!arrayIsEmpty(tableColumnNames)) {
+      const currentColumnNames = mapGet(schemaMap, tableName);
+      const unaccountedColumnNames = setNew(collValues(currentColumnNames));
+      if (!arrayIsEmpty(settingColumnNames)) {
         if (!collHas(schemaMap, tableName)) {
           yield databaseExecuteCommand(
             'CREATE ' +
@@ -706,28 +706,29 @@ const getCommandFunctions = (
               escapeId(tableName) +
               `(${escapeId(rowIdColumnName)}${columnType} PRIMARY KEY${arrayJoin(
                 arrayMap(
-                  tableColumnNames,
-                  (columnName) => COMMA + escapeId(columnName) + columnType,
+                  settingColumnNames,
+                  (settingColumnName) =>
+                    COMMA + escapeId(settingColumnName) + columnType,
                 ),
               )});`,
           );
           mapSet(
             schemaMap,
             tableName,
-            setNew([rowIdColumnName, ...tableColumnNames]),
+            setNew([rowIdColumnName, ...settingColumnNames]),
           );
         } else {
           yield promiseAll(
             arrayMap(
-              [rowIdColumnName, ...tableColumnNames],
-              (columnName, index) =>
+              [rowIdColumnName, ...settingColumnNames],
+              (settingColumnName, index) =>
                 __async$4(void 0, null, function* () {
-                  if (!collDel(currentColumnNames, columnName)) {
+                  if (!collDel(unaccountedColumnNames, settingColumnName)) {
                     yield databaseExecuteCommand(
                       ALTER_TABLE +
                         escapeId(tableName) +
                         'ADD' +
-                        escapeId(columnName) +
+                        escapeId(settingColumnName) +
                         columnType,
                     );
                     if (index == 0) {
@@ -737,7 +738,7 @@ const getCommandFunctions = (
                           `(${escapeId(rowIdColumnName)})`,
                       );
                     }
-                    setAdd(tableSchemaColumns, columnName);
+                    setAdd(currentColumnNames, settingColumnName);
                   }
                 }),
             ),
@@ -746,18 +747,20 @@ const getCommandFunctions = (
       }
       yield promiseAll([
         ...(!partial && deleteEmptyColumns
-          ? arrayMap(collValues(currentColumnNames), (columnName) =>
-              __async$4(void 0, null, function* () {
-                if (columnName != rowIdColumnName) {
-                  yield databaseExecuteCommand(
-                    ALTER_TABLE +
-                      escapeId(tableName) +
-                      'DROP' +
-                      escapeId(columnName),
-                  );
-                  collDel(tableSchemaColumns, columnName);
-                }
-              }),
+          ? arrayMap(
+              collValues(unaccountedColumnNames),
+              (unaccountedColumnName) =>
+                __async$4(void 0, null, function* () {
+                  if (unaccountedColumnName != rowIdColumnName) {
+                    yield databaseExecuteCommand(
+                      ALTER_TABLE +
+                        escapeId(tableName) +
+                        'DROP' +
+                        escapeId(unaccountedColumnName),
+                    );
+                    collDel(currentColumnNames, unaccountedColumnName);
+                  }
+                }),
             )
           : []),
       ]);
@@ -779,19 +782,18 @@ const getCommandFunctions = (
                       '=$1',
                     [rowId],
                   );
-                } else if (!arrayIsEmpty(tableColumnNames)) {
+                } else if (!arrayIsEmpty(settingColumnNames)) {
                   yield upsert(
                     databaseExecuteCommand,
                     tableName,
                     rowIdColumnName,
                     objIds(row),
-                    [
-                      rowId,
-                      ...(encode
+                    {
+                      [rowId]: encode
                         ? arrayMap(objValues(row), encode)
-                        : objValues(row)),
-                    ],
-                    orReplace,
+                        : objValues(row),
+                    },
+                    currentColumnNames,
                   );
                 }
               }),
@@ -799,24 +801,20 @@ const getCommandFunctions = (
           );
         }
       } else {
-        if (!arrayIsEmpty(tableColumnNames)) {
+        if (!arrayIsEmpty(settingColumnNames)) {
           const changingColumnNames = arrayFilter(
             collValues(mapGet(schemaMap, tableName)),
-            (columnName) => columnName != rowIdColumnName,
+            (changingColumnName) => changingColumnName != rowIdColumnName,
           );
-          const params = [];
+          const rows = {};
           const deleteRowIds = [];
           objToArray(content != null ? content : {}, (row, rowId) => {
-            arrayPush(
-              params,
-              rowId,
-              ...arrayMap(changingColumnNames, (cellId) =>
-                encode
-                  ? encode(row == null ? void 0 : row[cellId])
-                  : row == null
-                    ? void 0
-                    : row[cellId],
-              ),
+            rows[rowId] = arrayMap(changingColumnNames, (cellId) =>
+              encode
+                ? encode(row == null ? void 0 : row[cellId])
+                : row == null
+                  ? void 0
+                  : row[cellId],
             );
             arrayPush(deleteRowIds, rowId);
           });
@@ -825,8 +823,7 @@ const getCommandFunctions = (
             tableName,
             rowIdColumnName,
             changingColumnNames,
-            params,
-            orReplace,
+            rows,
           );
           yield databaseExecuteCommand(
             DELETE_FROM +
@@ -857,63 +854,46 @@ const getCommandFunctions = (
     });
   return [refreshSchema, loadTable, saveTable, transaction];
 };
-const upsert = (
+const defaultUpsert = (
   executeCommand,
   tableName,
   rowIdColumnName,
   changingColumnNames,
-  params,
-  orReplace = 0,
+  rows,
 ) =>
   __async$4(void 0, null, function* () {
-    return yield executeCommand(
-      'INSERT ' +
-        (orReplace ? 'OR REPLACE ' : EMPTY_STRING) +
-        'INTO' +
+    const offset = [1];
+    yield executeCommand(
+      'INSERT INTO' +
         escapeId(tableName) +
         '(' +
+        escapeColumnNames(rowIdColumnName, ...changingColumnNames) +
+        ')VALUES' +
+        arrayJoin(
+          objToArray(
+            rows,
+            (row) =>
+              '($' + offset[0]++ + ',' + getPlaceholders(row, offset) + ')',
+          ),
+          COMMA,
+        ) +
+        'ON CONFLICT(' +
         escapeId(rowIdColumnName) +
+        ')DO UPDATE SET' +
         arrayJoin(
           arrayMap(
             changingColumnNames,
-            (columnName) => COMMA + escapeId(columnName),
+            (columnName) =>
+              escapeId(columnName) + '=excluded.' + escapeId(columnName),
           ),
-        ) +
-        ')VALUES' +
-        getUpsertPlaceholders(params, size(changingColumnNames) + 1) +
-        (orReplace
-          ? EMPTY_STRING
-          : 'ON CONFLICT(' +
-            escapeId(rowIdColumnName) +
-            ')DO UPDATE SET' +
-            arrayJoin(
-              arrayMap(
-                changingColumnNames,
-                (columnName) =>
-                  escapeId(columnName) + '=excluded.' + escapeId(columnName),
-              ),
-              COMMA,
-            )),
-      arrayMap(params, (param) => (param != null ? param : null)),
+          COMMA,
+        ),
+      objToArray(rows, (row, id) => [
+        id,
+        ...arrayMap(row, (value) => (value != null ? value : null)),
+      ]).flat(),
     );
   });
-const getUpsertPlaceholders = (array, columnCount) =>
-  arrayJoin(
-    arrayNew(
-      size(array) / columnCount,
-      (row) =>
-        '(' +
-        arrayJoin(
-          arrayNew(
-            columnCount,
-            (column) => '$' + (row * columnCount + column + 1),
-          ),
-          COMMA,
-        ) +
-        ')',
-    ),
-    COMMA,
-  );
 
 var __async$3 = (__this, __arguments, generator) => {
   return new Promise((resolve, reject) => {
@@ -952,7 +932,7 @@ const createJsonPersister = (
   thing,
   getThing,
   columnType,
-  orReplace,
+  upsert,
 ) => {
   const [refreshSchema, loadTable, saveTable, transaction] =
     getCommandFunctions(
@@ -961,7 +941,7 @@ const createJsonPersister = (
       querySchema,
       onIgnoredError,
       columnType,
-      orReplace,
+      upsert,
     );
   const getPersisted = () =>
     __async$3(void 0, null, function* () {
@@ -1064,7 +1044,7 @@ const createTabularPersister = (
   thing,
   getThing,
   columnType,
-  orReplace,
+  upsert,
   encode,
   decode,
 ) => {
@@ -1075,7 +1055,7 @@ const createTabularPersister = (
       querySchema,
       onIgnoredError,
       columnType,
-      orReplace,
+      upsert,
       encode,
       decode,
     );
@@ -1343,7 +1323,7 @@ const createCustomSqlitePersister = (
   persist,
   thing,
   getThing = 'getDb',
-  orReplace,
+  upsert,
 ) => {
   let dataVersion;
   let schemaVersion;
@@ -1418,7 +1398,7 @@ const createCustomSqlitePersister = (
     thing,
     getThing,
     EMPTY_STRING,
-    orReplace,
+    upsert,
     (cellOrValue) =>
       cellOrValue === true ? 1 : cellOrValue === false ? 0 : cellOrValue,
     void 0,
@@ -1474,9 +1454,9 @@ const createPowerSyncPersister = (
   configOrStoreTableName,
   onSqlCommand,
   onIgnoredError,
-  orReplace = 1,
-) =>
-  createCustomSqlitePersister(
+) => {
+  let tableListener;
+  return createCustomSqlitePersister(
     store,
     configOrStoreTableName,
     (_0, ..._1) =>
@@ -1503,7 +1483,9 @@ const createPowerSyncPersister = (
               more = false
             ) {
               const update = temp.value;
-              arrayMap(update.changedTables, listener);
+              if (tableListener) {
+                arrayMap(update.changedTables, tableListener);
+              }
             }
           } catch (temp) {
             error = [temp];
@@ -1515,9 +1497,13 @@ const createPowerSyncPersister = (
             }
           }
         }))();
+      tableListener = listener;
       return abortController;
     },
-    (abortController) => abortController.abort(),
+    (abortController) => {
+      tableListener = void 0;
+      abortController.abort();
+    },
     onSqlCommand,
     onIgnoredError,
     () => 0,
@@ -1525,7 +1511,86 @@ const createPowerSyncPersister = (
     // StoreOnly,
     powerSync,
     'getPowerSync',
-    orReplace,
+    viewUpsert,
   );
+};
+const viewUpsert = (
+  executeCommand,
+  tableName,
+  rowIdColumnName,
+  changingColumnNames,
+  rows,
+  currentColumnNames,
+) =>
+  __async(void 0, null, function* () {
+    const offset = [1];
+    const changingColumnNamesSet = setNew(changingColumnNames);
+    const unchangingColumnNames = currentColumnNames
+      ? arrayFilter(
+          [...currentColumnNames],
+          (currentColumnName) =>
+            currentColumnName != rowIdColumnName &&
+            !collHas(changingColumnNamesSet, currentColumnName),
+        )
+      : [];
+    if (!arrayIsEmpty(unchangingColumnNames)) {
+      const ids = objIds(rows);
+      const unchangingData = objNew(
+        arrayMap(
+          yield executeCommand(
+            'SELECT' +
+              escapeColumnNames(rowIdColumnName, ...unchangingColumnNames) +
+              'FROM' +
+              escapeId(tableName) +
+              'WHERE' +
+              escapeId(rowIdColumnName) +
+              'IN(' +
+              getPlaceholders(ids) +
+              ')',
+            ids,
+          ),
+          (unchangingRow) => [unchangingRow[rowIdColumnName], unchangingRow],
+        ),
+      );
+      arrayForEach(ids, (id) =>
+        arrayPush(
+          rows[id],
+          ...arrayMap(unchangingColumnNames, (unchangingColumnName) => {
+            var _a, _b;
+            return (_b =
+              (_a = unchangingData == null ? void 0 : unchangingData[id]) ==
+              null
+                ? void 0
+                : _a[unchangingColumnName]) != null
+              ? _b
+              : null;
+          }),
+        ),
+      );
+    }
+    yield executeCommand(
+      'INSERT OR REPLACE INTO' +
+        escapeId(tableName) +
+        '(' +
+        escapeColumnNames(
+          rowIdColumnName,
+          ...changingColumnNames,
+          ...unchangingColumnNames,
+        ) +
+        ')VALUES' +
+        arrayJoin(
+          objToArray(
+            rows,
+            (row) =>
+              '($' + offset[0]++ + ',' + getPlaceholders(row, offset) + ')',
+          ),
+          COMMA,
+        ),
+      objToArray(rows, (row, id) => [
+        id,
+        ...arrayMap(row, (value) => (value != null ? value : null)),
+      ]).flat(),
+    );
+  });
 
 export {createPowerSyncPersister};