import { create } from "zustand";
import { Command, EditorState } from "prosemirror-state";
import { EditorView } from "prosemirror-view";
export let useEditorStates = create(() => ({
  lastXPosition: 0,
  editorStates: {} as {
    [entity: string]:
      | {
          editor: InstanceType<typeof EditorState>;
          initial?: boolean;
          view: InstanceType<typeof EditorView>;
          keymap?: { [key: string]: Command };
        }
      | undefined;
  },
}));

export const setEditorState = (
  entityID: string,
  s: {
    editor: InstanceType<typeof EditorState>;
  },
) => {
  useEditorStates.setState((oldState) => {
    let existingState = oldState.editorStates[entityID];
    if (!existingState) return oldState;
    return {
      editorStates: {
        ...oldState.editorStates,
        [entityID]: { ...existingState, ...s },
      },
    };
  });
};