feat: 实现起卦、设置与积分系统

2026-04-03 16:56:47 +08:00
parent 31594558eb
commit f245eec5f6
170 changed files with 20728 additions and 328 deletions
@@ -0,0 +1,423 @@
+"""add profiles points ledger sessions messages schema
+
+Revision ID: 202604030002
+Revises: 202604020001
+Create Date: 2026-04-03 22:40:00
+"""
+
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+from sqlalchemy.dialects import postgresql
+
+revision: str = "202604030002"
+down_revision: Union[str, Sequence[str], None] = "202604020001"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    op.create_table(
+        "profiles",
+        sa.Column("id", sa.UUID(), nullable=False),
+        sa.Column("username", sa.String(length=30), nullable=False),
+        sa.Column("avatar_url", sa.Text(), nullable=True),
+        sa.Column("bio", sa.String(length=200), nullable=True),
+        sa.Column(
+            "settings",
+            postgresql.JSONB(astext_type=sa.Text()),
+            nullable=False,
+            server_default=sa.text("'{}'::jsonb"),
+        ),
+        sa.Column(
+            "created_at",
+            sa.DateTime(timezone=True),
+            server_default=sa.text("now()"),
+            nullable=False,
+        ),
+        sa.Column(
+            "updated_at",
+            sa.DateTime(timezone=True),
+            server_default=sa.text("now()"),
+            nullable=False,
+        ),
+        sa.Column("deleted_at", sa.DateTime(timezone=True), nullable=True),
+        sa.CheckConstraint(
+            "char_length(username) >= 1", name="ck_profiles_username_non_empty"
+        ),
+        sa.ForeignKeyConstraint(["id"], ["auth.users.id"], ondelete="CASCADE"),
+        sa.PrimaryKeyConstraint("id"),
+    )
+    op.create_index("ix_profiles_username", "profiles", ["username"], unique=False)
+    op.create_index(
+        "ix_profiles_settings_gin",
+        "profiles",
+        ["settings"],
+        unique=False,
+        postgresql_using="gin",
+    )
+    _enable_rls("profiles")
+
+    op.create_table(
+        "sessions",
+        sa.Column("id", sa.UUID(), nullable=False),
+        sa.Column("user_id", sa.UUID(), nullable=False),
+        sa.Column("session_type", sa.String(length=20), nullable=False),
+        sa.Column("job_id", sa.UUID(), nullable=True),
+        sa.Column("title", sa.String(length=255), nullable=True),
+        sa.Column("status", sa.String(length=20), nullable=False),
+        sa.Column(
+            "last_activity_at",
+            sa.DateTime(timezone=True),
+            server_default=sa.text("now()"),
+            nullable=False,
+        ),
+        sa.Column(
+            "message_count",
+            sa.Integer(),
+            server_default=sa.text("0"),
+            nullable=False,
+        ),
+        sa.Column(
+            "total_tokens",
+            sa.Integer(),
+            server_default=sa.text("0"),
+            nullable=False,
+        ),
+        sa.Column(
+            "total_cost",
+            sa.Numeric(12, 6),
+            server_default=sa.text("0"),
+            nullable=False,
+        ),
+        sa.Column(
+            "state_snapshot", postgresql.JSONB(astext_type=sa.Text()), nullable=True
+        ),
+        sa.Column(
+            "created_at",
+            sa.DateTime(timezone=True),
+            server_default=sa.text("now()"),
+            nullable=False,
+        ),
+        sa.Column(
+            "updated_at",
+            sa.DateTime(timezone=True),
+            server_default=sa.text("now()"),
+            nullable=False,
+        ),
+        sa.Column("deleted_at", sa.DateTime(timezone=True), nullable=True),
+        sa.CheckConstraint(
+            "session_type in ('chat', 'automation')",
+            name="ck_sessions_session_type",
+        ),
+        sa.CheckConstraint(
+            "status in ('pending', 'running', 'completed', 'failed')",
+            name="ck_sessions_status",
+        ),
+        sa.CheckConstraint(
+            "message_count >= 0", name="ck_sessions_message_count_non_negative"
+        ),
+        sa.CheckConstraint(
+            "total_tokens >= 0", name="ck_sessions_total_tokens_non_negative"
+        ),
+        sa.CheckConstraint(
+            "total_cost >= 0", name="ck_sessions_total_cost_non_negative"
+        ),
+        sa.ForeignKeyConstraint(["user_id"], ["auth.users.id"], ondelete="CASCADE"),
+        sa.PrimaryKeyConstraint("id"),
+    )
+    op.create_index("ix_sessions_user_id", "sessions", ["user_id"], unique=False)
+    op.create_index(
+        "ix_sessions_user_activity",
+        "sessions",
+        ["user_id", "last_activity_at"],
+        unique=False,
+    )
+    _enable_rls("sessions")
+
+    op.create_table(
+        "messages",
+        sa.Column("id", sa.UUID(), nullable=False),
+        sa.Column("session_id", sa.UUID(), nullable=False),
+        sa.Column("seq", sa.Integer(), nullable=False),
+        sa.Column("role", sa.String(length=20), nullable=False),
+        sa.Column("content", sa.Text(), nullable=False),
+        sa.Column("model_code", sa.String(length=50), nullable=True),
+        sa.Column("tool_name", sa.String(length=100), nullable=True),
+        sa.Column(
+            "input_tokens", sa.Integer(), server_default=sa.text("0"), nullable=False
+        ),
+        sa.Column(
+            "output_tokens", sa.Integer(), server_default=sa.text("0"), nullable=False
+        ),
+        sa.Column(
+            "cost", sa.Numeric(12, 6), server_default=sa.text("0"), nullable=False
+        ),
+        sa.Column("latency_ms", sa.Integer(), nullable=True),
+        sa.Column(
+            "visibility_mask",
+            sa.BigInteger(),
+            server_default=sa.text("0"),
+            nullable=False,
+        ),
+        sa.Column("metadata", postgresql.JSONB(astext_type=sa.Text()), nullable=True),
+        sa.Column(
+            "created_at",
+            sa.DateTime(timezone=True),
+            server_default=sa.text("now()"),
+            nullable=False,
+        ),
+        sa.Column(
+            "updated_at",
+            sa.DateTime(timezone=True),
+            server_default=sa.text("now()"),
+            nullable=False,
+        ),
+        sa.Column("deleted_at", sa.DateTime(timezone=True), nullable=True),
+        sa.CheckConstraint("seq > 0", name="ck_messages_seq_positive"),
+        sa.CheckConstraint(
+            "role in ('user', 'assistant', 'system', 'tool')",
+            name="ck_messages_role",
+        ),
+        sa.CheckConstraint(
+            "input_tokens >= 0", name="ck_messages_input_tokens_non_negative"
+        ),
+        sa.CheckConstraint(
+            "output_tokens >= 0", name="ck_messages_output_tokens_non_negative"
+        ),
+        sa.CheckConstraint("cost >= 0", name="ck_messages_cost_non_negative"),
+        sa.CheckConstraint(
+            "latency_ms is null or latency_ms >= 0",
+            name="ck_messages_latency_non_negative",
+        ),
+        sa.ForeignKeyConstraint(["session_id"], ["sessions.id"], ondelete="CASCADE"),
+        sa.PrimaryKeyConstraint("id"),
+        sa.UniqueConstraint("session_id", "seq", name="uq_messages_session_seq"),
+    )
+    op.create_index("ix_messages_session_id", "messages", ["session_id"], unique=False)
+    op.create_index(
+        "ix_messages_session_seq_visibility",
+        "messages",
+        ["session_id", "seq", "visibility_mask"],
+        unique=False,
+    )
+    _enable_rls("messages")
+
+    op.create_table(
+        "user_points",
+        sa.Column("user_id", sa.UUID(), nullable=False),
+        sa.Column(
+            "balance", sa.BigInteger(), server_default=sa.text("0"), nullable=False
+        ),
+        sa.Column(
+            "frozen_balance",
+            sa.BigInteger(),
+            server_default=sa.text("0"),
+            nullable=False,
+        ),
+        sa.Column(
+            "lifetime_earned",
+            sa.BigInteger(),
+            server_default=sa.text("0"),
+            nullable=False,
+        ),
+        sa.Column(
+            "lifetime_spent",
+            sa.BigInteger(),
+            server_default=sa.text("0"),
+            nullable=False,
+        ),
+        sa.Column("version", sa.Integer(), server_default=sa.text("0"), nullable=False),
+        sa.Column(
+            "created_at",
+            sa.DateTime(timezone=True),
+            server_default=sa.text("now()"),
+            nullable=False,
+        ),
+        sa.Column(
+            "updated_at",
+            sa.DateTime(timezone=True),
+            server_default=sa.text("now()"),
+            nullable=False,
+        ),
+        sa.CheckConstraint("balance >= 0", name="ck_user_points_balance_non_negative"),
+        sa.CheckConstraint(
+            "frozen_balance >= 0", name="ck_user_points_frozen_balance_non_negative"
+        ),
+        sa.CheckConstraint(
+            "lifetime_earned >= 0", name="ck_user_points_lifetime_earned_non_negative"
+        ),
+        sa.CheckConstraint(
+            "lifetime_spent >= 0", name="ck_user_points_lifetime_spent_non_negative"
+        ),
+        sa.CheckConstraint(
+            "frozen_balance <= balance", name="ck_user_points_frozen_le_balance"
+        ),
+        sa.ForeignKeyConstraint(["user_id"], ["auth.users.id"], ondelete="CASCADE"),
+        sa.PrimaryKeyConstraint("user_id"),
+    )
+    _enable_rls("user_points")
+
+    op.create_table(
+        "points_ledger",
+        sa.Column("id", sa.UUID(), nullable=False),
+        sa.Column("user_id", sa.UUID(), nullable=False),
+        sa.Column("direction", sa.SmallInteger(), nullable=False),
+        sa.Column("amount", sa.BigInteger(), nullable=False),
+        sa.Column("balance_after", sa.BigInteger(), nullable=False),
+        sa.Column("change_type", sa.String(length=16), nullable=False),
+        sa.Column("biz_type", sa.String(length=16), nullable=False),
+        sa.Column("biz_id", sa.UUID(), nullable=False),
+        sa.Column("event_id", sa.String(length=64), nullable=False),
+        sa.Column("operator_id", sa.UUID(), nullable=True),
+        sa.Column(
+            "metadata",
+            postgresql.JSONB(astext_type=sa.Text()),
+            nullable=False,
+            server_default=sa.text("'{}'::jsonb"),
+        ),
+        sa.Column(
+            "created_at",
+            sa.DateTime(timezone=True),
+            server_default=sa.text("now()"),
+            nullable=False,
+        ),
+        sa.Column(
+            "updated_at",
+            sa.DateTime(timezone=True),
+            server_default=sa.text("now()"),
+            nullable=False,
+        ),
+        sa.CheckConstraint("amount > 0", name="ck_points_ledger_amount_positive"),
+        sa.CheckConstraint(
+            "direction in (1, -1)", name="ck_points_ledger_direction_valid"
+        ),
+        sa.CheckConstraint(
+            "balance_after >= 0", name="ck_points_ledger_balance_after_non_negative"
+        ),
+        sa.CheckConstraint(
+            "change_type in ('register', 'consume', 'grant', 'adjust')",
+            name="ck_points_ledger_change_type",
+        ),
+        sa.CheckConstraint("biz_type = 'chat'", name="ck_points_ledger_biz_type"),
+        sa.CheckConstraint(
+            "((change_type in ('register', 'grant') and direction = 1) "
+            "or (change_type = 'consume' and direction = -1) "
+            "or (change_type = 'adjust' and direction in (1, -1)))",
+            name="ck_points_ledger_direction_by_change_type",
+        ),
+        sa.CheckConstraint(
+            "jsonb_typeof(metadata) = 'object'",
+            name="ck_points_ledger_metadata_object",
+        ),
+        sa.CheckConstraint(
+            "metadata->>'schema_version' = '1' and "
+            "metadata->>'reason_code' in ('REGISTER_WELCOME', 'CHAT_CONSUME', 'CHAT_GRANT', 'CHAT_ADJUST') and "
+            "metadata->>'operator_type' in ('user', 'system', 'admin') and "
+            "coalesce(metadata->>'run_id', '') <> '' and "
+            "(not (metadata ? 'ext') or jsonb_typeof(metadata->'ext') = 'object')",
+            name="ck_points_ledger_metadata_common",
+        ),
+        sa.CheckConstraint(
+            "(change_type <> 'register' or "
+            "(metadata->>'reason_code' = 'REGISTER_WELCOME' and not (metadata ? 'charge')))",
+            name="ck_points_ledger_metadata_register_shape",
+        ),
+        sa.CheckConstraint(
+            "(change_type <> 'consume' or "
+            "(metadata->>'reason_code' = 'CHAT_CONSUME' and "
+            "(metadata ? 'charge') and jsonb_typeof(metadata->'charge') = 'object' and "
+            "(metadata->'charge' ? 'message_id') and (metadata->'charge' ? 'message_seq') and "
+            "(metadata->'charge' ? 'model_code') and (metadata->'charge' ? 'input_tokens') and "
+            "(metadata->'charge' ? 'output_tokens') and (metadata->'charge' ? 'cost')))",
+            name="ck_points_ledger_metadata_consume_shape",
+        ),
+        sa.CheckConstraint(
+            "(change_type <> 'grant' or metadata->>'reason_code' = 'CHAT_GRANT')",
+            name="ck_points_ledger_metadata_grant_shape",
+        ),
+        sa.CheckConstraint(
+            "(change_type <> 'adjust' or "
+            "(metadata->>'reason_code' = 'CHAT_ADJUST' and "
+            "(metadata ? 'ext') and (metadata->'ext' ? 'ticket_id') and "
+            "coalesce(metadata #>> '{ext,ticket_id}', '') <> ''))",
+            name="ck_points_ledger_metadata_adjust_shape",
+        ),
+        sa.ForeignKeyConstraint(["biz_id"], ["sessions.id"], ondelete="RESTRICT"),
+        sa.ForeignKeyConstraint(
+            ["operator_id"], ["auth.users.id"], ondelete="SET NULL"
+        ),
+        sa.ForeignKeyConstraint(["user_id"], ["auth.users.id"], ondelete="CASCADE"),
+        sa.PrimaryKeyConstraint("id"),
+        sa.UniqueConstraint("user_id", "event_id", name="uq_points_ledger_user_event"),
+    )
+    op.create_index(
+        "ix_points_ledger_user_created_at",
+        "points_ledger",
+        ["user_id", sa.text("created_at DESC")],
+        unique=False,
+    )
+    op.create_index(
+        "ix_points_ledger_biz_type_biz_id",
+        "points_ledger",
+        ["biz_type", "biz_id"],
+        unique=False,
+    )
+    _enable_rls("points_ledger")
+
+
+def downgrade() -> None:
+    _drop_rls("points_ledger")
+    op.drop_index("ix_points_ledger_biz_type_biz_id", table_name="points_ledger")
+    op.drop_index("ix_points_ledger_user_created_at", table_name="points_ledger")
+    op.drop_table("points_ledger")
+
+    _drop_rls("user_points")
+    op.drop_table("user_points")
+
+    _drop_rls("messages")
+    op.drop_index("ix_messages_session_seq_visibility", table_name="messages")
+    op.drop_index("ix_messages_session_id", table_name="messages")
+    op.drop_table("messages")
+
+    _drop_rls("sessions")
+    op.drop_index("ix_sessions_user_activity", table_name="sessions")
+    op.drop_index("ix_sessions_user_id", table_name="sessions")
+    op.drop_table("sessions")
+
+    _drop_rls("profiles")
+    op.drop_index("ix_profiles_settings_gin", table_name="profiles")
+    op.drop_index("ix_profiles_username", table_name="profiles")
+    op.drop_table("profiles")
+
+
+def _enable_rls(table_name: str) -> None:
+    for role in ["anon", "authenticated"]:
+        for action in ["select", "insert", "update", "delete"]:
+            op.execute(
+                f"DROP POLICY IF EXISTS {role}_{action}_{table_name} ON {table_name}"
+            )
+    op.execute(f"ALTER TABLE {table_name} ENABLE ROW LEVEL SECURITY")
+    for role in ["anon", "authenticated"]:
+        op.execute(
+            f"CREATE POLICY {role}_select_{table_name} ON {table_name} FOR SELECT TO {role} USING (false)"
+        )
+        op.execute(
+            f"CREATE POLICY {role}_insert_{table_name} ON {table_name} FOR INSERT TO {role} WITH CHECK (false)"
+        )
+        op.execute(
+            f"CREATE POLICY {role}_update_{table_name} ON {table_name} FOR UPDATE TO {role} USING (false) WITH CHECK (false)"
+        )
+        op.execute(
+            f"CREATE POLICY {role}_delete_{table_name} ON {table_name} FOR DELETE TO {role} USING (false)"
+        )
+
+
+def _drop_rls(table_name: str) -> None:
+    for role in ["anon", "authenticated"]:
+        op.execute(f"DROP POLICY IF EXISTS {role}_delete_{table_name} ON {table_name}")
+        op.execute(f"DROP POLICY IF EXISTS {role}_update_{table_name} ON {table_name}")
+        op.execute(f"DROP POLICY IF EXISTS {role}_insert_{table_name} ON {table_name}")
+        op.execute(f"DROP POLICY IF EXISTS {role}_select_{table_name} ON {table_name}")
+    op.execute(f"ALTER TABLE {table_name} DISABLE ROW LEVEL SECURITY")
@@ -0,0 +1,163 @@
+"""allow register ledger without chat and add user init trigger
+
+Revision ID: 202604030003
+Revises: 202604030002
+Create Date: 2026-04-03 23:20:00
+"""
+
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+revision: str = "202604030003"
+down_revision: Union[str, Sequence[str], None] = "202604030002"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    op.alter_column(
+        "points_ledger", "biz_type", existing_type=sa.String(length=16), nullable=True
+    )
+    op.alter_column("points_ledger", "biz_id", existing_type=sa.UUID(), nullable=True)
+
+    op.drop_constraint("ck_points_ledger_biz_type", "points_ledger", type_="check")
+    op.create_check_constraint(
+        "ck_points_ledger_biz_type",
+        "points_ledger",
+        "biz_type is null or biz_type = 'chat'",
+    )
+    op.create_check_constraint(
+        "ck_points_ledger_biz_binding",
+        "points_ledger",
+        "((change_type = 'register' and biz_type is null and biz_id is null) or "
+        "(change_type in ('consume', 'grant', 'adjust') and biz_type = 'chat' and biz_id is not null))",
+    )
+
+    op.execute(
+        """
+        create or replace function public.initialize_profile_and_points_on_signup()
+        returns trigger
+        language plpgsql
+        security definer
+        set search_path = public
+        as $$
+        declare
+          v_username text;
+          v_ledger_id uuid;
+          v_event_id text;
+        begin
+          v_username := 'user_' || substring(md5(new.id::text || clock_timestamp()::text || random()::text) from 1 for 6);
+          v_ledger_id := md5(new.id::text || 'ledger' || clock_timestamp()::text || random()::text)::uuid;
+          v_event_id := 'register:' || new.id::text;
+
+          insert into public.profiles (id, username, avatar_url, bio, settings)
+          values (
+            new.id,
+            v_username,
+            null,
+            null,
+            jsonb_build_object(
+              'version', 1,
+              'preferences', jsonb_build_object(
+                'interface_language', 'zh-CN',
+                'ai_language', 'zh-CN',
+                'timezone', 'Asia/Shanghai',
+                'country', 'CN'
+              ),
+              'privacy', jsonb_build_object('profile_visibility', 'public'),
+              'notification', jsonb_build_object('push_enabled', true)
+            )
+          )
+          on conflict (id) do nothing;
+
+          insert into public.user_points (
+            user_id,
+            balance,
+            frozen_balance,
+            lifetime_earned,
+            lifetime_spent,
+            version
+          )
+          values (new.id, 100, 0, 100, 0, 0)
+          on conflict (user_id) do nothing;
+
+          insert into public.points_ledger (
+            id,
+            user_id,
+            direction,
+            amount,
+            balance_after,
+            change_type,
+            biz_type,
+            biz_id,
+            event_id,
+            operator_id,
+            metadata
+          )
+          values (
+            v_ledger_id,
+            new.id,
+            1,
+            100,
+            100,
+            'register',
+            null,
+            null,
+            v_event_id,
+            null,
+            jsonb_build_object(
+              'schema_version', 1,
+              'reason_code', 'REGISTER_WELCOME',
+              'operator_type', 'system',
+              'run_id', v_event_id,
+              'request_id', null,
+              'ext', jsonb_build_object('source', 'auth_signup')
+            )
+          )
+          on conflict (user_id, event_id) do nothing;
+
+          return new;
+        end;
+        $$;
+        """
+    )
+
+    op.execute(
+        "drop trigger if exists trg_initialize_profile_and_points_on_signup on auth.users"
+    )
+    op.execute(
+        """
+        create trigger trg_initialize_profile_and_points_on_signup
+        after insert on auth.users
+        for each row
+        execute function public.initialize_profile_and_points_on_signup();
+        """
+    )
+
+
+def downgrade() -> None:
+    op.execute(
+        "drop trigger if exists trg_initialize_profile_and_points_on_signup on auth.users"
+    )
+    op.execute(
+        "drop function if exists public.initialize_profile_and_points_on_signup()"
+    )
+
+    op.drop_constraint("ck_points_ledger_biz_binding", "points_ledger", type_="check")
+    op.drop_constraint("ck_points_ledger_biz_type", "points_ledger", type_="check")
+    op.create_check_constraint(
+        "ck_points_ledger_biz_type",
+        "points_ledger",
+        "biz_type = 'chat'",
+    )
+
+    op.execute(
+        "delete from points_ledger where change_type = 'register' and biz_id is null"
+    )
+
+    op.alter_column(
+        "points_ledger", "biz_type", existing_type=sa.String(length=16), nullable=False
+    )
+    op.alter_column("points_ledger", "biz_id", existing_type=sa.UUID(), nullable=False)
@@ -0,0 +1,316 @@
+"""remove redundant reason_code from points ledger metadata
+
+Revision ID: 202604030004
+Revises: 202604030003
+Create Date: 2026-04-03 23:35:00
+"""
+
+from typing import Sequence, Union
+
+from alembic import op
+
+revision: str = "202604030004"
+down_revision: Union[str, Sequence[str], None] = "202604030003"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    op.drop_constraint(
+        "ck_points_ledger_metadata_common", "points_ledger", type_="check"
+    )
+    op.drop_constraint(
+        "ck_points_ledger_metadata_register_shape", "points_ledger", type_="check"
+    )
+    op.drop_constraint(
+        "ck_points_ledger_metadata_consume_shape", "points_ledger", type_="check"
+    )
+    op.drop_constraint(
+        "ck_points_ledger_metadata_grant_shape", "points_ledger", type_="check"
+    )
+    op.drop_constraint(
+        "ck_points_ledger_metadata_adjust_shape", "points_ledger", type_="check"
+    )
+
+    op.create_check_constraint(
+        "ck_points_ledger_metadata_common",
+        "points_ledger",
+        "metadata->>'schema_version' = '1' and "
+        "metadata->>'operator_type' in ('user', 'system', 'admin') and "
+        "coalesce(metadata->>'run_id', '') <> '' and "
+        "(not (metadata ? 'ext') or jsonb_typeof(metadata->'ext') = 'object')",
+    )
+    op.create_check_constraint(
+        "ck_points_ledger_metadata_register_shape",
+        "points_ledger",
+        "(change_type <> 'register' or not (metadata ? 'charge'))",
+    )
+    op.create_check_constraint(
+        "ck_points_ledger_metadata_consume_shape",
+        "points_ledger",
+        "(change_type <> 'consume' or ("
+        "(metadata ? 'charge') and jsonb_typeof(metadata->'charge') = 'object' and "
+        "(metadata->'charge' ? 'message_id') and (metadata->'charge' ? 'message_seq') and "
+        "(metadata->'charge' ? 'model_code') and (metadata->'charge' ? 'input_tokens') and "
+        "(metadata->'charge' ? 'output_tokens') and (metadata->'charge' ? 'cost')))",
+    )
+    op.create_check_constraint(
+        "ck_points_ledger_metadata_adjust_shape",
+        "points_ledger",
+        "(change_type <> 'adjust' or ("
+        "(metadata ? 'ext') and (metadata->'ext' ? 'ticket_id') and "
+        "coalesce(metadata #>> '{ext,ticket_id}', '') <> ''))",
+    )
+
+    op.execute(
+        "update points_ledger set metadata = metadata - 'reason_code' where metadata ? 'reason_code'"
+    )
+
+    op.execute(
+        """
+        create or replace function public.initialize_profile_and_points_on_signup()
+        returns trigger
+        language plpgsql
+        security definer
+        set search_path = public
+        as $$
+        declare
+          v_username text;
+          v_ledger_id uuid;
+          v_event_id text;
+        begin
+          v_username := 'user_' || substring(md5(new.id::text || clock_timestamp()::text || random()::text) from 1 for 6);
+          v_ledger_id := md5(new.id::text || 'ledger' || clock_timestamp()::text || random()::text)::uuid;
+          v_event_id := 'register:' || new.id::text;
+
+          insert into public.profiles (id, username, avatar_url, bio, settings)
+          values (
+            new.id,
+            v_username,
+            null,
+            null,
+            jsonb_build_object(
+              'version', 1,
+              'preferences', jsonb_build_object(
+                'interface_language', 'zh-CN',
+                'ai_language', 'zh-CN',
+                'timezone', 'Asia/Shanghai',
+                'country', 'CN'
+              ),
+              'privacy', jsonb_build_object('profile_visibility', 'public'),
+              'notification', jsonb_build_object('push_enabled', true)
+            )
+          )
+          on conflict (id) do nothing;
+
+          insert into public.user_points (
+            user_id,
+            balance,
+            frozen_balance,
+            lifetime_earned,
+            lifetime_spent,
+            version
+          )
+          values (new.id, 100, 0, 100, 0, 0)
+          on conflict (user_id) do nothing;
+
+          insert into public.points_ledger (
+            id,
+            user_id,
+            direction,
+            amount,
+            balance_after,
+            change_type,
+            biz_type,
+            biz_id,
+            event_id,
+            operator_id,
+            metadata
+          )
+          values (
+            v_ledger_id,
+            new.id,
+            1,
+            100,
+            100,
+            'register',
+            null,
+            null,
+            v_event_id,
+            null,
+            jsonb_build_object(
+              'schema_version', 1,
+              'operator_type', 'system',
+              'run_id', v_event_id,
+              'request_id', null,
+              'ext', jsonb_build_object('source', 'auth_signup')
+            )
+          )
+          on conflict (user_id, event_id) do nothing;
+
+          return new;
+        end;
+        $$;
+        """
+    )
+
+
+def downgrade() -> None:
+    op.drop_constraint(
+        "ck_points_ledger_metadata_adjust_shape", "points_ledger", type_="check"
+    )
+    op.drop_constraint(
+        "ck_points_ledger_metadata_consume_shape", "points_ledger", type_="check"
+    )
+    op.drop_constraint(
+        "ck_points_ledger_metadata_register_shape", "points_ledger", type_="check"
+    )
+    op.drop_constraint(
+        "ck_points_ledger_metadata_common", "points_ledger", type_="check"
+    )
+
+    op.execute(
+        """
+        update points_ledger
+        set metadata = jsonb_set(
+          metadata,
+          '{reason_code}',
+          to_jsonb(
+            case change_type
+              when 'register' then 'REGISTER_WELCOME'
+              when 'consume' then 'CHAT_CONSUME'
+              when 'grant' then 'CHAT_GRANT'
+              when 'adjust' then 'CHAT_ADJUST'
+              else 'CHAT_ADJUST'
+            end
+          ),
+          true
+        )
+        where not (metadata ? 'reason_code');
+        """
+    )
+
+    op.create_check_constraint(
+        "ck_points_ledger_metadata_common",
+        "points_ledger",
+        "metadata->>'schema_version' = '1' and "
+        "metadata->>'reason_code' in ('REGISTER_WELCOME', 'CHAT_CONSUME', 'CHAT_GRANT', 'CHAT_ADJUST') and "
+        "metadata->>'operator_type' in ('user', 'system', 'admin') and "
+        "coalesce(metadata->>'run_id', '') <> '' and "
+        "(not (metadata ? 'ext') or jsonb_typeof(metadata->'ext') = 'object')",
+    )
+    op.create_check_constraint(
+        "ck_points_ledger_metadata_register_shape",
+        "points_ledger",
+        "(change_type <> 'register' or (metadata->>'reason_code' = 'REGISTER_WELCOME' and not (metadata ? 'charge')))",
+    )
+    op.create_check_constraint(
+        "ck_points_ledger_metadata_consume_shape",
+        "points_ledger",
+        "(change_type <> 'consume' or (metadata->>'reason_code' = 'CHAT_CONSUME' and "
+        "(metadata ? 'charge') and jsonb_typeof(metadata->'charge') = 'object' and "
+        "(metadata->'charge' ? 'message_id') and (metadata->'charge' ? 'message_seq') and "
+        "(metadata->'charge' ? 'model_code') and (metadata->'charge' ? 'input_tokens') and "
+        "(metadata->'charge' ? 'output_tokens') and (metadata->'charge' ? 'cost')))",
+    )
+    op.create_check_constraint(
+        "ck_points_ledger_metadata_grant_shape",
+        "points_ledger",
+        "(change_type <> 'grant' or metadata->>'reason_code' = 'CHAT_GRANT')",
+    )
+    op.create_check_constraint(
+        "ck_points_ledger_metadata_adjust_shape",
+        "points_ledger",
+        "(change_type <> 'adjust' or (metadata->>'reason_code' = 'CHAT_ADJUST' and "
+        "(metadata ? 'ext') and (metadata->'ext' ? 'ticket_id') and "
+        "coalesce(metadata #>> '{ext,ticket_id}', '') <> ''))",
+    )
+
+    op.execute(
+        """
+        create or replace function public.initialize_profile_and_points_on_signup()
+        returns trigger
+        language plpgsql
+        security definer
+        set search_path = public
+        as $$
+        declare
+          v_username text;
+          v_ledger_id uuid;
+          v_event_id text;
+        begin
+          v_username := 'user_' || substring(md5(new.id::text || clock_timestamp()::text || random()::text) from 1 for 6);
+          v_ledger_id := md5(new.id::text || 'ledger' || clock_timestamp()::text || random()::text)::uuid;
+          v_event_id := 'register:' || new.id::text;
+
+          insert into public.profiles (id, username, avatar_url, bio, settings)
+          values (
+            new.id,
+            v_username,
+            null,
+            null,
+            jsonb_build_object(
+              'version', 1,
+              'preferences', jsonb_build_object(
+                'interface_language', 'zh-CN',
+                'ai_language', 'zh-CN',
+                'timezone', 'Asia/Shanghai',
+                'country', 'CN'
+              ),
+              'privacy', jsonb_build_object('profile_visibility', 'public'),
+              'notification', jsonb_build_object('push_enabled', true)
+            )
+          )
+          on conflict (id) do nothing;
+
+          insert into public.user_points (
+            user_id,
+            balance,
+            frozen_balance,
+            lifetime_earned,
+            lifetime_spent,
+            version
+          )
+          values (new.id, 100, 0, 100, 0, 0)
+          on conflict (user_id) do nothing;
+
+          insert into public.points_ledger (
+            id,
+            user_id,
+            direction,
+            amount,
+            balance_after,
+            change_type,
+            biz_type,
+            biz_id,
+            event_id,
+            operator_id,
+            metadata
+          )
+          values (
+            v_ledger_id,
+            new.id,
+            1,
+            100,
+            100,
+            'register',
+            null,
+            null,
+            v_event_id,
+            null,
+            jsonb_build_object(
+              'schema_version', 1,
+              'reason_code', 'REGISTER_WELCOME',
+              'operator_type', 'system',
+              'run_id', v_event_id,
+              'request_id', null,
+              'ext', jsonb_build_object('source', 'auth_signup')
+            )
+          )
+          on conflict (user_id, event_id) do nothing;
+
+          return new;
+        end;
+        $$;
+        """
+    )