@atlaskit/editor-common
Version:
A package that contains common classes and components for editor and renderer
186 lines (183 loc) • 5.94 kB
JavaScript
import { NodeSelection, Selection, TextSelection } from '@atlaskit/editor-prosemirror/state';
import { CellSelection } from '@atlaskit/editor-tables/cell-selection';
export function addParagraphAtEnd(tr) {
const {
doc: {
type: {
schema: {
nodes: {
paragraph
}
}
}
},
doc
} = tr;
if (doc.lastChild && !(doc.lastChild.type === paragraph && doc.lastChild.content.size === 0)) {
if (paragraph) {
tr.insert(doc.content.size, paragraph.createAndFill());
}
}
tr.setSelection(TextSelection.create(tr.doc, tr.doc.content.size - 1));
tr.scrollIntoView();
}
export function createParagraphAtEnd() {
return function (state, dispatch) {
const {
tr
} = state;
addParagraphAtEnd(tr);
if (dispatch) {
dispatch(tr);
}
return true;
};
}
export const changeImageAlignment = align => (state, dispatch) => {
const {
from,
to
} = state.selection;
const tr = state.tr;
state.doc.nodesBetween(from, to, (node, pos) => {
if (node.type === state.schema.nodes.mediaSingle) {
tr.setNodeMarkup(pos, undefined, {
...node.attrs,
layout: align === 'center' ? 'center' : `align-${align}`
});
}
});
if (tr.docChanged && dispatch) {
dispatch(tr.scrollIntoView());
return true;
}
return false;
};
export const createToggleBlockMarkOnRange = (markType, getAttrs, allowedBlocks) => (from, to, tr, state) => {
let markApplied = false;
state.doc.nodesBetween(from, to, (node, pos, parent) => {
if (!node.type.isBlock) {
return false;
}
if ((!allowedBlocks || (Array.isArray(allowedBlocks) ? allowedBlocks.indexOf(node.type) > -1 : allowedBlocks(state.schema, node, parent))) && parent !== null && parent !== void 0 && parent.type.allowsMarkType(markType)) {
const oldMarks = node.marks.filter(mark => mark.type === markType);
const prevAttrs = oldMarks.length ? oldMarks[0].attrs : undefined;
const newAttrs = getAttrs(prevAttrs, node);
if (newAttrs !== undefined) {
tr.setNodeMarkup(pos, node.type, node.attrs, node.marks.filter(mark => !markType.excludes(mark.type)).concat(newAttrs === false ? [] : markType.create(newAttrs)));
markApplied = true;
}
}
return;
});
return markApplied;
};
export const createToggleInlineMarkOnRange = (markType, getAttrs) => (from, to, tr, state) => {
let markApplied = false;
state.doc.nodesBetween(from, to, (node, pos, parent) => {
if (parent !== null && parent !== void 0 && parent.type.allowsMarkType(markType)) {
const oldMarks = node.marks.filter(mark => mark.type === markType);
const prevAttrs = oldMarks.length ? oldMarks[0].attrs : undefined;
const newAttrs = getAttrs(prevAttrs, node);
if (newAttrs !== undefined) {
tr.setNodeMarkup(pos, node.type, node.attrs, node.marks.filter(mark => !markType.excludes(mark.type)).concat(newAttrs === false ? [] : markType.create(newAttrs)));
tr.setSelection(NodeSelection.create(tr.doc, state.selection.from));
markApplied = true;
}
}
return;
});
return markApplied;
};
/**
* Toggles block mark based on the return type of `getAttrs`.
* This is similar to ProseMirror's `getAttrs` from `AttributeSpec`
* return `false` to remove the mark.
* return `undefined for no-op.
* return an `object` to update the mark.
*/
export const toggleBlockMark = (markType, getAttrs, allowedBlocks) => (state, dispatch) => {
let markApplied = false;
const tr = state.tr;
const toggleBlockMarkOnRange = createToggleBlockMarkOnRange(markType, getAttrs, allowedBlocks);
if (state.selection instanceof CellSelection) {
state.selection.forEachCell((cell, pos) => {
markApplied = toggleBlockMarkOnRange(pos, pos + cell.nodeSize, tr, state);
});
} else {
const {
from,
to
} = state.selection;
markApplied = toggleBlockMarkOnRange(from, to, tr, state);
}
if (markApplied && tr.docChanged) {
if (dispatch) {
dispatch(tr.scrollIntoView());
}
return true;
}
return false;
};
export const clearEditorContent = (state, dispatch) => {
const tr = state.tr;
tr.replace(0, state.doc.nodeSize - 2);
tr.setSelection(Selection.atStart(tr.doc));
if (dispatch) {
dispatch(tr);
return true;
}
return false;
};
// https://github.com/ProseMirror/prosemirror-commands/blob/master/src/commands.js#L90
// Keep going left up the tree, without going across isolating boundaries, until we
// can go along the tree at that same level
//
// You can think of this as, if you could construct each document like we do in the tests,
// return the position of the first ) backwards from the current selection.
export function findCutBefore($pos) {
// parent is non-isolating, so we can look across this boundary
if (!$pos.parent.type.spec.isolating) {
// search up the tree from the pos's *parent*
for (let i = $pos.depth - 1; i >= 0; i--) {
// starting from the inner most node's parent, find out
// if we're not its first child
if ($pos.index(i) > 0) {
return $pos.doc.resolve($pos.before(i + 1));
}
if ($pos.node(i).type.spec.isolating) {
break;
}
}
}
return null;
}
/**
* @deprecated
*
* This method is no longer needed and can be accessed via the
* `editor-plugin-block-type` command of `setTextLevel`
*/
export function setHeading(level) {
return function (state, dispatch) {
const {
selection,
schema,
tr
} = state;
const ranges = selection instanceof CellSelection ? selection.ranges : [selection];
ranges.forEach(({
$from,
$to
}) => {
tr.setBlockType($from.pos, $to.pos, schema.nodes.heading, {
level
});
});
if (dispatch) {
dispatch(tr);
}
return true;
};
}
export { insertBlock } from './insert-block';