# shopmate/logging/audit.py -- every ShopMate API call logged for compliance
import anthropic, json, hashlib, uuid
from datetime import datetime, timezone
from pathlib import Path

LOG_FILE = Path("logs/shopmate_audit.jsonl")
LOG_FILE.parent.mkdir(exist_ok=True)
client = anthropic.Anthropic()

def logged_create(brand_id: str, feature: str, **kwargs):
    """Wrap every ShopMate Claude call with audit logging."""
    resp = client.messages.create(**kwargs)
    entry = {
        "id":        str(uuid.uuid4()),
        "ts":        datetime.now(timezone.utc).isoformat(),
        "brand":     brand_id,
        "feature":   feature,
        "model":     kwargs.get("model"),
        "tokens_in": resp.usage.input_tokens,
        "tokens_out":resp.usage.output_tokens,
        "cost_usd":  round(resp.usage.input_tokens/1e6*0.80 + resp.usage.output_tokens/1e6*4.00, 6),
    }
    with LOG_FILE.open("a") as f:
        f.write(json.dumps(entry) + "
")
    return resp

# Monthly cost report per brand
from collections import defaultdict

def monthly_report():
    logs = [json.loads(l) for l in LOG_FILE.read_text().splitlines()]
    by_brand = defaultdict(lambda: {"calls":0,"cost":0})
    for log in logs:
        by_brand[log["brand"]]["calls"]  += 1
        by_brand[log["brand"]]["cost"]   += log["cost_usd"]
    print(f"{'Brand':<15} {'Calls':>6} {'Cost':>10}")
    for brand, s in sorted(by_brand.items(), key=lambda x: -x[1]["cost"]):
        print(f"{brand:<15} {s['calls']:>6,} ${s['cost']:>9.2f}")