mirror of
https://github.com/vichan-devel/vichan.git
synced 2024-11-28 01:10:51 +01:00
132 lines
3.8 KiB
PHP
132 lines
3.8 KiB
PHP
<?php
|
|
namespace Vichan\Data\Driver;
|
|
|
|
defined('TINYBOARD') or exit;
|
|
|
|
|
|
/**
|
|
* Honestly this is just a wrapper for cURL. Still useful to mock it and have an OOP API on PHP 7.
|
|
*/
|
|
class HttpDriver {
|
|
private mixed $inner;
|
|
private int $timeout;
|
|
private int $max_file_size;
|
|
|
|
|
|
private function resetTowards(string $url, int $timeout): void {
|
|
\curl_reset($this->inner);
|
|
\curl_setopt_array($this->inner, [
|
|
\CURLOPT_URL => $url,
|
|
\CURLOPT_TIMEOUT => $timeout,
|
|
\CURLOPT_USERAGENT => 'Tinyboard',
|
|
\CURLOPT_PROTOCOLS => \CURLPROTO_HTTP | \CURLPROTO_HTTPS,
|
|
]);
|
|
}
|
|
|
|
public function __construct(int $timeout, int $max_file_size) {
|
|
$this->inner = \curl_init();
|
|
$this->timeout = $timeout;
|
|
$this->max_file_size = $max_file_size;
|
|
}
|
|
|
|
public function __destruct() {
|
|
\curl_close($this->inner);
|
|
}
|
|
|
|
/**
|
|
* Execute a GET request.
|
|
*
|
|
* @param string $endpoint Uri endpoint.
|
|
* @param ?array $data Optional GET parameters.
|
|
* @param int $timeout Optional request timeout in seconds. Use the default timeout if 0.
|
|
* @return string Returns the body of the response.
|
|
* @throws RuntimeException Throws on IO error.
|
|
*/
|
|
public function requestGet(string $endpoint, ?array $data, int $timeout = 0): string {
|
|
if (!empty($data)) {
|
|
$endpoint .= '?' . \http_build_query($data);
|
|
}
|
|
if ($timeout == 0) {
|
|
$timeout = $this->timeout;
|
|
}
|
|
|
|
$this->resetTowards($endpoint, $timeout);
|
|
\curl_setopt($this->inner, \CURLOPT_RETURNTRANSFER, true);
|
|
$ret = \curl_exec($this->inner);
|
|
|
|
if ($ret === false) {
|
|
throw new \RuntimeException(\curl_error($this->inner));
|
|
}
|
|
return $ret;
|
|
}
|
|
|
|
/**
|
|
* Execute a POST request.
|
|
*
|
|
* @param string $endpoint Uri endpoint.
|
|
* @param ?array $data Optional POST parameters.
|
|
* @param int $timeout Optional request timeout in seconds. Use the default timeout if 0.
|
|
* @return string Returns the body of the response.
|
|
* @throws RuntimeException Throws on IO error.
|
|
*/
|
|
public function requestPost(string $endpoint, ?array $data, int $timeout = 0): string {
|
|
if ($timeout == 0) {
|
|
$timeout = $this->timeout;
|
|
}
|
|
|
|
$this->resetTowards($endpoint, $timeout);
|
|
\curl_setopt($this->inner, \CURLOPT_POST, true);
|
|
if (!empty($data)) {
|
|
\curl_setopt($this->inner, \CURLOPT_POSTFIELDS, \http_build_query($data));
|
|
}
|
|
\curl_setopt($this->inner, \CURLOPT_RETURNTRANSFER, true);
|
|
$ret = \curl_exec($this->inner);
|
|
|
|
if ($ret === false) {
|
|
throw new \RuntimeException(\curl_error($this->inner));
|
|
}
|
|
return $ret;
|
|
}
|
|
|
|
/**
|
|
* Download the url's target with curl.
|
|
*
|
|
* @param string $url Url to the file to download.
|
|
* @param ?array $data Optional GET parameters.
|
|
* @param resource $fd File descriptor to save the content to.
|
|
* @param int $timeout Optional request timeout in seconds. Use the default timeout if 0.
|
|
* @return bool Returns true on success, false if the file was too large.
|
|
* @throws RuntimeException Throws on IO error.
|
|
*/
|
|
public function requestGetInto(string $endpoint, ?array $data, mixed $fd, int $timeout = 0): bool {
|
|
if (!empty($data)) {
|
|
$endpoint .= '?' . \http_build_query($data);
|
|
}
|
|
if ($timeout == 0) {
|
|
$timeout = $this->timeout;
|
|
}
|
|
|
|
$this->resetTowards($endpoint, $timeout);
|
|
// Adapted from: https://stackoverflow.com/a/17642638
|
|
$opt = (\PHP_MAJOR_VERSION >= 8 && \PHP_MINOR_VERSION >= 2) ? \CURLOPT_XFERINFOFUNCTION : \CURLOPT_PROGRESSFUNCTION;
|
|
\curl_setopt_array($this->inner, [
|
|
\CURLOPT_NOPROGRESS => false,
|
|
$opt => fn($res, $next_dl, $dl, $next_up, $up) => (int)($dl <= $this->max_file_size),
|
|
\CURLOPT_FAILONERROR => true,
|
|
\CURLOPT_FOLLOWLOCATION => false,
|
|
\CURLOPT_FILE => $fd,
|
|
\CURLOPT_IPRESOLVE => CURL_IPRESOLVE_V4,
|
|
]);
|
|
$ret = \curl_exec($this->inner);
|
|
|
|
if ($ret === false) {
|
|
if (\curl_errno($this->inner) === CURLE_ABORTED_BY_CALLBACK) {
|
|
return false;
|
|
}
|
|
|
|
throw new \RuntimeException(\curl_error($this->inner));
|
|
}
|
|
return true;
|
|
}
|
|
}
|