"""add notifications table, schedule_id to executions, webhook_url to agent_schedules, feishu_open_id to users

Revision ID: 009_add_notifications_and_schedule_fields
Revises: 008_add_agent_budget_config
Create Date: 2026-05-02
"""
from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects.mysql import CHAR


revision = "009_notif_sched_feishu"
down_revision = "008_add_agent_budget_config"
branch_labels = None
depends_on = None


def upgrade() -> None:
    # 1. 新增 notifications 表
    op.create_table(
        "notifications",
        sa.Column("id", CHAR(36), primary_key=True),
        sa.Column("user_id", CHAR(36), sa.ForeignKey("users.id"), nullable=False, index=True),
        sa.Column("title", sa.String(200), nullable=False),
        sa.Column("content", sa.Text(), nullable=True),
        sa.Column("category", sa.String(32), default="system"),
        sa.Column("ref_type", sa.String(32), nullable=True),
        sa.Column("ref_id", sa.String(36), nullable=True),
        sa.Column("is_read", sa.Boolean(), default=False),
        sa.Column("created_at", sa.DateTime(), default=sa.func.now()),
    )

    # 2. executions 表添加 schedule_id 字段
    op.add_column(
        "executions",
        sa.Column(
            "schedule_id",
            CHAR(36),
            sa.ForeignKey("agent_schedules.id"),
            nullable=True,
            comment="定时任务ID",
        ),
    )

    # 3. agent_schedules 表添加 webhook_url 字段
    op.add_column(
        "agent_schedules",
        sa.Column(
            "webhook_url",
            sa.String(512),
            nullable=True,
            comment="飞书机器人 Webhook URL（可选），执行完成后推送通知",
        ),
    )

    # 4. users 表添加 feishu_open_id 字段
    op.add_column(
        "users",
        sa.Column(
            "feishu_open_id",
            sa.String(64),
            nullable=True,
            comment="飞书用户 open_id，用于推送通知",
        ),
    )

    # 5. users 表添加 feishu_default_agent_id 字段
    op.add_column(
        "users",
        sa.Column(
            "feishu_default_agent_id",
            CHAR(36),
            nullable=True,
            comment="飞书对话默认 Agent ID",
        ),
    )


def downgrade() -> None:
    op.drop_column("users", "feishu_default_agent_id")
    op.drop_column("users", "feishu_open_id")
    op.drop_column("agent_schedules", "webhook_url")
    op.drop_column("executions", "schedule_id")
    op.drop_table("notifications")