howardjones/network-weathermap

View on GitHub
lib/Weathermap/CLI/Runner.php

Summary

Maintainability
B
6 hrs
Test Coverage
<?php

namespace Weathermap\CLI;

use GetOpt\GetOpt;
use GetOpt\Option;
use GetOpt\ArgumentException;
use Weathermap\Core\Map;
use Weathermap\Core\MapUtility;

/**
 * The CLI map-creation tool
 *
 * @package Weathermap\CLI
 */
class Runner
{
    /** @var GetOpt $getOpt */
    private $getOpt;
    private $defines = array();
    private $optionsOutput = array();

    /** @var Map $map */
    private $map;
    private $configFile;
    private $imageFile = "";
    private $htmlFile = "";

    public function run()
    {
        $rrdtool = "/usr/bin/rrdtool";

        $this->getOptions();
        $this->translateRuntimeOptionsToSettings();
        $this->translateDebugOptionsToSettings();

        $this->map = new Map();
        $this->map->rrdtool = $rrdtool;
        $this->map->context = "cli";

        if ($this->makeMap()) {
            $this->postRun();
        }
    }

    private function getOptions()
    {
        $this->getOpt = new GetOpt(null, [GetOpt::SETTING_STRICT_OPERANDS => true]);

        $this->addMainOptions($this->getOpt);
        $this->addExpertOptions($this->getOpt);
        $this->addDevOptions($this->getOpt);

        // process arguments and catch user errors
        try {
            $this->getOpt->process();
        } catch (ArgumentException $exception) {
            file_put_contents('php://stderr', $exception->getMessage() . PHP_EOL);
            echo PHP_EOL . $this->getOpt->getHelpText();
            exit;
        }

        // show version and quit
        if ($this->getOpt->getOption('version')) {
            echo sprintf('PHP Network Weathermap %s' . PHP_EOL, WEATHERMAP_VERSION);
            exit;
        }

        // show help and quit
        if ($this->getOpt->getOption('help')) {
            echo $this->getOpt->getHelpText();
            exit;
        }
    }

    /**
     * @param GetOpt $opt
     */
    private function addMainOptions($opt)
    {
        $opt->addOptions(
            array(
                Option::create(null, 'version', GetOpt::NO_ARGUMENT)
                    ->setDescription('Show version info and quit'),
                Option::create('h', 'help', GetOpt::NO_ARGUMENT)
                    ->setDescription('Show this help and quit'),

                Option::create(null, 'config', GetOpt::REQUIRED_ARGUMENT)
                    ->setDescription('filename to read from. Default weathermap.conf')
                    ->setArgumentName('filename')
                    ->setDefaultValue('weathermap.conf'),
                Option::create(null, 'output', GetOpt::REQUIRED_ARGUMENT)
                    ->setDescription('filename to write image. Default weathermap.png')
                    ->setArgumentName('filename')
                    ->setDefaultValue(''),
                Option::create(null, 'htmloutput', GetOpt::REQUIRED_ARGUMENT)
                    ->setDescription('filename to write HTML. Default weathermap.html')
                    ->setArgumentName('filename')
                    ->setDefaultValue(''),
                Option::create(null, 'image-uri', GetOpt::REQUIRED_ARGUMENT)
                    ->setArgumentName('uri')
                    ->setDescription('URI to prefix <img> tags in HTML output'),
            )
        );
    }

    /**
     * @param GetOpt $opt
     */
    private function addExpertOptions($opt)
    {
        $opt->addOptions(
            array(
            Option::create(null, 'define', GetOpt::MULTIPLE_ARGUMENT)
                ->setArgumentName('name=value')
                ->setDescription('Define internal variables (equivalent to global SET in config file)'),

            Option::create(null, 'bulge', GetOpt::NO_ARGUMENT)
                ->setDescription('Enable link-bulging mode. See manual.'),
            Option::create(null, 'no-data', GetOpt::NO_ARGUMENT)
                ->setDescription('skip the data-reading process (just a \'grey\' map)'),
            Option::create(null, 'sizedebug', GetOpt::NO_ARGUMENT)
                ->setDescription('show the maximum possible value instead of the current one'),

            Option::create(null, 'debug', GetOpt::NO_ARGUMENT)
                ->setDescription('produce (LOTS) of debugging information during run'),
            Option::create(null, 'no-warn', GetOpt::REQUIRED_ARGUMENT)
                ->setDescription('suppress warnings with listed errorcodes (comma-separated)')
                ->setArgumentName('WMxxx_errorcode'),
            )
        );
    }

    /**
     * @param GetOpt $opt
     */
    private function addDevOptions($opt)
    {
        $opt->addOptions(
            array(
            Option::create(null, 'randomdata', GetOpt::NO_ARGUMENT)
                ->setDescription('skip the data-reading process, generate random data'),
            Option::create(null, 'uberdebug', GetOpt::NO_ARGUMENT)
                ->setDescription('produce even more debug information'),
            Option::create(null, 'setdebug', GetOpt::NO_ARGUMENT)
                ->setDescription('produce debug information related to map variables (SET)'),

            Option::create(null, 'dump-config', GetOpt::REQUIRED_ARGUMENT)
                ->setArgumentName('filename')
                ->setDescription('(development) dump config to a new file (testing editor)'),
            Option::create(null, 'dump-json', GetOpt::REQUIRED_ARGUMENT)
                ->setArgumentName('filename')
                ->setDescription('(development) dump JSON config to a new file'),
            )
        );
    }

    private function translateRuntimeOptionsToSettings()
    {
        global $weathermap_error_suppress;

        $this->configFile = $this->getOpt->getOption('config');
        $this->htmlFile = $this->getOpt->getOption('htmloutput');
        $this->imageFile = $this->getOpt->getOption('output');
        $this->optionsOutput['imageuri'] = $this->getOpt->getOption('image-uri');

        if ($this->getOpt->getOption('bulge') === 1) {
            $this->optionsOutput['widthmod'] = true;
        }
        if ($this->getOpt->getOption('sizedebug') === 1) {
            $this->optionsOutput['sizedebug'] = true;
        }
        if ($this->getOpt->getOption('no-data') === 1) {
            $this->optionsOutput['sizedebug'] = true;
        }

        if ($this->getOpt->getOption('no-warn') != '') {
            // allow disabling of warnings from the command-line, too (mainly for the rrdtool warning)
            $suppressedWarnings = explode(",", $this->getOpt->getOption('no-warn'));
            foreach ($suppressedWarnings as $s) {
                $weathermap_error_suppress[] = strtoupper($s);
            }
        }

        $defineList = $this->getOpt->getOption('define');
        foreach ($defineList as $define) {
            preg_match("/^([^=]+)=(.*)\s*$/", $define, $matches);
            if (isset($matches[2])) {
                $varname = $matches[1];
                $value = $matches[2];
                MapUtility::debug(">> $varname = '$value'\n");
                // save this for later, so that when the map object exists, it can be defined
                $this->defines[$varname] = $value;
            } else {
                print "WARNING: --define format is:  --define name=value\n";
            }
        }
    }

    private function translateDebugOptionsToSettings()
    {
        global $weathermap_debug_suppress;
        global $weathermap_debugging;

        if ($this->getOpt->getOption('debug') === 1) {
            $this->optionsOutput['debugging'] = true;
        }
        if ($this->getOpt->getOption('uberdebug') === 1) {
            $this->optionsOutput['debugging'] = true;
            // allow ALL trace messages (normally we block some of the chatty ones)
            $weathermap_debug_suppress = array();
        }

        // set this BEFORE we create the map object, so we get the debug output from Reset(), as well
        if (isset($this->optionsOutput['debugging']) && $this->optionsOutput['debugging']) {
            $weathermap_debugging = true;
            // enable assertion handling
            assert_options(ASSERT_ACTIVE, 1);
            assert_options(ASSERT_WARNING, 0);
            assert_options(ASSERT_QUIET_EVAL, 1);

            // Set up the callback
            assert_options(ASSERT_CALLBACK, 'my_assert_handler');

            MapUtility::debug("------------------------------------\n");
            MapUtility::debug("Starting PHP-Weathermap run, with config: $this->configFile\n");
            MapUtility::debug("------------------------------------\n");
        }
    }

    private function makeMap()
    {
        // now stuff in all the others, that we got from getopts
        foreach ($this->optionsOutput as $key => $value) {
            $this->map->$key = $value;
        }

        if ($this->map->readConfig($this->configFile)) {
            $this->mapFileSettings();
            $this->mapSettingsPostConfig();
            $this->getMapData();

            // TODO: it would be good if this used MapRuntime (would need a stub ApplicationInterface)

            if ($this->imageFile != '') {
                $this->map->drawMap($this->imageFile);
                $this->map->imagefile = $this->imageFile;
            }

            $this->outputHTML();

            return true;
        }
        return false;
    }

    private function mapFileSettings()
    {
        // allow command-lines to override the config file, but provide a default if neither are present
        $this->imageFile = $this->imageFile ?: $this->map->imageoutputfile ?: "weathermap.png";
        $this->htmlFile = $this->htmlFile ?: $this->map->htmloutputfile ?: "";
    }

    private function mapSettingsPostConfig()
    {
        // feed in any command-line defaults, so that they appear as if SET lines in the config
        foreach ($this->defines as $hintname => $hint) {
            $this->map->addHint($hintname, $hint);
        }

        // now stuff in all the others, that we got from getopts
        foreach ($this->optionsOutput as $key => $value) {
            $this->map->addHint($key, $value);
        }
    }

    private function getMapData()
    {
        if ($this->map->sizedebug) {
            return;
        }

        if ($this->getOpt->getOption('randomdata') === 1) {
            $this->map->randomData();
            return;
        }

        $this->map->ReadData();
    }

    private function outputHTML()
    {
        if ($this->htmlFile != '') {
            MapUtility::debug("Writing HTML to $this->htmlFile\n");

            $fd = fopen($this->htmlFile, 'w');
            fwrite(
                $fd,
                '<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd"><html xmlns="http://www.w3.org/1999/xhtml"><head>'
            );
            if ($this->map->htmlstylesheet != '') {
                fwrite($fd, '<link rel="stylesheet" type="text/css" href="' . $this->map->htmlstylesheet . '" />');
            }
            fwrite(
                $fd,
                '<meta http-equiv="refresh" content="300" /><title>' . $this->map->processString(
                    $this->map->title,
                    $this->map
                ) . '</title></head><body>'
            );

            if ($this->map->htmlstyle == "overlib") {
                fwrite(
                    $fd,
                    "<div id=\"overDiv\" style=\"position:absolute; visibility:hidden; z-index:1000;\"></div>\n"
                );
                fwrite(
                    $fd,
                    "<script type=\"text/javascript\" src=\"overlib.js\"><!-- overLIB (c) Erik Bosrup --></script> \n"
                );
            }

            fwrite($fd, $this->map->makeHTML());
            fwrite(
                $fd,
                '<hr /><span id="byline">Network Map created with <a href="http://www.network-weathermap.com/?vs=' . WEATHERMAP_VERSION . '">PHP Network Weathermap v' . WEATHERMAP_VERSION . '</a></span></body></html>'
            );
            fclose($fd);
        }
    }

    private function postRun()
    {
        if ($this->getOpt->getOption('dump-config') != '') {
            $this->map->writeConfig($this->getOpt->getOption('dump-config'));
        }

        if ($this->getOpt->getOption('dump-json') != '') {
            $fd = fopen($this->getOpt->getOption('dump-json'), "w");
            fputs($fd, $this->map->getJSONConfig());
            fclose($fd);
        }

        if ($this->map->dataoutputfile != '') {
            $this->map->writeDataFile($this->map->dataoutputfile);
        }

        if ($this->getOpt->getOption('setdebug') === 1) {
            $this->dumpSetDebugInfo();
        }
    }

    private function dumpSetDebugInfo()
    {
        foreach ($this->map->buildAllItemsList() as $item) {
            print "$item->name :\n";
            foreach ($item->hints as $n => $v) {
                print "  SET $n = $v\n";
            }
            foreach ($item->notes as $n => $v) {
                print "  -> $n = $v\n";
            }
            print "\n";
        }
    }
}