From 867e587f36526f7413d7bf951d85ebe8dd88ec7c Mon Sep 17 00:00:00 2001
From: arunshenoy99 <devarunshenoy99@gmail.com>
Date: Wed, 21 Feb 2024 12:57:14 +0530
Subject: [PATCH 1/2] Replace menu items in previews with those from the
 sitemap.

---
 includes/Patterns.php                | 16 +++++++++++++---
 includes/Services/SiteGenService.php | 26 ++++++++++++++++++++++++++
 2 files changed, 39 insertions(+), 3 deletions(-)

diff --git a/includes/Patterns.php b/includes/Patterns.php
index 653195e..a74d0be 100644
--- a/includes/Patterns.php
+++ b/includes/Patterns.php
@@ -1,6 +1,7 @@
 <?php
 namespace NewfoldLabs\WP\Module\Onboarding\Data;
 
+use NewfoldLabs\WP\Module\Onboarding\Data\Services\SiteGenService;
 use NewfoldLabs\WP\Module\Onboarding\Data\Services\WonderBlocksService;
 
 /**
@@ -13,8 +14,8 @@ final class Patterns {
 	 *
 	 * @return array
 	 */
-	public static function get_dummy_menu_items() {
-		return array(
+	public static function get_dummy_navigation_menu_items() {
+		$default_items = array(
 			__( 'Home', 'wp-module-onboarding' ),
 			__( 'About', 'wp-module-onboarding' ),
 			__( 'Contact', 'wp-module-onboarding' ),
@@ -22,6 +23,15 @@ public static function get_dummy_menu_items() {
 			__( 'Privacy', 'wp-module-onboarding' ),
 			__( 'Careers', 'wp-module-onboarding' ),
 		);
+
+		$flow_to_dummy_items = array(
+			'sitegen'   => SiteGenService::get_dummy_navigation_menu_items(),
+			'wp-setup'  => $default_items,
+			'ecommerce' => $default_items,
+		);
+
+		$current_flow = Data::current_flow();
+		return isset( $flow_to_dummy_items[ $current_flow ] ) ? $flow_to_dummy_items[ $current_flow ] : $default_items;
 	}
 
 	/**
@@ -357,7 +367,7 @@ private static function replace_split_menu_items( $pattern_content ) {
 		$dummy_menu_grammar      = '';
 		$menu_navigation_grammar = '<!-- wp:navigation-link {"isTopLevelLink":true} /-->';
 
-		foreach ( self::get_dummy_menu_items() as $item ) {
+		foreach ( self::get_dummy_navigation_menu_items() as $item ) {
 			$dummy_menu_grammar = '<!-- wp:navigation-link {
 				"isTopLevelLink":true, 
 				"label":"' . strtolower( $item ) . '", 
diff --git a/includes/Services/SiteGenService.php b/includes/Services/SiteGenService.php
index ef73fe3..2cd183a 100644
--- a/includes/Services/SiteGenService.php
+++ b/includes/Services/SiteGenService.php
@@ -819,4 +819,30 @@ public static function publish_sitemap_pages( $site_description, $content_style,
 
 		return true;
 	}
+
+	/**
+	 * Get the dummy navigation menu items for the Sitegen previews.
+	 *
+	 * @return array
+	 */
+	public static function get_dummy_navigation_menu_items() {
+		$prompt    = self::get_prompt();
+		$site_info = array( 'site_description' => $prompt );
+		$sitemap   = self::instantiate_site_meta( $site_info, 'sitemap' );
+		if ( is_wp_error( $sitemap ) ) {
+			return $sitemap;
+		}
+
+		$dummy_items = array();
+		foreach ( $sitemap as $page ) {
+			if ( ! isset( $page['title'] ) ) {
+				continue;
+			}
+
+			array_push( $dummy_items, $page['title'] );
+
+		}
+
+		return $dummy_items;
+	}
 }

From b08573f8604021c9e03317d934756744bbdbca09 Mon Sep 17 00:00:00 2001
From: arunshenoy99 <devarunshenoy99@gmail.com>
Date: Wed, 21 Feb 2024 18:57:41 +0530
Subject: [PATCH 2/2] Enhance checks

---
 includes/Patterns.php                | 2 +-
 includes/Services/SiteGenService.php | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/includes/Patterns.php b/includes/Patterns.php
index a74d0be..ba53de5 100644
--- a/includes/Patterns.php
+++ b/includes/Patterns.php
@@ -31,7 +31,7 @@ public static function get_dummy_navigation_menu_items() {
 		);
 
 		$current_flow = Data::current_flow();
-		return isset( $flow_to_dummy_items[ $current_flow ] ) ? $flow_to_dummy_items[ $current_flow ] : $default_items;
+		return ! empty( $flow_to_dummy_items[ $current_flow ] ) ? $flow_to_dummy_items[ $current_flow ] : $default_items;
 	}
 
 	/**
diff --git a/includes/Services/SiteGenService.php b/includes/Services/SiteGenService.php
index 2cd183a..dd620d5 100644
--- a/includes/Services/SiteGenService.php
+++ b/includes/Services/SiteGenService.php
@@ -830,7 +830,7 @@ public static function get_dummy_navigation_menu_items() {
 		$site_info = array( 'site_description' => $prompt );
 		$sitemap   = self::instantiate_site_meta( $site_info, 'sitemap' );
 		if ( is_wp_error( $sitemap ) ) {
-			return $sitemap;
+			return array();
 		}
 
 		$dummy_items = array();