<?php

/**
 * Copyright 2015 Google Inc. All Rights Reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
namespace DeliciousBrains\WP_Offload_Media\Gcp\Google\Cloud\Core;

use DeliciousBrains\WP_Offload_Media\Gcp\Google\Auth\HttpHandler\HttpHandlerFactory;
use DeliciousBrains\WP_Offload_Media\Gcp\Google\Cloud\Core\Exception;
use DeliciousBrains\WP_Offload_Media\Gcp\Google\ApiCore\ApiException;
use DeliciousBrains\WP_Offload_Media\Gcp\Google\ApiCore\OperationResponse;
use DeliciousBrains\WP_Offload_Media\Gcp\Google\ApiCore\PagedListResponse;
use DeliciousBrains\WP_Offload_Media\Gcp\Google\ApiCore\Serializer;
use DeliciousBrains\WP_Offload_Media\Gcp\Google\ApiCore\ServerStream;
use DeliciousBrains\WP_Offload_Media\Gcp\Google\GAX\ApiException as GaxApiException;
use DeliciousBrains\WP_Offload_Media\Gcp\Google\GAX\OperationResponse as GaxOperationResponse;
use DeliciousBrains\WP_Offload_Media\Gcp\Google\GAX\PagedListResponse as GaxPagedListResponse;
use DeliciousBrains\WP_Offload_Media\Gcp\Google\GAX\Serializer as GaxSerializer;
use DeliciousBrains\WP_Offload_Media\Gcp\Google\GAX\ServerStream as GaxServerStream;
use DeliciousBrains\WP_Offload_Media\Gcp\Google\Protobuf\Internal\Message;
use DeliciousBrains\WP_Offload_Media\Gcp\Google\Rpc\BadRequest;
use DeliciousBrains\WP_Offload_Media\Gcp\Google\Rpc\Code;
use DeliciousBrains\WP_Offload_Media\Gcp\Google\Rpc\RetryInfo;
use Grpc;
/**
 * The GrpcRequestWrapper is responsible for delivering gRPC requests.
 */
class GrpcRequestWrapper
{
    use RequestWrapperTrait;
    /**
     * @var callable A handler used to deliver Psr7 requests specifically for
     * authentication.
     */
    private $authHttpHandler;
    /**
     * @var Serializer A serializer used to encode responses.
     */
    private $serializer;
    /**
     * @var array gRPC specific configuration options passed off to the ApiCore
     * library.
     */
    private $grpcOptions;
    /**
     * @var array gRPC retry codes.
     */
    private $grpcRetryCodes = [\DeliciousBrains\WP_Offload_Media\Gcp\Google\Rpc\Code::UNKNOWN, \DeliciousBrains\WP_Offload_Media\Gcp\Google\Rpc\Code::INTERNAL, \DeliciousBrains\WP_Offload_Media\Gcp\Google\Rpc\Code::UNAVAILABLE, \DeliciousBrains\WP_Offload_Media\Gcp\Google\Rpc\Code::DATA_LOSS];
    /**
     * @var array Map of error metadata types to RPC wrappers.
     */
    private $metadataTypes = ['google.rpc.retryinfo-bin' => \DeliciousBrains\WP_Offload_Media\Gcp\Google\Rpc\RetryInfo::class, 'google.rpc.badrequest-bin' => \DeliciousBrains\WP_Offload_Media\Gcp\Google\Rpc\BadRequest::class];
    /**
     * @param array $config [optional] {
     *     Configuration options. Please see
     *     {@see Google\Cloud\Core\RequestWrapperTrait::setCommonDefaults()} for
     *     the other available options.
     *
     *     @type callable $authHttpHandler A handler used to deliver Psr7
     *           requests specifically for authentication.
     *     @type Serializer $serializer A serializer used to encode responses.
     *     @type array $grpcOptions gRPC specific configuration options passed
     *           off to the ApiCore library.
     * }
     */
    public function __construct(array $config = [])
    {
        $this->setCommonDefaults($config);
        $config += ['authHttpHandler' => null, 'serializer' => $this->buildSerializer(), 'grpcOptions' => []];
        $this->authHttpHandler = $config['authHttpHandler'] ?: \DeliciousBrains\WP_Offload_Media\Gcp\Google\Auth\HttpHandler\HttpHandlerFactory::build();
        $this->serializer = $config['serializer'];
        $this->grpcOptions = $config['grpcOptions'];
    }
    /**
     * Deliver the request.
     *
     * @param callable $request The request to execute.
     * @param array $args The arguments for the request.
     * @param array $options [optional] {
     *     Request options.
     *
     *     @type float $requestTimeout Seconds to wait before timing out the
     *           request. **Defaults to** `60`.
     *     @type int $retries Number of retries for a failed request.
     *           **Defaults to** `3`.
     *     @type array $grpcOptions gRPC specific configuration options.
     * }
     * @return array
     */
    public function send(callable $request, array $args, array $options = [])
    {
        $retries = isset($options['retries']) ? $options['retries'] : $this->retries;
        $grpcOptions = isset($options['grpcOptions']) ? $options['grpcOptions'] : $this->grpcOptions;
        $timeout = isset($options['requestTimeout']) ? $options['requestTimeout'] : $this->requestTimeout;
        $backoff = new \DeliciousBrains\WP_Offload_Media\Gcp\Google\Cloud\Core\ExponentialBackoff($retries, function (\Exception $ex) {
            $statusCode = $ex->getCode();
            return in_array($statusCode, $this->grpcRetryCodes);
        });
        if (!isset($grpcOptions['retrySettings'])) {
            $retrySettings = ['retriesEnabled' => false];
            if ($timeout) {
                $retrySettings['noRetriesRpcTimeoutMillis'] = $timeout * 1000;
            }
            $grpcOptions['retrySettings'] = $retrySettings;
        }
        $optionalArgs =& $args[count($args) - 1];
        $optionalArgs += $grpcOptions;
        try {
            return $this->handleResponse($backoff->execute($request, $args));
        } catch (\Exception $ex) {
            if ($ex instanceof ApiException || $ex instanceof GaxApiException) {
                throw $this->convertToGoogleException($ex);
            }
            throw $ex;
        }
    }
    /**
     * Serializes a gRPC response.
     *
     * @param mixed $response
     * @return \Generator|array|null
     */
    private function handleResponse($response)
    {
        if ($response instanceof PagedListResponse || $response instanceof GaxPagedListResponse) {
            $response = $response->getPage()->getResponseObject();
        }
        if ($response instanceof Message) {
            return $this->serializer->encodeMessage($response);
        }
        if ($response instanceof OperationResponse || $response instanceof GaxOperationResponse) {
            return $response;
        }
        if ($response instanceof ServerStream || $response instanceof GaxServerStream) {
            return $this->handleStream($response);
        }
        return null;
    }
    /**
     * Handles a streaming response.
     *
     * @param ServerStream|GaxServerStream $response
     * @return \Generator|array|null
     */
    private function handleStream($response)
    {
        try {
            foreach ($response->readAll() as $count => $result) {
                $res = $this->serializer->encodeMessage($result);
                (yield $res);
            }
        } catch (\Exception $ex) {
            throw $this->convertToGoogleException($ex);
        }
    }
    /**
     * Convert a ApiCore exception to a Google Exception.
     *
     * @param ApiException|GaxApiException $ex
     * @return Exception\ServiceException
     */
    private function convertToGoogleException($ex)
    {
        switch ($ex->getCode()) {
            case \DeliciousBrains\WP_Offload_Media\Gcp\Google\Rpc\Code::INVALID_ARGUMENT:
                $exception = \DeliciousBrains\WP_Offload_Media\Gcp\Google\Cloud\Core\Exception\BadRequestException::class;
                break;
            case \DeliciousBrains\WP_Offload_Media\Gcp\Google\Rpc\Code::NOT_FOUND:
            case \DeliciousBrains\WP_Offload_Media\Gcp\Google\Rpc\Code::UNIMPLEMENTED:
                $exception = \DeliciousBrains\WP_Offload_Media\Gcp\Google\Cloud\Core\Exception\NotFoundException::class;
                break;
            case \DeliciousBrains\WP_Offload_Media\Gcp\Google\Rpc\Code::ALREADY_EXISTS:
                $exception = \DeliciousBrains\WP_Offload_Media\Gcp\Google\Cloud\Core\Exception\ConflictException::class;
                break;
            case \DeliciousBrains\WP_Offload_Media\Gcp\Google\Rpc\Code::FAILED_PRECONDITION:
                $exception = \DeliciousBrains\WP_Offload_Media\Gcp\Google\Cloud\Core\Exception\FailedPreconditionException::class;
                break;
            case \DeliciousBrains\WP_Offload_Media\Gcp\Google\Rpc\Code::UNKNOWN:
                $exception = \DeliciousBrains\WP_Offload_Media\Gcp\Google\Cloud\Core\Exception\ServerException::class;
                break;
            case \DeliciousBrains\WP_Offload_Media\Gcp\Google\Rpc\Code::INTERNAL:
                $exception = \DeliciousBrains\WP_Offload_Media\Gcp\Google\Cloud\Core\Exception\ServerException::class;
                break;
            case \DeliciousBrains\WP_Offload_Media\Gcp\Google\Rpc\Code::ABORTED:
                $exception = \DeliciousBrains\WP_Offload_Media\Gcp\Google\Cloud\Core\Exception\AbortedException::class;
                break;
            case \DeliciousBrains\WP_Offload_Media\Gcp\Google\Rpc\Code::DEADLINE_EXCEEDED:
                $exception = \DeliciousBrains\WP_Offload_Media\Gcp\Google\Cloud\Core\Exception\DeadlineExceededException::class;
                break;
            default:
                $exception = \DeliciousBrains\WP_Offload_Media\Gcp\Google\Cloud\Core\Exception\ServiceException::class;
                break;
        }
        $metadata = [];
        if ($ex->getMetadata()) {
            foreach ($ex->getMetadata() as $type => $binaryValue) {
                if (!isset($this->metadataTypes[$type])) {
                    continue;
                }
                $metadataElement = new $this->metadataTypes[$type]();
                $metadataElement->mergeFromString($binaryValue[0]);
                $metadata[] = $this->serializer->encodeMessage($metadataElement);
            }
        }
        return new $exception($ex->getMessage(), $ex->getCode(), $ex, $metadata);
    }
    /**
     * @return Serializer|GaxSerializer
     */
    private function buildSerializer()
    {
        return class_exists(\DeliciousBrains\WP_Offload_Media\Gcp\Google\ApiCore\Serializer::class) ? new \DeliciousBrains\WP_Offload_Media\Gcp\Google\ApiCore\Serializer() : new \DeliciousBrains\WP_Offload_Media\Gcp\Google\GAX\Serializer();
    }
}