"""发票管理表迁移

创建 invoices（发票主表）和 invoice_orders（发票-订单关联表）。

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

from alembic import op
import sqlalchemy as sa


revision: str = '20260810_20260810'
down_revision: Union[str, None] = '2026080601'
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:
    # ============================================================
    # 1. 创建 invoices 表
    # ============================================================
    op.create_table(
        'invoices',
        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('invoice_no', sa.String(32), nullable=False, unique=True),
        sa.Column('header_type', sa.String(16), nullable=False),
        sa.Column('header_name', sa.String(128), nullable=False),
        sa.Column('header_tax_no', sa.String(32), nullable=True),
        sa.Column('header_register_address', sa.String(256), nullable=True),
        sa.Column('header_register_phone', sa.String(32), nullable=True),
        sa.Column('header_bank_name', sa.String(128), nullable=True),
        sa.Column('header_bank_account', sa.String(64), nullable=True),
        sa.Column('email', sa.String(128), nullable=False),
        sa.Column('total_amount', sa.Float, nullable=False, server_default='0'),
        sa.Column('total_credits', sa.Float, nullable=False, server_default='0'),
        sa.Column('status', sa.String(16), nullable=False, server_default='processing'),
        sa.Column('failure_reason', sa.Text, nullable=True),
        sa.Column('issued_at', sa.DateTime(timezone=True), nullable=True),
        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_invoices_user_created', 'invoices', ['user_id', 'created_at'])
    op.create_index('idx_invoices_status_created', 'invoices', ['status', 'created_at'])
    op.create_index('idx_invoices_invoice_no', 'invoices', ['invoice_no'], unique=True)

    # ============================================================
    # 2. 创建 invoice_orders 表
    # ============================================================
    op.create_table(
        'invoice_orders',
        sa.Column('id', sa.String(32), primary_key=True),
        sa.Column('invoice_id', sa.String(32), sa.ForeignKey('invoices.id', ondelete='CASCADE'), nullable=False),
        sa.Column('order_id', sa.String(32), sa.ForeignKey('payment_orders.id', ondelete='CASCADE'), nullable=False),
        sa.Column('order_no', sa.String(64), nullable=False),
        sa.Column('amount', sa.Float, nullable=False, server_default='0'),
        sa.Column('credits', sa.Float, nullable=False, server_default='0'),
        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_orders_invoice', 'invoice_orders', ['invoice_id'])
    op.create_index('idx_invoice_orders_order', 'invoice_orders', ['order_id'])
    op.create_unique_constraint('uq_invoice_orders', 'invoice_orders', ['invoice_id', 'order_id'])

    # ============================================================
    # 3. 表注释和字段注释
    # ============================================================
    _comment_table('invoices', '发票主表')
    _comment_column('invoices', 'id', '主键')
    _comment_column('invoices', 'user_id', '申请用户ID')
    _comment_column('invoices', 'invoice_no', '发票编号')
    _comment_column('invoices', 'header_type', '抬头类型: personal/company')
    _comment_column('invoices', 'header_name', '抬头名称')
    _comment_column('invoices', 'header_tax_no', '税号')
    _comment_column('invoices', 'header_register_address', '注册地址')
    _comment_column('invoices', 'header_register_phone', '注册电话')
    _comment_column('invoices', 'header_bank_name', '开户行')
    _comment_column('invoices', 'header_bank_account', '银行账号')
    _comment_column('invoices', 'email', '电子邮箱（必填）')
    _comment_column('invoices', 'total_amount', '开票总金额')
    _comment_column('invoices', 'total_credits', '总积分')
    _comment_column('invoices', 'status', '状态: processing/success/failed')
    _comment_column('invoices', 'failure_reason', '失败原因')
    _comment_column('invoices', 'issued_at', '开票成功时间')

    _comment_table('invoice_orders', '发票-订单关联表')
    _comment_column('invoice_orders', 'id', '主键')
    _comment_column('invoice_orders', 'invoice_id', '发票ID')
    _comment_column('invoice_orders', 'order_id', '订单ID')
    _comment_column('invoice_orders', 'order_no', '订单号（冗余）')
    _comment_column('invoice_orders', 'amount', '订单金额（冗余）')
    _comment_column('invoice_orders', 'credits', '订单积分（冗余）')


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