"""Repository for investment goals and reviewed goal-book content."""

from datetime import datetime
from typing import cast

from sqlalchemy import select, update
from sqlalchemy.ext.asyncio import AsyncSession

from app.model.investment_goal import AdvisorInvestmentGoal, ClientFacingContent


class InvestmentGoalRepository:
    def __init__(self, session: AsyncSession) -> None:
        self.session = session

    def add_goal(self, goal: AdvisorInvestmentGoal) -> None:
        self.session.add(goal)

    def add_goal_book(self, content: ClientFacingContent) -> None:
        self.session.add(content)

    async def goal(self, goal_no: str, *, lock: bool = False) -> AdvisorInvestmentGoal | None:
        query = select(AdvisorInvestmentGoal).where(AdvisorInvestmentGoal.goal_no == goal_no)
        if lock:
            query = query.with_for_update()
        return cast(AdvisorInvestmentGoal | None, await self.session.scalar(query))

    async def latest_for_customer(self, customer_id: int) -> AdvisorInvestmentGoal | None:
        return cast(AdvisorInvestmentGoal | None, await self.session.scalar(
            select(AdvisorInvestmentGoal)
            .where(
                AdvisorInvestmentGoal.customer_id == customer_id,
                AdvisorInvestmentGoal.status != "superseded",
            )
            .order_by(AdvisorInvestmentGoal.created_at.desc(), AdvisorInvestmentGoal.id.desc())
            .limit(1)
        ))

    async def goal_book(self, content_id: int) -> ClientFacingContent | None:
        return await self.session.get(ClientFacingContent, content_id)

    async def supersede_confirmed(
        self, customer_id: int, except_goal_no: str, now: datetime
    ) -> None:
        await self.session.execute(
            update(AdvisorInvestmentGoal)
            .where(
                AdvisorInvestmentGoal.customer_id == customer_id,
                AdvisorInvestmentGoal.status == "confirmed",
                AdvisorInvestmentGoal.goal_no != except_goal_no,
            )
            .values(status="superseded", updated_at=now)
        )