icingaweb2/application/controllers/TestController.php

<?php

namespace Icinga\Controllers;

use Icinga\Application\Benchmark;
use Icinga\Application\Icinga;
use Icinga\Data\Filter\Filter;
use Icinga\Module\Icingadb\Common\Database;
use Icinga\Module\Icingadb\Model\Host;
use Icinga\Module\Icingadb\Model\Service;
use Icinga\Util\Environment;
use Icinga\Util\Format;
use Icinga\Util\Json;
use Icinga\Web\Notification;
use ipl\Html\Html;
use ipl\Sql\Expression;
use ipl\Web\Compat\CompatController;
use ipl\Web\Filter\QueryString;
use Psr\Http\Message\ResponseInterface;
use React\Http\Browser;
use React\Socket\FixedUriConnector;
use React\Socket\UnixConnector;

use function Clue\React\Block\await;

class TestController extends CompatController
{
    public function separateRefreshAction()
    {
        $this->view->content = 'About to be...';

        $this->getTabs()->add('separate_refresh', [
            'label'     => 'Separate Refresh',
            'url'       => 'test/separate-refresh',
            'active'    => true
        ]);
    }

    public function separateContentAction()
    {
        $this->view->content = 'Refreshed!';

        $this->setAutorefreshInterval(10);
        $this->_helper->layout()->disableLayout();
    }

    public function memErrorAction()
    {
        ini_set('memory_limit', Format::unpackShorthandBytes('10MB'));

        $iterations = 10;

        $s = '';
        while (true) {
            $s .= 'aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa';
        }


        $images = [];
        for ($i = 0; $i < $iterations; $i++) {
            //$images[] = file_get_contents(Icinga::app()->getBaseDir(). '/public/img/icingaweb2-background.jpg');
        }

        die('works');
    }

    public function nullfilterAction()
    {
        $obj = (object) ['foo' => null];
        //$filter = Filter::expression('foo', '=', '*');
        $filter = Filter::where('foo', '*');
        $filter->setCaseSensitive(false);
        $filter->matches($obj);
        exit;
    }

    public function notificationTestAction()
    {
        Notification::error('Lorem ipsum dolor sit amet, consetetur sadipscing elitr, sed diam nonumy eirmod tempor invidunt ut l');
        Notification::info('Lorem ipsum dolor sit amet, consetetur sadipscing elitr, sed diam nonumy eirmod tempor invidunt ut l');
        Notification::warning('Lorem ipsum dolor sit amet, consetetur sadipscing elitr, sed diam nonumy eirmod tempor invidunt ut l');
        Notification::success('Lorem ipsum dolor sit amet, consetetur sadipscing elitr, sed diam nonumy eirmod tempor invidunt ut l');
    }

    public function testAction()
    {
        $this->view->title = 'Test';
    }

    public function forwardAction()
    {
        $this->view->title = 'foo';
        $this->forward('test');
    }

    public function daemonAction()
    {
        $connector = new FixedUriConnector(
            'unix:///run/icingawebd.socket',
            new UnixConnector()
        );

        $browser = new Browser($connector);

        $response = await($browser->get('http://localhost/info'));
        /** @var ResponseInterface $response */
        echo $response->getBody(); exit;
    }



    use Database;
    protected $requiresAuthentication = false;

    public function objectIdsAction()
    {
        $type = $this->params->shiftRequired('type');
        $objectId = $this->params->shift('object_id');
        $limit = $this->params->shift('limit');

        $roles = Json::decode(file_get_contents(Icinga::app()->getConfigDir() . '/' . 'roles.json'));

        switch ($type) {
            case 'host':
                $query = Host::on($this->getDb());
                break;
            case 'service':
                $query = Service::on($this->getDb());
                break;
            default:
                $this->httpBadRequest('invalid type');
        }

        $query->columns('id');
        if ($limit) {
            $query->limit((int) $limit);
        }

        $userQueries = [];
        foreach ($roles as $username => $restrictions) {
            if (substr($username, 0, 1) === '_') {
                continue;
            }

            $filter = \ipl\Stdlib\Filter::any();
            foreach ($restrictions as $restriction) {
                if (substr($restriction, 0, 1) === '_') {
                    continue;
                }

                $filter->add(QueryString::parse($restriction));
            }

            $subQuery = clone $query;
            $subQuery->getResolver()->setAliasPrefix("${username}_");
            $linkToParentQuery = [sprintf('%s.id = %s.id', $query->getResolver()->getAlias($query->getModel()), $subQuery->getResolver()->getAlias($subQuery->getModel()))];

            list($stmt, $values) = $this->getDb()->getQueryBuilder()->assembleSelect(
                $subQuery
                    ->columns([new Expression('1')])
                    ->filter($filter)
                    ->limit(1)
                    ->assembleSelect()
                    ->where($linkToParentQuery)
                    ->resetOrderBy()
            );
            $userQueries[$username] = new Expression($stmt, null, ...$values);
        }

        if ($objectId) {
            $query->filter(\ipl\Stdlib\Filter::equal('id', hex2bin($objectId)));
        }

        $query->withColumns($userQueries);
        $query->disableDefaultSort();

        //file_get_contents('http://localhost/');
        //usleep(200000);


        if ($this->params->get('export') === 'sql') {
            list($sql, $values) = $query->dump();

            $unused = [];
            foreach ($values as $value) {
                $pos = strpos($sql, '?');
                if ($pos !== false) {
                    if (is_string($value)) {
                        $value = "'" . $value . "'";
                    }

                    $sql = substr_replace($sql, $value, $pos, 1);
                } else {
                    $unused[] = $value;
                }
            }

            if (! empty($unused)) {
                $sql .= ' /* Unused values: "' . join('", "', $unused) . '" */';
            }

            $this->addContent(Html::tag('pre', $sql));
        } elseif ($this->params->get('export') === 'json') {
            $this->getResponse()
                ->setHeader('Content-Type', 'application/json')
                ->setHeader('Cache-Control', 'no-store')
                ->setHeader('Content-Disposition', 'inline')
                ->sendResponse();

            ob_end_clean();
            Environment::raiseExecutionTime();
            Environment::raiseMemoryLimit('1024M');

            echo '[';
            foreach ($query as $i => $result) {
                $users = [];
                foreach ($roles as $username => $_) {
                    if (substr($username, 0, 1) === '_') {
                        continue;
                    }

                    if ($result->$username) {
                        $users[] = $username;
                    }
                }

                if ($i > 0) {
                    echo PHP_EOL . ',';
                }

                echo json_encode([
                    'id' => bin2hex($result->id),
                    'users' => $users
                ]);
            }

            echo ']';
            exit;
        } else {
            Benchmark::measure('starting query');
            foreach ($query as $i => $result) {
                if ($i === 0) {
                    Benchmark::measure('starting loop');
                }
            }

            Benchmark::measure('finished loop');

            Benchmark::dump();
            exit;
        }
    }
}