Sitegen flow related data

composer.json

@@ -23,8 +23,11 @@
     },
     "require": {
         "newfold-labs/wp-module-installer": "^1.1",
-        "newfold-labs/wp-module-data": "^2.4.16",
+        "newfold-labs/wp-module-patterns": "^0.1.12",
+        "newfold-labs/wp-module-ai": "^1.0.4",
         "wp-forge/wp-upgrade-handler": "^1.0",
+        "mustache/mustache": "^2.14",
+        "newfold-labs/wp-module-data": "^2.4.16",
         "newfold-labs/wp-module-coming-soon": "^1.1.13"
     },
     "require-dev": {
@@ -35,4 +38,4 @@
             "dealerdirect/phpcodesniffer-composer-installer": true
         }
     }
-}
+}

includes/Brands.php

@@ -37,6 +37,9 @@ public static function get_default_brand() {
 	 */
 	public static function get_brands() {
 
+		// Checks if customer has acess to AI Sitegen
+		$has_ai_sitegen = Config::has_ai_sitegen();
+
 		return array(
 			'bluehost'       => array(
 				'brand'                       => 'bluehost',
@@ -103,6 +106,7 @@ public static function get_brands() {
 					'enabled_flows'  => array(
 						'ecommerce' => true,
 						'wp-setup'  => true,
+						'sitegen'   => $has_ai_sitegen,
 					),
 					'wonder_blocks'  => true,
 					'prioritization' => false,
@@ -165,6 +169,7 @@ public static function get_brands() {
 					'enabled_flows'  => array(
 						'ecommerce' => true,
 						'wp-setup'  => false,
+						'sitegen'   => $has_ai_sitegen,
 					),
 					'wonder_blocks'  => true,
 					'prioritization' => false,
@@ -225,6 +230,7 @@ public static function get_brands() {
 					'enabled_flows'  => array(
 						'ecommerce' => false,
 						'wp-setup'  => false,
+						'sitegen'   => $has_ai_sitegen,
 					),
 					'wonder_blocks'  => true,
 					'prioritization' => false,
@@ -285,6 +291,7 @@ public static function get_brands() {
 					'enabled_flows'  => array(
 						'ecommerce' => true,
 						'wp-setup'  => true,
+						'sitegen'   => $has_ai_sitegen,
 					),
 					'wonder_blocks'  => true,
 					'prioritization' => false,
@@ -352,6 +359,7 @@ public static function get_brands() {
 					'enabled_flows'  => array(
 						'ecommerce' => true,
 						'wp-setup'  => true,
+						'sitegen'   => $has_ai_sitegen,
 					),
 					'wonder_blocks'  => true,
 					'prioritization' => false,
@@ -437,6 +445,7 @@ public static function get_brands() {
 					'enabled_flows'  => array(
 						'ecommerce' => true,
 						'wp-setup'  => true,
+						'sitegen'   => $has_ai_sitegen,
 					),
 					'wonder_blocks'  => true,
 					'prioritization' => false,

includes/Config.php

@@ -48,4 +48,13 @@ public static function get_site_capability( $capability ) {
 	public static function is_jarvis() {
 		return self::get_site_capability( 'isJarvis' );
 	}
+
+	/**
+	 * Gets the current customer capability if he has access to AI Sitegen.
+	 *
+	 * @return boolean
+	 */
+	public static function has_ai_sitegen() {
+		return self::get_site_capability( 'hasAISiteGen' );
+	}
 }

includes/Data.php

@@ -16,18 +16,23 @@ final class Data {
 	 */
 	public static function runtime() {
 		return array(
-			'buildUrl'          => \NFD_ONBOARDING_BUILD_URL,
-			'siteUrl'           => \get_site_url(),
-			'restUrl'           => \get_home_url() . '/index.php?rest_route=',
-			'adminUrl'          => \admin_url(),
-			'currentBrand'      => self::current_brand(),
-			'currentPlan'       => self::current_plan(),
-			'currentFlow'       => self::current_flow(),
-			'pluginInstallHash' => PluginInstaller::rest_get_plugin_install_hash(),
-			'previewSettings'   => array(
+			'buildUrl'           => \NFD_ONBOARDING_BUILD_URL,
+			'siteUrl'            => \get_site_url(),
+			'restUrl'            => \get_home_url() . '/index.php?rest_route=',
+			'adminUrl'           => \admin_url(),
+			'currentBrand'       => self::current_brand(),
+			'currentPlan'        => self::current_plan(),
+			'currentFlow'        => self::current_flow(),
+			'pluginInstallHash'  => PluginInstaller::rest_get_plugin_install_hash(),
+			'previewSettings'    => array(
 				'settings'        => Preview::get_settings(),
 				'stepPreviewData' => Themes::step_preview_data(),
 			),
+			'aiPreviewSettings'  => array(
+				'settings'        => Preview::get_settings(),
+				'stepPreviewData' => Themes::step_preview_data(),
+			),
+			'currentUserDetails' => self::wp_current_user_details(),
 		);
 	}
 
@@ -51,6 +56,15 @@ public static function current_brand() {
 	 * @return array
 	 */
 	public static function current_plan() {
+		$is_sitegen = Flows::is_sitegen();
+		if ( $is_sitegen ) {
+			return array(
+				'flow'    => 'sitegen',
+				'subtype' => null,
+				'type'    => null,
+			);
+		}
+
 		$customer_data = self::customer_data();
 
 		$current_flow = Flows::get_flow_from_customer_data( $customer_data );
@@ -140,4 +154,25 @@ public static function coming_soon() {
 		$coming_soon_service = container()->get( 'comingSoon' );
 		return $coming_soon_service->is_enabled();
 	}
+
+	/**
+	 * Get the current WordPress admin user details.
+	 *
+	 * @return array
+	 */
+	public static function wp_current_user_details() {
+		$user = wp_get_current_user();
+		if ( $user->exists() ) {
+			return array(
+				'displayName' => $user->display_name,
+				'avatarUrl'   => get_avatar_url( $user->ID ),
+			);
+		}
+
+		// If no user is found, return an empty array or default values as appropriate
+		return array(
+			'displayName' => '',
+			'avatarUrl'   => '',
+		);
+	}
 }

includes/Flows/Flows.php

@@ -16,7 +16,7 @@ final class Flows {
 	 * @var array
 	 */
 	protected static $data = array(
-		'version'              => '1.0.3',
+		'version'              => '2.0.0',
 
 		// Each time step is viewed, insert GMT timestamp to array.
 		'isViewed'             => array(),
@@ -67,8 +67,10 @@ final class Flows {
 
 			// This integer will map to the attachment ID for an uploaded image to the WordPress media library
 			'siteLogo'        => array(
-				'id'  => 0,
-				'url' => '',
+				'id'       => 0,
+				'url'      => '',
+				'fileName' => '',
+				'fileSize' => 0,
 			),
 
 			// key-value store for social media accounts
@@ -110,6 +112,8 @@ final class Flows {
 			'comingSoon'      => false,
 		),
 
+		'activeFlow'           => '',
+
 		// we will store active flows (abandoned wp-setup, abandoned wp-commerce) with their identifier and use as a reference to access currentStep and data
 		'currentFlows'         => array(),
 
@@ -132,6 +136,37 @@ final class Flows {
 				'product_types' => array(),
 			),
 		),
+
+		'sitegen'              => array(
+			'siteDetails'           => array(
+				'name'                => '',
+				'type'                => '',
+				'style'               => '',
+				'prompt'              => '',
+				'uniqueAboutBusiness' => '',
+				'mode'                => 'simple',
+			),
+			'siteLogo'              => array(
+				'id'       => 0,
+				'url'      => '',
+				'fileName' => '',
+				'fileSize' => 0,
+			),
+			'experience'            => array(
+				'level' => 0,
+			),
+			'siteGenMetaStatus'     => array(
+				'currentStatus' => 0,
+				'totalCount'    => 8,
+			),
+			'homepages'             => array(
+				'active' => array(),
+				'data'   => array(),
+			),
+			'skipCache'             => true,
+			'sitemapPagesGenerated' => false,
+			'customDesign'          => false,
+		),
 	);
 
 	/**
@@ -184,6 +219,7 @@ public static function get_flows() {
 		? $current_brand['config']['enabled_flows'] : array(
 			'wp-setup'  => false,
 			'ecommerce' => false,
+			'sitegen'   => false,
 		);
 	}
 
@@ -273,6 +309,25 @@ public static function get_flow_from_plan_subtype( $plan_subtype ) {
 		}
 		return false;
 	}
+
+	/**
+	 * Determines whether SiteGen flow was ever visited.
+	 *
+	 * @return boolean
+	 */
+	public static function is_sitegen() {
+		if ( ! self::get_flows()['sitegen'] ) {
+			return false;
+		}
+
+		$flow_data = FlowService::read_data_from_wp_option();
+		if ( ! $flow_data || empty( $flow_data['activeFlow'] ) ) {
+			return false;
+		}
+
+		return 'sitegen' === $flow_data['activeFlow'];
+	}
+
 	/**
 	 * Get the corresponding flow from the top priority in flow data.
 	 *

includes/Mustache/Mustache.php

@@ -0,0 +1,37 @@
+<?php
+namespace NewfoldLabs\WP\Module\Onboarding\Data\Mustache;
+
+/**
+ * Class Mustache
+ */
+class Mustache {
+
+	/**
+	 * Mustache Engine.
+	 *
+	 * @var array
+	 */
+	protected $mustache_engine;
+
+	/**
+	 * Setup mustache engine.
+	 */
+	public function __construct() {
+		$this->mustache_engine = new \Mustache_Engine(
+			array(
+				'loader' => new \Mustache_Loader_FilesystemLoader( __DIR__ . '/Templates' ),
+			)
+		);
+	}
+
+	/**
+	 * Render respective template data.
+	 *
+	 * @param string $template_name Template Name
+	 * @param array  $data Data
+	 * @return string
+	 */
+	public function render_template( $template_name, $data ) {
+		return $this->mustache_engine->loadTemplate( $template_name )->render( $data );
+	}
+}

includes/Mustache/Templates/themeStylesheet.mustache

@@ -0,0 +1,9 @@
+/*
+Theme Name: {{ theme_name }}
+Author: {{ site_title }} & {{ brand_name }}
+Author URI: {{ site_url }}
+Description: A Custom WordPress Theme built for {{ site_title }} by {{ brand_name }} 
+Version: 1.0.0
+Template: {{ parent_theme_slug }}
+Text Domain: {{ child_theme_slug }}
+*/

includes/Options.php

@@ -55,6 +55,7 @@ final class Options {
 		'wc_calc_taxes'                 => 'woocommerce_calc_taxes',
 		'wc_no_sales_tax'               => 'woocommerce_no_sales_tax',
 		'filter_active_plugins'         => 'filter_active_plugins',
+		'sitegen_regenerated_homepages' => 'sitegen_regenerated_homepages',
 		'status'                        => 'status',
 	);
 

includes/Plugins.php

@@ -1,6 +1,10 @@
 <?php
 namespace NewfoldLabs\WP\Module\Onboarding\Data;
 
+use NewfoldLabs\WP\Module\Installer\Data\Plugins as PluginsInstaller;
+
+use function NewfoldLabs\WP\ModuleLoader\container;
+
 /**
  * List of Plugin Slugs/URLs/Domains
  */
@@ -232,4 +236,16 @@ public static function get_init() {
 	public static function wc_prevent_redirect_on_activation() {
 		\delete_transient( '_wc_activation_redirect' );
 	}
+
+	/**
+	 * List of plugins that should stay active even with the filter option
+	 *
+	 * @return array
+	 */
+	public static function get_active_plugins_list() {
+		return array(
+			container()->plugin()->basename,
+			isset( PluginsInstaller::get_wp_slugs()['woocommerce']['path'] ) ? PluginsInstaller::get_wp_slugs()['woocommerce']['path'] : false,
+		);
+	}
 }

includes/Services/FlowService.php

@@ -390,7 +390,7 @@ private static function update_data_for_ecommerce( $data ) {
 	 *
 	 * @return array
 	 */
-	private static function update_data_in_wp_option( $data ) {
+	public static function update_data_in_wp_option( $data ) {
 		return \update_option( Options::get_option_name( 'flow' ), $data );
 	}