"""journal search and rotation

Revision ID: 0003
Revises: 0002
"""
from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects import postgresql

revision = '0003'
down_revision = '0002'
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.create_table('app_settings',
    sa.Column('key', sa.String(length=50), nullable=False),
    sa.Column('value', postgresql.JSONB(astext_type=sa.Text()), nullable=False),
    sa.PrimaryKeyConstraint('key')
    )
    op.create_table('clear_attempts',
    sa.Column('id', sa.Integer(), nullable=False),
    sa.Column('user_id', sa.Integer(), nullable=False),
    sa.Column('ts', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
    sa.ForeignKeyConstraint(['user_id'], ['users.id'], ondelete='CASCADE'),
    sa.PrimaryKeyConstraint('id')
    )
    op.create_index(op.f('ix_clear_attempts_user_id'), 'clear_attempts', ['user_id'], unique=False)
    op.add_column('audit_log', sa.Column('uid', sa.UUID(), server_default=sa.text('gen_random_uuid()'), nullable=False))
    op.add_column('audit_log', sa.Column('message', sa.Text(), server_default='', nullable=False))
    op.create_index('ix_audit_log_entity_type_action', 'audit_log', ['entity_type', 'action'], unique=False)
    op.create_unique_constraint("audit_log_uid_key", "audit_log", ["uid"])

    # бэкфилл человекочитаемого сообщения для существующих записей (с согласованием рода глагола)
    nouns = {"organization": ("Организация", "f"), "vrf": ("VRF", "m"), "prefix": ("Префикс", "m"), "address": ("Адрес", "m"),
             "device": ("Устройство", "n"), "device_type": ("Тип устройства", "m"), "isp": ("Оператор", "m")}
    verbs = {"created": ("создан", "создана", "создано"), "updated": ("изменён", "изменена", "изменено"),
             "deleted": ("удалён", "удалена", "удалено"), "assigned": ("назначен", "назначена", "назначено")}
    for entity, (noun, gender) in nouns.items():
        for action, forms in verbs.items():
            verb = forms[{"m": 0, "f": 1, "n": 2}[gender]]
            op.get_bind().execute(sa.text(
                "UPDATE audit_log SET message = :noun || ' ' || entity_label || ' ' || :verb WHERE entity_type = :e AND action = :a"),
                {"noun": noun, "verb": verb, "e": entity, "a": action})
    op.execute("""INSERT INTO app_settings (key, value) VALUES ('journal', '{"retention_days": 90, "max_entries": 100000}')""")


def downgrade() -> None:
    op.drop_constraint("audit_log_uid_key", "audit_log", type_="unique")
    op.drop_index('ix_audit_log_entity_type_action', table_name='audit_log')
    op.drop_column('audit_log', 'message')
    op.drop_column('audit_log', 'uid')
    op.drop_index(op.f('ix_clear_attempts_user_id'), table_name='clear_attempts')
    op.drop_table('clear_attempts')
    op.drop_table('app_settings')