import os

from pydantic import Field

from liteyuki.utils.data import LiteModel, Database as DB

DATA_PATH = "data/liteyuki"

user_db = DB(os.path.join(DATA_PATH, 'users.ldb'))
group_db = DB(os.path.join(DATA_PATH, 'groups.ldb'))
plugin_db = DB(os.path.join(DATA_PATH, 'plugins.ldb'))
common_db = DB(os.path.join(DATA_PATH, 'common.ldb'))


class User(LiteModel):
    user_id: str = Field(str(), alias='user_id')
    username: str = Field(str(), alias='username')
    profile: dict[str, str] = Field(dict(), alias='profile')
    enabled_plugins: list[str] = Field(list(), alias='enabled_plugins')
    disabled_plugins: list[str] = Field(list(), alias='disabled_plugins')


class GroupChat(LiteModel):
    # Group是一个关键字，所以这里用GroupChat
    group_id: str = Field(str(), alias='group_id')
    group_name: str = Field(str(), alias='group_name')
    enabled_plugins: list[str] = Field([], alias='enabled_plugins')
    disabled_plugins: list[str] = Field([], alias='disabled_plugins')


class InstalledPlugin(LiteModel):
    module_name: str = Field(str(), alias='module_name')
    version: str = Field(str(), alias='version')


class GlobalPlugin(LiteModel):
    module_name: str = Field(str(), alias='module_name')
    enabled: bool = Field(True, alias='enabled')


def auto_migrate():
    user_db.auto_migrate(User)
    group_db.auto_migrate(GroupChat)
    plugin_db.auto_migrate(InstalledPlugin)
    common_db.auto_migrate(GlobalPlugin)