Nuša Pukšič
3 months ago
20 changed files with 419 additions and 218 deletions
@ -0,0 +1,168 @@ |
|||||||
|
<?php |
||||||
|
declare(strict_types=1); |
||||||
|
|
||||||
|
namespace App\Util\NostrPhp; |
||||||
|
|
||||||
|
use swentel\nostr\Message\AuthMessage; |
||||||
|
use swentel\nostr\Message\CloseMessage; |
||||||
|
use swentel\nostr\MessageInterface; |
||||||
|
use swentel\nostr\Nip42\AuthEvent; |
||||||
|
use swentel\nostr\Relay\Relay; |
||||||
|
use swentel\nostr\Relay\RelaySet; |
||||||
|
use swentel\nostr\RelayResponse\RelayResponse; |
||||||
|
use swentel\nostr\RequestInterface; |
||||||
|
use swentel\nostr\Sign\Sign; |
||||||
|
use WebSocket\Client as WsClient; |
||||||
|
use WebSocket\Message\Pong; |
||||||
|
use WebSocket\Message\Text; |
||||||
|
|
||||||
|
/** |
||||||
|
* A deterministic "stop on first matching EVENT" request. |
||||||
|
* Implements RequestInterface so we can DI-substitute it for the vendor Request. |
||||||
|
*/ |
||||||
|
final class TweakedRequest implements RequestInterface |
||||||
|
{ |
||||||
|
private RelaySet $relays; |
||||||
|
private string $payload; |
||||||
|
private array $responses = []; |
||||||
|
|
||||||
|
/** Optional: when set, CLOSE & disconnect immediately once this id arrives */ |
||||||
|
private ?string $stopOnEventId = null; |
||||||
|
|
||||||
|
public function __construct(Relay|RelaySet $relay, MessageInterface $message) |
||||||
|
{ |
||||||
|
if ($relay instanceof RelaySet) { |
||||||
|
$this->relays = $relay; |
||||||
|
} else { |
||||||
|
$set = new RelaySet(); |
||||||
|
$set->setRelays([$relay]); |
||||||
|
$this->relays = $set; |
||||||
|
} |
||||||
|
$this->payload = $message->generate(); |
||||||
|
} |
||||||
|
|
||||||
|
public function stopOnEventId(?string $hexId): self |
||||||
|
{ |
||||||
|
$this->stopOnEventId = $hexId; |
||||||
|
return $this; |
||||||
|
} |
||||||
|
|
||||||
|
/** @return array<string, array|RelayResponse> */ |
||||||
|
public function send(): array |
||||||
|
{ |
||||||
|
$result = []; |
||||||
|
|
||||||
|
foreach ($this->relays->getRelays() as $relay) { |
||||||
|
$this->responses = []; // reset per relay |
||||||
|
try { |
||||||
|
if (!$relay->isConnected()) { |
||||||
|
$relay->connect(); |
||||||
|
} |
||||||
|
|
||||||
|
$client = $relay->getClient(); |
||||||
|
$client->setTimeout(15); // seconds per receive call (keep it small if you want responsiveness) |
||||||
|
|
||||||
|
// Send subscription payload |
||||||
|
$client->text($this->payload); |
||||||
|
|
||||||
|
// Loop until: first match, EOSE/CLOSED/ERROR, or socket ends |
||||||
|
while ($resp = $client->receive()) { |
||||||
|
if ($resp instanceof \WebSocket\Message\Ping) { |
||||||
|
$client->text((new Pong())->getPayload()); |
||||||
|
continue; |
||||||
|
} |
||||||
|
if (!$resp instanceof Text) { |
||||||
|
continue; |
||||||
|
} |
||||||
|
|
||||||
|
$decoded = json_decode($resp->getContent()); |
||||||
|
$relayResponse = RelayResponse::create($decoded); |
||||||
|
$this->responses[] = $relayResponse; |
||||||
|
|
||||||
|
// Early exit on matching EVENT |
||||||
|
if ($relayResponse->type === 'EVENT') { |
||||||
|
// Safest: decode again to array to grab [ "EVENT", subId, event ] |
||||||
|
$raw = json_decode($resp->getContent(), true); |
||||||
|
$sub = $raw[1] ?? null; |
||||||
|
$evt = $raw[2] ?? null; |
||||||
|
$evtId = is_array($evt) ? ($evt['id'] ?? null) : null; |
||||||
|
|
||||||
|
if ($this->stopOnEventId !== null && $evtId === $this->stopOnEventId) { |
||||||
|
if ($sub) { |
||||||
|
$this->sendClose($client, $sub); |
||||||
|
} |
||||||
|
$relay->disconnect(); |
||||||
|
$result[$relay->getUrl()] = $this->responses; |
||||||
|
// stop the outer foreach too (we're done) |
||||||
|
return $result; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// Tear-down conditions |
||||||
|
if ($relayResponse->type === 'EOSE') { |
||||||
|
$sub = $relayResponse->subscriptionId ?? null; |
||||||
|
if ($sub) { |
||||||
|
$this->sendClose($client, $sub); |
||||||
|
} |
||||||
|
$relay->disconnect(); |
||||||
|
break; |
||||||
|
} |
||||||
|
|
||||||
|
if ($relayResponse->type === 'NOTICE' && str_starts_with($relayResponse->message ?? '', 'ERROR:')) { |
||||||
|
$relay->disconnect(); |
||||||
|
break; |
||||||
|
} |
||||||
|
|
||||||
|
if ($relayResponse->type === 'CLOSED') { |
||||||
|
$relay->disconnect(); |
||||||
|
break; |
||||||
|
} |
||||||
|
|
||||||
|
// NIP-42: if relay requests AUTH, perform it once, then continue. |
||||||
|
if ($relayResponse->type === 'OK' && isset($relayResponse->message) && str_starts_with($relayResponse->message, 'auth-required:')) { |
||||||
|
$this->performAuth($relay, $client); |
||||||
|
// After AUTH, re-send the original payload |
||||||
|
$client->text($this->payload); |
||||||
|
// continue loop |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// Save what we got for this relay |
||||||
|
$result[$relay->getUrl()] = $this->responses; |
||||||
|
} catch (\Throwable $e) { |
||||||
|
$result[$relay->getUrl()][] = ['ERROR', '', false, $e->getMessage()]; |
||||||
|
// best-effort disconnect |
||||||
|
try { $relay->disconnect(); } catch (\Throwable) {} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
return $result; |
||||||
|
} |
||||||
|
|
||||||
|
private function sendClose(WsClient $client, string $subscriptionId): void |
||||||
|
{ |
||||||
|
try { |
||||||
|
$close = new CloseMessage($subscriptionId); |
||||||
|
$client->text($close->generate()); |
||||||
|
} catch (\Throwable) {} |
||||||
|
} |
||||||
|
|
||||||
|
/** Very lightweight NIP-42 auth flow: sign challenge and send AUTH + resume. */ |
||||||
|
private function performAuth(Relay $relay, WsClient $client): void |
||||||
|
{ |
||||||
|
// NOTE: This reuses the vendor types, but uses a dummy secret. You should inject your real sec key. |
||||||
|
if (!isset($_SESSION['challenge'])) { |
||||||
|
return; |
||||||
|
} |
||||||
|
try { |
||||||
|
$authEvent = new AuthEvent($relay->getUrl(), $_SESSION['challenge']); |
||||||
|
$sec = '0000000000000000000000000000000000000000000000000000000000000001'; // TODO inject your real sec |
||||||
|
(new Sign())->signEvent($authEvent, $sec); |
||||||
|
|
||||||
|
$authMsg = new AuthMessage($authEvent); |
||||||
|
$client->text($authMsg->generate()); |
||||||
|
} catch (\Throwable) { |
||||||
|
// ignore and continue; some relays won’t require it |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
Loading…
Reference in new issue