unfulvio / wp-api-menus

includes/wp-api-menus-v2.php

Indentation +347 added lines, -347 removed lines

@@ -6,299 +6,299 @@ 
  */
 
 if ( ! defined( 'ABSPATH' ) ) {
-    exit; // Exit if accessed directly
+	exit; // Exit if accessed directly
 }
 
 if ( ! class_exists( 'WP_REST_Menus' ) ) :
 
 
-    /**
-     * WP REST Menus class.
-     *
-     * WP API Menus support for WP API v2.
-     *
-     * @package WP_API_Menus
-     * @since 1.2.0
-     */
-    class WP_REST_Menus {
-
-
-	    /**
-	     * Get WP API namespace.
-	     *
-	     * @since 1.2.0
-	     * @return string
-	     */
-        public static function get_api_namespace() {
-            return 'wp/v2';
-        }
-
-
-	    /**
-	     * Get WP API Menus namespace.
-	     *
-	     * @since 1.2.1
-	     * @return string
-	     */
-	    public static function get_plugin_namespace() {
-		    return 'wp-api-menus/v2';
-	    }
-
-
-        /**
-         * Register menu routes for WP API v2.
-         *
-         * @since  1.2.0
-         */
-        public function register_routes() {
-
-            register_rest_route( self::get_plugin_namespace(), '/menus', array(
-                array(
-                    'methods'  => WP_REST_Server::READABLE,
-                    'callback' => array( $this, 'get_menus' ),
-                    'args'     => array(
-                        'include' => array(
-                            'required' => false,
-                            'validate_callback' => function($val){
-                                $list = explode(',',$val);
-                                if(!is_array( $list))
-                                    return new WP_Error( 'rest_invalid_param', 'include should be an array of menu ID numbers', array( 'status' => 500 ) );
-                                foreach($list as $id){
-                                    if(!is_nav_menu($id))
-                                        return new WP_Error( 'rest_invalid_param', $id.' is not a nav menu', array( 'status' => 500 ) );
-                                }
-                                return true;
-                            }
-                        ),
-                        'include_menu_items' => array(
-                            'required' => false
-                        ),
-                    ),
-                    'permission_callback' => '__return_true',
-                )
-            ) );
-
-            register_rest_route( self::get_plugin_namespace(), '/menus/(?P<id>\d+)', array(
-                array(
-                    'methods'  => WP_REST_Server::READABLE,
-                    'callback' => array( $this, 'get_menu' ),
-                    'permission_callback' => '__return_true',
-                    'args'     => array(
-                        'context' => array(
-                        'default' => 'view',
-                        ),
-                    ),
-                )
-            ) );
-
-            register_rest_route( self::get_plugin_namespace(), '/menu-locations', array(
-                array(
-                    'methods'  => WP_REST_Server::READABLE,
-                    'callback' => array( $this, 'get_menu_locations' ),
-                    'permission_callback' => '__return_true',
-                )
-            ) );
-
-            register_rest_route( self::get_plugin_namespace(), '/menu-locations/(?P<location>[a-zA-Z0-9_-]+)', array(
-                array(
-                    'methods'  => WP_REST_Server::READABLE,
-                    'callback' => array( $this, 'get_menu_location' ),
-                    'permission_callback' => '__return_true',
-                )
-            ) );
-        }
-
-
-        /**
-         * Get menus.
-         *
-         * @since  1.2.0
-         * @return array All registered menus
-         */
-        public static function get_menus($request =false) {
-            if(!$request)
-                return [];
-
-            $params = $request->get_params();
+	/**
+	 * WP REST Menus class.
+	 *
+	 * WP API Menus support for WP API v2.
+	 *
+	 * @package WP_API_Menus
+	 * @since 1.2.0
+	 */
+	class WP_REST_Menus {
+
+
+		/**
+		 * Get WP API namespace.
+		 *
+		 * @since 1.2.0
+		 * @return string
+		 */
+		public static function get_api_namespace() {
+			return 'wp/v2';
+		}
+
+
+		/**
+		 * Get WP API Menus namespace.
+		 *
+		 * @since 1.2.1
+		 * @return string
+		 */
+		public static function get_plugin_namespace() {
+			return 'wp-api-menus/v2';
+		}
+
+
+		/**
+		 * Register menu routes for WP API v2.
+		 *
+		 * @since  1.2.0
+		 */
+		public function register_routes() {
+
+			register_rest_route( self::get_plugin_namespace(), '/menus', array(
+				array(
+					'methods'  => WP_REST_Server::READABLE,
+					'callback' => array( $this, 'get_menus' ),
+					'args'     => array(
+						'include' => array(
+							'required' => false,
+							'validate_callback' => function($val){
+								$list = explode(',',$val);
+								if(!is_array( $list))
+									return new WP_Error( 'rest_invalid_param', 'include should be an array of menu ID numbers', array( 'status' => 500 ) );
+								foreach($list as $id){
+									if(!is_nav_menu($id))
+										return new WP_Error( 'rest_invalid_param', $id.' is not a nav menu', array( 'status' => 500 ) );
+								}
+								return true;
+							}
+						),
+						'include_menu_items' => array(
+							'required' => false
+						),
+					),
+					'permission_callback' => '__return_true',
+				)
+			) );
+
+			register_rest_route( self::get_plugin_namespace(), '/menus/(?P<id>\d+)', array(
+				array(
+					'methods'  => WP_REST_Server::READABLE,
+					'callback' => array( $this, 'get_menu' ),
+					'permission_callback' => '__return_true',
+					'args'     => array(
+						'context' => array(
+						'default' => 'view',
+						),
+					),
+				)
+			) );
+
+			register_rest_route( self::get_plugin_namespace(), '/menu-locations', array(
+				array(
+					'methods'  => WP_REST_Server::READABLE,
+					'callback' => array( $this, 'get_menu_locations' ),
+					'permission_callback' => '__return_true',
+				)
+			) );
+
+			register_rest_route( self::get_plugin_namespace(), '/menu-locations/(?P<location>[a-zA-Z0-9_-]+)', array(
+				array(
+					'methods'  => WP_REST_Server::READABLE,
+					'callback' => array( $this, 'get_menu_location' ),
+					'permission_callback' => '__return_true',
+				)
+			) );
+		}
+
+
+		/**
+		 * Get menus.
+		 *
+		 * @since  1.2.0
+		 * @return array All registered menus
+		 */
+		public static function get_menus($request =false) {
+			if(!$request)
+				return [];
+
+			$params = $request->get_params();
             
-            $query_args = [];
-            if(isset($params['include']))
-                $query_args['include'] = explode(',',$params['include']);
+			$query_args = [];
+			if(isset($params['include']))
+				$query_args['include'] = explode(',',$params['include']);
 
-            $rest_url = trailingslashit( get_rest_url() . self::get_plugin_namespace() . '/menus/' );
-            $wp_menus = wp_get_nav_menus($query_args);
+			$rest_url = trailingslashit( get_rest_url() . self::get_plugin_namespace() . '/menus/' );
+			$wp_menus = wp_get_nav_menus($query_args);
 
-            // check if we should also include the actual menu; default to false
-            $include_items= (isset($params['include_menu_items']) && ($params['include_menu_items'] == "true")) ? true : false;
+			// check if we should also include the actual menu; default to false
+			$include_items= (isset($params['include_menu_items']) && ($params['include_menu_items'] == "true")) ? true : false;
             
-            $i = 0;
-            $rest_menus = array();
-            foreach ( $wp_menus as $wp_menu ) :
+			$i = 0;
+			$rest_menus = array();
+			foreach ( $wp_menus as $wp_menu ) :
 
-                $rest_menus[ $i ] = array_merge((array)$wp_menu,(new WP_REST_Menus)->get_menu(['id'=>$wp_menu->term_id],$include_items));
-                $i ++;
-            endforeach;
+				$rest_menus[ $i ] = array_merge((array)$wp_menu,(new WP_REST_Menus)->get_menu(['id'=>$wp_menu->term_id],$include_items));
+				$i ++;
+			endforeach;
 
-            return apply_filters( 'rest_menus_format_menus', $rest_menus );
-        }
+			return apply_filters( 'rest_menus_format_menus', $rest_menus );
+		}
 
 
-        /**
-         * Get a menu.
-         *
-         * @since  1.2.0
-         * @param  $request
-         * @return array Menu data
-         */
-        public function get_menu( $request,$include_items=true ) {
+		/**
+		 * Get a menu.
+		 *
+		 * @since  1.2.0
+		 * @param  $request
+		 * @return array Menu data
+		 */
+		public function get_menu( $request,$include_items=true ) {
 
-            $id             = (int) $request['id'];
-            $rest_url       = get_rest_url() . self::get_api_namespace() . '/menus/';
-            $wp_menu_object = $id ? wp_get_nav_menu_object( $id ) : array();            
+			$id             = (int) $request['id'];
+			$rest_url       = get_rest_url() . self::get_api_namespace() . '/menus/';
+			$wp_menu_object = $id ? wp_get_nav_menu_object( $id ) : array();            
 
-            $rest_menu = array();
+			$rest_menu = array();
 
-            if ( $wp_menu_object ) :
+			if ( $wp_menu_object ) :
 
-                $menu = (array) $wp_menu_object;
-                $rest_menu['ID']          = abs( $menu['term_id'] );
-                $rest_menu['name']        = $menu['name'];
-                $rest_menu['slug']        = $menu['slug'];
-                $rest_menu['description'] = $menu['description'];
-                $rest_menu['count']       = abs( $menu['count'] );
+				$menu = (array) $wp_menu_object;
+				$rest_menu['ID']          = abs( $menu['term_id'] );
+				$rest_menu['name']        = $menu['name'];
+				$rest_menu['slug']        = $menu['slug'];
+				$rest_menu['description'] = $menu['description'];
+				$rest_menu['count']       = abs( $menu['count'] );
 
 
-                if($include_items){
-                    $wp_menu_items  = $id ? wp_get_nav_menu_items( $id ) : array();
+				if($include_items){
+					$wp_menu_items  = $id ? wp_get_nav_menu_items( $id ) : array();
 
-                    $rest_menu_items = array();
-                    foreach ( $wp_menu_items as $item_object ) {
+					$rest_menu_items = array();
+					foreach ( $wp_menu_items as $item_object ) {
                     
-	                    $rest_menu_items[] = $this->format_menu_item( $item_object );
-                    }
+						$rest_menu_items[] = $this->format_menu_item( $item_object );
+					}
 
                 
-                    $rest_menu_items = $this->nested_menu_items($rest_menu_items, 0);
+					$rest_menu_items = $this->nested_menu_items($rest_menu_items, 0);
+
+					$rest_menu['items']                       = $rest_menu_items;
+				}
+
+				// wp_die(print_r($rest_menu));
+				$rest_menu['meta']['links']['collection'] = $rest_url;
+				$rest_menu['meta']['links']['self']       = $rest_url . $id;
+
+			endif;
+
+			return apply_filters( 'rest_menus_format_menu', $rest_menu );
+		}
+
+
+		/**
+		 * Handle nested menu items.
+		 *
+		 * Given a flat array of menu items, split them into parent/child items
+		 * and recurse over them to return children nested in their parent.
+		 *
+		 * @since  1.2.0
+		 * @param  $menu_items
+		 * @param  $parent
+		 * @return array
+		 */
+		private function nested_menu_items( &$menu_items, $parent = null ) {
+
+			$parents = array();
+			$children = array();
+
+			// Separate menu_items into parents & children.
+			array_map( function( $i ) use ( $parent, &$children, &$parents ){
+				if ( $i['id'] != $parent && $i['parent'] == $parent ) {
+					$parents[] = $i;
+				} else {
+					$children[] = $i;
+				}
+			}, $menu_items );
+
+			foreach ( $parents as &$parent ) {
+
+				if ( $this->has_children( $children, $parent['id'] ) ) {
+					$parent['children'] = $this->nested_menu_items( $children, $parent['id'] );
+				}
+			}
+
+			return $parents;
+		}
+
+
+		/**
+		 * Check if a collection of menu items contains an item that is the parent id of 'id'.
+		 *
+		 * @since  1.2.0
+		 * @param  array $items
+		 * @param  int $id
+		 * @return array
+		 */
+		private function has_children( $items, $id ) {
+			return array_filter( $items, function( $i ) use ( $id ) {
+				return $i['parent'] == $id;
+			} );
+		}
+
+
+		/**
+		 * Get menu locations.
+		 *
+		 * @since 1.2.0
+		 * @param  $request
+		 * @return array All registered menus locations
+		 */
+		public static function get_menu_locations( $request ) {
+
+			$locations        = get_nav_menu_locations();
+			$registered_menus = get_registered_nav_menus();
+			$rest_url         = get_rest_url() . self::get_api_namespace() . '/menu-locations/';
+			$rest_menus       = array();
+
+			if ( $locations && $registered_menus ) :
+
+				foreach ( $registered_menus as $slug => $label ) :
+
+					// Sanity check
+					if ( ! isset( $locations[ $slug ] ) ) {
+						continue;
+					}
+
+					$rest_menus[ $slug ]['ID']                          = $locations[ $slug ];
+					$rest_menus[ $slug ]['label']                       = $label;
+					$rest_menus[ $slug ]['meta']['links']['collection'] = $rest_url;
+					$rest_menus[ $slug ]['meta']['links']['self']       = $rest_url . $slug;
+
+				endforeach;
 
-                    $rest_menu['items']                       = $rest_menu_items;
-                }
-
-                // wp_die(print_r($rest_menu));
-                $rest_menu['meta']['links']['collection'] = $rest_url;
-                $rest_menu['meta']['links']['self']       = $rest_url . $id;
-
-            endif;
-
-            return apply_filters( 'rest_menus_format_menu', $rest_menu );
-        }
-
-
-        /**
-         * Handle nested menu items.
-         *
-         * Given a flat array of menu items, split them into parent/child items
-         * and recurse over them to return children nested in their parent.
-         *
-         * @since  1.2.0
-         * @param  $menu_items
-         * @param  $parent
-         * @return array
-         */
-        private function nested_menu_items( &$menu_items, $parent = null ) {
-
-            $parents = array();
-            $children = array();
-
-            // Separate menu_items into parents & children.
-            array_map( function( $i ) use ( $parent, &$children, &$parents ){
-                if ( $i['id'] != $parent && $i['parent'] == $parent ) {
-                    $parents[] = $i;
-                } else {
-                    $children[] = $i;
-                }
-            }, $menu_items );
-
-            foreach ( $parents as &$parent ) {
-
-                if ( $this->has_children( $children, $parent['id'] ) ) {
-                    $parent['children'] = $this->nested_menu_items( $children, $parent['id'] );
-                }
-            }
-
-            return $parents;
-        }
-
-
-        /**
-         * Check if a collection of menu items contains an item that is the parent id of 'id'.
-         *
-         * @since  1.2.0
-         * @param  array $items
-         * @param  int $id
-         * @return array
-         */
-        private function has_children( $items, $id ) {
-            return array_filter( $items, function( $i ) use ( $id ) {
-                return $i['parent'] == $id;
-            } );
-        }
-
-
-        /**
-         * Get menu locations.
-         *
-         * @since 1.2.0
-         * @param  $request
-         * @return array All registered menus locations
-         */
-        public static function get_menu_locations( $request ) {
-
-            $locations        = get_nav_menu_locations();
-            $registered_menus = get_registered_nav_menus();
-	        $rest_url         = get_rest_url() . self::get_api_namespace() . '/menu-locations/';
-            $rest_menus       = array();
-
-            if ( $locations && $registered_menus ) :
-
-                foreach ( $registered_menus as $slug => $label ) :
-
-	                // Sanity check
-	                if ( ! isset( $locations[ $slug ] ) ) {
-		                continue;
-	                }
-
-	                $rest_menus[ $slug ]['ID']                          = $locations[ $slug ];
-                    $rest_menus[ $slug ]['label']                       = $label;
-                    $rest_menus[ $slug ]['meta']['links']['collection'] = $rest_url;
-                    $rest_menus[ $slug ]['meta']['links']['self']       = $rest_url . $slug;
-
-                endforeach;
-
-            endif;
-
-            return $rest_menus;
-        }
-
-
-        /**
-         * Get menu for location.
-         *
-         * @since 1.2.0
-         * @param  $request
-         * @return array The menu for the corresponding location
-         */
-        public function get_menu_location( $request ) {
-
-            $params     = $request->get_params();
-            $location   = $params['location'];
-            $locations  = get_nav_menu_locations();
-
-            if ( ! isset( $locations[ $location ] ) ) {
-	            return array();
-            }
-
-            $wp_menu = wp_get_nav_menu_object( $locations[ $location ] );
-            $menu_items = wp_get_nav_menu_items( $wp_menu->term_id );
+			endif;
+
+			return $rest_menus;
+		}
+
+
+		/**
+		 * Get menu for location.
+		 *
+		 * @since 1.2.0
+		 * @param  $request
+		 * @return array The menu for the corresponding location
+		 */
+		public function get_menu_location( $request ) {
+
+			$params     = $request->get_params();
+			$location   = $params['location'];
+			$locations  = get_nav_menu_locations();
+
+			if ( ! isset( $locations[ $location ] ) ) {
+				return array();
+			}
+
+			$wp_menu = wp_get_nav_menu_object( $locations[ $location ] );
+			$menu_items = wp_get_nav_menu_items( $wp_menu->term_id );
 			/**
 			 * wp_get_nav_menu_items() outputs a list that's already sequenced correctly.
 			 * So the easiest thing to do is to reverse the list and then build our tree
@@ -323,7 +323,7 @@ 
 					'object_id'   => abs( $item->object_id ),
 					'object'      => $item->object,
 					'type'        => $item->type,
-                    'type_label'  => $item->type_label,
+					'type_label'  => $item->type_label,
 					'children'    => array(),
 				);
 
@@ -331,7 +331,7 @@ 
 					$formatted['children'] = array_reverse( $cache[ $item->ID ] );
 				}
 
-            	$formatted = apply_filters( 'rest_menus_format_menu_item', $formatted );
+				$formatted = apply_filters( 'rest_menus_format_menu_item', $formatted );
 
 				if ( $item->menu_item_parent != 0 ) {
 
@@ -349,82 +349,82 @@ 
 			endforeach;
 
 			return array_reverse ( $rev_menu );
-        }
-
-
-        /**
-         * Returns all child nav_menu_items under a specific parent.
-         *
-         * @since   1.2.0
-         * @param int   $parent_id      The parent nav_menu_item ID
-         * @param array $nav_menu_items Navigation menu items
-         * @param bool  $depth          Gives all children or direct children only
-         * @return array	returns filtered array of nav_menu_items
-         */
-        public function get_nav_menu_item_children( $parent_id, $nav_menu_items, $depth = true ) {
-
-            $nav_menu_item_list = array();
-
-            foreach ( (array) $nav_menu_items as $nav_menu_item ) :
-
-                if ( $nav_menu_item->menu_item_parent == $parent_id ) :
-
-                    $nav_menu_item_list[] = $this->format_menu_item( $nav_menu_item, true, $nav_menu_items );
-
-                    if ( $depth ) {
-                        if ( $children = $this->get_nav_menu_item_children( $nav_menu_item->ID, $nav_menu_items ) ) {
-                            $nav_menu_item_list = array_merge( $nav_menu_item_list, $children );
-                        }
-                    }
-
-                endif;
-
-            endforeach;
-
-            return $nav_menu_item_list;
-        }
-
-
-        /**
-         * Format a menu item for REST API consumption.
-         *
-         * @since  1.2.0
-         * @param  object|array $menu_item  The menu item
-         * @param  bool         $children   Get menu item children (default false)
-         * @param  array        $menu       The menu the item belongs to (used when $children is set to true)
-         * @return array	a formatted menu item for REST
-         */
-        public function format_menu_item( $menu_item, $children = false, $menu = array() ) {
-
-            $item = (array) $menu_item;
-            $id = (int)$item['ID'];
-            $menu_item = array(
-                'id'          => abs( $item['ID'] ),
-                'order'       => (int) $item['menu_order'],
-                'parent'      => abs( $item['menu_item_parent'] ),
-                'title'       => $item['title'],
-                'url'         => $item['url'],
-                'attr'        => $item['attr_title'],
-                'target'      => $item['target'],
-                'classes'     => implode( ' ', $item['classes'] ),
-                'xfn'         => $item['xfn'],
-                'description' => $item['description'],
-                'object_id'   => abs( $item['object_id'] ),
-                'object'      => $item['object'],
-                'object_slug' => $item['post_name'],
-                'type'        => $item['type'],
-                'type_label'  => $item['type_label']
-            );
-
-            if ( $children === true && ! empty( $menu ) ) {
-	            $menu_item['children'] = $this->get_nav_menu_item_children( $item['ID'], $menu );
-            }
-
-            return apply_filters( 'rest_menus_format_menu_item', $menu_item );
-        }
-
-
-    }
+		}
+
+
+		/**
+		 * Returns all child nav_menu_items under a specific parent.
+		 *
+		 * @since   1.2.0
+		 * @param int   $parent_id      The parent nav_menu_item ID
+		 * @param array $nav_menu_items Navigation menu items
+		 * @param bool  $depth          Gives all children or direct children only
+		 * @return array	returns filtered array of nav_menu_items
+		 */
+		public function get_nav_menu_item_children( $parent_id, $nav_menu_items, $depth = true ) {
+
+			$nav_menu_item_list = array();
+
+			foreach ( (array) $nav_menu_items as $nav_menu_item ) :
+
+				if ( $nav_menu_item->menu_item_parent == $parent_id ) :
+
+					$nav_menu_item_list[] = $this->format_menu_item( $nav_menu_item, true, $nav_menu_items );
+
+					if ( $depth ) {
+						if ( $children = $this->get_nav_menu_item_children( $nav_menu_item->ID, $nav_menu_items ) ) {
+							$nav_menu_item_list = array_merge( $nav_menu_item_list, $children );
+						}
+					}
+
+				endif;
+
+			endforeach;
+
+			return $nav_menu_item_list;
+		}
+
+
+		/**
+		 * Format a menu item for REST API consumption.
+		 *
+		 * @since  1.2.0
+		 * @param  object|array $menu_item  The menu item
+		 * @param  bool         $children   Get menu item children (default false)
+		 * @param  array        $menu       The menu the item belongs to (used when $children is set to true)
+		 * @return array	a formatted menu item for REST
+		 */
+		public function format_menu_item( $menu_item, $children = false, $menu = array() ) {
+
+			$item = (array) $menu_item;
+			$id = (int)$item['ID'];
+			$menu_item = array(
+				'id'          => abs( $item['ID'] ),
+				'order'       => (int) $item['menu_order'],
+				'parent'      => abs( $item['menu_item_parent'] ),
+				'title'       => $item['title'],
+				'url'         => $item['url'],
+				'attr'        => $item['attr_title'],
+				'target'      => $item['target'],
+				'classes'     => implode( ' ', $item['classes'] ),
+				'xfn'         => $item['xfn'],
+				'description' => $item['description'],
+				'object_id'   => abs( $item['object_id'] ),
+				'object'      => $item['object'],
+				'object_slug' => $item['post_name'],
+				'type'        => $item['type'],
+				'type_label'  => $item['type_label']
+			);
+
+			if ( $children === true && ! empty( $menu ) ) {
+				$menu_item['children'] = $this->get_nav_menu_item_children( $item['ID'], $menu );
+			}
+
+			return apply_filters( 'rest_menus_format_menu_item', $menu_item );
+		}
+
+
+	}
 
 
 endif;