CurlClientWithCookies.php

Go to the documentation of this file.

<?php
namespace Magento\TestFramework\TestCase\HttpClient;

use Magento\TestFramework\Helper\JsonSerializer;
use Magento\TestFramework\Helper\Bootstrap;

class CurlClientWithCookies
{
    const COOKIE_HEADER = 'Set-Cookie: ';

    protected $curlClient;

    protected $jsonSerializer;

    public function __construct(
        CurlClient $curlClient,
        \Magento\TestFramework\Helper\JsonSerializer $jsonSerializer
    ) {
        $objectManager = Bootstrap::getObjectManager();
        $this->curlClient = $curlClient ? : $objectManager->get(CurlClient::class);
        $this->jsonSerializer = $jsonSerializer ? : $objectManager->get(JsonSerializer::class);
    }

    public function constructResourceUrl($resourcePath)
    {
        return rtrim(TESTS_BASE_URL, '/') . '/' . ltrim($resourcePath, '/');
    }

    public function get($resourcePath, $data = [], $headers = [])
    {
        $url = $this->constructResourceUrl($resourcePath);
        if (!empty($data)) {
            $url .= '?' . http_build_query($data);
        }

        $curlOpts = [];
        $curlOpts[CURLOPT_CUSTOMREQUEST] = \Magento\Framework\Webapi\Rest\Request::HTTP_METHOD_GET;
        $curlOpts[CURLOPT_SSLVERSION] = 3;
        $response = $this->curlClient->invokeApi($url, $curlOpts, $headers);
        $response['cookies'] = $this->cookieParse($response['header']);
        return $response;
    }

    private function cookieParse($headerBlock)
    {
        $header = explode("\r\n", $headerBlock);
        $cookies = [];
        foreach ($header as $line) {
            $line = trim($line);
            if (substr($line, 0, strlen(self::COOKIE_HEADER)) == self::COOKIE_HEADER) {
                $line = trim(substr($line, strlen(self::COOKIE_HEADER)));
                $cookieData = [];
                // Check if cookie contains attributes
                if (strpos($line, ';') === false) {
                    // no attributes, just name and value
                    list($cookieData['name'], $cookieData['value']) = explode('=', $line);
                } else {
                    // has attributes, must parse them out and loop through
                    list($nvPair, $cookieMetadata) = explode(';', $line, 2);
                    list($cookieData['name'], $cookieData['value']) = explode('=', $nvPair);
                    $rawCookieData = explode(';', $cookieMetadata);
                    foreach ($rawCookieData as $keyValuePairs) {
                        list($key, $value) = array_merge(explode('=', $keyValuePairs), ['true']);
                        $cookieData[strtolower(trim($key))] = trim($value);
                    }
                }
                $cookies[] = $cookieData;
            }
        }
        return $cookies;
    }
}