// ==UserScript==
// @name        Иконки поисковых систем из расширения ContextSearch-web-ext в контекстном меню
// @namespace   cswem
// @version     2.1
// @description Редактирует контекстное меню Firefox
// @match       *://*/*
// @grant       none
// @icon        https://www.mozilla.org/favicon.ico
// ==/UserScript==

(function() {
    'use strict';
    
    // Конфигурация
    const SETTINGS = {
        initializedFlag: 'cswemInitialized',  // Флаг инициализации
        menuId: 'cswem-menugroup',           // ID группы меню
        styleId: 'cswem-styles',             // ID стилей
        hiddenItems: [                       // Элементы для скрытия
            '_5dd73bb9-e728-4d1e-990b-c77d8e03670f_-menuitem-_root_menu',
            'context-searchselect',
            'context-keywordfield'
        ]
    };

    // Проверяем, не запущен ли скрипт ранее
    if (window[SETTINGS.initializedFlag]) return;
    window[SETTINGS.initializedFlag] = true;

    try {
        // ██████████████████████████████████████████████████████████████
        // █ 1. ДОБАВЛЯЕМ СТИЛИ ДЛЯ СКРЫТИЯ ЛИШНИХ ЭЛЕМЕНТОВ И НАСТРОЙКИ МЕНЮ
        // ██████████████████████████████████████████████████████████████
        const style = document.createElement('style');
        style.id = SETTINGS.styleId;
        style.textContent = `
            #${SETTINGS.menuId} {
                padding-left: 30px;
                display: grid;
                grid-template-columns: repeat(auto-fill, 32px);
                grid-auto-rows: 26px;
                gap: 2px;
            }
            #${SETTINGS.menuId} > menuitem {
                -moz-box-pack: center;
                list-style-image: var(--image) !important;
            }
            ${SETTINGS.hiddenItems.map(id => `#${id}`).join(', ')},
            #${SETTINGS.menuId}:empty {
                display: none !important;
            }
        `;
        document.head.appendChild(style);

        // ██████████████████████████████████████████████████████████████
        // █ 2. СОЗДАЕМ КОНТЕЙНЕР ДЛЯ ИКОНОК
        // ██████████████████████████████████████████████████████████████
        const menugroup = document.createXULElement('menugroup');
        menugroup.id = SETTINGS.menuId;
        
        // Размещаем после поисковой строки или в начало меню
        const searchSelect = document.getElementById('context-searchselect');
        if (searchSelect) {
            searchSelect.after(menugroup);
        } else {
            document.getElementById('contentAreaContextMenu').prepend(menugroup);
        }

        // ██████████████████████████████████████████████████████████████
        // █ 3. ФУНКЦИЯ ФИЛЬТРАЦИИ: ОПРЕДЕЛЯЕМ, КАКИЕ ЭЛЕМЕНТЫ ОСТАВИТЬ
        // ██████████████████████████████████████████████████████████████
        const shouldKeepItem = (item) => {
            const forbiddenClasses = ['menuitem-iconic-webext', 'extension-menu-item'];
            const forbiddenIds = ['ublock', 'adguard', 'grammarly', 'ext-', 'extension-', ...SETTINGS.hiddenItems];
            
            return (
                item.tagName === 'menuitem' &&
                !item.hidden &&
                item.getAttribute('image') &&
                item.getAttribute('label') &&
                !item.id?.includes('_separator') &&
                !forbiddenClasses.some(c => item.classList.contains(c)) &&
                !forbiddenIds.some(id => item.id?.includes(id))
            );
        };

        // ██████████████████████████████████████████████████████████████
        // █ 4. ОБНОВЛЕНИЕ МЕНЮ: УДАЛЯЕМ СТАРЫЕ ЭЛЕМЕНТЫ И ДОБАВЛЯЕМ НОВЫЕ
        // ██████████████████████████████████████████████████████████████
        const refreshMenu = () => {
            // Полностью очищаем меню перед обновлением
            while (menugroup.firstChild) {
                menugroup.removeChild(menugroup.firstChild);
            }
            
            // Находим все элементы меню и фильтруем их
            const allMenuItems = Array.from(document.querySelectorAll('#contentAreaContextMenu menuitem'));
            const validItems = allMenuItems.filter(shouldKeepItem);
            
            // Клонируем и добавляем только нужные элементы
            validItems.forEach(item => {
                const clone = item.cloneNode(true);
                clone.id = ''; // Удаляем ID, чтобы избежать конфликтов
                
                // Назначаем обработчик клика
                clone.addEventListener('command', () => {
                    try {
                        item.click();
                    } catch (e) {
                        console.error('Ошибка при клике:', e);
                    }
                });
                
                menugroup.appendChild(clone);
            });
            
            // Скрываем группу, если нет элементов
            menugroup.hidden = menugroup.children.length === 0;
        };

        // ██████████████████████████████████████████████████████████████
        // █ 5. НАСТРОЙКА СОБЫТИЙ: ОБНОВЛЯЕМ МЕНЮ ПРИ КАЖДОМ ОТКРЫТИИ
        // ██████████████████████████████████████████████████████████████
        const contextMenu = document.getElementById('contentAreaContextMenu');
        contextMenu.addEventListener('popupshowing', refreshMenu);

        // ██████████████████████████████████████████████████████████████
        // █ 6. ОЧИСТКА: УДАЛЯЕМ ЭЛЕМЕНТЫ ПРИ ЗАКРЫТИИ СТРАНИЦЫ
        // ██████████████████████████████████████████████████████████████
        window.addEventListener('unload', () => {
            contextMenu.removeEventListener('popupshowing', refreshMenu);
            document.getElementById(SETTINGS.menuId)?.remove();
            document.getElementById(SETTINGS.styleId)?.remove();
            delete window[SETTINGS.initializedFlag];
        });

    } catch (error) {
        console.error('Ошибка в скрипте контекстного меню:', error);
    }
})();