Add filters on export queues

src/Jobs/ExportJob.php

@@ -4,7 +4,6 @@
 
 use Illuminate\Bus\{Batchable, Queueable};
 use Illuminate\Contracts\Queue\ShouldQueue;
-use Illuminate\Database\Eloquent\Builder;
 use Illuminate\Foundation\Bus\Dispatchable;
 use Illuminate\Queue\{InteractsWithQueue, SerializesModels};
 use PowerComponents\LivewirePowerGrid\PowerGridComponent;
@@ -35,25 +34,29 @@ public function __construct(
         $this->fileName        = $params['fileName'];
         $this->offset          = $params['offset'];
         $this->limit           = $params['limit'];
+        $this->filters         = $params['filters'];
 
         /** @var PowerGridComponent $componentTable */
         $this->componentTable = new $componentTable();
     }
 
     public function handle(): void
     {
-        /** @var Builder $query */
-        $query = $this->componentTable->datasource();
+        $exportable = new $this->exportableClass();
 
-        $query = $query->offset($this->offset)
-            ->limit($this->limit)
-            ->get();
+        $currentHiddenStates = collect($this->columns)
+            ->mapWithKeys(fn ($column) => [data_get($column, 'field') => data_get($column, 'hidden')]);
 
-        $exportable = new $this->exportableClass();
+        $columnsWithHiddenState = array_map(function ($column) use ($currentHiddenStates) {
+            $column->hidden = $currentHiddenStates[$column->field];
+
+            return $column;
+        }, $this->componentTable->columns());
 
         /** @phpstan-ignore-next-line  */
-        $exportable->fileName($this->getFilename())
-            ->setData($this->columns, $this->transform($query))
+        $exportable
+            ->fileName($this->getFilename())
+            ->setData($columnsWithHiddenState, $this->prepareToExport())
             ->download([]);
     }
 }

src/Traits/ExportableJob.php

@@ -2,9 +2,10 @@
 
 namespace PowerComponents\LivewirePowerGrid\Traits;
 
-use Illuminate\Database\Eloquent\Collection;
-use Illuminate\Support\{Str, Stringable};
-use PowerComponents\LivewirePowerGrid\PowerGridComponent;
+use Illuminate\Database\Eloquent as Eloquent;
+use Illuminate\Support\{Collection, Str, Stringable};
+use PowerComponents\LivewirePowerGrid\DataSource\Builder;
+use PowerComponents\LivewirePowerGrid\{PowerGridComponent, ProcessDataSource};
 
 /** @codeCoverageIgnore */
 trait ExportableJob
@@ -21,23 +22,42 @@ trait ExportableJob
 
     private int $limit;
 
+    private array $filters;
+
     private function getFilename(): Stringable
     {
         return Str::of($this->fileName)
             ->replace('.xlsx', '')
             ->replace('.csv', '');
     }
 
-    private function transform(Collection $collection): Collection
+    private function prepareToExport(): Eloquent\Collection|Collection
     {
-        return $collection->transform(function ($row) {
-            $columns = $this->componentTable->addColumns()->columns;
+        /** @phpstan-ignore-next-line */
+        $processDataSource = tap(ProcessDataSource::fillData($this->componentTable), fn ($datasource) => $datasource->get());
+
+        $inClause = $processDataSource->component->filtered ?? [];
+
+        /** @phpstan-ignore-next-line */
+        $this->componentTable->filters = $this->filters ?? [];
+
+        /** @phpstan-ignore-next-line */
+        $currentTable = $processDataSource->component->currentTable;
+
+        $sortField = Str::of($processDataSource->component->sortField)->contains('.') ? $processDataSource->component->sortField : $currentTable . '.' . $processDataSource->component->sortField;
 
-            foreach ($columns as $key => $column) {
-                $row->{$key} = $column($row);
-            }
+        $results = $processDataSource->prepareDataSource()
+            ->where(
+                fn ($query) => Builder::make($query, $this->componentTable)
+                    ->filterContains()
+                    ->filter()
+            )
+            ->when($inClause, function ($query, $inClause) use ($processDataSource) {
+                return $query->whereIn($processDataSource->component->primaryKey, $inClause);
+            })
+            ->orderBy($sortField, $processDataSource->component->sortDirection)
+            ->get();
 
-            return $row;
-        });
+        return $processDataSource->transform($results);
     }
 }

src/Traits/WithExport.php

@@ -98,10 +98,13 @@ public function runOnQueue(string $exportFileType, string $exportType): bool
 
     private function putQueuesToBus(string $exportableClass, string $fileExtension): Collection
     {
+        $processDataSource = tap(ProcessDataSource::fillData($this), fn ($datasource) => $datasource->get());
+
         $this->exportedFiles = [];
+        $filters             = $processDataSource?->component?->filters ?? [];
         $queues              = collect([]);
-        $countQueue          = $this->getQueuesCount();
-        $perPage             = $this->total / $countQueue;
+        $countQueue          = $this->total > $this->getQueuesCount() ? $this->getQueuesCount() : 1;
+        $perPage             = $this->total > $countQueue ? ($this->total / $countQueue) : 1;
         $offset              = 0;
         $limit               = $perPage;
 
@@ -117,6 +120,7 @@ private function putQueuesToBus(string $exportableClass, string $fileExtension):
                 'fileName'        => $fileName,
                 'offset'          => $offset,
                 'limit'           => $limit,
+                'filters'         => $filters,
             ];
 
             $queues->push(new $this->exportableJobClass(