From 43940fc53d9ce55e37803ce8feba289f29c9bc68 Mon Sep 17 00:00:00 2001
From: isvaljek <ivan.svaljek@gmail.com>
Date: Sun, 11 Jun 2017 14:56:46 +0200
Subject: [PATCH] refactor getting object slug into function

---
 includes/wp-api-menus-v2.php | 25 +++++++++++++++----------
 1 file changed, 15 insertions(+), 10 deletions(-)

diff --git a/includes/wp-api-menus-v2.php b/includes/wp-api-menus-v2.php
index 5ae69d4..ec84f2e 100644
--- a/includes/wp-api-menus-v2.php
+++ b/includes/wp-api-menus-v2.php
@@ -369,14 +369,7 @@ public function get_nav_menu_item_children( $parent_id, $nav_menu_items, $depth
         public function format_menu_item( $menu_item, $children = false, $menu = array() ) {
 
             $item = (array) $menu_item;
-			
-			if($item['type'] == 'taxonomy'){
-                $slug = get_term( $item['object_id'] )->slug;
-            }
-            else {
-                $slug = get_post( $item['object_id'] )->post_name;
-            }
-
+						
             $menu_item = array(
                 'id'          => abs( $item['ID'] ),
                 'order'       => (int) $item['menu_order'],
@@ -390,7 +383,7 @@ public function format_menu_item( $menu_item, $children = false, $menu = array()
                 'description' => $item['description'],
                 'object_id'   => abs( $item['object_id'] ),
                 'object'      => $item['object'],
-                'object_slug' => $slug,
+                'object_slug' => get_object_slug($item),
                 'type'        => $item['type'],
                 'type_label'  => $item['type_label'],
             );
@@ -401,7 +394,19 @@ public function format_menu_item( $menu_item, $children = false, $menu = array()
 
             return apply_filters( 'rest_menus_format_menu_item', $menu_item );
         }
-
+	
+		private function get_object_slug($item){
+			$slug = '';
+			
+			if($item['type'] == 'taxonomy'){
+                $slug = get_term( $item['object_id'] )->slug;
+            }
+            else {
+                $slug = get_post( $item['object_id'] )->post_name;
+            }
+			
+			return $slug;
+		}
 
     }