Merge pull request #945 from newfold-labs/add/ai-base-url-helper-methods

Add configurable API URL support for ContentGeneration service

Author: zlateska

includes/Services/Ai/ContentGeneration/ContentGenerationServiceRequest.php

@@ -16,12 +16,22 @@
  */
 class ContentGenerationServiceRequest {
 
+	/**
+	 * Default production base URL
+	 */
+	const DEFAULT_PRODUCTION_BASE_URL = 'https://patterns.hiive.cloud';
+
+	/**
+	 * Default local base URL
+	 */
+	const DEFAULT_LOCAL_BASE_URL = 'http://localhost:8888';
+
 	/**
 	 * API URL
 	 *
 	 * @var string
 	 */
-	private $url = 'https://patterns.hiive.cloud/api/v1/content-generation/';
+	private $url;
 
 	/**
 	 * API endpoint
@@ -54,20 +64,37 @@ class ContentGenerationServiceRequest {
 	/**
 	 * Constructor
 	 *
-	 * @param array $body    Request body data.
-	 * @param array $headers Additional headers to include in the request.
+	 * @param string $endpoint API endpoint.
+	 * @param array  $body     Request body data.
+	 * @param array  $headers  Additional headers to include in the request.
 	 */
 	public function __construct( string $endpoint, array $body, array $headers = array() ) {
 		$this->endpoint = $endpoint;
-		$this->body = $body;
-		$this->headers = array_merge(
+		$this->body     = $body;
+		$this->url      = $this->get_api_url();
+		$this->headers  = array_merge(
 			array(
 				'Content-Type'  => 'application/json',
 				'Authorization' => 'Bearer ' . HiiveConnection::get_auth_token(),
 			),
 			$headers
 		);
+	}
 
+	/**
+	 * Get the API URL based on configuration
+	 *
+	 * @return string The API URL to use.
+	 */
+	private function get_api_url(): string {
+
+		if ( defined( 'NFD_WB_DEV_MODE' ) && NFD_WB_DEV_MODE ) {
+			$base_url = defined( 'NFD_WB_LOCAL_BASE_URL' ) ? NFD_WB_LOCAL_BASE_URL : self::DEFAULT_LOCAL_BASE_URL;
+		} else {
+			$base_url = defined( 'NFD_WB_PRODUCTION_BASE_URL' ) ? NFD_WB_PRODUCTION_BASE_URL : self::DEFAULT_PRODUCTION_BASE_URL;
+		}
+
+		return $base_url . '/api/v1/content-generation/';
 	}
 
 	/**
@@ -88,7 +115,7 @@ public function send(): ContentGenerationServiceRequest {
 		$this->response = $response;
 		return $this;
 	}
-	
+
 	/**
 	 * Check if the request was successful
 	 *
@@ -98,7 +125,7 @@ public function is_successful(): bool {
 		if ( ! $this->response || is_wp_error( $this->response ) ) {
 			return false;
 		}
-		
+
 		$code = wp_remote_retrieve_response_code( $this->response );
 		return $code >= 200 && $code < 300;
 	}