from sqlalchemy import Column, Integer, String, Text, TIMESTAMP, Boolean, text
from sqlalchemy.dialects.mysql import LONGTEXT

from database.orm import Session, DB_LINK
from database.orm_base import Base
from core.utils.i18n import default_locale

is_mysql = DB_LINK.startswith('mysql')


class SenderInfo(Base):
    """发送者信息"""
    __tablename__ = "SenderInfo"
    id = Column(String(512), primary_key=True)
    isInBlockList = Column(Boolean, default=False)
    isInAllowList = Column(Boolean, default=False)
    isSuperUser = Column(Boolean, default=False)
    warns = Column(Integer, default='0')
    disable_typing = Column(Boolean, default=False)


class TargetInfo(Base):
    __tablename__ = "TargetInfo"
    targetId = Column(String(512), primary_key=True)
    enabledModules = Column(LONGTEXT if is_mysql else Text, default='[]')
    options = Column(LONGTEXT if is_mysql else Text, default='{}')
    custom_admins = Column(LONGTEXT if is_mysql else Text, default='[]')
    muted = Column(Boolean, default=False)
    locale = Column(String(512), default=default_locale)
    petal = Column(Integer, default=0)


class StoredData(Base):
    """数据存储"""
    __tablename__ = "StoredData"
    name = Column(String(512), primary_key=True)
    value = Column(LONGTEXT if is_mysql else Text)


class CommandTriggerTime(Base):
    """命令触发时间"""
    __tablename__ = "CommandTriggerTime"
    targetId = Column(String(512), primary_key=True)
    commandName = Column(String(512))
    timestamp = Column(TIMESTAMP, default=text('CURRENT_TIMESTAMP'))


class GroupAllowList(Base):
    __tablename__ = "GroupAllowList"
    targetId = Column(String(512), primary_key=True)


class AnalyticsData(Base):
    """统计信息"""
    __tablename__ = "Analytics"
    id = Column(Integer, primary_key=True)
    moduleName = Column(String(512))
    moduleType = Column(String(512))
    targetId = Column(String(512))
    senderId = Column(String(512))
    command = Column(LONGTEXT if is_mysql else Text)
    timestamp = Column(TIMESTAMP, default=text('CURRENT_TIMESTAMP'))

    __table_args__ = {'mysql_charset': 'utf8mb4'}


class DBVersion(Base):
    __tablename__ = "DBVersion"
    value = Column(String(512), primary_key=True)


class UnfriendlyActionsTable(Base):
    __tablename__ = "unfriendly_action"
    id = Column(Integer, primary_key=True)
    targetId = Column(String(512))
    senderId = Column(String(512))
    action = Column(String(512))
    detail = Column(String(512))
    timestamp = Column(TIMESTAMP, default=text('CURRENT_TIMESTAMP'))


class JobQueueTable(Base):
    __tablename__ = "job_queues"
    taskid = Column(String(512), primary_key=True)
    targetClient = Column(String(512))
    hasDone = Column(Boolean, default=False)
    action = Column(String(512))
    args = Column(LONGTEXT if is_mysql else Text, default='{}')
    returnVal = Column(LONGTEXT if is_mysql else Text, default='{}')
    timestamp = Column(TIMESTAMP, default=text('CURRENT_TIMESTAMP'))

    __table_args__ = {'mysql_charset': 'utf8mb4'}


Session.create()
__all__ = ["SenderInfo", "TargetInfo", "CommandTriggerTime", "GroupAllowList",
           "StoredData", "DBVersion", "AnalyticsData", "UnfriendlyActionsTable", "JobQueueTable"]