<?php

/*
 * This file is part of the Behat MinkExtension.
 * (c) Konstantin Kudryashov <ever.zet@gmail.com>
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

namespace Behat\MinkExtension\ServiceContainer\Driver;

use Symfony\Component\Config\Definition\Builder\ArrayNodeDefinition;
use Symfony\Component\DependencyInjection\Definition;

class SauceLabsFactory extends Selenium2Factory
{
    public function getDriverName(): string
    {
        return 'sauce_labs';
    }

    public function configure(ArrayNodeDefinition $builder): void
    {
        $builder
            ->children()
                ->scalarNode('username')->defaultValue(getenv('SAUCE_USERNAME'))->end()
                ->scalarNode('access_key')->defaultValue(getenv('SAUCE_ACCESS_KEY'))->end()
                ->booleanNode('connect')->defaultFalse()->end()
                ->scalarNode('browser')->defaultValue('firefox')->end()
                ->append($this->getCapabilitiesNode())
            ->end()
        ;
    }

    /**
     * @param array<mixed> $config
     */
    public function buildDriver(array $config): Definition
    {
        $host = 'ondemand.saucelabs.com';
        if ((bool) $config['connect']) {
            $host = 'localhost:4445';
        }

        $config['wd_host'] = sprintf('%s:%s@%s/wd/hub', is_string($config['username']) ? $config['username'] : '', is_string($config['access_key']) ? $config['access_key'] : '', $host);

        return parent::buildDriver($config);
    }

    protected function getCapabilitiesNode(): ArrayNodeDefinition
    {
        $node = parent::getCapabilitiesNode();

        $node
            ->children()
                ->scalarNode('platform')->defaultValue('Linux')->end()
                ->scalarNode('selenium-version')->end()
                ->scalarNode('max-duration')->end()
                ->scalarNode('command-timeout')->end()
                ->scalarNode('idle-timeout')->end()
                ->scalarNode('build')->info('will be set automatically based on the TRAVIS_BUILD_NUMBER environment variable if available')->end()
                ->arrayNode('custom-data')
                    ->useAttributeAsKey('')
                    ->prototype('variable')->end()
                ->end()
                ->scalarNode('screen-resolution')->end()
                ->scalarNode('tunnel-identifier')->info('will be set automatically based on the TRAVIS_JOB_NUMBER environment variable if available')->end()
                ->arrayNode('prerun')
                    ->children()
                        ->scalarNode('executable')->isRequired()->end()
                        ->arrayNode('args')->prototype('scalar')->end()->end()
                        ->booleanNode('background')->defaultFalse()->end()
                    ->end()
                ->end()
                ->booleanNode('record-video')->end()
                ->booleanNode('record-screenshots')->end()
                ->booleanNode('capture-html')->end()
                ->booleanNode('disable-popup-handler')->end()
            ->end()
            ->validate()
                ->ifTrue(function (mixed $v): bool {return !is_array($v) || empty($v['custom-data']); })
                ->then(function (mixed $v): mixed {
                    if (is_array($v)) {
                        unset($v['custom-data']);
                    }

                    return $v;
                })
            ->end()
        ;

        return $node;
    }
}