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

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

  fetchAvailabilities: async () => {
    set({ loading: true, error: null });
    try {
      const data = await prafrotService.getAvailability();
      set({ availabilities: Array.isArray(data) ? data : (data.data || []) });
    } catch (error) {
      set({ error: error.message });
      toast.error('Erro ao carregar disponibilidades: ' + error.message);
    } finally {
      set({ loading: false });
    }
  },

  createAvailability: async (payload) => {
    set({ loading: true });
    try {
      await prafrotService.createAvailability(payload);
      toast.success('Disponibilidade registrada com sucesso!');
      get().fetchAvailabilities();
      return true;
    } catch (error) {
      toast.error('Erro ao registrar disponibilidade: ' + error.message);
      return false;
    } finally {
      set({ loading: false });
    }
  },

  updateAvailability: async (id, payload) => {
    set({ loading: true });
    try {
      await prafrotService.updateAvailability(id, payload);
      toast.success('Disponibilidade atualizada com sucesso!');
      get().fetchAvailabilities();
      return true;
    } catch (error) {
      toast.error('Erro ao atualizar disponibilidade: ' + error.message);
      return false;
    } finally {
      set({ loading: false });
    }
  }
}));
