"""add knowledge iteration tables: conversation_annotations and knowledge_suggestions

Revision ID: 043_knowledge_iteration
Revises: 042_add_employee_avatar_updated_at
Create Date: 2026-07-06

"""
from typing import Sequence, Union

from alembic import op
import sqlalchemy as sa


# revision identifiers, used by Alembic.
revision: str = '043_knowledge_iteration'
down_revision: Union[str, None] = '042_add_employee_avatar_updated_at'
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    # --------------------------------------------------------------------------
    # 1. 创建会话标注表 conversation_annotations
    # --------------------------------------------------------------------------
    op.create_table(
        'conversation_annotations',
        sa.Column('id', sa.String(36), primary_key=True),
        sa.Column('conversation_id', sa.String(36), nullable=False, index=True),
        sa.Column('agent_id', sa.String(36), nullable=False),
        sa.Column('message_id', sa.String(36), nullable=False),
        sa.Column('feedback', sa.String(20), nullable=False),
        sa.Column('comment', sa.Text(), nullable=True),
        sa.Column('created_at', sa.DateTime(timezone=True), nullable=False),
    )
    op.create_index('idx_annotation_conversation', 'conversation_annotations', ['conversation_id'])
    op.create_index('idx_annotation_message', 'conversation_annotations', ['message_id'])

    # --------------------------------------------------------------------------
    # 2. 创建知识库优化建议表 knowledge_suggestions
    # --------------------------------------------------------------------------
    op.create_table(
        'knowledge_suggestions',
        sa.Column('id', sa.String(36), primary_key=True),
        sa.Column('suggestion_type', sa.String(20), nullable=False, server_default='new_faq'),
        sa.Column('status', sa.String(20), nullable=False, server_default='pending', index=True),
        sa.Column('title', sa.String(256), nullable=False),
        sa.Column('content', sa.Text(), nullable=False),
        sa.Column('category', sa.String(64), nullable=False, server_default='其他'),
        sa.Column('tags', sa.JSON(), nullable=False, server_default='[]'),
        sa.Column('source_type', sa.String(30), nullable=False),
        sa.Column('source_data', sa.JSON(), nullable=True),
        sa.Column('reason', sa.Text(), nullable=True),
        sa.Column('reject_reason', sa.Text(), nullable=True),
        sa.Column('reviewer_id', sa.String(36), nullable=True),
        sa.Column('reviewed_at', sa.DateTime(timezone=True), nullable=True),
        sa.Column('created_at', sa.DateTime(timezone=True), nullable=False),
        sa.Column('updated_at', sa.DateTime(timezone=True), nullable=False),
    )
    op.create_index('idx_suggestion_status', 'knowledge_suggestions', ['status'])
    op.create_index('idx_suggestion_type', 'knowledge_suggestions', ['suggestion_type'])
    op.create_index('idx_suggestion_created', 'knowledge_suggestions', ['created_at'])


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