<?php

namespace App\Filament\Widgets;

use App\Enums\MediaClassification;
use App\Models\PartnerMedia;
use Filament\Widgets\ChartWidget;

class MediaClassificationChart extends ChartWidget
{
    protected ?string $heading = 'Klasifikasi Media';

    protected static ?int $sort = 3;

    protected int|string|array $columnSpan = 1;

    protected function getData(): array
    {
        $data = PartnerMedia::query()
            ->selectRaw('classification, count(*) as total')
            ->groupBy('classification')
            ->get();

        $labels = [];
        $counts = [];

        foreach (MediaClassification::cases() as $case) {
            $labels[] = $case->getLabel();
            $counts[] = $data->where('classification', $case)->first()?->total ?? 0;
        }

        return [
            'datasets' => [
                [
                    'label' => 'Jumlah Media',
                    'data' => $counts,
                    'backgroundColor' => '#4F46E5',
                ],
            ],
            'labels' => $labels,
        ];
    }

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