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

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

  fetchMaintenances: async () => {
    set({ loading: true, error: null });
    try {
      const data = await prafrotService.getMaintenance();
      set({ maintenances: Array.isArray(data) ? data : (data.data || []) });
    } catch (error) {
      set({ error: error.message });
      toast.error('Erro ao carregar manutenções: ' + error.message);
    } finally {
      set({ loading: false });
    }
  },

  createMaintenance: async (payload) => {
    set({ loading: true });
    try {
      await prafrotService.createMaintenance(payload);
      toast.success('Manutenção registrada com sucesso!');
      get().fetchMaintenances();
      return true;
    } catch (error) {
      toast.error('Erro ao criar manutenção: ' + error.message);
      return false;
    } finally {
      set({ loading: false });
    }
  },

  updateMaintenance: async (id, payload) => {
    set({ loading: true });
    try {
      await prafrotService.updateMaintenance(id, payload);
      toast.success('Manutenção atualizada com sucesso!');
      get().fetchMaintenances();
      return true;
    } catch (error) {
      toast.error('Erro ao atualizar manutenção: ' + error.message);
      return false;
    } finally {
      set({ loading: false });
    }
  }
}));
