import os
import glob

def update_menu():
    template_dir = r"c:\xampp\htdocs\New_Stram_Site_web\templates"
    files = glob.glob(os.path.join(template_dir, "**", "*.html.twig"), recursive=True)
    
    desktop_search = r"<li class=\"menu-item\"><a href=\"{{ path('app_approach') }}\">{{ 'menu.approach'|trans }}</a></li>"
    desktop_replace = r"""<li class="menu-item"><a href="{{ path('app_approach') }}">{{ 'menu.approach'|trans }}</a></li>
                                    <li class="menu-item"><a href="{{ path('app_quality') }}">{{ 'menu.quality'|trans|default('Qualité') }}</a></li>"""
                                    
    mobile_search = r"<li class=\"iphonemenu_item\"><a href=\"{{ path('stram_about_frontend') }}\">{{ 'menu.about'|trans }}</a></li>"
    mobile_replace = r"""<li class="iphonemenu_item"><a href="{{ path('stram_about_frontend') }}">{{ 'menu.about'|trans }}</a></li>
        <li class="iphonemenu_item"><a href="{{ path('app_approach') }}">{{ 'menu.approach'|trans|default('Approche') }}</a></li>
        <li class="iphonemenu_item"><a href="{{ path('app_quality') }}">{{ 'menu.quality'|trans|default('Qualité') }}</a></li>"""
        
    footer_search = r"<li class=\"menu-item\"><a href=\"{{ path('stram_about_frontend') }}\">Qui sommes nous</a></li>"
    footer_replace = r"""<li class="menu-item"><a href="{{ path('stram_about_frontend') }}">Qui sommes nous</a></li>
                                <li class="menu-item"><a href="{{ path('app_quality') }}">Qualité</a></li>"""

    for file_path in files:
        if 'adminstram' in file_path:
            continue
            
        try:
            with open(file_path, 'r', encoding='utf-8') as f:
                content = f.read()
                
            original_content = content
            
            # Simple replacements where possible
            if "app_quality" not in content:
                content = content.replace(desktop_search, desktop_replace)
                content = content.replace(mobile_search, mobile_replace)
                content = content.replace(footer_search, footer_replace)
                
                if content != original_content:
                    with open(file_path, 'w', encoding='utf-8') as f:
                        f.write(content)
                    print(f"Updated menu in {file_path}")
        except Exception as e:
            print(f"Error reading {file_path}: {e}")

if __name__ == '__main__':
    update_menu()