Respects the $keep_alive param for WebFetchApi::fetch()

Signed-off-by: Jon Stovell <jonstovell@gmail.com>

Author: Sesquipedalian

Sources/WebFetch/APIs/CurlFetcher.php

@@ -122,6 +122,13 @@ class CurlFetcher extends WebFetchApi
 	 */
 	public $headers;
 
+	/**
+	 * @var bool
+	 *
+	 * Whether to keep the connection open after the initial request.
+	 */
+	public bool $keep_alive = false;
+
 	/*********************
 	 * Internal properties
 	 *********************/
@@ -192,6 +199,14 @@ public function __construct(array $options = [], int $max_redirect = 3)
 			$this->max_redirect = max(3, $this->max_redirect);
 			$this->user_options[CURLOPT_FOLLOWLOCATION] = false;
 		}
+
+		// Do we want to keep the connection open after the initial request?
+		if (
+			version_compare(curl_version()['version'], '7.25.0', '>=')
+			&& !empty($this->user_options[CURLOPT_TCP_KEEPALIVE])
+		) {
+			$this->keep_alive = true;
+		}
 	}
 
 	/**
@@ -243,6 +258,12 @@ public function request(string|Url $url, array|string $post_data = []): object
 		}
 
 		// Set the options and get it.
+		if (version_compare(curl_version()['version'], '7.25.0', '>=')) {
+			$this->user_options[CURLOPT_TCP_KEEPALIVE] = (int) $this->keep_alive;
+		} else {
+			$this->keep_alive = false;
+		}
+
 		$this->setOptions();
 		$this->sendRequest(str_replace(' ', '%20', strval($url)));
 

Sources/WebFetch/APIs/SocketFetcher.php

@@ -84,16 +84,16 @@ class SocketFetcher extends WebFetchApi
 	 */
 	public $response = [];
 
-	/*********************
-	 * Internal properties
-	 *********************/
-
 	/**
 	 * @var bool
 	 *
 	 * Whether to keep the socket connection open after the initial request.
 	 */
-	private bool $keep_alive;
+	public bool $keep_alive = false;
+
+	/*********************
+	 * Internal properties
+	 *********************/
 
 	/**
 	 * @var bool

Sources/WebFetch/WebFetchApi.php

@@ -61,6 +61,17 @@ abstract class WebFetchApi implements WebFetchApiInterface
 		'https' => ['SocketFetcher', 'CurlFetcher'],
 	];
 
+	/****************************
+	 * Internal static properties
+	 ****************************/
+
+	/**
+	 * @var array
+	 *
+	 * Fetchers that still have an open connection after the initial request.
+	 */
+	private static array $still_alive = [];
+
 	/****************
 	 * Public methods
 	 ****************/
@@ -121,14 +132,35 @@ public static function fetch(Url|string $url, string|array $post_data = [], bool
 			$data = false;
 		}
 
-		foreach (self::$scheme_handlers[$url->scheme] as $class) {
-			$class = __NAMESPACE__ . '\\APIs\\' . $class;
-
-			$fetcher = new $class();
+		if (isset(self::$still_alive[(string) $url])) {
+			$fetcher = self::$still_alive[(string) $url];
 			$fetcher->request($url, $post_data);
+		} else {
+			foreach (self::$scheme_handlers[$url->scheme] as $class) {
+				// Get an instance of the desired class.
+				$class = __NAMESPACE__ . '\\APIs\\' . $class;
 
-			if ($fetcher->result('success')) {
-				break;
+				$fetcher = new $class();
+
+				// Do we want to keep this connection alive, and can we do so?
+				if ($keep_alive && property_exists($fetcher, 'keep_alive')) {
+					$fetcher->keep_alive = $keep_alive;
+					self::$still_alive[(string) $url] = $fetcher;
+				}
+
+				// Make the request.
+				$fetcher->request($url, $post_data);
+
+				// If keep_alive was turned off during the request, we don't
+				// need to maintain this instance after we're done the request.
+				if (!($fetcher->keep_alive ?? false)) {
+					unset(self::$still_alive[(string) $url]);
+				}
+
+				// If the request worked, we can stop looping.
+				if ($fetcher->result('success')) {
+					break;
+				}
 			}
 		}