import { create } from 'zustand';
import { prafrotService } from '../services/prafrotService';
import { toast } from 'sonner';

export const useVehicles = create((set, get) => ({
  vehicles: [],
  loading: false,
  error: null,

  fetchVehicles: async () => {
    set({ loading: true, error: null });
    try {
      const data = await prafrotService.getVehicles();
      // Ensure data is an array
      set({ vehicles: Array.isArray(data) ? data : (data.data || []) });
    } catch (error) {
      set({ error: error.message });
      toast.error('Erro ao carregar veículos: ' + error.message);
    } finally {
      set({ loading: false });
    }
  },

  createVehicle: async (vehicleData) => {
    set({ loading: true });
    try {
      await prafrotService.createVehicle(vehicleData);
      toast.success('Veículo cadastrado com sucesso!');
      get().fetchVehicles(); // Refresh list
      return true;
    } catch (error) {
      toast.error('Erro ao cadastrar veículo: ' + error.message);
      return false;
    } finally {
      set({ loading: false });
    }
  },

  updateVehicle: async (id, vehicleData) => {
    set({ loading: true });
    try {
      await prafrotService.updateVehicle(id, vehicleData);
      toast.success('Veículo atualizado com sucesso!');
      get().fetchVehicles();
      return true;
    } catch (error) {
      toast.error('Erro ao atualizar veículo: ' + error.message);
      return false;
    } finally {
      set({ loading: false });
    }
  },

  updateVehicleStatus: async (id, statusFrota) => {
    // Não colocamos loading global aqui para não travar a UI toda durante edição inline
    try {
      await prafrotService.updateVehicleStatus(id, statusFrota);
      toast.success('Status atualizado!');
      // Atualiza localmente para ser mais rápido (otimista) ou busca tudo de novo
      get().fetchVehicles(); 
      return true;
    } catch (error) {
      toast.error('Erro ao atualizar status: ' + error.message);
      return false;
    }
  }
}));
