# -*- coding: utf-8 -*-
"""站点数据异步读写仓储（FastAPI 栈用）。

直接读写 auth_system 建立的同一份 SQLite（同一 DB_PATH / 同一 schema），
但走 core.db 的 aiosqlite 异步层——读并发、写串行、WAL 读写同时进行。

设计约束：本模块**不改动** auth_system 既有 schema，只做读写；字段口径与
auth_system 保持一致（键名、默认值）。迁移期两条栈共存、共库。
"""
from typing import Any

from core import db

# ---------- settings ----------

async def get_setting(key: str, default: str = '') -> str:
    """读单个站点设置（对应 auth_system.get_setting）。"""
    row = await db.fetch_one('SELECT value FROM settings WHERE key = ?', (key,))
    return row['value'] if row is not None else default


async def set_setting(key: str, value: str) -> None:
    """写站点设置（upsert）。"""
    await db.execute(
        'INSERT INTO settings (key, value) VALUES (?, ?) '
        'ON CONFLICT(key) DO UPDATE SET value = excluded.value',
        (key, value),
    )


async def all_settings() -> dict[str, str]:
    rows = await db.fetch_all('SELECT key, value FROM settings')
    return {r['key']: r['value'] for r in rows}


# ---------- announcements ----------

async def list_announcements(limit: int = 20) -> list[dict[str, Any]]:
    rows = await db.fetch_all(
        'SELECT * FROM announcements ORDER BY rowid DESC LIMIT ?', (limit,))
    return [dict(r) for r in rows]


# ---------- users（只读统计；写操作请走 auth_system 的事务口径） ----------

async def count_users(role: str | None = None) -> int:
    if role:
        return int(await db.fetch_val('SELECT COUNT(*) FROM users WHERE role = ?', (role,), 0))
    return int(await db.fetch_val('SELECT COUNT(*) FROM users', (), 0))


async def get_user_by_name(username: str) -> dict[str, Any] | None:
    row = await db.fetch_one('SELECT * FROM users WHERE username = ?', (username,))
    return dict(row) if row is not None else None


# ---------- billing ledger（只读对账） ----------

async def ledger_open_count() -> int:
    return int(await db.fetch_val(
        "SELECT COUNT(*) FROM billing_ledger WHERE status = 'open'", (), 0))