"""initial schema

Revision ID: 854325b9cbe6
Revises: 
Create Date: 2026-03-14 13:48:44.859635

"""
from typing import Sequence, Union

from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects import postgresql

# revision identifiers, used by Alembic.
revision: str = '854325b9cbe6'
down_revision: Union[str, Sequence[str], None] = None
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    """Upgrade schema."""
    # ### commands auto generated by Alembic - please adjust! ###
    op.create_table('proxy_sources',
    sa.Column('url', sa.String(length=2048), nullable=False),
    sa.Column('parser_name', sa.String(length=64), nullable=False),
    sa.Column('cron_schedule', sa.String(length=64), nullable=True),
    sa.Column('default_protocol', postgresql.ENUM('HTTP', 'HTTPS', 'SOCKS4', 'SOCKS5', name='proxy_protocol'), nullable=False),
    sa.Column('is_active', sa.Boolean(), nullable=False),
    sa.Column('last_scraped_at', sa.DateTime(), nullable=True),
    sa.Column('id', sa.Uuid(), nullable=False),
    sa.Column('created_at', sa.DateTime(), server_default=sa.text('now()'), nullable=False),
    sa.PrimaryKeyConstraint('id'),
    sa.UniqueConstraint('url')
    )
    op.create_table('users',
    sa.Column('email', sa.String(length=320), nullable=False),
    sa.Column('display_name', sa.String(length=128), nullable=True),
    sa.Column('is_active', sa.Boolean(), nullable=False),
    sa.Column('id', sa.Uuid(), nullable=False),
    sa.Column('created_at', sa.DateTime(), server_default=sa.text('now()'), nullable=False),
    sa.PrimaryKeyConstraint('id'),
    sa.UniqueConstraint('email')
    )
    op.create_table('api_keys',
    sa.Column('user_id', sa.Uuid(), nullable=False),
    sa.Column('key_hash', sa.String(length=128), nullable=False),
    sa.Column('prefix', sa.String(length=8), nullable=False),
    sa.Column('label', sa.String(length=128), nullable=True),
    sa.Column('is_active', sa.Boolean(), nullable=False),
    sa.Column('last_used_at', sa.DateTime(), nullable=True),
    sa.Column('expires_at', sa.DateTime(), nullable=True),
    sa.Column('id', sa.Uuid(), nullable=False),
    sa.Column('created_at', sa.DateTime(), server_default=sa.text('now()'), nullable=False),
    sa.ForeignKeyConstraint(['user_id'], ['users.id'], ondelete='CASCADE'),
    sa.PrimaryKeyConstraint('id')
    )
    op.create_index('ix_api_keys_hash', 'api_keys', ['key_hash'], unique=True)
    op.create_index('ix_api_keys_prefix', 'api_keys', ['prefix'], unique=False)
    op.create_table('credit_ledger',
    sa.Column('user_id', sa.Uuid(), nullable=False),
    sa.Column('amount', sa.Integer(), nullable=False),
    sa.Column('tx_type', postgresql.ENUM('PURCHASE', 'ACQUIRE', 'REFUND', 'ADMIN_ADJUST', name='credit_tx_type'), nullable=False),
    sa.Column('description', sa.Text(), nullable=True),
    sa.Column('reference_id', sa.Uuid(), nullable=True),
    sa.Column('id', sa.Uuid(), nullable=False),
    sa.Column('created_at', sa.DateTime(), server_default=sa.text('now()'), nullable=False),
    sa.ForeignKeyConstraint(['user_id'], ['users.id'], ondelete='CASCADE'),
    sa.PrimaryKeyConstraint('id')
    )
    op.create_index('ix_ledger_user_created', 'credit_ledger', ['user_id', 'created_at'], unique=False)
    op.create_table('proxies',
    sa.Column('ip', postgresql.INET(), nullable=False),
    sa.Column('port', sa.Integer(), nullable=False),
    sa.Column('protocol', postgresql.ENUM('HTTP', 'HTTPS', 'SOCKS4', 'SOCKS5', name='proxy_protocol'), nullable=False),
    sa.Column('source_id', sa.Uuid(), nullable=False),
    sa.Column('status', postgresql.ENUM('UNCHECKED', 'ACTIVE', 'DEAD', name='proxy_status'), nullable=False),
    sa.Column('anonymity', postgresql.ENUM('TRANSPARENT', 'ANONYMOUS', 'ELITE', name='anonymity_level'), nullable=True),
    sa.Column('exit_ip', postgresql.INET(), nullable=True),
    sa.Column('country', sa.String(length=2), nullable=True),
    sa.Column('score', sa.Float(), nullable=False),
    sa.Column('avg_latency_ms', sa.Float(), nullable=True),
    sa.Column('uptime_pct', sa.Float(), nullable=True),
    sa.Column('first_seen_at', sa.DateTime(), server_default=sa.text('now()'), nullable=False),
    sa.Column('last_checked_at', sa.DateTime(), nullable=True),
    sa.Column('id', sa.Uuid(), nullable=False),
    sa.Column('created_at', sa.DateTime(), server_default=sa.text('now()'), nullable=False),
    sa.ForeignKeyConstraint(['source_id'], ['proxy_sources.id'], ),
    sa.PrimaryKeyConstraint('id')
    )
    op.create_index('ix_proxies_ip_port_proto', 'proxies', ['ip', 'port', 'protocol'], unique=True)
    op.create_index('ix_proxies_status_score', 'proxies', ['status', 'score'], unique=False)
    op.create_table('proxy_checks',
    sa.Column('proxy_id', sa.Uuid(), nullable=False),
    sa.Column('checker_name', sa.String(length=64), nullable=False),
    sa.Column('stage', sa.Integer(), nullable=False),
    sa.Column('passed', sa.Boolean(), nullable=False),
    sa.Column('latency_ms', sa.Float(), nullable=True),
    sa.Column('detail', sa.Text(), nullable=True),
    sa.Column('exit_ip', postgresql.INET(), nullable=True),
    sa.Column('id', sa.Uuid(), nullable=False),
    sa.Column('created_at', sa.DateTime(), server_default=sa.text('now()'), nullable=False),
    sa.ForeignKeyConstraint(['proxy_id'], ['proxies.id'], ondelete='CASCADE'),
    sa.PrimaryKeyConstraint('id')
    )
    op.create_index('ix_checks_proxy_created', 'proxy_checks', ['proxy_id', 'created_at'], unique=False)
    op.create_table('proxy_leases',
    sa.Column('user_id', sa.Uuid(), nullable=False),
    sa.Column('proxy_id', sa.Uuid(), nullable=False),
    sa.Column('acquired_at', sa.DateTime(), server_default=sa.text('now()'), nullable=False),
    sa.Column('expires_at', sa.DateTime(), nullable=False),
    sa.Column('is_released', sa.Boolean(), nullable=False),
    sa.Column('id', sa.Uuid(), nullable=False),
    sa.ForeignKeyConstraint(['proxy_id'], ['proxies.id'], ),
    sa.ForeignKeyConstraint(['user_id'], ['users.id'], ),
    sa.PrimaryKeyConstraint('id')
    )
    op.create_index('ix_leases_proxy_active', 'proxy_leases', ['proxy_id', 'is_released'], unique=False)
    op.create_index('ix_leases_user', 'proxy_leases', ['user_id'], unique=False)
    op.create_table('proxy_tags',
    sa.Column('proxy_id', sa.Uuid(), nullable=False),
    sa.Column('key', sa.String(length=64), nullable=False),
    sa.Column('value', sa.String(length=256), nullable=False),
    sa.Column('id', sa.Uuid(), nullable=False),
    sa.ForeignKeyConstraint(['proxy_id'], ['proxies.id'], ondelete='CASCADE'),
    sa.PrimaryKeyConstraint('id')
    )
    op.create_index('ix_tags_proxy_key', 'proxy_tags', ['proxy_id', 'key'], unique=True)
    # ### end Alembic commands ###


def downgrade() -> None:
    """Downgrade schema."""
    # ### commands auto generated by Alembic - please adjust! ###
    op.drop_index('ix_tags_proxy_key', table_name='proxy_tags')
    op.drop_table('proxy_tags')
    op.drop_index('ix_leases_user', table_name='proxy_leases')
    op.drop_index('ix_leases_proxy_active', table_name='proxy_leases')
    op.drop_table('proxy_leases')
    op.drop_index('ix_checks_proxy_created', table_name='proxy_checks')
    op.drop_table('proxy_checks')
    op.drop_index('ix_proxies_status_score', table_name='proxies')
    op.drop_index('ix_proxies_ip_port_proto', table_name='proxies')
    op.drop_table('proxies')
    op.drop_index('ix_ledger_user_created', table_name='credit_ledger')
    op.drop_table('credit_ledger')
    op.drop_index('ix_api_keys_prefix', table_name='api_keys')
    op.drop_index('ix_api_keys_hash', table_name='api_keys')
    op.drop_table('api_keys')
    op.drop_table('users')
    op.drop_table('proxy_sources')
    # ### end Alembic commands ###