icingaweb2-module-businessprocess/library/Businessprocess/Storage/LegacyStorage.php

<?php

namespace Icinga\Module\Businessprocess\Storage;

use Icinga\Application\Icinga;
use Icinga\Data\ConfigObject;
use Icinga\Exception\ConfigurationError;
use Icinga\Module\Businessprocess\BusinessProcess;
use Icinga\Module\Businessprocess\BpNode;
use Icinga\Module\Businessprocess\Storage\Storage;
use DirectoryIterator;
use Icinga\Exception\SystemPermissionException;
use Icinga\Application\Benchmark;

class LegacyStorage extends Storage
{
    protected $configDir;

    protected $parsing_line_number;

    protected $currentFilename;

    public function getConfigDir()
    {
        if ($this->configDir === null) {
            $dir = Icinga::app()
                ->getModuleManager()
                ->getModule('businessprocess')
                ->getConfigDir();

            // TODO: This is silly. We need Config::requireDirectory().
            if (! is_dir($dir)) {
                if (! is_dir(dirname($dir))) {
                    if (! @mkdir(dirname($dir))) {
                        throw new SystemPermissionException('Could not create config directory "%s"', dirname($dir));
                    }
                }
                if (! mkdir($dir)) {
                    throw new SystemPermissionException('Could not create config directory "%s"', $dir);
                }
            }
            $dir = $dir . '/processes';
            if (! is_dir($dir)) {
                if (! mkdir($dir)) {
                    throw new SystemPermissionException('Could not create config directory "%s"', $dir);
                }
            }
            $this->configDir = $dir;
        }
        return $this->configDir;
    }

    /**
     * @return array
     */
    public function listProcesses()
    {
        $files = array();
        foreach (new DirectoryIterator($this->getConfigDir()) as $file) {
            if($file->isDot()) continue;
            $filename = $file->getFilename();
            if (substr($filename, -5) === '.conf') {
                $name = substr($filename, 0, -5);
                $header = $this->readHeader($file->getPathname(), $name);
                if ($header['Title'] === null) {
                    $files[$name] = $name;
                } else {
                    $files[$name] = sprintf('%s (%s)', $header['Title'], $name);
                }
            }
        }
        return $files;
    }

    protected function readHeader($file, $name)
    {
        $fh = fopen($file, 'r');
        $cnt = 0;
        $header = array(
            'Title'     => null,
            'Owner'     => null,
            'Backend'   => null,
            'Statetype' => 'soft',
            'SLA Hosts' => null
        );
        while ($cnt < 15 && false !== ($line = fgets($fh))) {
            $cnt++;
            if (preg_match('/^\s*#\s+(.+?)\s*:\s*(.+)$/', $line, $m)) {
                if (array_key_exists($m[1], $header)) {
                    $header[$m[1]] = $m[2];
                }
            }
        }
        return $header;
    }

    /**
     */
    public function storeProcess(BusinessProcess $process)
    {
        $filename = $this->getFilename($process->getName());
        $content = $process->toLegacyConfigString();
        file_put_contents(
            $filename,
            $content
        );
    }

    public function getSource($name)
    {
        return file_get_contents($this->getFilename($name));
    }

    public function getFilename($name)
    {
        return $this->getConfigDir() . '/' . $name . '.conf';
    }

    public function loadFromString($name, $string)
    {
        $bp = new BusinessProcess();
        $bp->setName($name);
        $this->parseString($string, $bp);
        $this->loadHeader($name, $bp);
        return $bp;
    }

    public function deleteProcess($name)
    {
        unlink($this->getFilename($name));
    }

    /**
     * @return BusinessProcess
     */
    public function loadProcess($name)
    {
        Benchmark::measure('Loading business process ' . $name);
        $bp = new BusinessProcess();
        $bp->setName($name);
        $this->parseFile($name, $bp);
        $this->loadHeader($name, $bp);
        Benchmark::measure('Business process ' . $name . ' loaded');
        return $bp;
    }

    protected function loadHeader($name, $bp)
    {
        // TODO: do not open twice, this is quick and dirty based on existing code
        $file = $this->currentFilename = $this->getFilename($name);
        $header = $this->readHeader($file, $name);
        $bp->setTitle($header['Title']);
        if ($header['Backend']) {
            $bp->setBackendName($header['Backend']);
        }
        if ($header['Statetype'] === 'soft') {
            $bp->useSoftStates();
        }
    }

    protected function parseFile($name, $bp)
    {
        $file = $this->currentFilename = $this->getFilename($name);
        $fh = @fopen($file, 'r');
        if (! $fh) {
            throw new SystemPermissionException('Could not open "%s"', $file);
        }

        $this->parsing_line_number = 0;
        while ($line = fgets($fh)) {
            $this->parseLine($line, $bp);
        }

        fclose($fh);
        unset($this->parsing_line_number);
        unset($this->currentFilename);
    }

    protected function parseString($string, $bp)
    {
        foreach (preg_split('/\n/', $string) as $line) {
            $this->parseLine($line, $bp);
        }
    }

    protected function parseLine(& $line, $bp)
    {
        $line = trim($line);

        $this->parsing_line_number++;

        if (empty($line)) {
            return;
        }
        if ($line[0] === '#') {
            return;
        }

        // TODO: substr?
        if (preg_match('~^display~', $line)) {
            list($display, $name, $desc) = preg_split('~\s*;\s*~', substr($line, 8), 3);
            $node = $bp->getNode($name)->setAlias($desc)->setDisplay($display);
            if ($display > 0) {
                $bp->addRootNode($name);
            }
            return;
        }

        if (preg_match('~^external_info~', $line)) {
            list($name, $script) = preg_split('~\s*;\s*~', substr($line, 14), 2);
            $node = $bp->getNode($name)->setInfoCommand($script);
            return;
        }

        // New feature:
        // if (preg_match('~^extra_info~', $line)) {
        //     list($name, $script) = preg_split('~\s*;\s*~', substr($line, 14), 2);
        //     $node = $this->getNode($name)->setExtraInfo($script);
        // }

        if (preg_match('~^info_url~', $line)) {
            list($name, $url) = preg_split('~\s*;\s*~', substr($line, 9), 2);
            $node = $bp->getNode($name)->setInfoUrl($url);
            return;
        }

        list($name, $value) = preg_split('~\s*=\s*~', $line, 2);

        if (strpos($name, ';') !== false) {
            $this->parseError('No semicolon allowed in varname');
        }

        $op = '&';
        if (preg_match_all('~([\|\+&\!])~', $value, $m)) {
            $op = implode('', $m[1]);
            for ($i = 1; $i < strlen($op); $i++) {
                if ($op[$i] !== $op[$i - 1]) {
                    $this->parseError('Mixing operators is not allowed');
                }
            }
        }
        $op = $op[0];
        $op_name = $op;

        if ($op === '+') {
            if (! preg_match('~^(\d+)\s*of:\s*(.+?)$~', $value, $m)) {
                $this->parseError('syntax: <var> = <num> of: <var1> + <var2> [+ <varn>]*');
            }
            $op_name = $m[1];
            $value   = $m[2];
        }
        $cmps = preg_split('~\s*\\' . $op . '\s*~', $value, -1, PREG_SPLIT_NO_EMPTY);

        foreach ($cmps as & $val) {
            if (strpos($val, ';') !== false) {
                if ($bp->hasNode($val)) continue;

                list($host, $service) = preg_split('~;~', $val, 2);
                if ($service === 'Hoststatus') {
                    $bp->createHost($host);
                } else {
                    $bp->createService($host, $service);
                }
            }
            if ($val[0] === '@' && strpos($val, ':') !== false) {
                list($config, $nodeName) = preg_split('~:\s*~', substr($val, 1), 2);
                $bp->createImportedNode($config, $nodeName);
                $val = $nodeName;
            }
        }

        $node = new BpNode($bp, (object) array(
            'name'        => $name,
            'operator'    => $op_name,
            'child_names' => $cmps
        ));
        $bp->addNode($name, $node);
    }

    protected function parseError($msg)
    {
        throw new ConfigurationError(
            sprintf(
                'Parse error on %s:%s: %s',
                $this->currentFilename,
                $this->parsing_line_number,
                $msg
            )
        );
    }
}