@webiny/api-headless-cms-ddb 0.0.0-unstable.99666aeb00 → 0.0.0-unstable.9bd236cf5e

package/operations/entry/index.js

@@ -5,19 +5,17 @@ Object.defineProperty(exports, "__esModule", {
   value: true
 });
 exports.createEntriesStorageOperations = void 0;
-var _objectSpread2 = _interopRequireDefault(require("@babel/runtime/helpers/objectSpread2"));
 var _error = _interopRequireDefault(require("@webiny/error"));
 var _dataLoaders = require("./dataLoaders");
 var _types = require("@webiny/api-headless-cms/types");
 var _keys = require("./keys");
-var _batchWrite = require("@webiny/db-dynamodb/utils/batchWrite");
-var _query = require("@webiny/db-dynamodb/utils/query");
-var _cleanup = require("@webiny/db-dynamodb/utils/cleanup");
+var _dbDynamodb = require("@webiny/db-dynamodb");
 var _cursor = require("@webiny/utils/cursor");
 var _zeroPad = require("@webiny/utils/zeroPad");
 var _apiHeadlessCms = require("@webiny/api-headless-cms");
 var _createFields = require("./filtering/createFields");
 var _filtering = require("./filtering");
+var _constants = require("@webiny/api-headless-cms/constants");
 const createType = () => {
   return "cms.entry";
 };
@@ -36,9 +34,10 @@ const convertToStorageEntry = params => {
     fields: model.fields,
     values: storageEntry.values
   });
-  return (0, _objectSpread2.default)((0, _objectSpread2.default)({}, storageEntry), {}, {
+  return {
+    ...storageEntry,
     values
-  });
+  };
 };
 const convertFromStorageEntry = params => {
   const {
@@ -49,11 +48,12 @@ const convertFromStorageEntry = params => {
     fields: model.fields,
     values: storageEntry.values
   });
-  return (0, _objectSpread2.default)((0, _objectSpread2.default)({}, storageEntry), {}, {
+  return {
+    ...storageEntry,
     values
-  });
+  };
 };
-const MAX_LIST_LIMIT = 10000;
+const MAX_LIST_LIMIT = 1000000;
 const createEntriesStorageOperations = params => {
   const {
     entity,
@@ -74,11 +74,12 @@ const createEntriesStorageOperations = params => {
   const dataLoaders = new _dataLoaders.DataLoadersHandler({
     entity
   });
-  const storageTransformPlugins = plugins.byType(_apiHeadlessCms.StorageTransformPlugin.type).reduce((collection, plugin) => {
-    collection[plugin.fieldType] = plugin;
-    return collection;
-  }, {});
   const createStorageTransformCallable = model => {
+    // Cache StorageTransformPlugin to optimize execution.
+    const storageTransformPlugins = plugins.byType(_apiHeadlessCms.StorageTransformPlugin.type).reduce((collection, plugin) => {
+      collection[plugin.fieldType] = plugin;
+      return collection;
+    }, {});
     return (field, value) => {
       const plugin = storageTransformPlugins[field.type];
       if (!plugin) {
@@ -117,40 +118,43 @@ const createEntriesStorageOperations = params => {
      *  - create new main entry item
      *  - create new or update the latest entry item
      */
-    const items = [entity.putBatch((0, _objectSpread2.default)((0, _objectSpread2.default)({}, storageEntry), {}, {
-      locked,
-      PK: partitionKey,
-      SK: (0, _keys.createRevisionSortKey)(entry),
-      TYPE: createType(),
-      GSI1_PK: (0, _keys.createGSIPartitionKey)(model, "A"),
-      GSI1_SK: (0, _keys.createGSISortKey)(storageEntry)
-    })), entity.putBatch((0, _objectSpread2.default)((0, _objectSpread2.default)({}, storageEntry), {}, {
-      locked,
-      PK: partitionKey,
-      SK: (0, _keys.createLatestSortKey)(),
-      TYPE: createLatestType(),
-      GSI1_PK: (0, _keys.createGSIPartitionKey)(model, "L"),
-      GSI1_SK: (0, _keys.createGSISortKey)(storageEntry)
-    }))];
+    const entityBatch = (0, _dbDynamodb.createEntityWriteBatch)({
+      entity,
+      put: [{
+        ...storageEntry,
+        locked,
+        PK: partitionKey,
+        SK: (0, _keys.createRevisionSortKey)(entry),
+        TYPE: createType(),
+        GSI1_PK: (0, _keys.createGSIPartitionKey)(model, "A"),
+        GSI1_SK: (0, _keys.createGSISortKey)(storageEntry)
+      }, {
+        ...storageEntry,
+        locked,
+        PK: partitionKey,
+        SK: (0, _keys.createLatestSortKey)(),
+        TYPE: createLatestType(),
+        GSI1_PK: (0, _keys.createGSIPartitionKey)(model, "L"),
+        GSI1_SK: (0, _keys.createGSISortKey)(storageEntry)
+      }]
+    });
 
     /**
      * We need to create published entry if
      */
     if (isPublished) {
-      items.push(entity.putBatch((0, _objectSpread2.default)((0, _objectSpread2.default)({}, storageEntry), {}, {
+      entityBatch.put({
+        ...storageEntry,
         locked,
         PK: partitionKey,
         SK: (0, _keys.createPublishedSortKey)(),
         TYPE: createLatestType(),
         GSI1_PK: (0, _keys.createGSIPartitionKey)(model, "P"),
         GSI1_SK: (0, _keys.createGSISortKey)(storageEntry)
-      })));
+      });
     }
     try {
-      await (0, _batchWrite.batchWriteAll)({
-        table: entity.table,
-        items
-      });
+      await entityBatch.execute();
       dataLoaders.clearAll({
         model
       });
@@ -177,29 +181,63 @@ const createEntriesStorageOperations = params => {
       storageEntry: initialStorageEntry,
       model
     });
+
     /**
      * We need to:
      *  - create the main entry item
-     *  - update the last entry item to a current one
+     *  - update the latest entry item to the current one
+     *  - if the entry's status was set to "published":
+     *      - update the published entry item to the current one
+     *      - unpublish previously published revision (if any)
      */
-    const items = [entity.putBatch((0, _objectSpread2.default)((0, _objectSpread2.default)({}, storageEntry), {}, {
-      PK: partitionKey,
-      SK: (0, _keys.createRevisionSortKey)(storageEntry),
-      TYPE: createType(),
-      GSI1_PK: (0, _keys.createGSIPartitionKey)(model, "A"),
-      GSI1_SK: (0, _keys.createGSISortKey)(storageEntry)
-    })), entity.putBatch((0, _objectSpread2.default)((0, _objectSpread2.default)({}, storageEntry), {}, {
-      PK: partitionKey,
-      SK: (0, _keys.createLatestSortKey)(),
-      TYPE: createLatestType(),
-      GSI1_PK: (0, _keys.createGSIPartitionKey)(model, "L"),
-      GSI1_SK: (0, _keys.createGSISortKey)(storageEntry)
-    }))];
-    try {
-      await (0, _batchWrite.batchWriteAll)({
-        table: entity.table,
-        items
+    const entityBatch = (0, _dbDynamodb.createEntityWriteBatch)({
+      entity,
+      put: [{
+        ...storageEntry,
+        PK: partitionKey,
+        SK: (0, _keys.createRevisionSortKey)(storageEntry),
+        TYPE: createType(),
+        GSI1_PK: (0, _keys.createGSIPartitionKey)(model, "A"),
+        GSI1_SK: (0, _keys.createGSISortKey)(storageEntry)
+      }, {
+        ...storageEntry,
+        PK: partitionKey,
+        SK: (0, _keys.createLatestSortKey)(),
+        TYPE: createLatestType(),
+        GSI1_PK: (0, _keys.createGSIPartitionKey)(model, "L"),
+        GSI1_SK: (0, _keys.createGSISortKey)(storageEntry)
+      }]
+    });
+    const isPublished = entry.status === "published";
+    if (isPublished) {
+      entityBatch.put({
+        ...storageEntry,
+        PK: partitionKey,
+        SK: (0, _keys.createPublishedSortKey)(),
+        TYPE: createPublishedType(),
+        GSI1_PK: (0, _keys.createGSIPartitionKey)(model, "P"),
+        GSI1_SK: (0, _keys.createGSISortKey)(storageEntry)
+      });
+
+      // Unpublish previously published revision (if any).
+      const [publishedRevisionStorageEntry] = await dataLoaders.getPublishedRevisionByEntryId({
+        model,
+        ids: [entry.id]
       });
+      if (publishedRevisionStorageEntry) {
+        entityBatch.put({
+          ...publishedRevisionStorageEntry,
+          PK: partitionKey,
+          SK: (0, _keys.createRevisionSortKey)(publishedRevisionStorageEntry),
+          TYPE: createType(),
+          status: _types.CONTENT_ENTRY_STATUS.UNPUBLISHED,
+          GSI1_PK: (0, _keys.createGSIPartitionKey)(model, "A"),
+          GSI1_SK: (0, _keys.createGSISortKey)(publishedRevisionStorageEntry)
+        });
+      }
+    }
+    try {
+      await entityBatch.execute();
       dataLoaders.clearAll({
         model
       });
@@ -228,7 +266,6 @@ const createEntriesStorageOperations = params => {
     });
     const isPublished = entry.status === "published";
     const locked = isPublished ? true : entry.locked;
-    const items = [];
     const storageEntry = convertToStorageEntry({
       model,
       storageEntry: initialStorageEntry
@@ -238,44 +275,81 @@ const createEntriesStorageOperations = params => {
      *  - update the current entry
      *  - update the latest entry if the current entry is the latest one
      */
-    items.push(entity.putBatch((0, _objectSpread2.default)((0, _objectSpread2.default)({}, storageEntry), {}, {
-      locked,
-      PK: partitionKey,
-      SK: (0, _keys.createRevisionSortKey)(storageEntry),
-      TYPE: createType(),
-      GSI1_PK: (0, _keys.createGSIPartitionKey)(model, "A"),
-      GSI1_SK: (0, _keys.createGSISortKey)(storageEntry)
-    })));
+
+    const entityBatch = (0, _dbDynamodb.createEntityWriteBatch)({
+      entity,
+      put: [{
+        ...storageEntry,
+        locked,
+        PK: partitionKey,
+        SK: (0, _keys.createRevisionSortKey)(storageEntry),
+        TYPE: createType(),
+        GSI1_PK: (0, _keys.createGSIPartitionKey)(model, "A"),
+        GSI1_SK: (0, _keys.createGSISortKey)(storageEntry)
+      }]
+    });
     if (isPublished) {
-      items.push(entity.putBatch((0, _objectSpread2.default)((0, _objectSpread2.default)({}, storageEntry), {}, {
+      entityBatch.put({
+        ...storageEntry,
         locked,
         PK: partitionKey,
         SK: (0, _keys.createPublishedSortKey)(),
         TYPE: createPublishedType(),
         GSI1_PK: (0, _keys.createGSIPartitionKey)(model, "P"),
         GSI1_SK: (0, _keys.createGSISortKey)(storageEntry)
-      })));
+      });
     }
 
     /**
      * We need the latest entry to update it as well if necessary.
      */
     const latestStorageEntry = await getLatestRevisionByEntryId(model, entry);
-    if (latestStorageEntry && latestStorageEntry.id === entry.id) {
-      items.push(entity.putBatch((0, _objectSpread2.default)((0, _objectSpread2.default)({}, storageEntry), {}, {
-        locked,
-        PK: partitionKey,
-        SK: (0, _keys.createLatestSortKey)(),
-        TYPE: createLatestType(),
-        GSI1_PK: (0, _keys.createGSIPartitionKey)(model, "L"),
-        GSI1_SK: (0, _keys.createGSISortKey)(entry)
-      })));
+    if (latestStorageEntry) {
+      const updatingLatestRevision = latestStorageEntry.id === entry.id;
+      if (updatingLatestRevision) {
+        entityBatch.put({
+          ...storageEntry,
+          locked,
+          PK: partitionKey,
+          SK: (0, _keys.createLatestSortKey)(),
+          TYPE: createLatestType(),
+          GSI1_PK: (0, _keys.createGSIPartitionKey)(model, "L"),
+          GSI1_SK: (0, _keys.createGSISortKey)(entry)
+        });
+      } else {
+        /**
+         * If not updating latest revision, we still want to update the latest revision's
+         * entry-level meta fields to match the current revision's entry-level meta fields.
+         */
+        const updatedEntryLevelMetaFields = (0, _constants.pickEntryMetaFields)(entry, _constants.isEntryLevelEntryMetaField);
+
+        /**
+         * First we update the regular DynamoDB table. Two updates are needed:
+         * - one for the actual revision record
+         * - one for the latest record
+         */
+        entityBatch.put({
+          ...latestStorageEntry,
+          ...updatedEntryLevelMetaFields,
+          PK: partitionKey,
+          SK: (0, _keys.createRevisionSortKey)(latestStorageEntry),
+          TYPE: createType(),
+          GSI1_PK: (0, _keys.createGSIPartitionKey)(model, "A"),
+          GSI1_SK: (0, _keys.createGSISortKey)(latestStorageEntry)
+        });
+        entityBatch.put({
+          ...latestStorageEntry,
+          ...updatedEntryLevelMetaFields,
+          PK: partitionKey,
+          SK: (0, _keys.createLatestSortKey)(),
+          TYPE: createLatestType(),
+          GSI1_PK: (0, _keys.createGSIPartitionKey)(model, "L"),
+          GSI1_SK: (0, _keys.createGSISortKey)(latestStorageEntry)
+        });
+      }
     }
     try {
-      await (0, _batchWrite.batchWriteAll)({
-        table: entity.table,
-        items
-      });
+      await entityBatch.execute();
       dataLoaders.clearAll({
         model
       });
@@ -288,6 +362,130 @@ const createEntriesStorageOperations = params => {
       });
     }
   };
+  const move = async (initialModel, id, folderId) => {
+    /**
+     * We need to:
+     * - load all the revisions of the entry, including published and latest
+     * - update all the revisions (published and latest ) of the entry with new folderId
+     */
+    const model = getStorageOperationsModel(initialModel);
+    /**
+     * First we need to load all the revisions and published / latest entry.
+     */
+    const queryAllParams = {
+      entity,
+      partitionKey: (0, _keys.createPartitionKey)({
+        id,
+        locale: model.locale,
+        tenant: model.tenant
+      }),
+      options: {
+        gte: " "
+      }
+    };
+    const records = await (0, _dbDynamodb.queryAll)(queryAllParams);
+    /**
+     * Then create the batch writes for the DynamoDB, with the updated folderId.
+     */
+    const entityBatch = (0, _dbDynamodb.createEntityWriteBatch)({
+      entity,
+      put: records.map(item => {
+        return {
+          ...item,
+          location: {
+            ...item.location,
+            folderId
+          }
+        };
+      })
+    });
+
+    /**
+     * And finally write it...
+     */
+    try {
+      await entityBatch.execute();
+    } catch (ex) {
+      throw _error.default.from(ex, {
+        message: "Could not move records to a new folder.",
+        data: {
+          id,
+          folderId
+        }
+      });
+    }
+  };
+  const moveToBin = async (initialModel, params) => {
+    const {
+      entry,
+      storageEntry: initialStorageEntry
+    } = params;
+    const model = getStorageOperationsModel(initialModel);
+
+    /**
+     * First we need to load all the revisions and published / latest entries.
+     */
+    const queryAllParams = {
+      entity,
+      partitionKey: (0, _keys.createPartitionKey)({
+        id: entry.id,
+        locale: model.locale,
+        tenant: model.tenant
+      }),
+      options: {
+        gte: " "
+      }
+    };
+    let records = [];
+    try {
+      records = await (0, _dbDynamodb.queryAll)(queryAllParams);
+    } catch (ex) {
+      throw new _error.default(ex.message || "Could not load all records.", ex.code || "LOAD_ALL_RECORDS_ERROR", {
+        error: ex,
+        id: entry.id
+      });
+    }
+    if (records.length === 0) {
+      return;
+    }
+    const storageEntry = convertToStorageEntry({
+      model,
+      storageEntry: initialStorageEntry
+    });
+
+    /**
+     * Let's pick the `deleted` meta fields from the storage entry.
+     */
+    const updatedDeletedMetaFields = (0, _constants.pickEntryMetaFields)(storageEntry, _constants.isDeletedEntryMetaField);
+
+    /**
+     * Then create the batch writes for the DynamoDB, with the updated data.
+     */
+    const entityBatch = (0, _dbDynamodb.createEntityWriteBatch)({
+      entity,
+      put: records.map(record => {
+        return {
+          ...record,
+          ...updatedDeletedMetaFields,
+          wbyDeleted: storageEntry.wbyDeleted,
+          location: storageEntry.location,
+          binOriginalFolderId: storageEntry.binOriginalFolderId
+        };
+      })
+    });
+    /**
+     * And finally write it...
+     */
+    try {
+      await entityBatch.execute();
+    } catch (ex) {
+      throw new _error.default(ex.message || "Could not move the entry to the bin.", ex.code || "MOVE_ENTRY_TO_BIN_ERROR", {
+        error: ex,
+        entry,
+        storageEntry
+      });
+    }
+  };
   const deleteEntry = async (initialModel, params) => {
     const {
       entry
@@ -307,24 +505,24 @@ const createEntriesStorageOperations = params => {
     };
     let records = [];
     try {
-      records = await (0, _query.queryAll)(queryAllParams);
+      records = await (0, _dbDynamodb.queryAll)(queryAllParams);
     } catch (ex) {
       throw new _error.default(ex.message || "Could not load all records.", ex.code || "LOAD_ALL_RECORDS_ERROR", {
         error: ex,
         id
       });
     }
-    const items = records.map(item => {
-      return entity.deleteBatch({
-        PK: item.PK,
-        SK: item.SK
-      });
+    const entityBatch = (0, _dbDynamodb.createEntityWriteBatch)({
+      entity,
+      delete: records.map(item => {
+        return {
+          PK: item.PK,
+          SK: item.SK
+        };
+      })
     });
     try {
-      await (0, _batchWrite.batchWriteAll)({
-        table: entity.table,
-        items
-      });
+      await entityBatch.execute();
       dataLoaders.clearAll({
         model
       });
@@ -336,6 +534,78 @@ const createEntriesStorageOperations = params => {
       });
     }
   };
+  const restoreFromBin = async (initialModel, params) => {
+    const {
+      entry,
+      storageEntry: initialStorageEntry
+    } = params;
+    const model = getStorageOperationsModel(initialModel);
+
+    /**
+     * First we need to load all the revisions and published / latest entries.
+     */
+    const queryAllParams = {
+      entity,
+      partitionKey: (0, _keys.createPartitionKey)({
+        id: entry.id,
+        locale: model.locale,
+        tenant: model.tenant
+      }),
+      options: {
+        gte: " "
+      }
+    };
+    let records = [];
+    try {
+      records = await (0, _dbDynamodb.queryAll)(queryAllParams);
+    } catch (ex) {
+      throw new _error.default(ex.message || "Could not load all records.", ex.code || "LOAD_ALL_RECORDS_ERROR", {
+        error: ex,
+        id: entry.id
+      });
+    }
+    if (records.length === 0) {
+      return initialStorageEntry;
+    }
+    const storageEntry = convertToStorageEntry({
+      model,
+      storageEntry: initialStorageEntry
+    });
+
+    /**
+     * Let's pick the `restored` meta fields from the storage entry.
+     */
+    const updatedRestoredMetaFields = (0, _constants.pickEntryMetaFields)(storageEntry, _constants.isRestoredEntryMetaField);
+    const entityBatch = (0, _dbDynamodb.createEntityWriteBatch)({
+      entity,
+      put: records.map(record => {
+        return {
+          ...record,
+          ...updatedRestoredMetaFields,
+          wbyDeleted: storageEntry.wbyDeleted,
+          location: storageEntry.location,
+          binOriginalFolderId: storageEntry.binOriginalFolderId
+        };
+      })
+    });
+
+    /**
+     * And finally write it...
+     */
+    try {
+      await entityBatch.execute();
+      dataLoaders.clearAll({
+        model
+      });
+      return initialStorageEntry;
+    } catch (ex) {
+      throw new _error.default(ex.message || "Could not restore the entry from the bin.", ex.code || "RESTORE_ENTRY_ERROR", {
+        error: ex,
+        entry,
+        storageEntry
+      });
+    }
+  };
   const deleteRevision = async (initialModel, params) => {
     const {
       entry,
@@ -348,39 +618,51 @@ const createEntriesStorageOperations = params => {
       locale: model.locale,
       tenant: model.tenant
     });
-    const items = [entity.deleteBatch({
-      PK: partitionKey,
-      SK: (0, _keys.createRevisionSortKey)(entry)
-    })];
+    const entityBatch = (0, _dbDynamodb.createEntityWriteBatch)({
+      entity,
+      delete: [{
+        PK: partitionKey,
+        SK: (0, _keys.createRevisionSortKey)(entry)
+      }]
+    });
     const publishedStorageEntry = await getPublishedRevisionByEntryId(model, entry);
 
     /**
      * If revision we are deleting is the published one as well, we need to delete those records as well.
      */
     if (publishedStorageEntry && entry.id === publishedStorageEntry.id) {
-      items.push(entity.deleteBatch({
+      entityBatch.delete({
         PK: partitionKey,
         SK: (0, _keys.createPublishedSortKey)()
-      }));
+      });
     }
     if (initialLatestStorageEntry) {
       const latestStorageEntry = convertToStorageEntry({
         storageEntry: initialLatestStorageEntry,
         model
       });
-      items.push(entity.putBatch((0, _objectSpread2.default)((0, _objectSpread2.default)({}, latestStorageEntry), {}, {
+      entityBatch.put({
+        ...latestStorageEntry,
         PK: partitionKey,
         SK: (0, _keys.createLatestSortKey)(),
         TYPE: createLatestType(),
         GSI1_PK: (0, _keys.createGSIPartitionKey)(model, "L"),
         GSI1_SK: (0, _keys.createGSISortKey)(latestStorageEntry)
-      })));
+      });
+
+      // Do an update on the latest revision. We need to update the latest revision's
+      // entry-level meta fields to match the previous revision's entry-level meta fields.
+      entityBatch.put({
+        ...latestStorageEntry,
+        PK: partitionKey,
+        SK: (0, _keys.createRevisionSortKey)(initialLatestStorageEntry),
+        TYPE: createType(),
+        GSI1_PK: (0, _keys.createGSIPartitionKey)(model, "A"),
+        GSI1_SK: (0, _keys.createGSISortKey)(initialLatestStorageEntry)
+      });
     }
     try {
-      await (0, _batchWrite.batchWriteAll)({
-        table: entity.table,
-        items
-      });
+      entityBatch.execute();
       dataLoaders.clearAll({
         model
       });
@@ -407,36 +689,30 @@ const createEntriesStorageOperations = params => {
     /**
      * Then we need to construct the queries for all the revisions and entries.
      */
-    const items = [];
+
+    const entityBatch = (0, _dbDynamodb.createEntityWriteBatch)({
+      entity
+    });
     for (const id of entries) {
-      /**
-       * Latest item.
-       */
-      items.push(entity.deleteBatch({
-        PK: (0, _keys.createPartitionKey)({
-          id,
-          locale: model.locale,
-          tenant: model.tenant
-        }),
+      const partitionKey = (0, _keys.createPartitionKey)({
+        id,
+        locale: model.locale,
+        tenant: model.tenant
+      });
+      entityBatch.delete({
+        PK: partitionKey,
         SK: "L"
-      }));
-      /**
-       * Published item.
-       */
-      items.push(entity.deleteBatch({
-        PK: (0, _keys.createPartitionKey)({
-          id,
-          locale: model.locale,
-          tenant: model.tenant
-        }),
+      });
+      entityBatch.delete({
+        PK: partitionKey,
         SK: "P"
-      }));
+      });
     }
     /**
      * Exact revisions of all the entries
      */
     for (const revision of revisions) {
-      items.push(entity.deleteBatch({
+      entityBatch.delete({
         PK: (0, _keys.createPartitionKey)({
           id: revision.id,
           locale: model.locale,
@@ -445,12 +721,9 @@ const createEntriesStorageOperations = params => {
         SK: (0, _keys.createRevisionSortKey)({
           version: revision.version
         })
-      }));
+      });
     }
-    await (0, _batchWrite.batchWriteAll)({
-      table: entity.table,
-      items
-    });
+    await entityBatch.execute();
   };
   const getLatestRevisionByEntryId = async (initialModel, params) => {
     const model = getStorageOperationsModel(initialModel);
@@ -578,8 +851,8 @@ const createEntriesStorageOperations = params => {
       }
     };
     try {
-      const result = await (0, _query.queryOne)(queryParams);
-      const storageEntry = (0, _cleanup.cleanupItem)(entity, result);
+      const result = await (0, _dbDynamodb.queryOne)(queryParams);
+      const storageEntry = (0, _dbDynamodb.cleanupItem)(entity, result);
       if (!storageEntry) {
         return null;
       }
@@ -588,12 +861,13 @@ const createEntriesStorageOperations = params => {
         model
       });
     } catch (ex) {
-      throw new _error.default(ex.message || "Could not get previous version of given entry.", ex.code || "GET_PREVIOUS_VERSION_ERROR", (0, _objectSpread2.default)((0, _objectSpread2.default)({}, params), {}, {
+      throw new _error.default(ex.message || "Could not get previous version of given entry.", ex.code || "GET_PREVIOUS_VERSION_ERROR", {
+        ...params,
         error: ex,
         partitionKey: queryParams.partitionKey,
         options: queryParams.options,
         model
-      }));
+      });
     }
   };
   const list = async (initialModel, params) => {
@@ -618,7 +892,7 @@ const createEntriesStorageOperations = params => {
     };
     let storageEntries = [];
     try {
-      storageEntries = await (0, _query.queryAll)(queryAllParams);
+      storageEntries = await (0, _dbDynamodb.queryAll)(queryAllParams);
     } catch (ex) {
       throw new _error.default(ex.message, "QUERY_ENTRIES_ERROR", {
         error: ex,
@@ -634,7 +908,9 @@ const createEntriesStorageOperations = params => {
         items: []
       };
     }
-    const where = (0, _objectSpread2.default)({}, initialWhere);
+    const where = {
+      ...initialWhere
+    };
     delete where["published"];
     delete where["latest"];
     /**
@@ -696,21 +972,22 @@ const createEntriesStorageOperations = params => {
      * Although we do not need a cursor here, we will use it as such to keep it standardized.
      * Number is simply encoded.
      */
-    const cursor = totalCount > start + limit ? (0, _cursor.encodeCursor)(`${start + limit}`) : null;
+    const cursor = (0, _cursor.encodeCursor)(`${start + limit}`);
     return {
       hasMoreItems,
       totalCount,
       cursor,
-      items: (0, _cleanup.cleanupItems)(entity, slicedItems)
+      items: (0, _dbDynamodb.cleanupItems)(entity, slicedItems)
     };
   };
   const get = async (initialModel, params) => {
     const model = getStorageOperationsModel(initialModel);
     const {
       items
-    } = await list(model, (0, _objectSpread2.default)((0, _objectSpread2.default)({}, params), {}, {
+    } = await list(model, {
+      ...params,
       limit: 1
-    }));
+    });
     return items.shift() || null;
   };
   const publish = async (initialModel, params) => {
@@ -729,59 +1006,129 @@ const createEntriesStorageOperations = params => {
      * We need the latest and published entries to see if something needs to be updated alongside the publishing one.
      */
     const initialLatestStorageEntry = await getLatestRevisionByEntryId(model, entry);
+    if (!initialLatestStorageEntry) {
+      throw new _error.default(`Could not publish entry. Could not load latest ("L") record.`, "PUBLISH_ERROR", {
+        entry
+      });
+    }
     const initialPublishedStorageEntry = await getPublishedRevisionByEntryId(model, entry);
     const storageEntry = convertToStorageEntry({
       model,
       storageEntry: initialStorageEntry
     });
-    /**
-     * We need to update:
-     *  - current entry revision sort key
-     *  - published sort key
-     *  - the latest sort key - if entry updated is actually latest
-     *  - previous published entry to unpublished status - if any previously published entry
-     */
-    const items = [entity.putBatch((0, _objectSpread2.default)((0, _objectSpread2.default)({}, storageEntry), {}, {
-      PK: partitionKey,
-      SK: (0, _keys.createRevisionSortKey)(entry),
-      TYPE: createType(),
-      GSI1_PK: (0, _keys.createGSIPartitionKey)(model, "A"),
-      GSI1_SK: (0, _keys.createGSISortKey)(entry)
-    })), entity.putBatch((0, _objectSpread2.default)((0, _objectSpread2.default)({}, storageEntry), {}, {
-      PK: partitionKey,
-      SK: (0, _keys.createPublishedSortKey)(),
-      TYPE: createPublishedType(),
-      GSI1_PK: (0, _keys.createGSIPartitionKey)(model, "P"),
-      GSI1_SK: (0, _keys.createGSISortKey)(entry)
-    }))];
-    if (initialLatestStorageEntry && entry.id === initialLatestStorageEntry.id) {
-      items.push(entity.putBatch((0, _objectSpread2.default)((0, _objectSpread2.default)({}, storageEntry), {}, {
+
+    // 1. Update REV# and P records with new data.
+    const entityBatch = (0, _dbDynamodb.createEntityWriteBatch)({
+      entity,
+      put: [{
+        ...storageEntry,
+        PK: partitionKey,
+        SK: (0, _keys.createRevisionSortKey)(entry),
+        TYPE: createType(),
+        GSI1_PK: (0, _keys.createGSIPartitionKey)(model, "A"),
+        GSI1_SK: (0, _keys.createGSISortKey)(entry)
+      }, {
+        ...storageEntry,
+        PK: partitionKey,
+        SK: (0, _keys.createPublishedSortKey)(),
+        TYPE: createPublishedType(),
+        GSI1_PK: (0, _keys.createGSIPartitionKey)(model, "P"),
+        GSI1_SK: (0, _keys.createGSISortKey)(entry)
+      }]
+    });
+
+    // 2. When it comes to the latest record, we need to perform a couple of different
+    // updates, based on whether the entry being published is the latest revision or not.
+    const publishedRevisionId = initialPublishedStorageEntry?.id;
+    const publishingLatestRevision = entry.id === initialLatestStorageEntry.id;
+    if (publishingLatestRevision) {
+      // 2.1 If we're publishing the latest revision, we first need to update the L record.
+      entityBatch.put({
+        ...storageEntry,
         PK: partitionKey,
         SK: (0, _keys.createLatestSortKey)(),
         TYPE: createLatestType(),
         GSI1_PK: (0, _keys.createGSIPartitionKey)(model, "L"),
         GSI1_SK: (0, _keys.createGSISortKey)(entry)
-      })));
-    }
-    if (initialPublishedStorageEntry && initialPublishedStorageEntry.id !== entry.id) {
-      const publishedStorageEntry = convertToStorageEntry({
-        storageEntry: initialPublishedStorageEntry,
+      });
+
+      // 2.2 Additionally, if we have a previously published entry, we need to mark it as unpublished.
+      if (publishedRevisionId && publishedRevisionId !== entry.id) {
+        const publishedStorageEntry = convertToStorageEntry({
+          storageEntry: initialPublishedStorageEntry,
+          model
+        });
+        entityBatch.put({
+          ...publishedStorageEntry,
+          PK: partitionKey,
+          SK: (0, _keys.createRevisionSortKey)(publishedStorageEntry),
+          TYPE: createType(),
+          status: _types.CONTENT_ENTRY_STATUS.UNPUBLISHED,
+          GSI1_PK: (0, _keys.createGSIPartitionKey)(model, "A"),
+          GSI1_SK: (0, _keys.createGSISortKey)(publishedStorageEntry)
+        });
+      }
+    } else {
+      // 2.3 If the published revision is not the latest one, the situation is a bit
+      // more complex. We first need to update the L and REV# records with the new
+      // values of *only entry-level* meta fields.
+      const updatedEntryLevelMetaFields = (0, _constants.pickEntryMetaFields)(entry, _constants.isEntryLevelEntryMetaField);
+      const latestStorageEntry = convertToStorageEntry({
+        storageEntry: initialLatestStorageEntry,
         model
       });
-      items.push(entity.putBatch((0, _objectSpread2.default)((0, _objectSpread2.default)({}, publishedStorageEntry), {}, {
+
+      // 2.3.1 Update L record. Apart from updating the entry-level meta fields, we also need
+      //    to change the status from "published" to "unpublished" (if the status is set to "published").
+      let latestRevisionStatus = latestStorageEntry.status;
+      if (latestRevisionStatus === _types.CONTENT_ENTRY_STATUS.PUBLISHED) {
+        latestRevisionStatus = _types.CONTENT_ENTRY_STATUS.UNPUBLISHED;
+      }
+      const latestStorageEntryFields = {
+        ...latestStorageEntry,
+        ...updatedEntryLevelMetaFields,
+        status: latestRevisionStatus
+      };
+      entityBatch.put({
+        ...latestStorageEntryFields,
         PK: partitionKey,
-        SK: (0, _keys.createRevisionSortKey)(publishedStorageEntry),
+        SK: (0, _keys.createLatestSortKey)(),
+        TYPE: createLatestType(),
+        GSI1_PK: (0, _keys.createGSIPartitionKey)(model, "L"),
+        GSI1_SK: (0, _keys.createGSISortKey)(latestStorageEntry)
+      });
+
+      // 2.3.2 Update REV# record.
+      entityBatch.put({
+        ...latestStorageEntryFields,
+        PK: partitionKey,
+        SK: (0, _keys.createRevisionSortKey)(latestStorageEntry),
         TYPE: createType(),
-        status: _types.CONTENT_ENTRY_STATUS.UNPUBLISHED,
         GSI1_PK: (0, _keys.createGSIPartitionKey)(model, "A"),
-        GSI1_SK: (0, _keys.createGSISortKey)(publishedStorageEntry)
-      })));
+        GSI1_SK: (0, _keys.createGSISortKey)(latestStorageEntry)
+      });
+
+      // 2.3.3 Finally, if we got a published entry, but it wasn't the latest one, we need to take
+      //    an extra step and mark it as unpublished.
+      const publishedRevisionDifferentFromLatest = publishedRevisionId && publishedRevisionId !== latestStorageEntry.id;
+      if (publishedRevisionDifferentFromLatest) {
+        const publishedStorageEntry = convertToStorageEntry({
+          storageEntry: initialPublishedStorageEntry,
+          model
+        });
+        entityBatch.put({
+          ...publishedStorageEntry,
+          PK: partitionKey,
+          SK: (0, _keys.createRevisionSortKey)(publishedStorageEntry),
+          TYPE: createType(),
+          status: _types.CONTENT_ENTRY_STATUS.UNPUBLISHED,
+          GSI1_PK: (0, _keys.createGSIPartitionKey)(model, "A"),
+          GSI1_SK: (0, _keys.createGSISortKey)(publishedStorageEntry)
+        });
+      }
     }
     try {
-      await (0, _batchWrite.batchWriteAll)({
-        table: entity.table,
-        items
-      });
+      await entityBatch.execute();
       dataLoaders.clearAll({
         model
       });
@@ -815,35 +1162,72 @@ const createEntriesStorageOperations = params => {
      *  - update current entry revision with new data
      *  - update the latest entry status - if entry being unpublished is latest
      */
-    const items = [entity.deleteBatch({
-      PK: partitionKey,
-      SK: (0, _keys.createPublishedSortKey)()
-    }), entity.putBatch((0, _objectSpread2.default)((0, _objectSpread2.default)({}, storageEntry), {}, {
-      PK: partitionKey,
-      SK: (0, _keys.createRevisionSortKey)(entry),
-      TYPE: createType(),
-      GSI1_PK: (0, _keys.createGSIPartitionKey)(model, "A"),
-      GSI1_SK: (0, _keys.createGSISortKey)(entry)
-    }))];
+    const entityBatch = (0, _dbDynamodb.createEntityWriteBatch)({
+      entity,
+      delete: [{
+        PK: partitionKey,
+        SK: (0, _keys.createPublishedSortKey)()
+      }],
+      put: [{
+        ...storageEntry,
+        PK: partitionKey,
+        SK: (0, _keys.createRevisionSortKey)(entry),
+        TYPE: createType(),
+        GSI1_PK: (0, _keys.createGSIPartitionKey)(model, "A"),
+        GSI1_SK: (0, _keys.createGSISortKey)(entry)
+      }]
+    });
 
     /**
      * We need the latest entry to see if something needs to be updated alongside the unpublishing one.
      */
-    const latestStorageEntry = await getLatestRevisionByEntryId(model, entry);
-    if (latestStorageEntry && entry.id === latestStorageEntry.id) {
-      items.push(entity.putBatch((0, _objectSpread2.default)((0, _objectSpread2.default)({}, storageEntry), {}, {
-        PK: partitionKey,
-        SK: (0, _keys.createLatestSortKey)(),
-        TYPE: createLatestType(),
-        GSI1_PK: (0, _keys.createGSIPartitionKey)(model, "L"),
-        GSI1_SK: (0, _keys.createGSISortKey)(entry)
-      })));
+    const initialLatestStorageEntry = await getLatestRevisionByEntryId(model, entry);
+    if (initialLatestStorageEntry) {
+      const unpublishingLatestRevision = entry.id === initialLatestStorageEntry.id;
+      if (unpublishingLatestRevision) {
+        entityBatch.put({
+          ...storageEntry,
+          PK: partitionKey,
+          SK: (0, _keys.createLatestSortKey)(),
+          TYPE: createLatestType(),
+          GSI1_PK: (0, _keys.createGSIPartitionKey)(model, "L"),
+          GSI1_SK: (0, _keys.createGSISortKey)(entry)
+        });
+      } else {
+        const latestStorageEntry = convertToStorageEntry({
+          storageEntry: initialLatestStorageEntry,
+          model
+        });
+
+        // If the unpublished revision is not the latest one, we still need to
+        // update the latest record with the new values of entry-level meta fields.
+        const updatedEntryLevelMetaFields = (0, _constants.pickEntryMetaFields)(entry, _constants.isEntryLevelEntryMetaField);
+
+        // 1. Update actual revision record.
+        entityBatch.put({
+          ...latestStorageEntry,
+          ...updatedEntryLevelMetaFields,
+          PK: partitionKey,
+          SK: (0, _keys.createRevisionSortKey)(latestStorageEntry),
+          TYPE: createType(),
+          GSI1_PK: (0, _keys.createGSIPartitionKey)(model, "A"),
+          GSI1_SK: (0, _keys.createGSISortKey)(latestStorageEntry)
+        });
+
+        // 2. Update latest record.
+        entityBatch.put({
+          ...latestStorageEntry,
+          ...updatedEntryLevelMetaFields,
+          PK: partitionKey,
+          SK: (0, _keys.createLatestSortKey)(),
+          TYPE: createLatestType(),
+          GSI1_PK: (0, _keys.createGSIPartitionKey)(model, "L"),
+          GSI1_SK: (0, _keys.createGSISortKey)(latestStorageEntry)
+        });
+      }
     }
     try {
-      await (0, _batchWrite.batchWriteAll)({
-        table: entity.table,
-        items
-      });
+      await entityBatch.execute();
       dataLoaders.clearAll({
         model
       });
@@ -883,10 +1267,9 @@ const createEntriesStorageOperations = params => {
         continue;
       }
       for (const value of values) {
-        var _result$value;
         result[value] = {
           value,
-          count: (((_result$value = result[value]) === null || _result$value === void 0 ? void 0 : _result$value.count) || 0) + 1
+          count: (result[value]?.count || 0) + 1
         };
       }
     }
@@ -896,7 +1279,10 @@ const createEntriesStorageOperations = params => {
     create,
     createRevisionFrom,
     update,
+    move,
     delete: deleteEntry,
+    moveToBin,
+    restoreFromBin,
     deleteRevision,
     deleteMultipleEntries,
     getPreviousRevision,
@@ -915,4 +1301,6 @@ const createEntriesStorageOperations = params => {
     getUniqueFieldValues
   };
 };
-exports.createEntriesStorageOperations = createEntriesStorageOperations;
+exports.createEntriesStorageOperations = createEntriesStorageOperations;
+
+//# sourceMappingURL=index.js.map