wip budgets

app/Filament/Company/Resources/Accounting/BudgetResource.php

@@ -67,6 +67,19 @@ class BudgetResource extends Resource
                                     ->columnSpan(1)
                                     ->required(),
 
+                                Forms\Components\TextInput::make('total_amount')
+                                    ->label('Total Amount')
+                                    ->numeric()
+                                    ->required()
+                                    ->columnSpan(1)
+                                    ->suffixAction(
+                                        Forms\Components\Actions\Action::make('disperse')
+                                            ->label('Disperse')
+                                            ->icon('heroicon-m-bars-arrow-down')
+                                            ->color('primary')
+                                            ->action(fn (Forms\Set $set, Forms\Get $get, $state) => self::disperseTotalAmount($set, $get, $state))
+                                    ),
+
                                 CustomSection::make('Budget Allocations')
                                     ->contained(false)
                                     ->columns(4)
@@ -98,6 +111,66 @@ class BudgetResource extends Resource
             ]);
     }
 
+    /**
+     * Disperses the total amount across the budget items based on the selected interval.
+     */
+    private static function disperseTotalAmount(Forms\Set $set, Forms\Get $get, float $totalAmount): void
+    {
+        $startDate = $get('../../start_date');
+        $endDate = $get('../../end_date');
+        $intervalType = $get('../../interval_type');
+
+        if (! $startDate || ! $endDate || ! $intervalType || $totalAmount <= 0) {
+            return;
+        }
+
+        // Generate labels based on interval type (must match `getAllocationFields()`)
+        $labels = self::generateFormattedLabels($startDate, $endDate, $intervalType);
+        $numPeriods = count($labels);
+
+        if ($numPeriods === 0) {
+            return;
+        }
+
+        // Calculate base allocation and handle rounding
+        $baseAmount = floor($totalAmount / $numPeriods);
+        $remainder = $totalAmount - ($baseAmount * $numPeriods);
+
+        // Assign amounts to the correct fields using labels
+        foreach ($labels as $index => $label) {
+            $amount = $baseAmount + ($index === 0 ? $remainder : 0);
+            $set("amounts.{$label}", $amount); // Now correctly assigns to the right field
+        }
+    }
+
+    /**
+     * Generates formatted labels for the budget allocation fields based on the selected interval type.
+     */
+    private static function generateFormattedLabels(string $startDate, string $endDate, string $intervalType): array
+    {
+        $start = Carbon::parse($startDate);
+        $end = Carbon::parse($endDate);
+        $labels = [];
+
+        while ($start->lte($end)) {
+            $labels[] = match ($intervalType) {
+                'month' => $start->format('M'), // Example: Jan, Feb, Mar
+                'quarter' => 'Q' . $start->quarter, // Example: Q1, Q2, Q3
+                'year' => (string) $start->year, // Example: 2024, 2025
+                default => '',
+            };
+
+            match ($intervalType) {
+                'month' => $start->addMonth(),
+                'quarter' => $start->addQuarter(),
+                'year' => $start->addYear(),
+                default => null,
+            };
+        }
+
+        return $labels;
+    }
+
     private static function getAllocationFields(?string $startDate, ?string $endDate, ?string $intervalType): array
     {
         if (! $startDate || ! $endDate || ! $intervalType) {

app/Filament/Company/Resources/Accounting/BudgetResource/Pages/CreateBudget.php

@@ -4,8 +4,16 @@ namespace App\Filament\Company\Resources\Accounting\BudgetResource\Pages;
 
 use App\Filament\Company\Resources\Accounting\BudgetResource;
 use Filament\Resources\Pages\CreateRecord;
+use Illuminate\Database\Eloquent\Model;
 
 class CreateBudget extends CreateRecord
 {
     protected static string $resource = BudgetResource::class;
+
+    protected function handleRecordCreation(array $data): Model
+    {
+        ray($data);
+
+        return parent::handleRecordCreation($data); // TODO: Change the autogenerated stub
+    }
 }

app/Models/Company.php

@@ -93,6 +93,21 @@ class Company extends FilamentCompaniesCompany implements HasAvatar
         return $this->hasMany(Accounting\Bill::class, 'company_id');
     }
 
+    public function budgets(): HasMany
+    {
+        return $this->hasMany(Accounting\Budget::class, 'company_id');
+    }
+
+    public function budgetItems(): HasMany
+    {
+        return $this->hasMany(Accounting\BudgetItem::class, 'company_id');
+    }
+
+    public function budgetAllocations(): HasMany
+    {
+        return $this->hasMany(Accounting\BudgetAllocation::class, 'company_id');
+    }
+
     public function accountSubtypes(): HasMany
     {
         return $this->hasMany(AccountSubtype::class, 'company_id');