"""add lorebook tables

Revision ID: 004
Revises: 003
Create Date: 2026-07-24 12:20:38.453954

"""
from collections.abc import Sequence

import sqlalchemy as sa

from alembic import op

# revision identifiers, used by Alembic.
revision: str = '004'
down_revision: str | Sequence[str] | None = '003'
branch_labels: str | Sequence[str] | None = None
depends_on: str | Sequence[str] | None = None


def upgrade() -> None:
    """Upgrade schema."""
    # ### commands auto generated by Alembic - please adjust! ###
    op.create_table('lorebooks',
    sa.Column('id', sa.String(length=80), nullable=False),
    sa.Column('character_id', sa.String(length=80), nullable=True),
    sa.Column('name', sa.String(length=200), nullable=False),
    sa.Column('description', sa.Text(), nullable=False),
    sa.Column('scan_depth', sa.Integer(), nullable=True),
    sa.Column('token_budget', sa.Integer(), nullable=True),
    sa.Column('recursive_scanning', sa.Boolean(), nullable=False),
    sa.Column('scope', sa.String(length=40), nullable=False),
    sa.Column('status', sa.String(length=40), nullable=False),
    sa.Column('owner_scope', sa.String(length=80), nullable=False),
    sa.Column('created_at', sa.DateTime(timezone=True), nullable=False),
    sa.Column('updated_at', sa.DateTime(timezone=True), nullable=False),
    sa.ForeignKeyConstraint(['character_id'], ['characters.id'], ),
    sa.PrimaryKeyConstraint('id')
    )
    op.create_index(op.f('ix_lorebooks_character_id'), 'lorebooks', ['character_id'], unique=False)
    op.create_index(op.f('ix_lorebooks_owner_scope'), 'lorebooks', ['owner_scope'], unique=False)
    op.create_index(op.f('ix_lorebooks_scope'), 'lorebooks', ['scope'], unique=False)
    op.create_index(op.f('ix_lorebooks_status'), 'lorebooks', ['status'], unique=False)
    op.create_table('lorebook_entries',
    sa.Column('id', sa.String(length=80), nullable=False),
    sa.Column('lorebook_id', sa.String(length=80), nullable=False),
    sa.Column('keys_json', sa.Text(), nullable=False),
    sa.Column('content', sa.Text(), nullable=False),
    sa.Column('secondary_keys_json', sa.Text(), nullable=False),
    sa.Column('constant', sa.Boolean(), nullable=False),
    sa.Column('selective', sa.Boolean(), nullable=False),
    sa.Column('enabled', sa.Boolean(), nullable=False),
    sa.Column('case_sensitive', sa.Boolean(), nullable=False),
    sa.Column('use_regex', sa.Boolean(), nullable=False),
    sa.Column('position', sa.String(length=40), nullable=False),
    sa.Column('insertion_order', sa.Integer(), nullable=False),
    sa.Column('priority', sa.Integer(), nullable=False),
    sa.Column('comment', sa.String(length=500), nullable=False),
    sa.Column('bound_entity_id', sa.String(length=80), nullable=True),
    sa.Column('created_at', sa.DateTime(timezone=True), nullable=False),
    sa.Column('updated_at', sa.DateTime(timezone=True), nullable=False),
    sa.ForeignKeyConstraint(['lorebook_id'], ['lorebooks.id'], ),
    sa.PrimaryKeyConstraint('id')
    )
    op.create_index(op.f('ix_lorebook_entries_bound_entity_id'), 'lorebook_entries', ['bound_entity_id'], unique=False)
    op.create_index(op.f('ix_lorebook_entries_lorebook_id'), 'lorebook_entries', ['lorebook_id'], unique=False)
    # ### end Alembic commands ###


def downgrade() -> None:
    """Downgrade schema."""
    # ### commands auto generated by Alembic - please adjust! ###
    op.drop_index(op.f('ix_lorebook_entries_lorebook_id'), table_name='lorebook_entries')
    op.drop_index(op.f('ix_lorebook_entries_bound_entity_id'), table_name='lorebook_entries')
    op.drop_table('lorebook_entries')
    op.drop_index(op.f('ix_lorebooks_status'), table_name='lorebooks')
    op.drop_index(op.f('ix_lorebooks_scope'), table_name='lorebooks')
    op.drop_index(op.f('ix_lorebooks_owner_scope'), table_name='lorebooks')
    op.drop_index(op.f('ix_lorebooks_character_id'), table_name='lorebooks')
    op.drop_table('lorebooks')
    # ### end Alembic commands ###