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',
        },
    ),
)