123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668 |
- <?php
- /*
- * This file is part of the Symfony package.
- *
- * (c) Fabien Potencier <fabien@symfony.com>
- *
- * For the full copyright and license information, please view the LICENSE
- * file that was distributed with this source code.
- */
- namespace Symfony\Component\Console\Helper;
- use Symfony\Component\Console\Output\OutputInterface;
- use Symfony\Component\Console\Exception\InvalidArgumentException;
- /**
- * Provides helpers to display a table.
- *
- * @author Fabien Potencier <fabien@symfony.com>
- * @author Саша Стаменковић <umpirsky@gmail.com>
- * @author Abdellatif Ait boudad <a.aitboudad@gmail.com>
- * @author Max Grigorian <maxakawizard@gmail.com>
- */
- class Table
- {
- /**
- * Table headers.
- *
- * @var array
- */
- private $headers = array();
- /**
- * Table rows.
- *
- * @var array
- */
- private $rows = array();
- /**
- * Column widths cache.
- *
- * @var array
- */
- private $columnWidths = array();
- /**
- * Number of columns cache.
- *
- * @var array
- */
- private $numberOfColumns;
- /**
- * @var OutputInterface
- */
- private $output;
- /**
- * @var TableStyle
- */
- private $style;
- /**
- * @var array
- */
- private $columnStyles = array();
- private static $styles;
- public function __construct(OutputInterface $output)
- {
- $this->output = $output;
- if (!self::$styles) {
- self::$styles = self::initStyles();
- }
- $this->setStyle('default');
- }
- /**
- * Sets a style definition.
- *
- * @param string $name The style name
- * @param TableStyle $style A TableStyle instance
- */
- public static function setStyleDefinition($name, TableStyle $style)
- {
- if (!self::$styles) {
- self::$styles = self::initStyles();
- }
- self::$styles[$name] = $style;
- }
- /**
- * Gets a style definition by name.
- *
- * @param string $name The style name
- *
- * @return TableStyle
- */
- public static function getStyleDefinition($name)
- {
- if (!self::$styles) {
- self::$styles = self::initStyles();
- }
- if (isset(self::$styles[$name])) {
- return self::$styles[$name];
- }
- throw new InvalidArgumentException(sprintf('Style "%s" is not defined.', $name));
- }
- /**
- * Sets table style.
- *
- * @param TableStyle|string $name The style name or a TableStyle instance
- *
- * @return $this
- */
- public function setStyle($name)
- {
- $this->style = $this->resolveStyle($name);
- return $this;
- }
- /**
- * Gets the current table style.
- *
- * @return TableStyle
- */
- public function getStyle()
- {
- return $this->style;
- }
- /**
- * Sets table column style.
- *
- * @param int $columnIndex Column index
- * @param TableStyle|string $name The style name or a TableStyle instance
- *
- * @return $this
- */
- public function setColumnStyle($columnIndex, $name)
- {
- $columnIndex = intval($columnIndex);
- $this->columnStyles[$columnIndex] = $this->resolveStyle($name);
- return $this;
- }
- /**
- * Gets the current style for a column.
- *
- * If style was not set, it returns the global table style.
- *
- * @param int $columnIndex Column index
- *
- * @return TableStyle
- */
- public function getColumnStyle($columnIndex)
- {
- if (isset($this->columnStyles[$columnIndex])) {
- return $this->columnStyles[$columnIndex];
- }
- return $this->getStyle();
- }
- public function setHeaders(array $headers)
- {
- $headers = array_values($headers);
- if (!empty($headers) && !is_array($headers[0])) {
- $headers = array($headers);
- }
- $this->headers = $headers;
- return $this;
- }
- public function setRows(array $rows)
- {
- $this->rows = array();
- return $this->addRows($rows);
- }
- public function addRows(array $rows)
- {
- foreach ($rows as $row) {
- $this->addRow($row);
- }
- return $this;
- }
- public function addRow($row)
- {
- if ($row instanceof TableSeparator) {
- $this->rows[] = $row;
- return $this;
- }
- if (!is_array($row)) {
- throw new InvalidArgumentException('A row must be an array or a TableSeparator instance.');
- }
- $this->rows[] = array_values($row);
- return $this;
- }
- public function setRow($column, array $row)
- {
- $this->rows[$column] = $row;
- return $this;
- }
- /**
- * Renders table to output.
- *
- * Example:
- * +---------------+-----------------------+------------------+
- * | ISBN | Title | Author |
- * +---------------+-----------------------+------------------+
- * | 99921-58-10-7 | Divine Comedy | Dante Alighieri |
- * | 9971-5-0210-0 | A Tale of Two Cities | Charles Dickens |
- * | 960-425-059-0 | The Lord of the Rings | J. R. R. Tolkien |
- * +---------------+-----------------------+------------------+
- */
- public function render()
- {
- $this->calculateNumberOfColumns();
- $rows = $this->buildTableRows($this->rows);
- $headers = $this->buildTableRows($this->headers);
- $this->calculateColumnsWidth(array_merge($headers, $rows));
- $this->renderRowSeparator();
- if (!empty($headers)) {
- foreach ($headers as $header) {
- $this->renderRow($header, $this->style->getCellHeaderFormat());
- $this->renderRowSeparator();
- }
- }
- foreach ($rows as $row) {
- if ($row instanceof TableSeparator) {
- $this->renderRowSeparator();
- } else {
- $this->renderRow($row, $this->style->getCellRowFormat());
- }
- }
- if (!empty($rows)) {
- $this->renderRowSeparator();
- }
- $this->cleanup();
- }
- /**
- * Renders horizontal header separator.
- *
- * Example: +-----+-----------+-------+
- */
- private function renderRowSeparator()
- {
- if (0 === $count = $this->numberOfColumns) {
- return;
- }
- if (!$this->style->getHorizontalBorderChar() && !$this->style->getCrossingChar()) {
- return;
- }
- $markup = $this->style->getCrossingChar();
- for ($column = 0; $column < $count; ++$column) {
- $markup .= str_repeat($this->style->getHorizontalBorderChar(), $this->columnWidths[$column]).$this->style->getCrossingChar();
- }
- $this->output->writeln(sprintf($this->style->getBorderFormat(), $markup));
- }
- /**
- * Renders vertical column separator.
- */
- private function renderColumnSeparator()
- {
- return sprintf($this->style->getBorderFormat(), $this->style->getVerticalBorderChar());
- }
- /**
- * Renders table row.
- *
- * Example: | 9971-5-0210-0 | A Tale of Two Cities | Charles Dickens |
- *
- * @param array $row
- * @param string $cellFormat
- */
- private function renderRow(array $row, $cellFormat)
- {
- if (empty($row)) {
- return;
- }
- $rowContent = $this->renderColumnSeparator();
- foreach ($this->getRowColumns($row) as $column) {
- $rowContent .= $this->renderCell($row, $column, $cellFormat);
- $rowContent .= $this->renderColumnSeparator();
- }
- $this->output->writeln($rowContent);
- }
- /**
- * Renders table cell with padding.
- *
- * @param array $row
- * @param int $column
- * @param string $cellFormat
- */
- private function renderCell(array $row, $column, $cellFormat)
- {
- $cell = isset($row[$column]) ? $row[$column] : '';
- $width = $this->columnWidths[$column];
- if ($cell instanceof TableCell && $cell->getColspan() > 1) {
- // add the width of the following columns(numbers of colspan).
- foreach (range($column + 1, $column + $cell->getColspan() - 1) as $nextColumn) {
- $width += $this->getColumnSeparatorWidth() + $this->columnWidths[$nextColumn];
- }
- }
- // str_pad won't work properly with multi-byte strings, we need to fix the padding
- if (false !== $encoding = mb_detect_encoding($cell, null, true)) {
- $width += strlen($cell) - mb_strwidth($cell, $encoding);
- }
- $style = $this->getColumnStyle($column);
- if ($cell instanceof TableSeparator) {
- return sprintf($style->getBorderFormat(), str_repeat($style->getHorizontalBorderChar(), $width));
- }
- $width += Helper::strlen($cell) - Helper::strlenWithoutDecoration($this->output->getFormatter(), $cell);
- $content = sprintf($style->getCellRowContentFormat(), $cell);
- return sprintf($cellFormat, str_pad($content, $width, $style->getPaddingChar(), $style->getPadType()));
- }
- /**
- * Calculate number of columns for this table.
- */
- private function calculateNumberOfColumns()
- {
- if (null !== $this->numberOfColumns) {
- return;
- }
- $columns = array(0);
- foreach (array_merge($this->headers, $this->rows) as $row) {
- if ($row instanceof TableSeparator) {
- continue;
- }
- $columns[] = $this->getNumberOfColumns($row);
- }
- $this->numberOfColumns = max($columns);
- }
- private function buildTableRows($rows)
- {
- $unmergedRows = array();
- for ($rowKey = 0; $rowKey < count($rows); ++$rowKey) {
- $rows = $this->fillNextRows($rows, $rowKey);
- // Remove any new line breaks and replace it with a new line
- foreach ($rows[$rowKey] as $column => $cell) {
- if (!strstr($cell, "\n")) {
- continue;
- }
- $lines = explode("\n", $cell);
- foreach ($lines as $lineKey => $line) {
- if ($cell instanceof TableCell) {
- $line = new TableCell($line, array('colspan' => $cell->getColspan()));
- }
- if (0 === $lineKey) {
- $rows[$rowKey][$column] = $line;
- } else {
- $unmergedRows[$rowKey][$lineKey][$column] = $line;
- }
- }
- }
- }
- $tableRows = array();
- foreach ($rows as $rowKey => $row) {
- $tableRows[] = $this->fillCells($row);
- if (isset($unmergedRows[$rowKey])) {
- $tableRows = array_merge($tableRows, $unmergedRows[$rowKey]);
- }
- }
- return $tableRows;
- }
- /**
- * fill rows that contains rowspan > 1.
- *
- * @param array $rows
- * @param int $line
- *
- * @return array
- */
- private function fillNextRows($rows, $line)
- {
- $unmergedRows = array();
- foreach ($rows[$line] as $column => $cell) {
- if ($cell instanceof TableCell && $cell->getRowspan() > 1) {
- $nbLines = $cell->getRowspan() - 1;
- $lines = array($cell);
- if (strstr($cell, "\n")) {
- $lines = explode("\n", $cell);
- $nbLines = count($lines) > $nbLines ? substr_count($cell, "\n") : $nbLines;
- $rows[$line][$column] = new TableCell($lines[0], array('colspan' => $cell->getColspan()));
- unset($lines[0]);
- }
- // create a two dimensional array (rowspan x colspan)
- $unmergedRows = array_replace_recursive(array_fill($line + 1, $nbLines, array()), $unmergedRows);
- foreach ($unmergedRows as $unmergedRowKey => $unmergedRow) {
- $value = isset($lines[$unmergedRowKey - $line]) ? $lines[$unmergedRowKey - $line] : '';
- $unmergedRows[$unmergedRowKey][$column] = new TableCell($value, array('colspan' => $cell->getColspan()));
- if ($nbLines === $unmergedRowKey - $line) {
- break;
- }
- }
- }
- }
- foreach ($unmergedRows as $unmergedRowKey => $unmergedRow) {
- // we need to know if $unmergedRow will be merged or inserted into $rows
- if (isset($rows[$unmergedRowKey]) && is_array($rows[$unmergedRowKey]) && ($this->getNumberOfColumns($rows[$unmergedRowKey]) + $this->getNumberOfColumns($unmergedRows[$unmergedRowKey]) <= $this->numberOfColumns)) {
- foreach ($unmergedRow as $cellKey => $cell) {
- // insert cell into row at cellKey position
- array_splice($rows[$unmergedRowKey], $cellKey, 0, array($cell));
- }
- } else {
- $row = $this->copyRow($rows, $unmergedRowKey - 1);
- foreach ($unmergedRow as $column => $cell) {
- if (!empty($cell)) {
- $row[$column] = $unmergedRow[$column];
- }
- }
- array_splice($rows, $unmergedRowKey, 0, array($row));
- }
- }
- return $rows;
- }
- /**
- * fill cells for a row that contains colspan > 1.
- *
- * @param array $row
- *
- * @return array
- */
- private function fillCells($row)
- {
- $newRow = array();
- foreach ($row as $column => $cell) {
- $newRow[] = $cell;
- if ($cell instanceof TableCell && $cell->getColspan() > 1) {
- foreach (range($column + 1, $column + $cell->getColspan() - 1) as $position) {
- // insert empty value at column position
- $newRow[] = '';
- }
- }
- }
- return $newRow ?: $row;
- }
- /**
- * @param array $rows
- * @param int $line
- *
- * @return array
- */
- private function copyRow($rows, $line)
- {
- $row = $rows[$line];
- foreach ($row as $cellKey => $cellValue) {
- $row[$cellKey] = '';
- if ($cellValue instanceof TableCell) {
- $row[$cellKey] = new TableCell('', array('colspan' => $cellValue->getColspan()));
- }
- }
- return $row;
- }
- /**
- * Gets number of columns by row.
- *
- * @param array $row
- *
- * @return int
- */
- private function getNumberOfColumns(array $row)
- {
- $columns = count($row);
- foreach ($row as $column) {
- $columns += $column instanceof TableCell ? ($column->getColspan() - 1) : 0;
- }
- return $columns;
- }
- /**
- * Gets list of columns for the given row.
- *
- * @param array $row
- *
- * @return array
- */
- private function getRowColumns($row)
- {
- $columns = range(0, $this->numberOfColumns - 1);
- foreach ($row as $cellKey => $cell) {
- if ($cell instanceof TableCell && $cell->getColspan() > 1) {
- // exclude grouped columns.
- $columns = array_diff($columns, range($cellKey + 1, $cellKey + $cell->getColspan() - 1));
- }
- }
- return $columns;
- }
- /**
- * Calculates columns widths.
- *
- * @param array $rows
- */
- private function calculateColumnsWidth($rows)
- {
- for ($column = 0; $column < $this->numberOfColumns; ++$column) {
- $lengths = array();
- foreach ($rows as $row) {
- if ($row instanceof TableSeparator) {
- continue;
- }
- foreach ($row as $i => $cell) {
- if ($cell instanceof TableCell) {
- $textLength = Helper::strlenWithoutDecoration($this->output->getFormatter(), $cell);
- if ($textLength > 0) {
- $contentColumns = str_split($cell, ceil($textLength / $cell->getColspan()));
- foreach ($contentColumns as $position => $content) {
- $row[$i + $position] = $content;
- }
- }
- }
- }
- $lengths[] = $this->getCellWidth($row, $column);
- }
- $this->columnWidths[$column] = max($lengths) + strlen($this->style->getCellRowContentFormat()) - 2;
- }
- }
- /**
- * Gets column width.
- *
- * @return int
- */
- private function getColumnSeparatorWidth()
- {
- return strlen(sprintf($this->style->getBorderFormat(), $this->style->getVerticalBorderChar()));
- }
- /**
- * Gets cell width.
- *
- * @param array $row
- * @param int $column
- *
- * @return int
- */
- private function getCellWidth(array $row, $column)
- {
- if (isset($row[$column])) {
- $cell = $row[$column];
- $cellWidth = Helper::strlenWithoutDecoration($this->output->getFormatter(), $cell);
- return $cellWidth;
- }
- return 0;
- }
- /**
- * Called after rendering to cleanup cache data.
- */
- private function cleanup()
- {
- $this->columnWidths = array();
- $this->numberOfColumns = null;
- }
- private static function initStyles()
- {
- $borderless = new TableStyle();
- $borderless
- ->setHorizontalBorderChar('=')
- ->setVerticalBorderChar(' ')
- ->setCrossingChar(' ')
- ;
- $compact = new TableStyle();
- $compact
- ->setHorizontalBorderChar('')
- ->setVerticalBorderChar(' ')
- ->setCrossingChar('')
- ->setCellRowContentFormat('%s')
- ;
- $styleGuide = new TableStyle();
- $styleGuide
- ->setHorizontalBorderChar('-')
- ->setVerticalBorderChar(' ')
- ->setCrossingChar(' ')
- ->setCellHeaderFormat('%s')
- ;
- return array(
- 'default' => new TableStyle(),
- 'borderless' => $borderless,
- 'compact' => $compact,
- 'symfony-style-guide' => $styleGuide,
- );
- }
- private function resolveStyle($name)
- {
- if ($name instanceof TableStyle) {
- return $name;
- }
- if (isset(self::$styles[$name])) {
- return self::$styles[$name];
- }
- throw new InvalidArgumentException(sprintf('Style "%s" is not defined.', $name));
- }
- }
|