summaryrefslogtreecommitdiffstats
path: root/library/Director/Import/ImportSourceSql.php
blob: b08a3f37691bb6f0e72ba10194a69b7bd05776f7 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
<?php

namespace Icinga\Module\Director\Import;

use gipfl\Web\Widget\Hint;
use Icinga\Data\Db\DbConnection;
use Icinga\Module\Director\Forms\ImportSourceForm;
use Icinga\Module\Director\Hook\ImportSourceHook;
use Icinga\Module\Director\Objects\ImportSource;
use Icinga\Module\Director\Util;
use Icinga\Module\Director\Web\Form\Filter\QueryColumnsFromSql;
use Icinga\Module\Director\Web\Form\QuickForm;
use ipl\Html\Html;

class ImportSourceSql extends ImportSourceHook
{
    protected $db;

    public function fetchData()
    {
        return $this->db()->fetchAll($this->settings['query']);
    }

    public function listColumns()
    {
        if ($columns = $this->getSetting('column_cache')) {
            return explode(', ', $columns);
        } else {
            return array_keys((array) current($this->fetchData()));
        }
    }

    public static function addSettingsFormFields(QuickForm $form)
    {
        /** @var ImportSourceForm $form */
        Util::addDbResourceFormElement($form, 'resource');
        /** @var ImportSource $current */
        $current = $form->getObject();

        $form->addElement('textarea', 'query', [
            'label'    => $form->translate('DB Query'),
            'required' => true,
            'rows'     => 15,
        ]);
        $form->addElement('hidden', 'column_cache', [
            'value'    => '',
            'filters'  => [new QueryColumnsFromSql($form)],
            'required' => true
        ]);
        if ($current) {
            if ($columns = $current->getSetting('column_cache')) {
                $form->addHtmlHint('Columns: ' . $columns);
            } else {
                $form->addHtmlHint(Hint::warning($form->translate(
                    'Please click "Store" once again to determine query columns'
                )));
            }
        }
        return $form;
    }

    protected function db()
    {
        if ($this->db === null) {
            $this->db = DbConnection::fromResourceName($this->settings['resource'])->getDbAdapter();
        }

        return $this->db;
    }
}