<?php

namespace App\Services;

use App\Models\Unit;
use Illuminate\Database\Eloquent\Collection;
use Illuminate\Http\UploadedFile;
use Illuminate\Support\Facades\Storage;

class UnitService
{
    public function getAll(): Collection
    {
        return Unit::with(['city', 'state'])
            ->orderBy('fantasy_name')
            ->get();
    }

    public function findById(int $id): ?Unit
    {
        return Unit::with(['city', 'state'])->find($id);
    }

    public function create(array $data): Unit
    {
        $data = $this->handleAvatar($data);
        return Unit::create($data);
    }

    public function update(int $id, array $data): ?Unit
    {
        $model = $this->findById($id);

        if (!$model) {
            return null;
        }

        $data = $this->handleAvatar($data, $model->avatar_url);
        $model->update($data);
        return $model->fresh(['city', 'state']);
    }

    public function delete(int $id): bool
    {
        $model = $this->findById($id);

        if (!$model) {
            return false;
        }

        if ($model->avatar_url) {
            Storage::disk('public')->delete($model->avatar_url);
        }

        return $model->delete();
    }

    private function handleAvatar(array $data, ?string $oldAvatarPath = null): array
    {
        if (!isset($data['avatar'])) {
            return $data;
        }

        if ($data['avatar'] instanceof UploadedFile) {
            if ($oldAvatarPath) {
                Storage::disk('public')->delete($oldAvatarPath);
            }
            $data['avatar_url'] = $data['avatar']->store('units/avatars', 'public');
        } elseif (is_null($data['avatar'])) {
            if ($oldAvatarPath) {
                Storage::disk('public')->delete($oldAvatarPath);
            }
            $data['avatar_url'] = null;
        }

        unset($data['avatar']);
        return $data;
    }
}