import type { StateCreator } from "zustand";
import { create } from "zustand";

const storeResetFns = new Set<() => void>();

const resetAllStores = () => {
    storeResetFns.forEach((resetFn) => {
        resetFn();
    });
};

export const actualCreate = (<T>() => {
    return (stateCreator: StateCreator<T>) => {
        const store = create(stateCreator);
        const initialState = store.getInitialState();
        storeResetFns.add(() => {
            store.setState(initialState, true);
        });
        return store;
    };
}) as typeof create;