<?php
namespace Abdiel_AMM;

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

/**
 * Switch classic theme menus by language.
 *
 * Conservative behavior:
 * - Only runs when enabled.
 * - Only switches when a menu is explicitly assigned for the current language.
 * - If a language has no assigned menu, the theme's default menu remains.
 */
class Menu_Switcher {

    public function hooks() {
        add_filter( 'theme_mod_nav_menu_locations', [ $this, 'filter_locations' ], 20, 1 );
        // Optional: also apply the same mapping to Elementor's Nav Menu widget (which typically passes a specific menu term).
        add_filter( 'wp_nav_menu_args', [ $this, 'filter_nav_menu_args_for_elementor' ], 20, 1 );
    }

    /**
     * Apply menu switching to menu render calls that pass a specific menu term (common in Elementor Nav Menu widget).
     *
     * Behavior:
     * - Only runs when menu switching is enabled AND the "apply_elementor" flag is enabled.
     * - Only switches when the current menu matches ANY menu in a configured location mapping.
     * - Switches to the menu assigned for the current language for that same location.
     * - If no menu is assigned for the current language, does nothing.
     *
     * @param array $args
     * @return array
     */
    public function filter_nav_menu_args_for_elementor( $args ) {
        if ( ! is_array( $args ) ) { return $args; }

        $opt = get_option( 'abdiel_amm_menu_switching', [] );
        if ( ! is_array( $opt ) ) { return $args; }
        if ( empty( $opt['enabled'] ) || empty( $opt['apply_elementor'] ) ) { return $args; }

        $map = isset( $opt['map'] ) && is_array( $opt['map'] ) ? $opt['map'] : [];
        if ( empty( $map ) ) { return $args; }

        $lang = abdiel_amm_boot()->content->get_current_language_code();
        $lang = is_string( $lang ) ? sanitize_key( $lang ) : '';
        if ( '' === $lang ) { return $args; }

        // Resolve current menu term id from args.
        if ( empty( $args['menu'] ) ) { return $args; }

        $current_menu_id = 0;
        $menu = $args['menu'];
        if ( is_numeric( $menu ) ) {
            $current_menu_id = absint( $menu );
        } elseif ( is_object( $menu ) && isset( $menu->term_id ) ) {
            $current_menu_id = absint( $menu->term_id );
        } elseif ( is_string( $menu ) ) {
            $obj = wp_get_nav_menu_object( $menu );
            if ( $obj && isset( $obj->term_id ) ) {
                $current_menu_id = absint( $obj->term_id );
            }
        }

        if ( $current_menu_id <= 0 ) { return $args; }

        // Find the mapping "location" whose any-language menu matches the current menu.
        $matched_loc = '';
        foreach ( $map as $loc => $langs ) {
            if ( ! is_array( $langs ) ) { continue; }
            foreach ( $langs as $code => $mid ) {
                if ( absint( $mid ) === $current_menu_id ) {
                    $matched_loc = sanitize_key( (string) $loc );
                    break 2;
                }
            }
        }

        if ( '' === $matched_loc ) { return $args; }
        if ( empty( $map[ $matched_loc ][ $lang ] ) ) { return $args; }

        $target_menu_id = absint( $map[ $matched_loc ][ $lang ] );
        if ( $target_menu_id > 0 && $target_menu_id !== $current_menu_id ) {
            $args['menu'] = $target_menu_id;
        }

        return $args;
    }

    /**
     * @param array $locations location => menu_id
     * @return array
     */
    public function filter_locations( $locations ) {
        if ( ! is_array( $locations ) ) { return $locations; }

        $opt = get_option( 'abdiel_amm_menu_switching', [] );
        if ( ! is_array( $opt ) ) { return $locations; }
        if ( empty( $opt['enabled'] ) ) { return $locations; }

        $map = isset( $opt['map'] ) && is_array( $opt['map'] ) ? $opt['map'] : [];
        if ( empty( $map ) ) { return $locations; }

        $lang = abdiel_amm_boot()->content->get_current_language_code();
        $lang = is_string( $lang ) ? sanitize_key( $lang ) : '';
        if ( '' === $lang ) { return $locations; }

        // If the current language has no explicit menu assignments, do nothing.
        foreach ( $locations as $loc => $menu_id ) {
            $loc = sanitize_key( (string) $loc );
            if ( ! isset( $map[ $loc ] ) || ! is_array( $map[ $loc ] ) ) { continue; }
            if ( ! empty( $map[ $loc ][ $lang ] ) ) {
                $locations[ $loc ] = absint( $map[ $loc ][ $lang ] );
            }
        }

        return $locations;
    }
}