Optimize grid total row

Author: z-song

resources/views/grid/total-row.blade.php

@@ -1,7 +1,7 @@
 <tfoot>
     <tr>
         @foreach($columns as $column)
-            <td>{!! $column !!}</td>
+            <td class="{{ $column['class'] }}">{!! $column['value'] !!}</td>
         @endforeach
     </tr>
 </tfoot>

src/Grid/Concerns/HasTotalRow.php

@@ -28,7 +28,7 @@ public function addTotalRow($column, $callback)
     /**
      * @return \Illuminate\Contracts\View\Factory|\Illuminate\View\View|string
      */
-    public function renderTotalRow()
+    public function renderTotalRow($columns = null)
     {
         if (empty($this->totalRowColumns)) {
             return '';
@@ -40,6 +40,10 @@ public function renderTotalRow()
 
         $totalRow->setGrid($this);
 
+        if ($columns) {
+            $totalRow->setVisibleColumns($columns);
+        }
+
         return $totalRow->render();
     }
 }

src/Grid/Tools/TotalRow.php

@@ -5,6 +5,7 @@
 use Encore\Admin\Grid\Column;
 use Illuminate\Database\Query\Builder;
 use Illuminate\Support\Arr;
+use Illuminate\Support\Collection;
 
 class TotalRow extends AbstractTool
 {
@@ -18,6 +19,11 @@ class TotalRow extends AbstractTool
      */
     protected $columns;
 
+    /**
+     * @var Collection
+     */
+    protected $visibleColumns;
+
     /**
      * TotalRow constructor.
      *
@@ -54,14 +60,34 @@ protected function total($column, $display = null)
         return $sum;
     }
 
+    /**
+     * @param Collection $columns
+     */
+    public function setVisibleColumns($columns)
+    {
+        $this->visibleColumns = $columns;
+    }
+
+    /**
+     * @return Collection|static
+     */
+    public function getVisibleColumns()
+    {
+        if ($this->visibleColumns) {
+            return $this->visibleColumns;
+        }
+
+        return $this->getGrid()->visibleColumns();
+    }
+
     /**
      * Render total-row.
      *
      * @return \Illuminate\Contracts\View\Factory|\Illuminate\View\View
      */
     public function render()
     {
-        $columns = $this->getGrid()->visibleColumns()->flatMap(function (Column $column) {
+        $columns = $this->getVisibleColumns()->map(function (Column $column) {
             $name = $column->getName();
 
             $total = '';
@@ -70,8 +96,11 @@ public function render()
                 $total = $this->total($name, Arr::get($this->columns, $name));
             }
 
-            return [$name => $total];
-        })->toArray();
+            return [
+                'class' => $column->getClassName(),
+                'value' => $total
+            ];
+        });
 
         return view('admin::grid.total-row', compact('columns'));
     }