"""create_initial_schema

从 Base.metadata 创建所有表（新库部署用）。
对已有数据库：若已 stamp 0001_baseline，本迁移会跳过已存在的表（create_all checkfirst）。

Revision ID: 0002_schema
Revises: 0001_baseline
Create Date: 2026-03-18

"""

from typing import Sequence, Union

from alembic import op
from sqlalchemy import text

revision: str = "0002_schema"
down_revision: Union[str, None] = "0001_baseline"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    conn = op.get_bind()
    conn.execute(text("CREATE EXTENSION IF NOT EXISTS vector"))

    # 导入所有 model 以注册到 Base.metadata
    from app.core.db import Base
    from app.features.auth import models as _auth_models  # noqa: F401
    from app.features.conversation import models as _conv_models  # noqa: F401
    from app.features.memory import models as _memory_models  # noqa: F401
    from app.features.memoir import models as _memoir_models  # noqa: F401
    from app.features.payment import models as _payment_models  # noqa: F401
    from app.features.user import models as _user_models  # noqa: F401

    Base.metadata.create_all(bind=conn)


def downgrade() -> None:
    conn = op.get_bind()
    from app.core.db import Base
    from app.features.auth import models as _auth_models  # noqa: F401
    from app.features.conversation import models as _conv_models  # noqa: F401
    from app.features.memory import models as _memory_models  # noqa: F401
    from app.features.memoir import models as _memoir_models  # noqa: F401
    from app.features.payment import models as _payment_models  # noqa: F401
    from app.features.user import models as _user_models  # noqa: F401

    Base.metadata.drop_all(bind=conn)