Exit early

WordPress · Nov 12, 2020 · 2bc91ab · 2bc91ab
1 parent ee50757
commit 2bc91ab
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/packages/components/src/navigation/use-create-navigation-tree.js b/packages/components/src/navigation/use-create-navigation-tree.js
@@ -48,20 +48,23 @@ export const useCreateNavigationTree = () => {
 			visited.push( current.menu );
 			queue = [ ...queue, ...getChildMenu( current.menu ) ];
 
-			callback( current );
+			if ( callback( current ) === false ) {
+				break;
+			}
 		}
 	};
 
 	const isMenuEmpty = ( menuToCheck ) => {
-		let count = 0;
+		let isEmpty = true;
 
 		traverseMenu( menuToCheck, ( current ) => {
 			if ( ! current.isEmpty ) {
-				count++;
+				isEmpty = false;
+				return false;
 			}
 		} );
 
-		return count === 0;
+		return isEmpty;
 	};
 
 	return {