2025-05-01 15:49:35 +02:00

73 lines
1.8 KiB
PHP

<?php
namespace App\Repositories\Daemon;
use App\Models\Node;
use Illuminate\Http\Client\PendingRequest;
use Illuminate\Support\Facades\Http;
use Webmozart\Assert\Assert;
use App\Models\Server;
use Illuminate\Http\Client\ConnectionException;
use Illuminate\Http\Client\Response;
abstract class DaemonRepository
{
protected ?Server $server;
protected ?Node $node;
/**
* Set the server model this request is stemming from.
*
* @return static
*/
public function setServer(Server $server): self
{
$this->server = $server;
$this->setNode($this->server->node);
return $this;
}
/**
* Set the node model this request is stemming from.
*/
public function setNode(Node $node): static
{
$this->node = $node;
return $this;
}
/**
* Return an instance of the Guzzle HTTP Client to be used for requests.
*
* @param array<string, string> $headers
*/
public function getHttpClient(array $headers = []): PendingRequest
{
Assert::isInstanceOf($this->node, Node::class);
return Http::daemon($this->node, $headers)->throwIf(fn ($condition) => $this->enforceValidNodeToken($condition));
}
protected function enforceValidNodeToken(Response|bool $condition): bool
{
if (is_bool($condition)) {
return $condition;
}
$header = $condition->header('User-Agent');
if (
empty($header) ||
preg_match('/^Pelican Wings\/v(?:\d+\.\d+\.\d+|develop) \(id:(\w*)\)$/', $header, $matches) &&
array_get($matches, 1, '') !== $this->node->daemon_token_id
) {
throw new ConnectionException($condition->effectiveUri()->__toString() . ' does not match node token_id !');
}
return true;
}
}