state.py

"""
AgentState: persisteert welke artikelen al verwerkt zijn en welke bets geplaatst zijn.
Slaat data op in een lokaal JSON-bestand (werkt gratis op elke hosting).
"""

import json
import logging
import os
from datetime import datetime, timedelta, timezone
from pathlib import Path

log = logging.getLogger(__name__)

STATE_FILE = os.getenv("STATE_FILE", "agent_state.json")


class AgentState:
    def __init__(self):
        self._path = Path(STATE_FILE)
        self._data = self._load()

    def _load(self) -> dict:
        if self._path.exists():
            try:
                with open(self._path) as f:
                    return json.load(f)
            except Exception as e:
                log.warning(f"State laden mislukt, nieuw bestand: {e}")
        return {"processed_articles": {}, "bets": []}

    def _save(self):
        with open(self._path, "w") as f:
            json.dump(self._data, f, indent=2, default=str)

    def already_processed(self, article_id: str) -> bool:
        return article_id in self._data["processed_articles"]

    def mark_processed(self, article_id: str):
        self._data["processed_articles"][article_id] = datetime.now(timezone.utc).isoformat()
        self._prune_old_articles()
        self._save()

    def record_bet(self, market_id: str, side: str, amount: float, price: float):
        self._data["bets"].append({
            "market_id": market_id,
            "side": side,
            "amount_usdc": amount,
            "entry_price": price,
            "placed_at": datetime.now(timezone.utc).isoformat(),
        })
        self._save()
        log.info(f"Bet opgeslagen: {side} ${amount} @ {price:.2%} op {market_id}")

    def _prune_old_articles(self):
        """Verwijder artikelen ouder dan 48 uur om bestand klein te houden."""
        cutoff = (datetime.now(timezone.utc) - timedelta(hours=48)).isoformat()
        self._data["processed_articles"] = {
            k: v for k, v in self._data["processed_articles"].items()
            if v >= cutoff
        }

    def get_bets(self) -> list:
        return self._data.get("bets", [])