Bex\Behat\ScreenshotExtension\Driver\AwsS3

Complexity

Total Complexity

10

Size/Duplication

Total Lines	101
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	3

Importance

Changes

0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	configure()	0	14	1
C	load()	0	31	7
A	upload()	0	7	1
A	createClient()	0	4	1

<?php

namespace Bex\Behat\ScreenshotExtension\Driver;

use Aws\S3\S3Client;
use Symfony\Component\Config\Definition\Builder\ArrayNodeDefinition;
use Symfony\Component\DependencyInjection\ContainerBuilder;

class AwsS3 implements ImageDriverInterface
{
    const CONFIG_PARAM_BUCKET = 'bucket';
    const CONFIG_PARAM_VERSION = 'version';
    const CONFIG_PARAM_REGION = 'region';
    const CONFIG_PARAM_CREDENTIALS_KEY = 'credentials_key';
    const CONFIG_PARAM_CREDENTIALS_SECRET = 'credentials_secret';
    const CONFIG_PARAM_CREDENTIALS_TOKEN = 'credentials_token';
    const CONFIG_PARAM_CLIENT_FACTORY = 'client_factory';
    const CONFIG_PARAM_NAMESPACE = 'namespace';
    /**
     * @var S3Client
     */
    private $api;
    /**
     * @var string
     */
    private $bucket;
    /**
     * @var string
     */
    private $namespace;

    /**
     * @param  ArrayNodeDefinition $builder
     */
    public function configure(ArrayNodeDefinition $builder)
    {
        $builder
            ->children()
                ->scalarNode(self::CONFIG_PARAM_BUCKET)->isRequired()->end()
                ->scalarNode(self::CONFIG_PARAM_VERSION)->defaultValue('latest')->end()
                ->scalarNode(self::CONFIG_PARAM_REGION)->isRequired()->end()
                ->scalarNode(self::CONFIG_PARAM_CREDENTIALS_KEY)->defaultNull()->end()
                ->scalarNode(self::CONFIG_PARAM_CREDENTIALS_SECRET)->defaultNull()->end()
                ->scalarNode(self::CONFIG_PARAM_CREDENTIALS_TOKEN)->defaultNull()->end()
                ->scalarNode(self::CONFIG_PARAM_CLIENT_FACTORY)->defaultNull()->end()
                ->scalarNode(self::CONFIG_PARAM_NAMESPACE)->defaultNull()->end()
            ->end();
    }

    /**
     * @param  ContainerBuilder $container
     * @param  array $config
     */
    public function load(ContainerBuilder $container, array $config)
    {
        $this->bucket = $config[self::CONFIG_PARAM_BUCKET];

        $this->namespace = $config[self::CONFIG_PARAM_NAMESPACE] ?: sprintf('%d%04d', time(), rand(0, 9999));
        $version = $config[self::CONFIG_PARAM_VERSION];
        $region = $config[self::CONFIG_PARAM_REGION];
        $credentials = null;
        if ($config[self::CONFIG_PARAM_CREDENTIALS_KEY] && $config[self::CONFIG_PARAM_CREDENTIALS_SECRET]) {
            $credentials = [
                'key' => $config[self::CONFIG_PARAM_CREDENTIALS_KEY],
                'secret' => $config[self::CONFIG_PARAM_CREDENTIALS_SECRET],
                'token' => $config[self::CONFIG_PARAM_CREDENTIALS_TOKEN],
            ];
        } elseif ($config[self::CONFIG_PARAM_CREDENTIALS_KEY] === false) {
            $credentials = false;
        }

        $clientFactory = $config[self::CONFIG_PARAM_CLIENT_FACTORY] ?: [$this, 'createClient'];
        if (!is_callable($clientFactory)) {
            throw new \RuntimeException('Invalid S3 API client factory callback');
        }

        $args = [
            'version' => $version,
            'region' => $region,
            'credentials' => $credentials,
        ];

        $this->api = call_user_func($clientFactory, $args);
    }

    /**
     * @param string $binaryImage
     * @param string $filename
     *
     * @return string URL to the image
     */
    public function upload($binaryImage, $filename)
    {
        $path = join('/', [$this->namespace, $filename]);
        $result = $this->api->upload($this->bucket, $path, $binaryImage, 'public-read');

        return $result['ObjectURL'];
    }

    /**
     * @param array $args
     *
     * @return S3Client
     */
    public function createClient($args)
    {
        return new S3Client($args);
    }
}