WIP

orders-events/app/app.py

@@ -13,6 +13,7 @@ from aws_lambda_powertools.logging import correlation_paths
 from aws_lambda_powertools.utilities.typing import LambdaContext
 from layercake.dateutils import now
 from layercake.dynamodb import DynamoDBPersistenceLayer, KeyPair
+from layercake.funcs import pick
 
 from boto3clients import dynamodb_client
 from config import ORDER_TABLE
@@ -29,10 +30,12 @@ class OrderNotFoundError(NotFoundError): ...
 class InvoiceNotFoundError(NotFoundError): ...
 
 
-class StatusAttr(Enum):
-    # Post-migration (orders): uncomment the following lines
+class StatusTimestampAttr(Enum):
+    # Post-migration (orders): uncomment the following 2 lines
     # PAID = 'paid_at'
     # EXTERNALLY_PAID = 'paid_at'
+
+    # Post-migration (orders): remove the following 2 lines
     EXTERNALLY_PAID = 'payment_date'
     PAID = 'payment_date'
 
@@ -41,19 +44,24 @@ class StatusAttr(Enum):
     EXPIRED = 'expired_at'
 
 
-def _status_attr(status: str) -> StatusAttr | None:
+def _timestamp_attr_for_status(status: str) -> str | None:
     try:
-        return StatusAttr[status]
+        return StatusTimestampAttr[status].value
     except KeyError:
         return None
 
 
 def _friendly_status(status: str) -> str:
-    if 'status' == 'EXTERNALLY_PAID':
+    if status == 'EXTERNALLY_PAID':
         return 'PAID'
     return status
 
 
+def _get_order_owner(order_id: str) -> dict:
+    r = dyn.get_item(KeyPair(order_id, '0'))
+    return pick(('user_id', 'org_id'), r)
+
+
 @app.post('/<order_id>/postback')
 @tracer.capture_method
 def postback(order_id: str):
@@ -62,31 +70,33 @@ def postback(order_id: str):
 
     now_ = now()
     event = decoded_body['event']
-    status = decoded_body.get('data[status]', '').upper()
-    status_attr = _status_attr(status)
+    raw_status = decoded_body.get('data[status]', '').upper()
+    status = _friendly_status(raw_status)
+    timestamp_attr = _timestamp_attr_for_status(raw_status)
 
-    if event != 'invoice.status_changed' or not status_attr:
-        return Response(status_code=HTTPStatus.NO_CONTENT)
+    if event != 'invoice.status_changed' or not timestamp_attr:
+        logger.debug('Event not acceptable', order_id=order_id)
+        return Response(status_code=HTTPStatus.NOT_ACCEPTABLE)
 
     with dyn.transact_writer() as transact:
         transact.update(
             key=KeyPair(order_id, '0'),
             update_expr='SET #status = :status, \
-                             #status_attr = :now, \
+                             #ts_attr = :now, \
                              updated_at = :now',
             cond_expr='attribute_exists(sk)',
             expr_attr_names={
                 '#status': 'status',
-                '#status_attr': status_attr.value,
+                '#ts_attr': timestamp_attr,
             },
             expr_attr_values={
-                ':status': _friendly_status(status),
+                ':status': status,
                 ':now': now_,
             },
             exc_cls=OrderNotFoundError,
         )
 
-        if status == 'EXTERNALLY_PAID':
+        if raw_status == 'EXTERNALLY_PAID':
             transact.update(
                 key=KeyPair(order_id, 'INVOICE'),
                 cond_expr='attribute_exists(sk)',
@@ -99,6 +109,51 @@ def postback(order_id: str):
                 exc_cls=InvoiceNotFoundError,
             )
 
+    if status == 'PAID':
+        try:
+            dyn.put_item(
+                item={
+                    'id': order_id,
+                    'sk': 'FULFILLMENT',
+                    'status': 'IN_PROGRESS',
+                    'created_at': now_,
+                    **_get_order_owner(order_id),
+                },
+                cond_expr='attribute_not_exists(sk)',
+            )
+        except Exception:
+            pass
+
+    if status in ('CANCELED', 'REFUNDED'):
+        try:
+            with dyn.transact_writer() as transact:
+                transact.condition(
+                    key=KeyPair(order_id, 'FULFILLMENT'),
+                    cond_expr=(
+                        'attribute_exists(sk) '
+                        'AND #status <> :in_progress '
+                        'AND #status <> :rollback'
+                    ),
+                    expr_attr_names={
+                        '#status': 'status',
+                    },
+                    expr_attr_values={
+                        ':in_progress': 'IN_PROGRESS',
+                        ':rollback': 'ROLLBACK',
+                    },
+                )
+                transact.put(
+                    item={
+                        'id': order_id,
+                        'sk': 'FULFILLMENT#ROLLBACK',
+                        'created_at': now_,
+                    },
+                    cond_expr='attribute_not_exists(sk)',
+                )
+                logger.debug('Fulfillment rollback event created', order_id=order_id)
+        except Exception:
+            logger.debug('Fulfillment rollback event already exists', order_id=order_id)
+
     return Response(status_code=HTTPStatus.NO_CONTENT)