from django.core.management.base import BaseCommand
from django.urls import get_resolver
import os
import json
from django.template.loaders.app_directories import get_app_template_dirs


class Command(BaseCommand):
    help = "Generate UI element map for IntroJS tours"

    def handle(self, *args, **kwargs):
        output_file = os.path.join('static', 'js', 'tours', 'ui_element_map.json')

        ui_map = {
            "pages": {},
            "common_elements": {
                "navigation": {
                    "inventory": "#inventory-menu, .inventory-nav, nav .inventory",
                    "finance": "#finance-menu, .finance-nav, nav .finance",
                    "customers": "#customers-menu, .customers-nav, nav .customers"
                },
                "actions": {
                    "add": ".btn-add, .add-button, button:contains('Add')",
                    "edit": ".btn-edit, .edit-button, button:contains('Edit')",
                    "save": ".btn-save, button[type='submit'], #save-button",
                    "cancel": ".btn-cancel, #cancel-button, button:contains('Cancel')",
                    "delete": ".btn-delete, .delete-button, button:contains('Delete')"
                },
                "forms": {
                    "search": "#search-form, .search-input, input[name='q']",
                    "date_range": ".date-range, input[type='date']",
                    "dropdown": "select, .dropdown, .select-field"
                }
            }
        }

        # Extract URL patterns to identify pages
        resolver = get_resolver()
        for url_pattern in resolver.url_patterns:
            if hasattr(url_pattern, 'name') and url_pattern.name:
                pattern_name = url_pattern.name
                # Skip admin and API URLs
                if pattern_name.startswith(('admin:', 'api:')):
                    continue

                ui_map["pages"][pattern_name] = {
                    "url_pattern": str(url_pattern.pattern),
                    "elements": {}
                }

        # Scan templates for UI elements with IDs
        template_dirs = get_app_template_dirs('templates')
        for template_dir in template_dirs:
            for root, dirs, files in os.walk(template_dir):
                for file in files:
                    if not file.endswith(('.html', '.htm')):
                        continue

                    try:
                        with open(os.path.join(root, file), 'r', encoding='utf-8') as f:
                            content = f.read()

                            # Try to identify the page/view this template is for
                            template_path = os.path.relpath(os.path.join(root, file), template_dir)
                            page_key = template_path.replace('/', '_').replace('.html', '')

                            # Create page entry if it doesn't exist
                            if page_key not in ui_map["pages"]:
                                ui_map["pages"][page_key] = {
                                    "template": template_path,
                                    "elements": {}
                                }

                            # Extract elements with IDs
                            import re
                            id_matches = re.findall(r'id=["\']([^"\']+)["\']', content)
                            for id_match in id_matches:
                                ui_map["pages"][page_key]["elements"][id_match] = f"#{id_match}"

                    except Exception as e:
                        self.stdout.write(self.style.WARNING(f"Error processing template {file}: {e}"))

        # Save UI map as JSON
        os.makedirs(os.path.dirname(output_file), exist_ok=True)
        with open(output_file, 'w', encoding='utf-8') as f:
            json.dump(ui_map, f, indent=2)

        self.stdout.write(self.style.SUCCESS(f"✅ UI element map saved to {output_file}"))