IMN-792 Add events service v2 in agreement-platformstate-writer

packages/agreement-platformstate-writer/src/consumerServiceV2.ts

@@ -1,27 +1,281 @@
 import { DynamoDBClient } from "@aws-sdk/client-dynamodb";
-import { AgreementEventEnvelopeV2 } from "pagopa-interop-models";
+import {
+  Agreement,
+  AgreementEventEnvelopeV2,
+  AgreementV2,
+  fromAgreementV2,
+  genericInternalError,
+  makeGSIPKConsumerIdEServiceId,
+  makeGSIPKEServiceIdDescriptorId,
+  makePlatformStatesAgreementPK,
+  makePlatformStatesEServiceDescriptorPK,
+  PlatformStatesAgreementEntry,
+  PlatformStatesCatalogEntry,
+} from "pagopa-interop-models";
 import { match } from "ts-pattern";
+import {
+  agreementStateToItemState,
+  deleteAgreementEntry,
+  isAgreementTheLatest,
+  readAgreementEntry,
+  readCatalogEntry,
+  updateAgreementStateInPlatformStatesEntry,
+  updateAgreementStateInTokenGenerationStatesTable,
+  updateAgreementStateInTokenGenerationStatesTablePlusDescriptorInfo,
+  writeAgreementEntry,
+} from "./utils.js";
 
 export async function handleMessageV2(
   message: AgreementEventEnvelopeV2,
-  _dynamoDBClient: DynamoDBClient
+  dynamoDBClient: DynamoDBClient
 ): Promise<void> {
   await match(message)
+    .with({ type: "AgreementActivated" }, async (msg) => {
+      const agreement = parseAgreement(msg.data.agreement);
+      const primaryKey = makePlatformStatesAgreementPK(agreement.id);
+
+      const existingAgreementEntry = await readAgreementEntry(
+        primaryKey,
+        dynamoDBClient
+      );
+      const GSIPK_consumerId_eserviceId = makeGSIPKConsumerIdEServiceId({
+        consumerId: agreement.consumerId,
+        eserviceId: agreement.eserviceId,
+      });
+
+      if (existingAgreementEntry) {
+        if (existingAgreementEntry.version > msg.version) {
+          // Stops processing if the message is older than the agreement entry
+          return Promise.resolve();
+        } else {
+          await updateAgreementStateInPlatformStatesEntry(
+            dynamoDBClient,
+            primaryKey,
+            agreementStateToItemState(agreement.state),
+            msg.version
+          );
+        }
+      } else {
+        const agreementEntry: PlatformStatesAgreementEntry = {
+          PK: primaryKey,
+          state: agreementStateToItemState(agreement.state),
+          version: msg.version,
+          updatedAt: new Date().toISOString(),
+          GSIPK_consumerId_eserviceId,
+          GSISK_agreementTimestamp:
+            // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
+            agreement.stamps.activation!.when.toISOString(),
+          agreementDescriptorId: agreement.descriptorId,
+        };
+
+        await writeAgreementEntry(agreementEntry, dynamoDBClient);
+      }
+
+      const pkCatalogEntry = makePlatformStatesEServiceDescriptorPK({
+        eserviceId: agreement.eserviceId,
+        descriptorId: agreement.descriptorId,
+      });
+      const catalogEntry = await readCatalogEntry(
+        pkCatalogEntry,
+        dynamoDBClient
+      );
+
+      const GSIPK_eserviceId_descriptorId = makeGSIPKEServiceIdDescriptorId({
+        eserviceId: agreement.eserviceId,
+        descriptorId: agreement.descriptorId,
+      });
+
+      // token-generation-states
+      await updateAgreementStateInTokenGenerationStatesTablePlusDescriptorInfo({
+        GSIPK_consumerId_eserviceId,
+        agreementState: agreement.state,
+        dynamoDBClient,
+        GSIPK_eserviceId_descriptorId,
+        catalogEntry,
+      });
+    })
     .with(
-      { type: "AgreementAdded" },
-      { type: "AgreementDeleted" },
-      { type: "DraftAgreementUpdated" },
-      { type: "AgreementSubmitted" },
-      { type: "AgreementActivated" },
       { type: "AgreementUnsuspendedByProducer" },
       { type: "AgreementUnsuspendedByConsumer" },
       { type: "AgreementUnsuspendedByPlatform" },
-      { type: "AgreementArchivedByConsumer" },
-      { type: "AgreementArchivedByUpgrade" },
-      { type: "AgreementUpgraded" },
       { type: "AgreementSuspendedByProducer" },
       { type: "AgreementSuspendedByConsumer" },
       { type: "AgreementSuspendedByPlatform" },
+      async (msg) => {
+        const agreement = parseAgreement(msg.data.agreement);
+        const primaryKey = makePlatformStatesAgreementPK(agreement.id);
+        const agreementEntry = await readAgreementEntry(
+          primaryKey,
+          dynamoDBClient
+        );
+
+        if (!agreementEntry || agreementEntry.version > msg.version) {
+          return Promise.resolve();
+        } else {
+          await updateAgreementStateInPlatformStatesEntry(
+            dynamoDBClient,
+            primaryKey,
+            agreementStateToItemState(agreement.state),
+            msg.version
+          );
+
+          const GSIPK_consumerId_eserviceId = makeGSIPKConsumerIdEServiceId({
+            consumerId: agreement.consumerId,
+            eserviceId: agreement.eserviceId,
+          });
+
+          if (
+            await isAgreementTheLatest(
+              GSIPK_consumerId_eserviceId,
+              agreement.id,
+              dynamoDBClient
+            )
+          ) {
+            // token-generation-states only if agreement is the latest
+
+            await updateAgreementStateInTokenGenerationStatesTable({
+              GSIPK_consumerId_eserviceId,
+              agreementState: agreement.state,
+              dynamoDBClient,
+            });
+          }
+        }
+      }
+    )
+    .with({ type: "AgreementUpgraded" }, async (msg) => {
+      const agreement = parseAgreement(msg.data.agreement);
+      const primaryKey = makePlatformStatesAgreementPK(agreement.id);
+      const agreementEntry = await readAgreementEntry(
+        primaryKey,
+        dynamoDBClient
+      );
+
+      const pkCatalogEntry = makePlatformStatesEServiceDescriptorPK({
+        eserviceId: agreement.eserviceId,
+        descriptorId: agreement.descriptorId,
+      });
+      const catalogEntry = await readCatalogEntry(
+        pkCatalogEntry,
+        dynamoDBClient
+      );
+      if (!catalogEntry) {
+        // TODO double-check
+        throw genericInternalError("Catalog entry not found");
+      }
+
+      const GSIPK_consumerId_eserviceId = makeGSIPKConsumerIdEServiceId({
+        consumerId: agreement.consumerId,
+        eserviceId: agreement.eserviceId,
+      });
+
+      if (agreementEntry) {
+        if (agreementEntry.version > msg.version) {
+          return Promise.resolve();
+        } else {
+          await updateAgreementStateInPlatformStatesEntry(
+            dynamoDBClient,
+            primaryKey,
+            agreementStateToItemState(agreement.state),
+            msg.version
+          );
+        }
+      } else {
+        const newAgreementEntry: PlatformStatesAgreementEntry = {
+          PK: primaryKey,
+          state: agreementStateToItemState(agreement.state),
+          version: msg.version,
+          updatedAt: new Date().toISOString(),
+          GSIPK_consumerId_eserviceId,
+          GSISK_agreementTimestamp:
+            // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
+            agreement.stamps.activation!.when.toISOString(),
+          agreementDescriptorId: agreement.descriptorId,
+        };
+
+        await writeAgreementEntry(newAgreementEntry, dynamoDBClient);
+      }
+
+      const doOperationOnTokenStates = async (
+        catalogEntry: PlatformStatesCatalogEntry
+      ): Promise<void> => {
+        if (
+          await isAgreementTheLatest(
+            GSIPK_consumerId_eserviceId,
+            agreement.id,
+            dynamoDBClient
+          )
+        ) {
+          // token-generation-states only if agreement is the latest
+          const GSIPK_eserviceId_descriptorId = makeGSIPKEServiceIdDescriptorId(
+            {
+              eserviceId: agreement.eserviceId,
+              descriptorId: agreement.descriptorId,
+            }
+          );
+
+          await updateAgreementStateInTokenGenerationStatesTablePlusDescriptorInfo(
+            {
+              GSIPK_consumerId_eserviceId,
+              agreementState: agreement.state,
+              dynamoDBClient,
+              GSIPK_eserviceId_descriptorId,
+              catalogEntry,
+            }
+          );
+        }
+      };
+
+      await doOperationOnTokenStates(catalogEntry);
+
+      const secondRetrievalCatalogEntry = await readCatalogEntry(
+        pkCatalogEntry,
+        dynamoDBClient
+      );
+      if (!secondRetrievalCatalogEntry) {
+        // TODO double-check
+        throw genericInternalError("Catalog entry not found");
+      }
+      if (secondRetrievalCatalogEntry.state !== catalogEntry.state) {
+        await doOperationOnTokenStates(secondRetrievalCatalogEntry);
+      }
+    })
+    .with({ type: "AgreementArchivedByUpgrade" }, async (msg) => {
+      const agreement = parseAgreement(msg.data.agreement);
+      const pk = makePlatformStatesAgreementPK(agreement.id);
+      await deleteAgreementEntry(pk, dynamoDBClient);
+    })
+    .with({ type: "AgreementArchivedByConsumer" }, async (msg) => {
+      const agreement = parseAgreement(msg.data.agreement);
+
+      const primaryKey = makePlatformStatesAgreementPK(agreement.id);
+      const GSIPK_consumerId_eserviceId = makeGSIPKConsumerIdEServiceId({
+        consumerId: agreement.consumerId,
+        eserviceId: agreement.eserviceId,
+      });
+
+      if (
+        await isAgreementTheLatest(
+          GSIPK_consumerId_eserviceId,
+          agreement.id,
+          dynamoDBClient
+        )
+      ) {
+        // token-generation-states only if agreement is the latest
+
+        await updateAgreementStateInTokenGenerationStatesTable({
+          GSIPK_consumerId_eserviceId,
+          agreementState: agreement.state,
+          dynamoDBClient,
+        });
+      }
+
+      await deleteAgreementEntry(primaryKey, dynamoDBClient);
+    })
+    .with(
+      { type: "AgreementAdded" },
+      { type: "AgreementDeleted" },
+      { type: "DraftAgreementUpdated" },
+      { type: "AgreementSubmitted" },
       { type: "AgreementRejected" },
       { type: "AgreementConsumerDocumentAdded" },
       { type: "AgreementConsumerDocumentRemoved" },
@@ -31,3 +285,11 @@ export async function handleMessageV2(
     )
     .exhaustive();
 }
+
+const parseAgreement = (agreementV2: AgreementV2 | undefined): Agreement => {
+  if (!agreementV2) {
+    throw genericInternalError(`Agreement not found in message data`);
+  }
+
+  return fromAgreementV2(agreementV2);
+};