from models import UserState
from redis_client import redis

class UserManager:
    def __init__(self):
        self.users = {}

    def get_user(self, user_id):
        if user_id not in self.users:
            self.users[user_id] = UserState()
        return self.users[user_id]

    async def increment_pomodoros(self, user_id):
        await redis.incr(f"user:{user_id}:pomodoros")

    async def get_pomodoros(self, user_id):
        val = await redis.get(f"user:{user_id}:pomodoros")
        return int(val) if val else 0

user_manager = UserManager()