Add Capability Check

newfold-labs · Jan 15, 2024 · 88274c9 · 88274c9
1 parent 048b3c4
commit 88274c9
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 5 deletions.
diff --git a/includes/Brands.php b/includes/Brands.php
@@ -103,7 +103,7 @@ public static function get_brands() {
 					'enabled_flows'  => array(
 						'ecommerce' => true,
 						'wp-setup'  => true,
-						'sitegen'   => true,
+						'sitegen'   => Config::has_ai_sitegen(),
 					),
 					'wonder_blocks'  => true,
 					'prioritization' => false,
@@ -166,7 +166,7 @@ public static function get_brands() {
 					'enabled_flows'  => array(
 						'ecommerce' => true,
 						'wp-setup'  => false,
-						'sitegen'   => false,
+						'sitegen'   => Config::has_ai_sitegen(),
 					),
 					'wonder_blocks'  => true,
 					'prioritization' => false,
@@ -227,7 +227,7 @@ public static function get_brands() {
 					'enabled_flows'  => array(
 						'ecommerce' => false,
 						'wp-setup'  => false,
-						'sitegen'   => false,
+						'sitegen'   => Config::has_ai_sitegen(),
 					),
 					'wonder_blocks'  => true,
 					'prioritization' => false,
@@ -288,7 +288,7 @@ public static function get_brands() {
 					'enabled_flows'  => array(
 						'ecommerce' => true,
 						'wp-setup'  => true,
-						'sitegen'   => false,
+						'sitegen'   => Config::has_ai_sitegen(),
 					),
 					'wonder_blocks'  => true,
 					'prioritization' => false,
@@ -356,7 +356,7 @@ public static function get_brands() {
 					'enabled_flows'  => array(
 						'ecommerce' => true,
 						'wp-setup'  => true,
-						'sitegen'   => false,
+						'sitegen'   => Config::has_ai_sitegen(),
 					),
 					'wonder_blocks'  => true,
 					'prioritization' => false,

diff --git a/includes/Config.php b/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' );
+	}
 }