icingaweb2-module-director/library/Director/Web/Table/ImportedrowsTable.php

66 lines
1.5 KiB
PHP

<?php
namespace Icinga\Module\Director\Web\Table;
use Icinga\Data\DataArray\ArrayDatasource;
use Icinga\Module\Director\Objects\ImportRun;
use ipl\Web\Table\SimpleQueryBasedTable;
class ImportedrowsTable extends SimpleQueryBasedTable
{
protected $columns;
/** @var ImportRun */
protected $importRun;
public static function load(ImportRun $run)
{
$table = new static();
$table->setImportRun($run);
return $table;
}
public function setImportRun(ImportRun $run)
{
$this->importRun = $run;
return $this;
}
public function setColumns($columns)
{
$this->columns = $columns;
return $this;
}
public function getColumns()
{
if ($this->columns === null) {
$cols = $this->importRun->listColumnNames();
$keyColumn = $this->importRun->importSource()->get('key_column');
if ($keyColumn !== null && ($pos = array_search($keyColumn, $cols)) !== false) {
unset($cols[$pos]);
array_unshift($cols, $keyColumn);
}
} else {
$cols = $this->columns;
}
return array_combine($cols, $cols);
}
public function getColumnsToBeRendered()
{
return $this->getColumns();
}
public function prepareQuery()
{
$ds = new ArrayDatasource(
$this->importRun->fetchRows($this->columns)
);
return $ds->select()->order('object_name');
}
}