feat: add untested stars purchases (#4)

Author: emptybutton

deploy/dev/docker-compose.yaml

@@ -16,6 +16,10 @@ services:
         condition: service_healthy
       redis:
         condition: service_healthy
+      nats:
+        condition: service_healthy
+      nats_streams:
+        condition: service_completed_successfully
     environment:
       TTT_POSTGRES_URL: postgresql+psycopg://root:root@postgres/root
       TTT_POSTGRES_POOL_SIZE: 8
@@ -27,6 +31,8 @@ services:
       TTT_REDIS_URL: redis://redis:6379/0
       TTT_REDIS_POOL_SIZE: 16
 
+      TTT_NATS_URL: nats://nats:4222
+
       TTT_GAME_WAITING_QUEUE_PULLING_TIMEOUT_MIN_MS: 100
       TTT_GAME_WAITING_QUEUE_PULLING_TIMEOUT_SALT_MS: 100
     secrets:
@@ -64,10 +70,36 @@ services:
       interval: 3s
     command: redis-server /mnt/dev/redis.conf
 
+  nats:
+    image: nats:2.11.5-alpine3.22
+    container_name: ttt-nats
+    volumes:
+      - nats-data:/data
+    ports:
+      - 4222:4222
+    command: nats-server -js
+    healthcheck:
+      test: wget http://nats:8222/healthz -q -O /dev/null
+      start_period: 2m
+      start_interval: 0.5s
+      interval: 3s
+
+  nats_streams:
+    image: bitnami/natscli:0.2.3-debian-12-r4
+    container_name: ttt-nats-streams
+    depends_on:
+      nats:
+        condition: service_healthy
+    volumes:
+      - ./nats:/mnt
+    entrypoint: null
+    command: ["bash", "/mnt/add_streams.sh"]
+
 volumes:
   backend-data: null
   postgres-data: null
   redis-data: null
+  nats-data: null
 
 secrets:
   secrets:

deploy/dev/nats/add_streams.sh

@@ -0,0 +1,3 @@
+#!/bin/bash
+
+nats stream add --config /mnt/streams/player.json

deploy/dev/nats/streams/player.json

@@ -0,0 +1,14 @@
+{
+    "name": "PLAYER",
+    "subjects": ["player.>"],
+    "retention": "limits",
+    "max_consumers": -1,
+    "max_msgs": -1,
+    "max_bytes": -1,
+    "max_age": 31536000000000000,
+    "max_msg_size": -1,
+    "storage": "file",
+    "discard": "old",
+    "num_replicas": 1,
+    "duplicate_window": 0
+}

pyproject.toml

@@ -20,6 +20,7 @@ dependencies = [
     "pydantic==2.10.6",
     "pydantic-settings[yaml]==2.9.1",
     "aiogram==3.20.0.post0",
+    "nats-py[nkeys]==2.10.0",
 ]
 
 [dependency-groups]

src/ttt/application/player/complete_stars_purshase.py

@@ -0,0 +1,53 @@
+from dataclasses import dataclass
+
+from ttt.application.common.ports.clock import Clock
+from ttt.application.common.ports.map import Map
+from ttt.application.common.ports.transaction import Transaction
+from ttt.application.player.ports.player_views import PlayerViews
+from ttt.application.player.ports.players import Players
+from ttt.application.player.ports.stars_purchase_payment_gateway import (
+    StarsPurchasePaymentGateway,
+)
+from ttt.entities.core.player.player import NoPurchaseError
+from ttt.entities.finance.payment.payment import PaymentAlreadyCompletedError
+from ttt.entities.tools.tracking import Tracking
+
+
+@dataclass(frozen=True, unsafe_hash=False)
+class CompleteStarsPurshase:
+    clock: Clock
+    payment_gateway: StarsPurchasePaymentGateway
+    players: Players
+    transaction: Transaction
+    map_: Map
+    player_views: PlayerViews
+
+    async def __call__(self) -> None:
+        current_datetime = await self.clock.current_datetime()
+
+        async for paid_payment in self.payment_gateway.paid_payment_stream():
+            async with self.transaction:
+                player = await self.players.player_with_id(
+                    paid_payment.location.player_id,
+                )
+
+                tracking = Tracking()
+                try:
+                    player.complete_stars_purchase(
+                        paid_payment.purshase_id,
+                        paid_payment.success,
+                        current_datetime,
+                        tracking,
+                    )
+                except (NoPurchaseError, PaymentAlreadyCompletedError):
+                    ...
+                else:
+                    await self.map_(tracking)
+                    await (
+                        self.player_views
+                        .render_completed_stars_purshase_view(
+                            player,
+                            paid_payment.purshase_id,
+                            paid_payment.location,
+                        )
+                    )

src/ttt/application/player/initiate_stars_purchase_payment.py

@@ -0,0 +1,64 @@
+from asyncio import gather
+from dataclasses import dataclass
+
+from ttt.application.common.ports.clock import Clock
+from ttt.application.common.ports.transaction import Transaction
+from ttt.application.common.ports.uuids import UUIDs
+from ttt.application.player.ports.player_fsm import PlayerFsm
+from ttt.application.player.ports.player_views import PlayerViews
+from ttt.application.player.ports.players import Players
+from ttt.application.player.ports.stars_purchase_payment_gateway import (
+    StarsPurchasePaymentGateway,
+)
+from ttt.entities.core.player.location import PlayerLocation
+from ttt.entities.core.player.stars_purchase import StarsPurchase
+from ttt.entities.core.stars import NonExchangeableRublesForStarsError
+from ttt.entities.finance.rubles import Rubles
+from ttt.entities.tools.tracking import Tracking
+
+
+@dataclass(frozen=True, unsafe_hash=False)
+class InitiateStarsPurchasePayment:
+    fsm: PlayerFsm
+    transaction: Transaction
+    players: Players
+    uuids: UUIDs
+    clock: Clock
+    player_views: PlayerViews
+    payment_gateway: StarsPurchasePaymentGateway
+
+    async def __call__(
+        self, location: PlayerLocation, rubles: Rubles,
+    ) -> None:
+        async with self.transaction:
+            player, purchase_id, payment_id, current_datetime = await gather(
+                self.players.player_with_id(location.player_id),
+                self.uuids.random_uuid(),
+                self.uuids.random_uuid(),
+                self.clock.current_datetime(),
+            )
+
+            tracking = Tracking()
+            try:
+                player.initiate_stars_purchase_payment(
+                    purchase_id,
+                    location.chat_id,
+                    payment_id,
+                    rubles,
+                    current_datetime,
+                    tracking,
+                )
+            except NonExchangeableRublesForStarsError:
+                await self.fsm.set(None)
+                await (
+                    self.player_views
+                    .render_non_exchangeable_rubles_for_stars_view(location)
+                )
+                return
+
+            await self.fsm.set(None)
+            await gather(*[
+                self.payment_gateway.process_payment(it, location)
+                for it in tracking.new
+                if isinstance(it, StarsPurchase)
+            ])

src/ttt/application/player/ports/player_views.py

@@ -1,6 +1,8 @@
 from abc import ABC, abstractmethod
+from uuid import UUID
 
 from ttt.entities.core.player.location import PlayerLocation
+from ttt.entities.core.player.player import Player
 from ttt.entities.core.stars import Stars
 
 
@@ -68,3 +70,18 @@ async def render_emoji_selected_view(
     async def render_selected_emoji_removed_view(
         self, location: PlayerLocation, /,
     ) -> None: ...
+
+    @abstractmethod
+    async def render_wait_rubles_to_start_stars_purshase_view(
+        self, location: PlayerLocation, /,
+    ) -> None: ...
+
+    @abstractmethod
+    async def render_non_exchangeable_rubles_for_stars_view(
+        self, location: PlayerLocation, /,
+    ) -> None: ...
+
+    @abstractmethod
+    async def render_completed_stars_purshase_view(
+        self, player: Player, purshase_id: UUID, location: PlayerLocation, /,
+    ) -> None: ...

src/ttt/application/player/ports/stars_purchase_payment_gateway.py

@@ -0,0 +1,29 @@
+from abc import ABC, abstractmethod
+from collections.abc import AsyncIterable
+from dataclasses import dataclass
+from uuid import UUID
+
+from ttt.entities.core.player.location import PlayerLocation
+from ttt.entities.core.player.stars_purchase import StarsPurchase
+from ttt.entities.finance.payment.success import PaymentSuccess
+
+
+@dataclass(frozen=True)
+class PaidStarsPurchasePayment:
+    purshase_id: UUID
+    location: PlayerLocation
+    success: PaymentSuccess
+
+
+class StarsPurchasePaymentGateway(ABC):
+    @abstractmethod
+    async def process_payment(
+        self,
+        purshase: StarsPurchase,
+        location: PlayerLocation,
+    ) -> None:
+        ...
+
+    @abstractmethod
+    def paid_payment_stream(self) -> AsyncIterable[PaidStarsPurchasePayment]:
+        ...

src/ttt/application/player/wait_rubles_to_start_stars_purshase.py

@@ -0,0 +1,17 @@
+from dataclasses import dataclass
+
+from ttt.application.player.ports.player_fsm import PlayerFsm
+from ttt.application.player.ports.player_views import PlayerViews
+from ttt.entities.core.player.location import PlayerLocation
+
+
+@dataclass(frozen=True, unsafe_hash=False)
+class WaitRublesToStartStarsPurshase:
+    fsm: PlayerFsm
+    player_views: PlayerViews
+
+    async def __call__(self, location: PlayerLocation) -> None:
+        await (
+            self.player_views
+            .render_wait_rubles_to_start_stars_purshase_view(location)
+        )

src/ttt/entities/aggregate.py

@@ -1,5 +1,6 @@
 from ttt.entities.core.game.game import GameAggregate
 from ttt.entities.core.player.player import PlayerAggregate
+from ttt.entities.finance.payment.payment import PaymentAggregate
 
 
-type Aggregate = GameAggregate | PlayerAggregate
+type Aggregate = GameAggregate | PlayerAggregate | PaymentAggregate