58 lines
1.4 KiB
TypeScript
58 lines
1.4 KiB
TypeScript
import { NodeType } from "prosemirror-model";
|
|
import { wrapInList, liftListItem } from "prosemirror-schema-list";
|
|
import { Command } from "prosemirror-state";
|
|
import chainTransactions from "../lib/chainTransactions";
|
|
import { findParentNode } from "../queries/findParentNode";
|
|
import isList from "../queries/isList";
|
|
import clearNodes from "./clearNodes";
|
|
|
|
export default function toggleList(
|
|
listType: NodeType,
|
|
itemType: NodeType
|
|
): Command {
|
|
return (state, dispatch) => {
|
|
const { schema, selection } = state;
|
|
const { $from, $to } = selection;
|
|
const range = $from.blockRange($to);
|
|
const { tr } = state;
|
|
|
|
if (!range) {
|
|
return false;
|
|
}
|
|
|
|
const parentList = findParentNode((node) => isList(node, schema))(
|
|
selection
|
|
);
|
|
|
|
if (range.depth >= 1 && parentList && range.depth - parentList.depth <= 1) {
|
|
if (parentList.node.type === listType) {
|
|
return liftListItem(itemType)(state, dispatch);
|
|
}
|
|
|
|
if (
|
|
isList(parentList.node, schema) &&
|
|
listType.validContent(parentList.node.content)
|
|
) {
|
|
tr.setNodeMarkup(parentList.pos, listType);
|
|
|
|
if (dispatch) {
|
|
dispatch(tr);
|
|
}
|
|
|
|
return false;
|
|
}
|
|
}
|
|
|
|
const canWrapInList = wrapInList(listType)(state);
|
|
|
|
if (canWrapInList) {
|
|
return wrapInList(listType)(state, dispatch);
|
|
}
|
|
|
|
return chainTransactions(clearNodes(), wrapInList(listType))(
|
|
state,
|
|
dispatch
|
|
);
|
|
};
|
|
}
|