"""create_gdpr_requests_table

Revision ID: d709b14aa24a
Revises: b1c4d7c154ec
Create Date: 2025-12-06 01:10:15.233886

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


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


def upgrade() -> None:
    # Check if table exists before creating
    bind = op.get_bind()
    inspector = sa.inspect(bind)
    tables = inspector.get_table_names()
    
    if 'gdpr_requests' not in tables:
        # Create GDPR requests table
        op.create_table(
            'gdpr_requests',
            sa.Column('id', sa.Integer(), autoincrement=True, nullable=False),
            sa.Column('request_type', sa.Enum('data_export', 'data_deletion', 'data_rectification', 'consent_withdrawal', name='gdprrequesttype'), nullable=False),
            sa.Column('status', sa.Enum('pending', 'processing', 'completed', 'rejected', 'cancelled', name='gdprrequeststatus'), nullable=False, server_default='pending'),
            sa.Column('user_id', sa.Integer(), nullable=True),
            sa.Column('user_email', sa.String(length=255), nullable=False),
            sa.Column('is_anonymous', sa.Boolean(), nullable=False, server_default='0'),
            sa.Column('request_data', sa.JSON(), nullable=True),
            sa.Column('verification_token', sa.String(length=255), nullable=True),
            sa.Column('verified_at', sa.DateTime(), nullable=True),
            sa.Column('processed_by', sa.Integer(), nullable=True),
            sa.Column('processed_at', sa.DateTime(), nullable=True),
            sa.Column('processing_notes', sa.Text(), nullable=True),
            sa.Column('export_file_path', sa.String(length=500), nullable=True),
            sa.Column('deletion_log', sa.JSON(), nullable=True),
            sa.Column('ip_address', sa.String(length=45), nullable=True),
            sa.Column('user_agent', sa.String(length=255), nullable=True),
            sa.Column('created_at', sa.DateTime(), nullable=False, server_default=sa.text('CURRENT_TIMESTAMP')),
            sa.Column('updated_at', sa.DateTime(), nullable=False, server_default=sa.text('CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP')),
            sa.Column('expires_at', sa.DateTime(), nullable=True),
            sa.ForeignKeyConstraint(['processed_by'], ['users.id'], ),
            sa.ForeignKeyConstraint(['user_id'], ['users.id'], ),
            sa.PrimaryKeyConstraint('id')
        )
        
        # Create indexes
        op.create_index(op.f('ix_gdpr_requests_id'), 'gdpr_requests', ['id'], unique=False)
        op.create_index(op.f('ix_gdpr_requests_request_type'), 'gdpr_requests', ['request_type'], unique=False)
        op.create_index(op.f('ix_gdpr_requests_status'), 'gdpr_requests', ['status'], unique=False)
        op.create_index(op.f('ix_gdpr_requests_user_id'), 'gdpr_requests', ['user_id'], unique=False)
        op.create_index(op.f('ix_gdpr_requests_is_anonymous'), 'gdpr_requests', ['is_anonymous'], unique=False)
        op.create_index(op.f('ix_gdpr_requests_created_at'), 'gdpr_requests', ['created_at'], unique=False)
        op.create_index(op.f('ix_gdpr_requests_verification_token'), 'gdpr_requests', ['verification_token'], unique=True)
    else:
        # Table already exists, ensure it has the is_anonymous column if missing
        columns = [col['name'] for col in inspector.get_columns('gdpr_requests')]
        if 'is_anonymous' not in columns:
            op.add_column('gdpr_requests', sa.Column('is_anonymous', sa.Boolean(), nullable=False, server_default='0'))
            op.create_index(op.f('ix_gdpr_requests_is_anonymous'), 'gdpr_requests', ['is_anonymous'], unique=False)
        
        # Ensure user_id is nullable
        # Note: This might fail if there are existing non-null values, but for a new table it should be fine
        try:
            op.alter_column('gdpr_requests', 'user_id', existing_type=sa.Integer(), nullable=True)
        except Exception:
            pass  # Column might already be nullable


def downgrade() -> None:
    # Drop indexes
    bind = op.get_bind()
    inspector = sa.inspect(bind)
    tables = inspector.get_table_names()
    
    if 'gdpr_requests' in tables:
        try:
            op.drop_index(op.f('ix_gdpr_requests_verification_token'), table_name='gdpr_requests')
        except Exception:
            pass
        try:
            op.drop_index(op.f('ix_gdpr_requests_created_at'), table_name='gdpr_requests')
        except Exception:
            pass
        try:
            op.drop_index(op.f('ix_gdpr_requests_is_anonymous'), table_name='gdpr_requests')
        except Exception:
            pass
        try:
            op.drop_index(op.f('ix_gdpr_requests_user_id'), table_name='gdpr_requests')
        except Exception:
            pass
        try:
            op.drop_index(op.f('ix_gdpr_requests_status'), table_name='gdpr_requests')
        except Exception:
            pass
        try:
            op.drop_index(op.f('ix_gdpr_requests_request_type'), table_name='gdpr_requests')
        except Exception:
            pass
        try:
            op.drop_index(op.f('ix_gdpr_requests_id'), table_name='gdpr_requests')
        except Exception:
            pass
        
        # Drop table
        op.drop_table('gdpr_requests')