<?php

namespace WPGraphQL\Data\Connection;

use Exception;

/**
 * Class MenuConnectionResolver
 *
 * @package WPGraphQL\Data\Connection
 */
class MenuConnectionResolver extends TermObjectConnectionResolver {

	/**
	 * Get the connection args for use in WP_Term_Query to query the menus
	 *
	 * @return array
	 * @throws Exception
	 */
	public function get_query_args() {
		$term_args = [
			'hide_empty' => false,
			'include'    => [],
			'taxonomy'   => 'nav_menu',
			'fields'     => 'ids',
		];

		if ( ! empty( $this->args['where']['slug'] ) ) {
			$term_args['slug']    = $this->args['where']['slug'];
			$term_args['include'] = null;
		}

		$theme_locations = get_nav_menu_locations();

		// If a location is specified in the args, use it
		if ( ! empty( $this->args['where']['location'] ) ) {
			if ( isset( $theme_locations[ $this->args['where']['location'] ] ) ) {
				$term_args['include'] = $theme_locations[ $this->args['where']['location'] ];
			}
		} else {
			// If the current user cannot edit theme options
			if ( ! current_user_can( 'edit_theme_options' ) ) {
				$term_args['include'] = array_values( $theme_locations );
			}
		}

		if ( ! empty( $this->args['where']['id'] ) ) {
			$term_args['include'] = $this->args['where']['id'];
		}

		$query_args = parent::get_query_args();

		return array_merge( $query_args, $term_args );
	}

}