<?php

namespace Codedivo\AdminOptimizer\Features;

if ( ! defined( 'ABSPATH' ) ) exit;

use Codedivo\AdminOptimizer\Core\SettingsManager;
use Codedivo\AdminOptimizer\Traits\Singleton;

class MenuEditor {
    use Singleton;

    private $settings;
    private $full_menu = [];

    public function __construct() {
        $this->settings = SettingsManager::get_instance();
        $this->init_hooks();
    }

    private function get_setting($key, $default = '') {
        return $this->settings->get('menu_editor', $key, $default);
    }

    private function init_hooks() {
        if ($this->get_setting('enable_menu_editor')) {
            // Re-ordering logic
            add_filter('custom_menu_order', '__return_true');
            add_filter('menu_order', [$this, 'apply_menu_order'], PHP_INT_MAX);
            
            // Capture menu state BEFORE modifications
            add_action('admin_menu', [$this, 'capture_menu'], PHP_INT_MAX - 100);

            // Apply modifications (Hiding/Renaming)
            add_action('admin_menu', [$this, 'apply_customizations'], PHP_INT_MAX);
        }
    }
    
    public function capture_menu() {
        global $menu;
        if (isset($menu) && is_array($menu)) {
            $this->full_menu = $menu;
        }
    }

    public function apply_menu_order($menu_order) {
        $custom_order = $this->get_setting('menu_order', []);
        
        if (empty($custom_order) || !is_array($custom_order)) {
            return $menu_order;
        }

        $new_order = [];
        // 1. Add items from custom order
        foreach ($custom_order as $slug) {
            if (in_array($slug, $menu_order)) {
                $new_order[] = $slug;
            }
        }
        // 2. Append remaining
        foreach ($menu_order as $slug) {
            if (!in_array($slug, $new_order)) {
                $new_order[] = $slug;
            }
        }
        return $new_order;
    }

    public function apply_customizations() {
        global $menu;
        
        // Ensure we captured the menu if for some reason capture_menu didn't run or array was empty then
        if (empty($this->full_menu) && !empty($menu)) {
            $this->full_menu = $menu;
        }

        if (empty($menu)) return;

        $renames = $this->get_setting('menu_renames', []);
        $hidden = $this->get_setting('menu_hidden', []);

        // 1. Hiding
        if (!empty($hidden)) {
             foreach ($hidden as $slug) {
                 // Ensure we NEVER hide our own plugin
                 if ($slug === 'admin-menu-optimizer') continue;
                 
                 remove_menu_page($slug);
             }
        }

        // 2. Renaming
        foreach ($menu as $index => $item) {
            if (!isset($item[2])) continue;
            
            $slug = $item[2];
            
            if (isset($renames[$slug]) && !empty($renames[$slug])) {
                if (preg_match('/<span.*<\/span>/', $item[0], $matches)) {
                   $menu[$index][0] = $renames[$slug] . ' ' . $matches[0];
                } else {
                   $menu[$index][0] = $renames[$slug];
                }
            }
        }
    }

    public function get_current_menus() {
        // Use the captured full menu if available, otherwise fallback to global menu
        $source_menu = !empty($this->full_menu) ? $this->full_menu : $GLOBALS['menu'];
        
        $items = [];
        if (!empty($source_menu)) {
            foreach ($source_menu as $item) {
                if (empty($item[0])) continue; // Skip separators
                
                // Clean label: remove notification spans first, then strip tags
                $raw_label = $item[0];
                if (strpos($raw_label, '<span') !== false) {
                    $raw_label = substr($raw_label, 0, strpos($raw_label, '<span'));
                }
                $label = wp_strip_all_tags($raw_label);
                // If it's the notification bubble span, strip_tags removes it, which is good for label display.
                // But wait, if we see the captured menu, it has original labels.
                
                $slug = $item[2];
                $icon = isset($item[6]) ? $item[6] : 'dashicons-admin-generic';
                
                $items[] = [
                    'label' => $label,
                    'slug' => $slug,
                    'icon' => $icon
                ];
            }
        }
        return $items;
    }

    public function render_settings() {
        // We need a JS-driven UI here.
        // We will output the current menu list as JSON in a script tag.
        
        $current_menus = $this->get_current_menus();
        $saved_order = $this->get_setting('menu_order', []);
        $saved_renames = $this->get_setting('menu_renames', []);
        $saved_hidden = $this->get_setting('menu_hidden', []);
        $enable_editor = $this->get_setting('enable_menu_editor');

        echo '<div class="cdvao-control-group">';
        echo '<label class="cdvao-control-label">' . esc_html( __('Enable Menu Editor', 'admin-menu-optimizer') ) . '</label>';
        echo '<div class="cdvao-toggle">';
        echo '<input type="checkbox" name="menu_editor[enable_menu_editor]" value="1" ' . checked(1, $enable_editor, false) . '>';
        echo '</div>';
        echo '</div>';

        echo '<div class="cdvao-menu-editor-ui" id="cdvao-menu-editor">';
        
        // Pass data to JS
        // Note: we need to handle sort order. If saved order exists, we should respect it in display.
        
        $display_items = [];
        if (!empty($saved_order)) {
            // First added saved/ordered items
            foreach ($saved_order as $slug) {
                foreach ($current_menus as $key => $menu_item) {
                    if ($menu_item['slug'] === $slug) {
                        $display_items[] = $menu_item;
                        unset($current_menus[$key]);
                        break;
                    }
                }
            }
        }
        // Append remaining
        $display_items = array_merge($display_items, $current_menus);

        echo '<ul class="cdvao-menu-list">';
        foreach ($display_items as $item) {
            $slug = $item['slug'];
            $label = $item['label'];
            $icon = $item['icon'];
            
            $is_hidden = in_array($slug, $saved_hidden);
            $rename_val = isset($saved_renames[$slug]) ? $saved_renames[$slug] : '';
            $is_protected = ($slug === 'admin-menu-optimizer');

            echo '<li class="cdvao-menu-item" data-slug="' . esc_attr($slug) . '">';
            echo '<div class="cdvao-menu-handle"><span class="dashicons dashicons-move"></span></div>';
            echo '<div class="cdvao-menu-icon"><span class="dashicons ' . esc_attr($icon) . '"></span></div>';
            echo '<div class="cdvao-menu-details">';
            echo '<span class="original-label">' . esc_html($label) . '</span>';
            echo '<input type="text" name="menu_editor[menu_renames][' . esc_attr($slug) . ']" value="' . esc_attr($rename_val) . '" placeholder="' . esc_attr( __('Rename to...', 'admin-menu-optimizer') ) . '">';
            echo '</div>';
            echo '<div class="cdvao-menu-actions">';
            if (!$is_protected) {
                echo '<label class="cdvao-visibility-toggle" title="' . esc_attr( __('Toggle Visibility', 'admin-menu-optimizer') ) . '">';
                echo '<input type="checkbox" name="menu_editor[menu_hidden][]" value="' . esc_attr($slug) . '" ' . checked(true, $is_hidden, false) . '>';
                echo '<span class="dashicons dashicons-visibility"></span>';
                echo '</label>';
            } else {
                echo '<span class="dashicons dashicons-lock" title="' . esc_attr( __('This menu cannot be hidden', 'admin-menu-optimizer') ) . '"></span>';
            }
            // Hidden input for order
            echo '<input type="hidden" name="menu_editor[menu_order][]" value="' . esc_attr($slug) . '">';
            echo '</div>';
            echo '</li>';
        }
        echo '</ul>';
        echo '</div>';
        
        echo '<p class="description">' . esc_html( __('Drag and drop to reorder. Uncheck the eye icon to hide. Enter text to rename.', 'admin-menu-optimizer') ) . '</p>';
    }
}