import json
import os
import time
from typing import List

CONTEXT_FILE = "data/memory/context.json"
MAX_MEMORY = 100


def load_context() -> List[dict]:
    if os.path.exists(CONTEXT_FILE):
        with open(CONTEXT_FILE, "r", encoding="utf-8") as f:
            return json.load(f)
    return []


def save_context(mem: List[dict]):
    with open(CONTEXT_FILE, "w", encoding="utf-8") as f:
        json.dump(mem[-MAX_MEMORY:], f, indent=2)


def add_to_context(text: str, source: str = "user"):
    mem = load_context()
    mem.append({
        "timestamp": time.time(),
        "source": source,
        "text": text
    })
    save_context(mem)


def get_recent_context(n: int = 5) -> List[str]:
    mem = load_context()
    return [entry["text"] for entry in mem[-n:]]