"""发票抬头表迁移

创建 invoice_headers（发票抬头表），用于用户管理常用发票抬头。

Revision ID: 20260811_20260811
Revises: 20260810_20260810
Create Date: 2026-08-11 00:00:00.000000
"""
from typing import Sequence, Union

from alembic import op
import sqlalchemy as sa


revision: str = '20260811_20260811'
down_revision: Union[str, None] = '20260810_20260810'
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def _comment_table(table_name: str, comment: str) -> None:
    op.execute(f"COMMENT ON TABLE {table_name} IS '{comment}'")


def _comment_column(table_name: str, column_name: str, comment: str) -> None:
    escaped = comment.replace("'", "''")
    op.execute(f"COMMENT ON COLUMN {table_name}.{column_name} IS '{escaped}'")


def upgrade() -> None:
    op.create_table(
        'invoice_headers',
        sa.Column('id', sa.String(32), primary_key=True),
        sa.Column('user_id', sa.String(32), sa.ForeignKey('users.id', ondelete='CASCADE'), nullable=False),
        sa.Column('type', sa.String(16), nullable=False),
        sa.Column('name', sa.String(128), nullable=False),
        sa.Column('tax_no', sa.String(32), nullable=True),
        sa.Column('register_address', sa.String(256), nullable=True),
        sa.Column('register_phone', sa.String(32), nullable=True),
        sa.Column('bank_name', sa.String(128), nullable=True),
        sa.Column('bank_account', sa.String(64), nullable=True),
        sa.Column('email', sa.String(128), nullable=True),
        sa.Column('is_default', sa.Boolean, nullable=False, server_default='false'),
        sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
        sa.Column('updated_at', sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
    )
    op.create_index('idx_invoice_headers_user', 'invoice_headers', ['user_id'])

    # 表注释和字段注释
    _comment_table('invoice_headers', '发票抬头表')
    _comment_column('invoice_headers', 'id', '主键')
    _comment_column('invoice_headers', 'user_id', '用户ID')
    _comment_column('invoice_headers', 'type', '抬头类型: personal/company')
    _comment_column('invoice_headers', 'name', '抬头名称')
    _comment_column('invoice_headers', 'tax_no', '税号')
    _comment_column('invoice_headers', 'register_address', '注册地址')
    _comment_column('invoice_headers', 'register_phone', '注册电话')
    _comment_column('invoice_headers', 'bank_name', '开户行')
    _comment_column('invoice_headers', 'bank_account', '银行账号')
    _comment_column('invoice_headers', 'email', '接收邮箱')
    _comment_column('invoice_headers', 'is_default', '是否默认')


def downgrade() -> None:
    op.drop_table('invoice_headers')