"""Initial database tables

Revision ID: 001
Revises: 
Create Date: 2025-01-02 16:51:00.000000

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

# revision identifiers, used by Alembic.
revision = '001'
down_revision = None
branch_labels = None
depends_on = None


def upgrade() -> None:
    """Create initial database tables."""
    
    # Create users table
    op.create_table(
        'users',
        sa.Column('id', postgresql.UUID(as_uuid=True), primary_key=True),
        sa.Column('username', sa.String(50), nullable=False, unique=True),
        sa.Column('email', sa.String(255), nullable=False, unique=True),
        sa.Column('password_hash', sa.String(255), nullable=False),
        sa.Column('first_name', sa.String(100)),
        sa.Column('last_name', sa.String(100)),
        sa.Column('is_active', sa.Boolean(), default=True, nullable=False),
        sa.Column('is_verified', sa.Boolean(), default=False, nullable=False),
        sa.Column('is_superuser', sa.Boolean(), default=False, nullable=False),
        sa.Column('avatar_url', sa.String(500)),
        sa.Column('bio', sa.Text()),
        sa.Column('last_login_at', sa.DateTime(timezone=True)),
        sa.Column('login_count', sa.Integer(), default=0),
        sa.Column('settings', postgresql.JSONB()),
        sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
        sa.Column('updated_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
    )
    
    # Create indexes for users
    op.create_index('ix_users_username', 'users', ['username'])
    op.create_index('ix_users_email', 'users', ['email'])
    op.create_index('ix_users_created_at', 'users', ['created_at'])
    op.create_index('ix_users_is_active', 'users', ['is_active'])
    
    # Create API keys table
    op.create_table(
        'api_keys',
        sa.Column('id', postgresql.UUID(as_uuid=True), primary_key=True),
        sa.Column('user_id', postgresql.UUID(as_uuid=True), sa.ForeignKey('users.id', ondelete='CASCADE'), nullable=False),
        sa.Column('name', sa.String(100), nullable=False),
        sa.Column('key_hash', sa.String(255), nullable=False, unique=True),
        sa.Column('key_prefix', sa.String(20), nullable=False),
        sa.Column('permissions', postgresql.JSONB(), default={}),
        sa.Column('is_active', sa.Boolean(), default=True, nullable=False),
        sa.Column('expires_at', sa.DateTime(timezone=True)),
        sa.Column('last_used_at', sa.DateTime(timezone=True)),
        sa.Column('usage_count', sa.Integer(), default=0),
        sa.Column('rate_limit', sa.Integer(), default=1000),
        sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
        sa.Column('updated_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
    )
    
    # Create indexes for API keys
    op.create_index('ix_api_keys_user_id', 'api_keys', ['user_id'])
    op.create_index('ix_api_keys_key_hash', 'api_keys', ['key_hash'])
    op.create_index('ix_api_keys_is_active', 'api_keys', ['is_active'])
    
    # Create user sessions table
    op.create_table(
        'user_sessions',
        sa.Column('id', postgresql.UUID(as_uuid=True), primary_key=True),
        sa.Column('user_id', postgresql.UUID(as_uuid=True), sa.ForeignKey('users.id', ondelete='CASCADE'), nullable=False),
        sa.Column('session_token', sa.String(255), nullable=False, unique=True),
        sa.Column('refresh_token', sa.String(255), nullable=False, unique=True),
        sa.Column('user_agent', sa.String(500)),
        sa.Column('ip_address', sa.String(45)),
        sa.Column('is_active', sa.Boolean(), default=True, nullable=False),
        sa.Column('expires_at', sa.DateTime(timezone=True), nullable=False),
        sa.Column('last_activity_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
        sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
    )
    
    # Create indexes for user sessions
    op.create_index('ix_user_sessions_user_id', 'user_sessions', ['user_id'])
    op.create_index('ix_user_sessions_session_token', 'user_sessions', ['session_token'])
    op.create_index('ix_user_sessions_is_active', 'user_sessions', ['is_active'])
    op.create_index('ix_user_sessions_expires_at', 'user_sessions', ['expires_at'])
    
    # Create content table
    op.create_table(
        'content',
        sa.Column('id', postgresql.UUID(as_uuid=True), primary_key=True),
        sa.Column('user_id', postgresql.UUID(as_uuid=True), sa.ForeignKey('users.id', ondelete='CASCADE'), nullable=False),
        sa.Column('title', sa.String(255), nullable=False),
        sa.Column('description', sa.Text()),
        sa.Column('content_type', sa.String(50), nullable=False),
        sa.Column('file_path', sa.String(500)),
        sa.Column('file_size', sa.BigInteger()),
        sa.Column('file_hash', sa.String(64)),
        sa.Column('mime_type', sa.String(100)),
        sa.Column('is_public', sa.Boolean(), default=False, nullable=False),
        sa.Column('is_featured', sa.Boolean(), default=False, nullable=False),
        sa.Column('view_count', sa.Integer(), default=0),
        sa.Column('download_count', sa.Integer(), default=0),
        sa.Column('like_count', sa.Integer(), default=0),
        sa.Column('metadata', postgresql.JSONB()),
        sa.Column('tags', postgresql.ARRAY(sa.String(50))),
        sa.Column('thumbnail_url', sa.String(500)),
        sa.Column('preview_url', sa.String(500)),
        sa.Column('status', sa.String(20), default='draft', nullable=False),
        sa.Column('published_at', sa.DateTime(timezone=True)),
        sa.Column('expires_at', sa.DateTime(timezone=True)),
        sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
        sa.Column('updated_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
    )
    
    # Create indexes for content
    op.create_index('ix_content_user_id', 'content', ['user_id'])
    op.create_index('ix_content_content_type', 'content', ['content_type'])
    op.create_index('ix_content_is_public', 'content', ['is_public'])
    op.create_index('ix_content_status', 'content', ['status'])
    op.create_index('ix_content_created_at', 'content', ['created_at'])
    op.create_index('ix_content_published_at', 'content', ['published_at'])
    op.create_index('ix_content_file_hash', 'content', ['file_hash'])
    op.create_index('ix_content_tags', 'content', ['tags'], postgresql_using='gin')
    
    # Create content versions table
    op.create_table(
        'content_versions',
        sa.Column('id', postgresql.UUID(as_uuid=True), primary_key=True),
        sa.Column('content_id', postgresql.UUID(as_uuid=True), sa.ForeignKey('content.id', ondelete='CASCADE'), nullable=False),
        sa.Column('version_number', sa.Integer(), nullable=False),
        sa.Column('title', sa.String(255), nullable=False),
        sa.Column('description', sa.Text()),
        sa.Column('file_path', sa.String(500)),
        sa.Column('file_size', sa.BigInteger()),
        sa.Column('file_hash', sa.String(64)),
        sa.Column('metadata', postgresql.JSONB()),
        sa.Column('change_summary', sa.Text()),
        sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
    )
    
    # Create indexes for content versions
    op.create_index('ix_content_versions_content_id', 'content_versions', ['content_id'])
    op.create_index('ix_content_versions_version_number', 'content_versions', ['version_number'])
    op.create_index('ix_content_versions_created_at', 'content_versions', ['created_at'])
    
    # Create file uploads table
    op.create_table(
        'file_uploads',
        sa.Column('id', postgresql.UUID(as_uuid=True), primary_key=True),
        sa.Column('user_id', postgresql.UUID(as_uuid=True), sa.ForeignKey('users.id', ondelete='CASCADE'), nullable=False),
        sa.Column('filename', sa.String(255), nullable=False),
        sa.Column('original_filename', sa.String(255), nullable=False),
        sa.Column('file_path', sa.String(500)),
        sa.Column('file_size', sa.BigInteger(), nullable=False),
        sa.Column('file_hash', sa.String(64)),
        sa.Column('mime_type', sa.String(100)),
        sa.Column('chunk_size', sa.Integer()),
        sa.Column('total_chunks', sa.Integer()),
        sa.Column('uploaded_chunks', sa.Integer(), default=0),
        sa.Column('upload_session_id', sa.String(100)),
        sa.Column('status', sa.String(20), default='pending', nullable=False),
        sa.Column('processed', sa.Boolean(), default=False, nullable=False),
        sa.Column('processing_started_at', sa.DateTime(timezone=True)),
        sa.Column('processing_completed_at', sa.DateTime(timezone=True)),
        sa.Column('error_message', sa.Text()),
        sa.Column('retry_count', sa.Integer(), default=0),
        sa.Column('metadata', postgresql.JSONB()),
        sa.Column('expires_at', sa.DateTime(timezone=True)),
        sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
        sa.Column('updated_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
    )
    
    # Create indexes for file uploads
    op.create_index('ix_file_uploads_user_id', 'file_uploads', ['user_id'])
    op.create_index('ix_file_uploads_status', 'file_uploads', ['status'])
    op.create_index('ix_file_uploads_processed', 'file_uploads', ['processed'])
    op.create_index('ix_file_uploads_upload_session_id', 'file_uploads', ['upload_session_id'])
    op.create_index('ix_file_uploads_file_hash', 'file_uploads', ['file_hash'])
    op.create_index('ix_file_uploads_expires_at', 'file_uploads', ['expires_at'])
    
    # Create user subscriptions table
    op.create_table(
        'user_subscriptions',
        sa.Column('id', postgresql.UUID(as_uuid=True), primary_key=True),
        sa.Column('user_id', postgresql.UUID(as_uuid=True), sa.ForeignKey('users.id', ondelete='CASCADE'), nullable=False),
        sa.Column('plan_name', sa.String(50), nullable=False),
        sa.Column('status', sa.String(20), default='active', nullable=False),
        sa.Column('storage_limit', sa.BigInteger(), nullable=False),
        sa.Column('bandwidth_limit', sa.BigInteger(), nullable=False),
        sa.Column('file_count_limit', sa.Integer(), nullable=False),
        sa.Column('features', postgresql.JSONB()),
        sa.Column('price', sa.Numeric(10, 2)),
        sa.Column('currency', sa.String(3), default='USD'),
        sa.Column('billing_cycle', sa.String(20), default='monthly'),
        sa.Column('starts_at', sa.DateTime(timezone=True), nullable=False),
        sa.Column('expires_at', sa.DateTime(timezone=True)),
        sa.Column('auto_renew', sa.Boolean(), default=True, nullable=False),
        sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
        sa.Column('updated_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
    )
    
    # Create indexes for user subscriptions
    op.create_index('ix_user_subscriptions_user_id', 'user_subscriptions', ['user_id'])
    op.create_index('ix_user_subscriptions_status', 'user_subscriptions', ['status'])
    op.create_index('ix_user_subscriptions_expires_at', 'user_subscriptions', ['expires_at'])
    
    # Create wallets table
    op.create_table(
        'wallets',
        sa.Column('id', postgresql.UUID(as_uuid=True), primary_key=True),
        sa.Column('user_id', postgresql.UUID(as_uuid=True), sa.ForeignKey('users.id', ondelete='CASCADE'), nullable=False),
        sa.Column('address', sa.String(100), nullable=False, unique=True),
        sa.Column('network', sa.String(20), default='mainnet', nullable=False),
        sa.Column('wallet_type', sa.String(20), default='ton', nullable=False),
        sa.Column('balance', sa.Numeric(20, 8), default=0),
        sa.Column('public_key', sa.String(200)),
        sa.Column('encrypted_private_key', sa.Text()),
        sa.Column('derivation_path', sa.String(100)),
        sa.Column('is_active', sa.Boolean(), default=True, nullable=False),
        sa.Column('is_primary', sa.Boolean(), default=False, nullable=False),
        sa.Column('last_sync_at', sa.DateTime(timezone=True)),
        sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
        sa.Column('updated_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
    )
    
    # Create indexes for wallets
    op.create_index('ix_wallets_user_id', 'wallets', ['user_id'])
    op.create_index('ix_wallets_address', 'wallets', ['address'])
    op.create_index('ix_wallets_network', 'wallets', ['network'])
    op.create_index('ix_wallets_is_active', 'wallets', ['is_active'])
    
    # Create transactions table
    op.create_table(
        'transactions',
        sa.Column('id', postgresql.UUID(as_uuid=True), primary_key=True),
        sa.Column('wallet_id', postgresql.UUID(as_uuid=True), sa.ForeignKey('wallets.id', ondelete='CASCADE'), nullable=False),
        sa.Column('tx_hash', sa.String(100), unique=True),
        sa.Column('from_address', sa.String(100), nullable=False),
        sa.Column('to_address', sa.String(100), nullable=False),
        sa.Column('amount', sa.Numeric(20, 8), nullable=False),
        sa.Column('fee', sa.Numeric(20, 8)),
        sa.Column('gas_limit', sa.BigInteger()),
        sa.Column('gas_used', sa.BigInteger()),
        sa.Column('gas_price', sa.Numeric(20, 8)),
        sa.Column('nonce', sa.BigInteger()),
        sa.Column('block_number', sa.BigInteger()),
        sa.Column('block_hash', sa.String(100)),
        sa.Column('transaction_index', sa.Integer()),
        sa.Column('status', sa.String(20), default='pending', nullable=False),
        sa.Column('transaction_type', sa.String(20), default='transfer', nullable=False),
        sa.Column('confirmations', sa.Integer(), default=0),
        sa.Column('metadata', postgresql.JSONB()),
        sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
        sa.Column('updated_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
    )
    
    # Create indexes for transactions
    op.create_index('ix_transactions_wallet_id', 'transactions', ['wallet_id'])
    op.create_index('ix_transactions_tx_hash', 'transactions', ['tx_hash'])
    op.create_index('ix_transactions_from_address', 'transactions', ['from_address'])
    op.create_index('ix_transactions_to_address', 'transactions', ['to_address'])
    op.create_index('ix_transactions_status', 'transactions', ['status'])
    op.create_index('ix_transactions_created_at', 'transactions', ['created_at'])
    op.create_index('ix_transactions_block_number', 'transactions', ['block_number'])
    
    # Create blockchain NFTs table
    op.create_table(
        'blockchain_nfts',
        sa.Column('id', postgresql.UUID(as_uuid=True), primary_key=True),
        sa.Column('wallet_id', postgresql.UUID(as_uuid=True), sa.ForeignKey('wallets.id', ondelete='CASCADE'), nullable=False),
        sa.Column('token_id', sa.String(100), nullable=False),
        sa.Column('collection_address', sa.String(100), nullable=False),
        sa.Column('owner_address', sa.String(100), nullable=False),
        sa.Column('token_uri', sa.String(500)),
        sa.Column('metadata', postgresql.JSONB()),
        sa.Column('name', sa.String(255)),
        sa.Column('description', sa.Text()),
        sa.Column('image_url', sa.String(500)),
        sa.Column('attributes', postgresql.JSONB()),
        sa.Column('rarity_score', sa.Numeric(10, 4)),
        sa.Column('last_price', sa.Numeric(20, 8)),
        sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
        sa.Column('updated_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
    )
    
    # Create unique constraint for NFTs
    op.create_unique_constraint('uq_blockchain_nfts_token_collection', 'blockchain_nfts', ['token_id', 'collection_address'])
    
    # Create indexes for blockchain NFTs
    op.create_index('ix_blockchain_nfts_wallet_id', 'blockchain_nfts', ['wallet_id'])
    op.create_index('ix_blockchain_nfts_collection_address', 'blockchain_nfts', ['collection_address'])
    op.create_index('ix_blockchain_nfts_owner_address', 'blockchain_nfts', ['owner_address'])
    
    # Create blockchain token balances table
    op.create_table(
        'blockchain_token_balances',
        sa.Column('id', postgresql.UUID(as_uuid=True), primary_key=True),
        sa.Column('wallet_id', postgresql.UUID(as_uuid=True), sa.ForeignKey('wallets.id', ondelete='CASCADE'), nullable=False),
        sa.Column('token_address', sa.String(100), nullable=False),
        sa.Column('token_name', sa.String(100)),
        sa.Column('token_symbol', sa.String(20)),
        sa.Column('balance', sa.Numeric(30, 18), default=0, nullable=False),
        sa.Column('decimals', sa.Integer(), default=18),
        sa.Column('usd_value', sa.Numeric(20, 8)),
        sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
        sa.Column('updated_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
    )
    
    # Create unique constraint for token balances
    op.create_unique_constraint('uq_token_balances_wallet_token', 'blockchain_token_balances', ['wallet_id', 'token_address'])
    
    # Create indexes for token balances
    op.create_index('ix_blockchain_token_balances_wallet_id', 'blockchain_token_balances', ['wallet_id'])
    op.create_index('ix_blockchain_token_balances_token_address', 'blockchain_token_balances', ['token_address'])
    
    # Create blockchain DeFi positions table
    op.create_table(
        'blockchain_defi_positions',
        sa.Column('id', postgresql.UUID(as_uuid=True), primary_key=True),
        sa.Column('wallet_id', postgresql.UUID(as_uuid=True), sa.ForeignKey('wallets.id', ondelete='CASCADE'), nullable=False),
        sa.Column('protocol_name', sa.String(100), nullable=False),
        sa.Column('position_type', sa.String(50), nullable=False),
        sa.Column('pool_address', sa.String(100)),
        sa.Column('token_symbols', postgresql.ARRAY(sa.String(20))),
        sa.Column('balance', sa.Numeric(30, 18), default=0),
        sa.Column('usd_value', sa.Numeric(20, 8)),
        sa.Column('yield_rate', sa.Numeric(10, 4)),
        sa.Column('metadata', postgresql.JSONB()),
        sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
        sa.Column('updated_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
    )
    
    # Create indexes for DeFi positions
    op.create_index('ix_blockchain_defi_positions_wallet_id', 'blockchain_defi_positions', ['wallet_id'])
    op.create_index('ix_blockchain_defi_positions_protocol_name', 'blockchain_defi_positions', ['protocol_name'])
    
    # Create blockchain staking table
    op.create_table(
        'blockchain_staking',
        sa.Column('id', postgresql.UUID(as_uuid=True), primary_key=True),
        sa.Column('wallet_id', postgresql.UUID(as_uuid=True), sa.ForeignKey('wallets.id', ondelete='CASCADE'), nullable=False),
        sa.Column('validator_address', sa.String(100), nullable=False),
        sa.Column('staked_amount', sa.Numeric(20, 8), nullable=False),
        sa.Column('rewards_earned', sa.Numeric(20, 8), default=0),
        sa.Column('status', sa.String(20), default='active', nullable=False),
        sa.Column('delegation_time', sa.DateTime(timezone=True), nullable=False),
        sa.Column('unlock_time', sa.DateTime(timezone=True)),
        sa.Column('apy', sa.Numeric(10, 4)),
        sa.Column('metadata', postgresql.JSONB()),
        sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
        sa.Column('updated_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
    )
    
    # Create indexes for staking
    op.create_index('ix_blockchain_staking_wallet_id', 'blockchain_staking', ['wallet_id'])
    op.create_index('ix_blockchain_staking_validator_address', 'blockchain_staking', ['validator_address'])
    op.create_index('ix_blockchain_staking_status', 'blockchain_staking', ['status'])


def downgrade() -> None:
    """Drop all database tables."""
    
    # Drop tables in reverse order to avoid foreign key constraints
    op.drop_table('blockchain_staking')
    op.drop_table('blockchain_defi_positions')
    op.drop_table('blockchain_token_balances')
    op.drop_table('blockchain_nfts')
    op.drop_table('transactions')
    op.drop_table('wallets')
    op.drop_table('user_subscriptions')
    op.drop_table('file_uploads')
    op.drop_table('content_versions')
    op.drop_table('content')
    op.drop_table('user_sessions')
    op.drop_table('api_keys')
    op.drop_table('users')