@webiny/api-headless-cms-ddb-es 5.40.0-beta.1 → 5.40.0-beta.3

package/operations/entry/index.js

@@ -557,6 +557,310 @@ const createEntriesStorageOperations = params => {
       });
     }
   };
+  const moveToBin = async (initialModel, params) => {
+    const {
+      entry: initialEntry,
+      storageEntry: initialStorageEntry
+    } = params;
+    const model = getStorageOperationsModel(initialModel);
+    const transformer = (0, _transformations.createTransformer)({
+      plugins,
+      model,
+      entry: initialEntry,
+      storageEntry: initialStorageEntry
+    });
+    const {
+      entry,
+      storageEntry
+    } = transformer.transformEntryKeys();
+    const partitionKey = (0, _keys.createPartitionKey)({
+      id: entry.id,
+      locale: model.locale,
+      tenant: model.tenant
+    });
+
+    /**
+     * First we need to fetch all the records in the regular DynamoDB table.
+     */
+    const queryAllParams = {
+      entity,
+      partitionKey,
+      options: {
+        gte: " "
+      }
+    };
+    const latestSortKey = (0, _keys.createLatestSortKey)();
+    const publishedSortKey = (0, _keys.createPublishedSortKey)();
+    const records = await (0, _query.queryAll)(queryAllParams);
+
+    /**
+     * Let's pick the `deleted` meta fields from the entry.
+     */
+    const updatedEntryMetaFields = (0, _constants.pickEntryMetaFields)(entry, _constants.isDeletedEntryMetaField);
+
+    /**
+     * Then update all the records with data received.
+     */
+    let latestRecord = undefined;
+    let publishedRecord = undefined;
+    const items = [];
+    for (const record of records) {
+      items.push(entity.putBatch({
+        ...record,
+        ...updatedEntryMetaFields,
+        wbyDeleted: storageEntry.wbyDeleted,
+        location: storageEntry.location,
+        binOriginalFolderId: storageEntry.binOriginalFolderId
+      }));
+      /**
+       * We need to get the published and latest records, so we can update the Elasticsearch.
+       */
+      if (record.SK === publishedSortKey) {
+        publishedRecord = record;
+      } else if (record.SK === latestSortKey) {
+        latestRecord = record;
+      }
+    }
+
+    /**
+     * We write the records back to the primary DynamoDB table.
+     */
+    try {
+      await (0, _batchWrite.batchWriteAll)({
+        table: entity.table,
+        items
+      });
+      dataLoaders.clearAll({
+        model
+      });
+    } catch (ex) {
+      throw new _error.default(ex.message || "Could mark as deleted all entry records from in the DynamoDB table.", ex.code || "MOVE_ENTRY_TO_BIN_ERROR", {
+        error: ex,
+        entry,
+        storageEntry
+      });
+    }
+
+    /**
+     * We need to get the published and latest records from Elasticsearch.
+     */
+    const esGetItems = [];
+    if (publishedRecord) {
+      esGetItems.push(esEntity.getBatch({
+        PK: partitionKey,
+        SK: publishedSortKey
+      }));
+    }
+    if (latestRecord) {
+      esGetItems.push(esEntity.getBatch({
+        PK: partitionKey,
+        SK: latestSortKey
+      }));
+    }
+    if (esGetItems.length === 0) {
+      return;
+    }
+    const esRecords = await (0, _dbDynamodb.batchReadAll)({
+      table: esEntity.table,
+      items: esGetItems
+    });
+    const esItems = (await Promise.all(esRecords.map(async record => {
+      if (!record) {
+        return null;
+      }
+      return {
+        ...record,
+        data: await (0, _apiElasticsearch.decompress)(plugins, record.data)
+      };
+    }))).filter(Boolean);
+    if (esItems.length === 0) {
+      return;
+    }
+
+    /**
+     * We update all ES records with data received.
+     */
+    const esUpdateItems = [];
+    for (const item of esItems) {
+      esUpdateItems.push(esEntity.putBatch({
+        ...item,
+        data: await (0, _apiElasticsearch.compress)(plugins, {
+          ...item.data,
+          ...updatedEntryMetaFields,
+          wbyDeleted: entry.wbyDeleted,
+          location: entry.location,
+          binOriginalFolderId: entry.binOriginalFolderId
+        })
+      }));
+    }
+
+    /**
+     * We write the records back to the primary DynamoDB Elasticsearch table.
+     */
+    try {
+      await (0, _batchWrite.batchWriteAll)({
+        table: esEntity.table,
+        items: esUpdateItems
+      });
+    } catch (ex) {
+      throw new _error.default(ex.message || "Could not mark as deleted entry records from DynamoDB Elasticsearch table.", ex.code || "MOVE_ENTRY_TO_BIN_ERROR", {
+        error: ex,
+        entry,
+        storageEntry
+      });
+    }
+  };
+  const restoreFromBin = async (initialModel, params) => {
+    const {
+      entry: initialEntry,
+      storageEntry: initialStorageEntry
+    } = params;
+    const model = getStorageOperationsModel(initialModel);
+    const transformer = (0, _transformations.createTransformer)({
+      plugins,
+      model,
+      entry: initialEntry,
+      storageEntry: initialStorageEntry
+    });
+    const {
+      entry,
+      storageEntry
+    } = transformer.transformEntryKeys();
+
+    /**
+     * Let's pick the `restored` meta fields from the storage entry.
+     */
+    const updatedEntryMetaFields = (0, _constants.pickEntryMetaFields)(entry, _constants.isRestoredEntryMetaField);
+    const partitionKey = (0, _keys.createPartitionKey)({
+      id: entry.id,
+      locale: model.locale,
+      tenant: model.tenant
+    });
+
+    /**
+     * First we need to fetch all the records in the regular DynamoDB table.
+     */
+    const queryAllParams = {
+      entity,
+      partitionKey,
+      options: {
+        gte: " "
+      }
+    };
+    const latestSortKey = (0, _keys.createLatestSortKey)();
+    const publishedSortKey = (0, _keys.createPublishedSortKey)();
+    const records = await (0, _query.queryAll)(queryAllParams);
+
+    /**
+     * Then update all the records with data received.
+     */
+    let latestRecord = undefined;
+    let publishedRecord = undefined;
+    const items = [];
+    for (const record of records) {
+      items.push(entity.putBatch({
+        ...record,
+        ...updatedEntryMetaFields,
+        wbyDeleted: storageEntry.wbyDeleted,
+        location: storageEntry.location,
+        binOriginalFolderId: storageEntry.binOriginalFolderId
+      }));
+      /**
+       * We need to get the published and latest records, so we can update the Elasticsearch.
+       */
+      if (record.SK === publishedSortKey) {
+        publishedRecord = record;
+      } else if (record.SK === latestSortKey) {
+        latestRecord = record;
+      }
+    }
+
+    /**
+     * We write the records back to the primary DynamoDB table.
+     */
+    try {
+      await (0, _batchWrite.batchWriteAll)({
+        table: entity.table,
+        items
+      });
+      dataLoaders.clearAll({
+        model
+      });
+    } catch (ex) {
+      throw new _error.default(ex.message || "Could not restore all entry records from in the DynamoDB table.", ex.code || "RESTORE_ENTRY_ERROR", {
+        error: ex,
+        entry,
+        storageEntry
+      });
+    }
+
+    /**
+     * We need to get the published and latest records from Elasticsearch.
+     */
+    const esGetItems = [];
+    if (publishedRecord) {
+      esGetItems.push(esEntity.getBatch({
+        PK: partitionKey,
+        SK: publishedSortKey
+      }));
+    }
+    if (latestRecord) {
+      esGetItems.push(esEntity.getBatch({
+        PK: partitionKey,
+        SK: latestSortKey
+      }));
+    }
+    const esRecords = await (0, _dbDynamodb.batchReadAll)({
+      table: esEntity.table,
+      items: esGetItems
+    });
+    const esItems = (await Promise.all(esRecords.map(async record => {
+      if (!record) {
+        return null;
+      }
+      return {
+        ...record,
+        data: await (0, _apiElasticsearch.decompress)(plugins, record.data)
+      };
+    }))).filter(Boolean);
+    if (esItems.length === 0) {
+      return initialStorageEntry;
+    }
+
+    /**
+     * We update all ES records with data received.
+     */
+    const esUpdateItems = [];
+    for (const item of esItems) {
+      esUpdateItems.push(esEntity.putBatch({
+        ...item,
+        data: await (0, _apiElasticsearch.compress)(plugins, {
+          ...item.data,
+          ...updatedEntryMetaFields,
+          wbyDeleted: entry.wbyDeleted,
+          location: entry.location,
+          binOriginalFolderId: entry.binOriginalFolderId
+        })
+      }));
+    }
+
+    /**
+     * We write the records back to the primary DynamoDB Elasticsearch table.
+     */
+    try {
+      await (0, _batchWrite.batchWriteAll)({
+        table: esEntity.table,
+        items: esUpdateItems
+      });
+    } catch (ex) {
+      throw new _error.default(ex.message || "Could not restore entry records from DynamoDB Elasticsearch table.", ex.code || "RESTORE_ENTRY_ERROR", {
+        error: ex,
+        entry,
+        storageEntry
+      });
+    }
+    return initialStorageEntry;
+  };
   const deleteEntry = async (initialModel, params) => {
     const {
       entry
@@ -603,7 +907,7 @@ const createEntriesStorageOperations = params => {
         model
       });
     } catch (ex) {
-      throw new _error.default(ex.message || "Could not delete entry records from DynamoDB table.", ex.code || "DELETE_ENTRY_ERROR", {
+      throw new _error.default(ex.message || "Could not destroy entry records from DynamoDB table.", ex.code || "DELETE_ENTRY_ERROR", {
         error: ex,
         id
       });
@@ -614,7 +918,7 @@ const createEntriesStorageOperations = params => {
         items: deleteEsItems
       });
     } catch (ex) {
-      throw new _error.default(ex.message || "Could not delete entry records from DynamoDB Elasticsearch table.", ex.code || "DELETE_ENTRY_ERROR", {
+      throw new _error.default(ex.message || "Could not destroy entry records from DynamoDB Elasticsearch table.", ex.code || "DELETE_ENTRY_ERROR", {
         error: ex,
         id
       });
@@ -1461,6 +1765,8 @@ const createEntriesStorageOperations = params => {
     update,
     move,
     delete: deleteEntry,
+    moveToBin,
+    restoreFromBin,
     deleteRevision,
     deleteMultipleEntries,
     get,