from django.core.management.base import BaseCommand
from django.db import connection
from django.contrib.auth.hashers import make_password
import uuid


class Command(BaseCommand):
    help = "Reset database and create admin user"

    def handle(self, *args, **options):
        self.stdout.write("Resetting database...")

        # Drop all tables
        with connection.cursor() as cursor:
            cursor.execute(
                """
                DO $$ DECLARE
                    r RECORD;
                BEGIN
                    FOR r IN (
                        SELECT tablename FROM pg_tables
                        WHERE schemaname = current_schema()
                    ) LOOP
                        EXECUTE 'DROP TABLE IF EXISTS ' || \
                                quote_ident(r.tablename) || ' CASCADE';
                    END LOOP;
                END $$;
            """
            )

            # Reset sequences
            cursor.execute(
                """
                DO $$ DECLARE
                    r RECORD;
                BEGIN
                    FOR r IN (
                        SELECT sequencename FROM pg_sequences
                        WHERE schemaname = current_schema()
                    ) LOOP
                        EXECUTE 'ALTER SEQUENCE ' || \
                                quote_ident(r.sequencename) || ' RESTART WITH 1';
                    END LOOP;
                END $$;
            """
            )

        self.stdout.write("All tables dropped and sequences reset.")

        # Run migrations
        from django.core.management import call_command

        call_command("migrate")

        self.stdout.write("Migrations applied.")

        # Create superuser using raw SQL
        try:
            with connection.cursor() as cursor:
                # Create user
                user_id = str(uuid.uuid4())[:10]
                cursor.execute(
                    """
                    INSERT INTO accounts_user (
                        username, password, email, is_superuser, is_staff,
                        is_active, date_joined, user_id, first_name,
                        last_name, role, is_banned, ban_reason,
                        theme_preference
                    ) VALUES (
                        'admin', %s, 'admin@thrillwiki.com', true, true,
                        true, NOW(), %s, '', '', 'SUPERUSER', false, '',
                        'light'
                    ) RETURNING id;
                """,
                    [make_password("admin"), user_id],
                )

                result = cursor.fetchone()
                if result is None:
                    raise Exception("Failed to create user - no ID returned")
                user_db_id = result[0]

                # Create profile
                profile_id = str(uuid.uuid4())[:10]
                cursor.execute(
                    """
                    INSERT INTO accounts_userprofile (
                        profile_id, display_name, pronouns, bio,
                        twitter, instagram, youtube, discord,
                        coaster_credits, dark_ride_credits,
                        flat_ride_credits, water_ride_credits,
                        user_id, avatar
                    ) VALUES (
                        %s, 'Admin', 'they/them', 'ThrillWiki Administrator',
                        '', '', '', '',
                        0, 0, 0, 0,
                        %s, ''
                    );
                """,
                    [profile_id, user_db_id],
                )

            self.stdout.write("Superuser created.")
        except Exception as e:
            self.stdout.write(
                self.style.ERROR(
                    f"Error creating superuser: {
                        str(e)}"
                )
            )
            raise

        self.stdout.write(self.style.SUCCESS("Database reset complete."))