Initial Commit

dependencies/amphp/dns/lib/Internal/UdpSocket.php

@@ -0,0 +1,51 @@
+<?php
+
+namespace WP_Ultimo\Dependencies\Amp\Dns\Internal;
+
+use WP_Ultimo\Dependencies\Amp\Dns\DnsException;
+use WP_Ultimo\Dependencies\Amp\Promise;
+use WP_Ultimo\Dependencies\Amp\Success;
+use WP_Ultimo\Dependencies\LibDNS\Decoder\DecoderFactory;
+use WP_Ultimo\Dependencies\LibDNS\Encoder\EncoderFactory;
+use WP_Ultimo\Dependencies\LibDNS\Messages\Message;
+use function WP_Ultimo\Dependencies\Amp\call;
+/** @internal */
+final class UdpSocket extends Socket
+{
+    /** @var \LibDNS\Encoder\Encoder */
+    private $encoder;
+    /** @var \LibDNS\Decoder\Decoder */
+    private $decoder;
+    public static function connect(string $uri) : Promise
+    {
+        if (!($socket = @\stream_socket_client($uri, $errno, $errstr, 0, \STREAM_CLIENT_ASYNC_CONNECT))) {
+            throw new DnsException(\sprintf("Connection to %s failed: [Error #%d] %s", $uri, $errno, $errstr));
+        }
+        return new Success(new self($socket));
+    }
+    protected function __construct($socket)
+    {
+        parent::__construct($socket);
+        $this->encoder = (new EncoderFactory())->create();
+        $this->decoder = (new DecoderFactory())->create();
+    }
+    protected function send(Message $message) : Promise
+    {
+        $data = $this->encoder->encode($message);
+        return $this->write($data);
+    }
+    protected function receive() : Promise
+    {
+        return call(function () {
+            $data = (yield $this->read());
+            if ($data === null) {
+                throw new DnsException("Reading from the server failed");
+            }
+            return $this->decoder->decode($data);
+        });
+    }
+    public function isAlive() : bool
+    {
+        return \true;
+    }
+}