Merge pull request #91 from saade/3.x

feat: add pivot data

Author: CodeWithDennis

Diff for: src/SelectTree.php

@@ -10,13 +10,15 @@
 use Filament\Forms\Components\Concerns\CanBeSearchable;
 use Filament\Forms\Components\Concerns\HasActions;
 use Filament\Forms\Components\Concerns\HasAffixes;
+use Filament\Forms\Components\Concerns\HasPivotData;
 use Filament\Forms\Components\Concerns\HasPlaceholder;
 use Filament\Forms\Components\Contracts\HasAffixActions;
 use Filament\Forms\Components\Field;
 use Filament\Forms\Form;
 use Filament\Support\Facades\FilamentIcon;
 use Illuminate\Database\Eloquent\Relations\BelongsTo;
 use Illuminate\Database\Eloquent\Relations\BelongsToMany;
+use Illuminate\Support\Arr;
 use Illuminate\Support\Collection;
 
 class SelectTree extends Field implements HasAffixActions
@@ -26,6 +28,7 @@ class SelectTree extends Field implements HasAffixActions
     use HasActions;
     use HasAffixes;
     use HasPlaceholder;
+    use HasPivotData;
 
     protected string $view = 'select-tree::select-tree';
 
@@ -101,10 +104,19 @@ protected function setUp(): void
             // Check if the component's relationship is a BelongsToMany relationship.
             if ($component->getRelationship() instanceof BelongsToMany) {
                 // Wrap the state in a collection and convert it to an array if it's not set.
-                $state = Collection::wrap($state ?? []);
+                $state = Arr::wrap($state ?? []);
+
+                $pivotData = $component->getPivotData();
 
                 // Sync the relationship with the provided state (IDs).
-                $component->getRelationship()->sync($state->toArray());
+                if ($pivotData === []) {
+                    $component->getRelationship()->sync($state ?? []);
+    
+                    return;
+                }
+
+                // Sync the relationship with the provided state (IDs) plus pivot data.
+                $component->getRelationship()->syncWithPivotValues($state ?? [], $pivotData);
             }
         });