import { MenuItem } from "prosemirror-menu";
import { buildMenuItems } from "prosemirror-example-setup";

import { writeFreelySchema } from "./schema";

function canInsert(state, nodeType, attrs) {
  let $from = state.selection.$from;
  for (let d = $from.depth; d >= 0; d--) {
    let index = $from.index(d);
    if ($from.node(d).canReplaceWith(index, index, nodeType, attrs))
      return true;
  }
  return false;
}

const ReadMoreItem = new MenuItem({
  label: "Read more",
  select: (state) => canInsert(state, writeFreelySchema.nodes.readmore),
  run(state, dispatch) {
    dispatch(
      state.tr.replaceSelectionWith(writeFreelySchema.nodes.readmore.create())
    );
  },
});

export const getMenu = () => {
  const menuContent = [
    ...buildMenuItems(writeFreelySchema).fullMenu,
    [ReadMoreItem],
  ];
  return menuContent;
};