add capability check again

newfold-labs · Dec 18, 2023 · 3d2d8e5 · 3d2d8e5
1 parent 86d5382
commit 3d2d8e5
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/bootstrap.php b/bootstrap.php
@@ -17,7 +17,7 @@ function () {
 
 			if ( ! defined( 'NFD_AI_SERVICE_BASE' ) ) {
 				define( 'NFD_AI_SERVICE_BASE', 'https://hiive.cloud/workers/ai-proxy/v1/' );
-				define( 'NFD_AI_BASE', 'https://hiive.cloud/workers/ai-proxy/' );
+				define( 'NFD_AI_BASE', 'http://localhost:8787/' );
 				define( 'NFD_PATTERNS_BASE', 'https://patterns.hiive.cloud/' );
 				define( 'NFD_SITEGEN_OPTION', 'nfd-ai-site-gen' );
 			}

diff --git a/includes/SiteGen/SiteGen.php b/includes/SiteGen/SiteGen.php
@@ -342,6 +342,11 @@ public static function generate_site_meta( $site_info, $identifier, $skip_cache
 	 * @param boolean $regenerate       If we need to regenerate.
 	 */
 	public static function get_home_pages( $site_description, $content_style, $target_audience, $regenerate = false ) {
+		if ( ! self::check_capabilities() ) {
+			return array(
+				'error' => __( 'You do not have the permissions to perform this action' ),
+			);
+		}
 
 		// Check if we have the response in cache already
 		if ( ! $regenerate ) {