<?php

namespace App\Filament\Widgets;

use App\Enums\ApprovalStatus;
use App\Models\Report;
use BezhanSalleh\FilamentShield\Traits\HasWidgetShield;
use Carbon\Carbon;
use Filament\Widgets\ChartWidget;
use Filament\Widgets\Concerns\InteractsWithPageFilters;
use Illuminate\Support\Facades\DB;

class TaskRealizationChart extends ChartWidget
{
    use HasWidgetShield, InteractsWithPageFilters;

    protected ?string $heading = 'Realisasi Penugasan (Laporan)';

    protected static ?int $sort = 10;

    protected int|string|array $columnSpan = 1;

    protected function getData(): array
    {
        $startDate = $this->filters['startDate'] ?? null;
        $endDate = $this->filters['endDate'] ?? null;

        $data = Report::query()
            ->select(
                DB::raw('COUNT(*) as count'),
                'status',
                DB::raw("DATE_FORMAT(created_at, '%Y-%m') as month")
            )
            ->when($startDate, fn ($q) => $q->whereDate('created_at', '>=', $startDate))
            ->when(! $startDate, fn ($q) => $q->where('created_at', '>=', now()->subMonths(6)))
            ->when($endDate, fn ($q) => $q->whereDate('created_at', '<=', $endDate))
            ->groupBy('month', 'status')
            ->orderBy('month')
            ->get();

        $months = $data->pluck('month')->unique()->sort()->values();

        $acceptedData = [];
        $pendingData = [];
        $rejectedData = [];

        foreach ($months as $month) {
            $monthData = $data->where('month', $month);
            $acceptedData[] = $monthData->where('status', ApprovalStatus::ACCEPTED)->first()?->count ?? 0;
            $pendingData[] = $monthData->where('status', ApprovalStatus::PENDING)->first()?->count ?? 0;
            $rejectedData[] = $monthData->where('status', ApprovalStatus::REJECTED)->first()?->count ?? 0;
        }

        return [
            'datasets' => [
                [
                    'label' => 'Disetujui',
                    'data' => $acceptedData,
                    'backgroundColor' => '#10B981',
                ],
                [
                    'label' => 'Menunggu',
                    'data' => $pendingData,
                    'backgroundColor' => '#F59E0B',
                ],
                [
                    'label' => 'Ditolak',
                    'data' => $rejectedData,
                    'backgroundColor' => '#EF4444',
                ],
            ],
            'labels' => $months->map(fn ($m) => Carbon::createFromFormat('Y-m', $m)->translatedFormat('M Y'))->toArray(),
        ];
    }

    protected function getType(): string
    {
        return 'bar';
    }
}
