"""
Management command to set up Jumbo company, users, and link existing data.
"""

from accounts.models import Company, CustomUser
from data_integration.models import ChatSession, ExternalDataSource
from django.core.management.base import BaseCommand


class Command(BaseCommand):
    help = "Set up Jumbo company, create users, and link existing external data"

    def handle(self, *_args, **_options):
        self.stdout.write("Setting up Jumbo company and data...")

        # 1. Create Jumbo company
        jumbo_company, created = Company.objects.get_or_create(
            name="Jumbo", defaults={"description": "Jumbo Supermarkets - External API Data"}
        )
        if created:
            self.stdout.write(self.style.SUCCESS("✓ Created Jumbo company"))
        else:
            self.stdout.write("  Jumbo company already exists")

        # 2. Create admin user for Jumbo
        admin_created = False
        if not CustomUser.objects.filter(username="jumbo_admin").exists():
            CustomUser.objects.create_user(  # nosec B106
                username="jumbo_admin",
                email="admin@jumbo.nl",
                password="jumbo123",
                company=jumbo_company,
                is_company_admin=True,
            )
            self.stdout.write(self.style.SUCCESS("✓ Created Jumbo admin: jumbo_admin / jumbo123"))
            admin_created = True
        else:
            self.stdout.write("  Jumbo admin already exists")

        # 3. Create regular users for Jumbo
        jumbo_users = [
            {
                "username": "jumbo_analyst",
                "email": "analyst@jumbo.nl",
                "password": "jumbo123",
                "is_company_admin": False,
            },
            {
                "username": "jumbo_manager",
                "email": "manager@jumbo.nl",
                "password": "jumbo123",
                "is_company_admin": False,
            },
        ]

        users_created = 0
        for user_data in jumbo_users:
            if not CustomUser.objects.filter(username=user_data["username"]).exists():
                CustomUser.objects.create_user(
                    username=user_data["username"],
                    email=user_data["email"],
                    password=user_data["password"],
                    company=jumbo_company,
                    is_company_admin=user_data["is_company_admin"],
                )
                users_created += 1

        if users_created:
            self.stdout.write(self.style.SUCCESS(f"✓ Created {users_created} Jumbo users"))
        else:
            self.stdout.write("  Jumbo users already exist")

        # 4. Link External Data Source to Jumbo company
        try:
            jumbo_ext_source = ExternalDataSource.objects.get(name="Jumbo API")
            if not jumbo_ext_source.company:
                jumbo_ext_source.company = jumbo_company
                jumbo_ext_source.save()
                self.stdout.write(self.style.SUCCESS("✓ Linked Jumbo API data source to company"))
            else:
                self.stdout.write("  Jumbo API data source already linked")
        except ExternalDataSource.DoesNotExist:
            self.stdout.write(
                self.style.WARNING("⚠ Jumbo API external data source not found. Create it in admin first.")
            )

        # 5. Link existing chat sessions to Jumbo company
        unlinked_sessions = ChatSession.objects.filter(company__isnull=True)
        if unlinked_sessions.exists():
            count = unlinked_sessions.update(company=jumbo_company)
            self.stdout.write(self.style.SUCCESS(f"✓ Linked {count} existing chat sessions to Jumbo company"))
        else:
            self.stdout.write("  All chat sessions already linked to companies")

        # 6. Summary
        total_sessions = ChatSession.objects.filter(company=jumbo_company).count()
        total_users = CustomUser.objects.filter(company=jumbo_company).count()

        self.stdout.write(
            self.style.SUCCESS(
                f"\n✓ Setup complete!"
                f"\n  Company: {jumbo_company.name}"
                f"\n  Users: {total_users} (including {1 if admin_created or CustomUser.objects.filter(username='jumbo_admin').exists() else 0} admin)"
                f"\n  Chat sessions: {total_sessions}"
            )
        )
        self.stdout.write("\nLogin as jumbo_admin/jumbo123 to view the dashboard with Jumbo data.")