"""
企业规则模型

用于存储企业自定义的映射规则，实现规则复用
"""

from datetime import datetime
from enum import Enum
from typing import Optional

from sqlalchemy import ForeignKey, Integer, String, Boolean, DateTime, JSON
from sqlalchemy.orm import Mapped, mapped_column, relationship

from app.models.base import BaseModel


class RuleType(str, Enum):
    """规则类型"""
    FIELD_MAPPING = "FIELD_MAPPING"  # 字段映射规则
    ACCOUNT_MAPPING = "ACCOUNT_MAPPING"  # 科目映射规则
    DEPARTMENT_MAPPING = "DEPARTMENT_MAPPING"  # 部门映射规则


class RuleStatus(str, Enum):
    """规则状态"""
    ACTIVE = "ACTIVE"  # 启用
    INACTIVE = "INACTIVE"  # 停用


class CompanyRule(BaseModel):
    """
    企业规则模型

    存储企业自定义的业务规则，用于自动应用
    """

    __tablename__ = "company_rules"

    company_id: Mapped[int] = mapped_column(Integer, ForeignKey("companies.id"), nullable=False, index=True)
    rule_type: Mapped[str] = mapped_column(String(50), nullable=False, index=True, comment="规则类型")
    match_condition: Mapped[dict] = mapped_column(JSON, nullable=False, comment="匹配条件")
    target_value: Mapped[str] = mapped_column(String(100), nullable=False, comment="目标值")
    priority: Mapped[int] = mapped_column(Integer, nullable=False, default=0, comment="优先级，数字越大优先级越高")
    status: Mapped[str] = mapped_column(String(20), nullable=False, default=RuleStatus.ACTIVE.value, comment="状态")
    description: Mapped[Optional[str]] = mapped_column(String(500), nullable=True, comment="规则描述")
    match_count: Mapped[int] = mapped_column(Integer, nullable=False, default=0, comment="匹配次数")
    last_used_at: Mapped[Optional[datetime]] = mapped_column(DateTime, nullable=True)
    created_by: Mapped[Optional[int]] = mapped_column(Integer, ForeignKey("users.id"), nullable=True)
    updated_by: Mapped[Optional[int]] = mapped_column(Integer, ForeignKey("users.id"), nullable=True)

    # 关系
    company = relationship("Company", back_populates="rules")
    creator = relationship("User", foreign_keys=[created_by])
    updater = relationship("User", foreign_keys=[updated_by])

    def __repr__(self) -> str:
        return f"<CompanyRule(type='{self.rule_type}', target='{self.target_value}', priority={self.priority})>"