whm\Smoke\Cli\Command\SmokeCommand

Complexity

Total Complexity

16

Size/Duplication

Total Lines	107
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	10

Importance

Changes

0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	init()	0	14	3
A	writeSmokeCredentials()	0	12	3
A	scan()	0	24	4
B	getConfigArray()	0	24	6

<?php

namespace whm\Smoke\Cli\Command;

use GuzzleHttp\Client;
use phmLabs\Components\Annovent\Dispatcher;
use PhmLabs\Components\Init\Init;
use Symfony\Component\Console\Command\Command;
use Symfony\Component\Console\Input\InputInterface;
use Symfony\Component\Console\Output\OutputInterface;
use whm\Smoke\Config\Configuration;
use whm\Smoke\Scanner\Scanner;
use whm\Smoke\Yaml\EnvAwareYaml;

class SmokeCommand extends Command
{
    /**
     * @var OutputInterface
     */
    protected $output;
    protected $eventDispatcher;

    /**
     * @var Configuration
     */
    protected $config;

    protected function init(InputInterface $input, OutputInterface $output, $url = null)
    {
        if ($input->hasOption('bootstrap') && !is_null($input->getOption('bootstrap'))) {
            include $input->getOption('bootstrap');
        }

        $this->output = $output;
        $this->eventDispatcher = new Dispatcher();

        Init::registerGlobalParameter('_eventDispatcher', $this->eventDispatcher);
        Init::registerGlobalParameter('_output', $output);

        $this->writeSmokeCredentials($url);
    }

    /**
     * This function creates the credentials header for the command line.
     *
     * @param null $url
     */
    protected function writeSmokeCredentials($url = null)
    {
        if (defined('SMOKE_CREDENTIALS')) {
            $this->output->writeln("\n " . SMOKE_CREDENTIALS . "\n");
        } else {
            $this->output->writeln("\n Smoke " . SMOKE_VERSION . " by Nils Langner\n");
        }

        if ($url) {
            $this->output->writeln(' <info>Scanning ' . $url . "</info>\n");
        }
    }

    /**
     * @return int
     * @throws \Exception
     */
    protected function scan()
    {
        $client = $this->config->getClient();

        $scanner = new Scanner($this->config->getRules(),
            $client,
            $this->eventDispatcher,
            $this->config->getExtension('_ResponseRetriever')->getRetriever());

        try {
            $scanner->scan();
        } catch (\Exception $e) {
            if (method_exists($client, 'close')) {
                $client->close();
            }
            throw $e;
        }

        if (method_exists($client, 'close')) {
            $client->close();
        }

        return $scanner->getStatus();
    }

    /**
     * Returns an array representing the configuration.
     *
     * @param $configFile
     *
     * @return array
     */
    protected function getConfigArray($configFile, $mandatory = false)
    {
        $configArray = array();

        if ($configFile) {
            if (strpos($configFile, 'http://') === 0 || strpos($configFile, 'https://') === 0) {
                $curlClient = new Client();
                $fileContent = (string)$curlClient->get($configFile)->getBody();
            } else {
                if (file_exists($configFile)) {
                    $fileContent = file_get_contents($configFile);
                } else {
                    throw new \RuntimeException("Config file was not found ('" . $configFile . "').");
                }
            }
            $configArray = EnvAwareYaml::parse($fileContent);
        } else {
            if ($mandatory) {
                throw new \RuntimeException('Config file was not defined.');
            }
        }

        return $configArray;
    }
}