# 一些全局运行时数据
from datetime import datetime
from typing import TYPE_CHECKING
import sqlite3
from threading import Lock

if TYPE_CHECKING:
    from core.base import Plugin

script_start_time = datetime.now()
llonebot_data_path = "/app/llonebot/server_data"    # 使用api是用这个地址
python_data_path = "./server_data"                  # 在python脚本中访问用这个地址
onebot_qq_volume = "/var/lib/docker/volumes/onebot_qq_volume/_data"
startup_changelog_sent = True
plugin_registry: list[type["Plugin"]] = []
DEFAULT_GROUP_ID = 296470819 # 在这里填写你想固定使用的群号

# 系统级插件（不可按群禁用，始终运行）
SYSTEM_PLUGINS = frozenset({
    "menu",
    "group_manager",
    "startup_changelog",
    "backup",
    "update",
    "auto_friend",
    "welcome",
})

# 跑团录制状态
recording_lock = Lock()
recording_sessions: dict[int, dict] = {}      # group_id → {"start": datetime, "messages": list, "participants": dict}
last_completed: dict[int, dict] = {}           # group_id → {"start": datetime, "end": datetime, "messages": list, "participants": dict}
RECORDING_ALLOWED_PLUGINS = frozenset({"trpg_dice", "trpg_session", "trpg_char"})

# 跑团角色：group_id → {user_id → "dm"|"ob"}
group_roles: dict[int, dict[str, str]] = {}

# 当前游戏规则系统（跑团骰子检定用），未来规则切换功能修改此值
GAME_SYSTEM = "dnd5e"

# 卧底游戏
game_lock = Lock()
game_rooms: dict[str, dict] = {}

def is_group_recording(group_id: int) -> bool:
    return group_id in recording_sessions

def get_recording_session(group_id: int) -> dict | None:
    return recording_sessions.get(group_id)

def get_last_completed(group_id: int) -> dict | None:
    return last_completed.get(group_id)

def pop_last_completed(group_id: int) -> dict | None:
    return last_completed.pop(group_id, None)

def is_plugin_allowed_during_recording(key: str) -> bool:
    return key in RECORDING_ALLOWED_PLUGINS

def plugin_key(plugin_cls: type["Plugin"]) -> str:
    return plugin_cls.__module__.split(".", 1)[1]

def is_plugin_enabled(plugin_cls: type["Plugin"], group_id: int | None) -> bool:
    key = plugin_key(plugin_cls)
    if key in SYSTEM_PLUGINS:
        return True
    gid = group_id if group_id is not None else 0
    try:
        conn = sqlite3.connect("data.db")
        cur = conn.execute(
            "SELECT 1 FROM group_plugin_config WHERE group_id = ? AND plugin_name = ?",
            (gid, key)
        )
        enabled = cur.fetchone() is not None
        conn.close()
        return enabled
    except sqlite3.Error:
        return True

def migrate_group_plugin_config():
    conn = sqlite3.connect("data.db")
    conn.execute("""
        CREATE TABLE IF NOT EXISTS group_plugin_config (
            group_id INTEGER NOT NULL,
            plugin_name TEXT NOT NULL,
            PRIMARY KEY (group_id, plugin_name)
        )
    """)
    rows = [(DEFAULT_GROUP_ID, plugin_key(cls)) for cls in plugin_registry
            if plugin_key(cls) not in SYSTEM_PLUGINS]
    if rows:
        conn.executemany(
            "INSERT OR IGNORE INTO group_plugin_config (group_id, plugin_name) VALUES (?, ?)", rows
        )
    conn.commit()
    conn.close()