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

type TokenStore = {
    token?: string;
    setToken: (token?: string) => void;
    removeToken: () => void;
};

export const useTokenStore = create<TokenStore>()(
    persist(
        (set, get) => ({
            token: undefined,
            setToken: (token?: string) => set({ token }),
            removeToken: () => set({ token: undefined }),
        }),
        {
            name: "token",
        },
    ),
);