[Summary view] [Print] [Text view]

   1  <?php
   2  /**
   3   * WP AI Client: WP_AI_Client_HTTP_Client class
   4   *
   5   * @package WordPress
   6   * @subpackage AI
   7   * @since 7.0.0
   8   */
   9  
  10  use WordPress\AiClientDependencies\Psr\Http\Client\ClientInterface;
  11  use WordPress\AiClientDependencies\Psr\Http\Message\RequestInterface;
  12  use WordPress\AiClientDependencies\Psr\Http\Message\ResponseInterface;
  13  use WordPress\AiClientDependencies\Psr\Http\Message\ResponseFactoryInterface;
  14  use WordPress\AiClientDependencies\Psr\Http\Message\StreamFactoryInterface;
  15  use WordPress\AiClient\Providers\Http\Contracts\ClientWithOptionsInterface;
  16  use WordPress\AiClient\Providers\Http\DTO\RequestOptions;
  17  use WordPress\AiClient\Providers\Http\Exception\NetworkException;
  18  
  19  /**
  20   * PSR-18 HTTP Client adapter using WordPress HTTP API.
  21   *
  22   * Allows WordPress HTTP functions to be used as a PSR-18 compliant HTTP client
  23   * for the AI Client SDK.
  24   *
  25   * @since 7.0.0
  26   * @internal Intended only to wire up the PHP AI Client SDK to WordPress's HTTP client.
  27   * @access private
  28   */
  29  class WP_AI_Client_HTTP_Client implements ClientInterface, ClientWithOptionsInterface {
  30  
  31      /**
  32       * Response factory instance.
  33       *
  34       * @since 7.0.0
  35       */
  36      private ResponseFactoryInterface $response_factory;
  37  
  38      /**
  39       * Stream factory instance.
  40       *
  41       * @since 7.0.0
  42       */
  43      private StreamFactoryInterface $stream_factory;
  44  
  45      /**
  46       * Constructor.
  47       *
  48       * @since 7.0.0
  49       *
  50       * @param ResponseFactoryInterface $response_factory PSR-17 Response factory.
  51       * @param StreamFactoryInterface   $stream_factory   PSR-17 Stream factory.
  52       */
  53  	public function __construct( ResponseFactoryInterface $response_factory, StreamFactoryInterface $stream_factory ) {
  54          $this->response_factory = $response_factory;
  55          $this->stream_factory   = $stream_factory;
  56      }
  57  
  58      /**
  59       * Sends a PSR-7 request and returns a PSR-7 response.
  60       *
  61       * @since 7.0.0
  62       *
  63       * @param RequestInterface $request The PSR-7 request.
  64       * @return ResponseInterface The PSR-7 response.
  65       *
  66       * @throws NetworkException If the WordPress HTTP request fails.
  67       */
  68  	public function sendRequest( RequestInterface $request ): ResponseInterface {
  69          $args = $this->prepare_wp_args( $request );
  70          $url  = (string) $request->getUri();
  71  
  72          $response = wp_safe_remote_request( $url, $args );
  73  
  74          if ( is_wp_error( $response ) ) {
  75              $message = sprintf(
  76                  /* translators: 1: HTTP method (e.g. GET, POST). 2: Request URL. 3: Error message. */
  77                  __( 'Network error occurred while sending %1$s request to %2$s: %3$s' ),
  78                  $request->getMethod(),
  79                  $url,
  80                  $response->get_error_message()
  81              );
  82              throw new NetworkException( $message ); // phpcs:ignore WordPress.Security.EscapeOutput.ExceptionNotEscaped
  83          }
  84  
  85          return $this->create_psr_response( $response );
  86      }
  87  
  88      /**
  89       * Sends a PSR-7 request with transport options and returns a PSR-7 response.
  90       *
  91       * @since 7.0.0
  92       *
  93       * @param RequestInterface $request The PSR-7 request.
  94       * @param RequestOptions   $options Transport options for the request.
  95       * @return ResponseInterface The PSR-7 response.
  96       *
  97       * @throws NetworkException If the WordPress HTTP request fails.
  98       */
  99  	public function sendRequestWithOptions( RequestInterface $request, RequestOptions $options ): ResponseInterface {
 100          $args = $this->prepare_wp_args( $request, $options );
 101          $url  = (string) $request->getUri();
 102  
 103          $response = wp_safe_remote_request( $url, $args );
 104  
 105          if ( is_wp_error( $response ) ) {
 106              $message = sprintf(
 107                  /* translators: 1: HTTP method (e.g. GET, POST). 2: Request URL. 3: Error message. */
 108                  __( 'Network error occurred while sending %1$s request to %2$s: %3$s' ),
 109                  $request->getMethod(),
 110                  $url,
 111                  $response->get_error_message()
 112              );
 113  
 114              throw new NetworkException(
 115                  $message, // phpcs:ignore WordPress.Security.EscapeOutput.ExceptionNotEscaped
 116                  $response->get_error_code() ? (int) $response->get_error_code() : 0
 117              );
 118          }
 119  
 120          return $this->create_psr_response( $response );
 121      }
 122  
 123      /**
 124       * Prepares WordPress HTTP API arguments from a PSR-7 request.
 125       *
 126       * @since 7.0.0
 127       *
 128       * @param RequestInterface    $request The PSR-7 request.
 129       * @param RequestOptions|null $options Optional transport options for the request.
 130       * @return array<string, mixed> WordPress HTTP API arguments.
 131       */
 132  	private function prepare_wp_args( RequestInterface $request, ?RequestOptions $options = null ): array {
 133          $args = array(
 134              'method'      => $request->getMethod(),
 135              'headers'     => $this->prepare_headers( $request ),
 136              'body'        => $this->prepare_body( $request ),
 137              'httpversion' => $request->getProtocolVersion(),
 138              'blocking'    => true,
 139          );
 140  
 141          if ( null !== $options ) {
 142              if ( null !== $options->getTimeout() ) {
 143                  $args['timeout'] = $options->getTimeout();
 144              }
 145  
 146              if ( null !== $options->getMaxRedirects() ) {
 147                  $args['redirection'] = $options->getMaxRedirects();
 148              }
 149          }
 150  
 151          return $args;
 152      }
 153  
 154      /**
 155       * Prepares headers for WordPress HTTP API.
 156       *
 157       * @since 7.0.0
 158       *
 159       * @param RequestInterface $request The PSR-7 request.
 160       * @return array<string, string> Headers array for WordPress HTTP API.
 161       */
 162  	private function prepare_headers( RequestInterface $request ): array {
 163          $headers = array();
 164  
 165          foreach ( $request->getHeaders() as $name => $values ) {
 166              $headers[ (string) $name ] = implode( ', ', $values );
 167          }
 168  
 169          return $headers;
 170      }
 171  
 172      /**
 173       * Prepares request body for WordPress HTTP API.
 174       *
 175       * @since 7.0.0
 176       *
 177       * @param RequestInterface $request The PSR-7 request.
 178       * @return string|null The request body.
 179       */
 180  	private function prepare_body( RequestInterface $request ): ?string {
 181          $body = $request->getBody();
 182  
 183          if ( $body->getSize() === 0 ) {
 184              return null;
 185          }
 186  
 187          if ( $body->isSeekable() ) {
 188              $body->rewind();
 189          }
 190  
 191          return (string) $body;
 192      }
 193  
 194      /**
 195       * Creates a PSR-7 response from a WordPress HTTP response.
 196       *
 197       * @since 7.0.0
 198       *
 199       * @param array<string, mixed> $wp_response WordPress HTTP API response array.
 200       * @return ResponseInterface PSR-7 response.
 201       */
 202  	private function create_psr_response( array $wp_response ): ResponseInterface {
 203          $status_code   = wp_remote_retrieve_response_code( $wp_response );
 204          $reason_phrase = wp_remote_retrieve_response_message( $wp_response );
 205          $headers       = wp_remote_retrieve_headers( $wp_response );
 206          $body          = wp_remote_retrieve_body( $wp_response );
 207  
 208          $response = $this->response_factory->createResponse( (int) $status_code, $reason_phrase );
 209  
 210          if ( $headers instanceof WP_HTTP_Requests_Response ) {
 211              $headers = $headers->get_headers();
 212          }
 213  
 214          if ( is_array( $headers ) || $headers instanceof Traversable ) {
 215              foreach ( $headers as $name => $value ) {
 216                  $response = $response->withHeader( $name, $value );
 217              }
 218          }
 219  
 220          if ( ! empty( $body ) ) {
 221              $stream   = $this->stream_factory->createStream( $body );
 222              $response = $response->withBody( $stream );
 223          }
 224  
 225          return $response;
 226      }
 227  }