"use client";

import { create } from "zustand";
import { persist } from "zustand/middleware";

interface WeatherStore {
  favorites: string[];
  selectedStationId: string | null;
  toggleFavorite: (id: string) => void;
  selectStation: (id: string) => void;
}

export const useWeatherStore = create<WeatherStore>()(
  persist(
    (set) => ({
      favorites: [],
      selectedStationId: null,
      toggleFavorite: (id) =>
        set((state) => ({
          favorites: state.favorites.includes(id)
            ? state.favorites.filter((favoriteId) => favoriteId !== id)
            : [...state.favorites, id],
        })),
      selectStation: (id) => set({ selectedStationId: id }),
    }),
    { name: "wtr:preferences" },
  ),
);