<?php

/*
 * This file is part of the Symfony package.
 *
 * (c) Fabien Potencier <fabien@symfony.com>
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

namespace Symfony\Component\HttpClient\Internal;

use Amp\Http\Client\Connection\Stream;
use Amp\Http\Client\EventListener;
use Amp\Http\Client\Request;
use Amp\Promise;
use Amp\Success;
use Symfony\Component\HttpClient\Exception\TransportException;

/**
 * @author Nicolas Grekas <p@tchwork.com>
 *
 * @internal
 */
class AmpListener implements EventListener
{
    /**
     * @var mixed[]
     */
    private $info;
    /**
     * @var mixed[]
     */
    private $pinSha256;
    /**
     * @var \Closure
     */
    private $onProgress;
    private $handle;

    /**
     * @param mixed[] $info
     * @param mixed[] $pinSha256
     * @param \Closure $onProgress
     */
    public function __construct(&$info, $pinSha256, $onProgress, &$handle)
    {
        $info += [
            'connect_time' => 0.0,
            'pretransfer_time' => 0.0,
            'starttransfer_time' => 0.0,
            'total_time' => 0.0,
            'namelookup_time' => 0.0,
            'primary_ip' => '',
            'primary_port' => 0,
        ];

        $this->info = &$info;
        $this->pinSha256 = $pinSha256;
        $this->onProgress = $onProgress;
        $this->handle = &$handle;
    }

    /**
     * @param \Amp\Http\Client\Request $request
     */
    public function startRequest($request)
    {
        $this->info['start_time'] = $this->info['start_time'] ?? microtime(true);
        ($this->onProgress)();

        return new Success();
    }

    /**
     * @param \Amp\Http\Client\Request $request
     */
    public function startDnsResolution($request)
    {
        ($this->onProgress)();

        return new Success();
    }

    /**
     * @param \Amp\Http\Client\Request $request
     */
    public function startConnectionCreation($request)
    {
        ($this->onProgress)();

        return new Success();
    }

    /**
     * @param \Amp\Http\Client\Request $request
     */
    public function startTlsNegotiation($request)
    {
        ($this->onProgress)();

        return new Success();
    }

    /**
     * @param \Amp\Http\Client\Request $request
     * @param \Amp\Http\Client\Connection\Stream $stream
     */
    public function startSendingRequest($request, $stream)
    {
        $host = $stream->getRemoteAddress()->getHost();

        if (strpos($host, ':') !== false) {
            $host = '['.$host.']';
        }

        $this->info['primary_ip'] = $host;
        $this->info['primary_port'] = $stream->getRemoteAddress()->getPort();
        $this->info['pretransfer_time'] = microtime(true) - $this->info['start_time'];
        $this->info['debug'] .= sprintf("* Connected to %s (%s) port %d\n", $request->getUri()->getHost(), $host, $this->info['primary_port']);

        if ((isset($this->info['peer_certificate_chain']) || $this->pinSha256) && null !== $tlsInfo = $stream->getTlsInfo()) {
            foreach ($tlsInfo->getPeerCertificates() as $cert) {
                $this->info['peer_certificate_chain'][] = openssl_x509_read($cert->toPem());
            }

            if ($this->pinSha256) {
                $pin = openssl_pkey_get_public($this->info['peer_certificate_chain'][0]);
                $pin = openssl_pkey_get_details($pin)['key'];
                $pin = \array_slice(explode("\n", $pin), 1, -2);
                $pin = base64_decode(implode('', $pin));
                $pin = base64_encode(hash('sha256', $pin, true));

                if (!\in_array($pin, $this->pinSha256, true)) {
                    throw new TransportException(sprintf('SSL public key does not match pinned public key for "%s".', $this->info['url']));
                }
            }
        }
        ($this->onProgress)();

        $uri = $request->getUri();
        $requestUri = $uri->getPath() ?: '/';

        if ('' !== $query = $uri->getQuery()) {
            $requestUri .= '?'.$query;
        }

        if ('CONNECT' === $method = $request->getMethod()) {
            $requestUri = $uri->getHost().': '.($uri->getPort() ?? ('https' === $uri->getScheme() ? 443 : 80));
        }

        $this->info['debug'] .= sprintf("> %s %s HTTP/%s \r\n", $method, $requestUri, $request->getProtocolVersions()[0]);

        foreach ($request->getRawHeaders() as list($name, $value)) {
            $this->info['debug'] .= $name.': '.$value."\r\n";
        }
        $this->info['debug'] .= "\r\n";

        return new Success();
    }

    /**
     * @param \Amp\Http\Client\Request $request
     * @param \Amp\Http\Client\Connection\Stream $stream
     */
    public function completeSendingRequest($request, $stream)
    {
        ($this->onProgress)();

        return new Success();
    }

    /**
     * @param \Amp\Http\Client\Request $request
     * @param \Amp\Http\Client\Connection\Stream $stream
     */
    public function startReceivingResponse($request, $stream)
    {
        $this->info['starttransfer_time'] = microtime(true) - $this->info['start_time'];
        ($this->onProgress)();

        return new Success();
    }

    /**
     * @param \Amp\Http\Client\Request $request
     * @param \Amp\Http\Client\Connection\Stream $stream
     */
    public function completeReceivingResponse($request, $stream)
    {
        $this->handle = null;
        ($this->onProgress)();

        return new Success();
    }

    /**
     * @param \Amp\Http\Client\Request $request
     */
    public function completeDnsResolution($request)
    {
        $this->info['namelookup_time'] = microtime(true) - $this->info['start_time'];
        ($this->onProgress)();

        return new Success();
    }

    /**
     * @param \Amp\Http\Client\Request $request
     */
    public function completeConnectionCreation($request)
    {
        $this->info['connect_time'] = microtime(true) - $this->info['start_time'];
        ($this->onProgress)();

        return new Success();
    }

    /**
     * @param \Amp\Http\Client\Request $request
     */
    public function completeTlsNegotiation($request)
    {
        ($this->onProgress)();

        return new Success();
    }

    /**
     * @param \Amp\Http\Client\Request $request
     * @param \Throwable $cause
     */
    public function abort($request, $cause)
    {
        return new Success();
    }
}
