from datetime import datetime
from flask_sqlalchemy import SQLAlchemy

db = SQLAlchemy()

class User(db.Model):
    __tablename__ = 'users'

    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(50), unique=True, nullable=False)
    email = db.Column(db.String(100), unique=True, nullable=False)
    password = db.Column(db.String(255), nullable=False)
    group_id = db.Column(db.Integer, db.ForeignKey('user_groups.id'), default=2)
    register_ip = db.Column(db.String(45))
    last_login_ip = db.Column(db.String(45))
    total_parse_count = db.Column(db.Integer, default=0)
    is_active = db.Column(db.Boolean, default=True)
    created_at = db.Column(db.DateTime, default=datetime.utcnow)
    updated_at = db.Column(db.DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)

    group = db.relationship('UserGroup', backref='users')

    @property
    def is_authenticated(self):
        return True

    def get_id(self):
        return str(self.id)

class UserGroup(db.Model):
    __tablename__ = 'user_groups'

    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(50), unique=True, nullable=False)
    daily_limit = db.Column(db.Integer, default=10)
    description = db.Column(db.String(255))
    created_at = db.Column(db.DateTime, default=datetime.utcnow)
    updated_at = db.Column(db.DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)

class Admin(db.Model):
    __tablename__ = 'admins'

    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(50), unique=True, nullable=False)
    password = db.Column(db.String(255), nullable=False)
    email = db.Column(db.String(100))
    totp_secret = db.Column(db.String(100))
    is_2fa_enabled = db.Column(db.Boolean, default=False)
    last_login_ip = db.Column(db.String(45))
    last_login_at = db.Column(db.DateTime)
    created_at = db.Column(db.DateTime, default=datetime.utcnow)
    updated_at = db.Column(db.DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)

class ParserAPI(db.Model):
    __tablename__ = 'parser_apis'

    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(100), nullable=False)
    platform = db.Column(db.String(50), nullable=False)
    api_url = db.Column(db.String(500), nullable=False)
    api_key = db.Column(db.String(255))
    weight = db.Column(db.Integer, default=1)
    is_enabled = db.Column(db.Boolean, default=True)
    health_status = db.Column(db.Boolean, default=True)
    last_check_at = db.Column(db.DateTime)
    fail_count = db.Column(db.Integer, default=0)
    total_calls = db.Column(db.Integer, default=0)
    success_calls = db.Column(db.Integer, default=0)
    avg_response_time = db.Column(db.Integer, default=0)
    created_at = db.Column(db.DateTime, default=datetime.utcnow)
    updated_at = db.Column(db.DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)

class SiteConfig(db.Model):
    __tablename__ = 'site_config'

    id = db.Column(db.Integer, primary_key=True)
    config_key = db.Column(db.String(100), unique=True, nullable=False)
    config_value = db.Column(db.Text)
    config_type = db.Column(db.String(50), default='string')
    description = db.Column(db.String(255))
    updated_at = db.Column(db.DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)

class SMTPConfig(db.Model):
    __tablename__ = 'smtp_config'

    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(100), nullable=False)
    host = db.Column(db.String(255), nullable=False)
    port = db.Column(db.Integer, nullable=False)
    username = db.Column(db.String(255), nullable=False)
    password = db.Column(db.String(255), nullable=False)
    from_email = db.Column(db.String(255), nullable=False)
    from_name = db.Column(db.String(100))
    use_tls = db.Column(db.Boolean, default=True)
    is_enabled = db.Column(db.Boolean, default=True)
    is_default = db.Column(db.Boolean, default=False)
    weight = db.Column(db.Integer, default=1)
    send_count = db.Column(db.Integer, default=0)
    fail_count = db.Column(db.Integer, default=0)
    created_at = db.Column(db.DateTime, default=datetime.utcnow)
    updated_at = db.Column(db.DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)

class EmailVerification(db.Model):
    __tablename__ = 'email_verification'

    id = db.Column(db.Integer, primary_key=True)
    email = db.Column(db.String(100), nullable=False)
    code = db.Column(db.String(10), nullable=False)
    purpose = db.Column(db.String(50), nullable=False)
    is_used = db.Column(db.Boolean, default=False)
    expires_at = db.Column(db.DateTime, nullable=False)
    created_at = db.Column(db.DateTime, default=datetime.utcnow)

class ParseLog(db.Model):
    __tablename__ = 'parse_logs'

    id = db.Column(db.BigInteger, primary_key=True)
    user_id = db.Column(db.Integer, db.ForeignKey('users.id'))
    ip_address = db.Column(db.String(45), nullable=False)
    platform = db.Column(db.String(50), nullable=False)
    video_url = db.Column(db.String(1000), nullable=False)
    parser_api_id = db.Column(db.Integer, db.ForeignKey('parser_apis.id'))
    status = db.Column(db.String(20), nullable=False)
    error_message = db.Column(db.Text)
    response_time = db.Column(db.Integer)
    created_at = db.Column(db.DateTime, default=datetime.utcnow)

class DailyParseStat(db.Model):
    __tablename__ = 'daily_parse_stats'

    id = db.Column(db.Integer, primary_key=True)
    user_id = db.Column(db.Integer, db.ForeignKey('users.id'))
    ip_address = db.Column(db.String(45))
    date = db.Column(db.Date, nullable=False)
    parse_count = db.Column(db.Integer, default=0)
    success_count = db.Column(db.Integer, default=0)
    fail_count = db.Column(db.Integer, default=0)
    created_at = db.Column(db.DateTime, default=datetime.utcnow)
    updated_at = db.Column(db.DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)

class HealthCheckConfig(db.Model):
    __tablename__ = 'health_check_config'

    id = db.Column(db.Integer, primary_key=True)
    platform = db.Column(db.String(50), nullable=False)
    test_url = db.Column(db.String(1000), nullable=False)
    check_interval = db.Column(db.Integer, default=300)
    is_enabled = db.Column(db.Boolean, default=True)
    alert_email = db.Column(db.String(255))
    created_at = db.Column(db.DateTime, default=datetime.utcnow)
    updated_at = db.Column(db.DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)

class HealthCheckLog(db.Model):
    __tablename__ = 'health_check_logs'

    id = db.Column(db.BigInteger, primary_key=True)
    parser_api_id = db.Column(db.Integer, db.ForeignKey('parser_apis.id'), nullable=False)
    status = db.Column(db.String(20), nullable=False)
    response_time = db.Column(db.Integer)
    error_message = db.Column(db.Text)
    checked_at = db.Column(db.DateTime, default=datetime.utcnow)


class UserApiKey(db.Model):
    """用户 API Key"""
    __tablename__ = 'user_api_keys'

    id = db.Column(db.Integer, primary_key=True)
    user_id = db.Column(db.Integer, db.ForeignKey('users.id'), nullable=False)
    name = db.Column(db.String(100), nullable=False)  # Key 名称，方便用户区分
    api_key = db.Column(db.String(64), unique=True, nullable=False)  # API Key
    is_active = db.Column(db.Boolean, default=True)
    daily_limit = db.Column(db.Integer, default=100)  # 每日调用限制
    total_calls = db.Column(db.Integer, default=0)  # 总调用次数
    last_used_at = db.Column(db.DateTime)  # 最后使用时间
    last_used_ip = db.Column(db.String(45))  # 最后使用IP
    created_at = db.Column(db.DateTime, default=datetime.utcnow)
    updated_at = db.Column(db.DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)

    user = db.relationship('User', backref='api_keys')


class ApiKeyDailyStat(db.Model):
    """API Key 每日统计"""
    __tablename__ = 'api_key_daily_stats'

    id = db.Column(db.Integer, primary_key=True)
    api_key_id = db.Column(db.Integer, db.ForeignKey('user_api_keys.id'), nullable=False)
    date = db.Column(db.Date, nullable=False)
    call_count = db.Column(db.Integer, default=0)
    success_count = db.Column(db.Integer, default=0)
    fail_count = db.Column(db.Integer, default=0)
    created_at = db.Column(db.DateTime, default=datetime.utcnow)
    updated_at = db.Column(db.DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)

    api_key = db.relationship('UserApiKey', backref='daily_stats')


class RedeemCode(db.Model):
    """兑换码"""
    __tablename__ = 'redeem_codes'

    id = db.Column(db.Integer, primary_key=True)
    code = db.Column(db.String(32), unique=True, nullable=False)  # 兑换码
    batch_id = db.Column(db.String(32))  # 批次ID，用于批量管理
    target_group_id = db.Column(db.Integer, db.ForeignKey('user_groups.id'), nullable=False)  # 兑换后的用户组
    duration_days = db.Column(db.Integer, default=30)  # 有效期天数
    is_used = db.Column(db.Boolean, default=False)  # 是否已使用
    used_by = db.Column(db.Integer, db.ForeignKey('users.id'))  # 使用者
    used_at = db.Column(db.DateTime)  # 使用时间
    expires_at = db.Column(db.DateTime)  # 兑换码过期时间
    remark = db.Column(db.String(255))  # 备注
    created_at = db.Column(db.DateTime, default=datetime.utcnow)

    target_group = db.relationship('UserGroup', backref='redeem_codes')
    user = db.relationship('User', backref='redeemed_codes')


class UserGroupExpiry(db.Model):
    """用户组到期时间记录"""
    __tablename__ = 'user_group_expiry'

    id = db.Column(db.Integer, primary_key=True)
    user_id = db.Column(db.Integer, db.ForeignKey('users.id'), nullable=False, unique=True)
    group_id = db.Column(db.Integer, db.ForeignKey('user_groups.id'), nullable=False)
    expires_at = db.Column(db.DateTime, nullable=False)  # 到期时间
    created_at = db.Column(db.DateTime, default=datetime.utcnow)
    updated_at = db.Column(db.DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)

    user = db.relationship('User', backref='group_expiry')
    group = db.relationship('UserGroup')