Ability to set optional|drop-able columns

jlslew · jlslew · commit 51a686f7dc70 · 2022-09-07T11:15:26.000+04:00
diff --git a/src/Symfony/Component/Console/Helper/Table.php b/src/Symfony/Component/Console/Helper/Table.php
@@ -50,6 +50,9 @@ class Table
     private array $columnStyles = [];
     private array $columnWidths = [];
     private array $columnMaxWidths = [];
+    private array $optionalColumns = [];
+    private array $droppedColumns = [];
+    private int $maxWidth = 0;
     private bool $rendered = false;
     private string $displayOrientation = self::DISPLAY_ORIENTATION_DEFAULT;
 
@@ -174,6 +177,18 @@ public function setColumnMaxWidth(int $columnIndex, int $width): static
         return $this;
     }
 
+    public function setOptionalColumns(array $columns): static
+    {
+        $this->optionalColumns = $columns;
+        return $this;
+    }
+
+    public function setMaxWidth(int $maxWidth): static
+    {
+        $this->maxWidth = $maxWidth;
+        return $this;
+    }
+
     /**
      * @return $this
      */
@@ -402,6 +417,16 @@ public function render()
                     continue;
                 }
 
+                if ($this->droppedColumns) {
+                    foreach ($this->droppedColumns as $column) {
+                        if ($this->numberOfColumns < count($row)) {
+                            unset($row[$column]);
+                        }
+                    }
+
+                    $row = array_values($row);
+                }
+
                 if ($isHeader && !$isHeaderSeparatorRendered) {
                     $this->renderRowSeparator(
                         $isHeader ? self::SEPARATOR_TOP : self::SEPARATOR_TOP_BOTTOM,
@@ -815,6 +840,36 @@ private function calculateColumnsWidth(iterable $groups)
 
             $this->effectiveColumnWidths[$column] = max($lengths) + Helper::width($this->style->getCellRowContentFormat()) - 2;
         }
+
+        $effectiveColumnWidths = $this->effectiveColumnWidths;
+
+        for ($column = $this->numberOfColumns; $this->numberOfColumns > 0; --$column) {
+            if ($this->maxWidth && $this->maxWidth > array_sum($effectiveColumnWidths)) {
+                break;
+            }
+
+            $droppedColumn = array_pop($this->optionalColumns);
+            unset($effectiveColumnWidths[$droppedColumn]);
+            $this->droppedColumns[] = $droppedColumn;
+        }
+
+        if ($this->droppedColumns) {
+            foreach ($this->droppedColumns as $column) {
+                unset(
+                    $this->effectiveColumnWidths[$column],
+                    $this->columnMaxWidths[$column],
+                    $this->columnStyles[$column],
+                    $this->columnWidths[$column],
+                );
+            }
+
+            $this->effectiveColumnWidths = array_values($this->effectiveColumnWidths);
+            $this->columnMaxWidths = array_values($this->columnMaxWidths);
+            $this->columnStyles = array_values($this->columnStyles);
+            $this->columnWidths = array_values($this->columnWidths);
+
+            $this->numberOfColumns -= count($this->droppedColumns);
+        }
     }
 
     private function getColumnSeparatorWidth(): int
