"""迁移 021: 为保司和产品表新增脱敏展示名字段。"""
import logging
from sqlalchemy import text

logger = logging.getLogger(__name__)


def _column_exists(db, table: str, column: str) -> bool:
    """检查列是否存在。"""
    try:
        result = db.session.execute(text(
            "SELECT COUNT(*) FROM information_schema.COLUMNS "
            "WHERE TABLE_NAME = :table AND COLUMN_NAME = :column"
        ), {"table": table, "column": column})
        return result.scalar() > 0
    except Exception:
        return False


def migrate():
    """执行迁移。"""
    from insurance.db.compat import db

    if not _column_exists(db, "insurance_ppt_companies", "masked_display_name"):
        db.session.execute(text(
            "ALTER TABLE insurance_ppt_companies ADD COLUMN masked_display_name VARCHAR(100)"
        ))
        logger.info("[migrate_021] 已为 insurance_ppt_companies 添加 masked_display_name 列")
    else:
        logger.info("[migrate_021] insurance_ppt_companies.masked_display_name 已存在，跳过")

    if not _column_exists(db, "insurance_ppt_products", "masked_display_name"):
        db.session.execute(text(
            "ALTER TABLE insurance_ppt_products ADD COLUMN masked_display_name VARCHAR(100)"
        ))
        logger.info("[migrate_021] 已为 insurance_ppt_products 添加 masked_display_name 列")
    else:
        logger.info("[migrate_021] insurance_ppt_products.masked_display_name 已存在，跳过")

    if not _column_exists(db, "poster_records", "extra_data"):
        db.session.execute(text(
            "ALTER TABLE poster_records ADD COLUMN extra_data TEXT"
        ))
        logger.info("[migrate_021] 已为 poster_records 添加 extra_data 列")
    else:
        logger.info("[migrate_021] poster_records.extra_data 已存在，跳过")

    db.session.commit()