103 lines
2.9 KiB
TypeScript
103 lines
2.9 KiB
TypeScript
import {
|
|
Decoration,
|
|
EditorState,
|
|
EditorView,
|
|
syntaxTree,
|
|
WidgetType,
|
|
} from "../deps.ts";
|
|
import {
|
|
decoratorStateField,
|
|
invisibleDecoration,
|
|
isCursorInRange,
|
|
} from "./util.ts";
|
|
|
|
import { renderMarkdownToHtml } from "../../plugs/markdown/markdown_render.ts";
|
|
import { ParseTree } from "$sb/lib/tree.ts";
|
|
import { lezerToParseTree } from "../../common/parse_tree.ts";
|
|
import type { Editor } from "../editor.tsx";
|
|
|
|
class TableViewWidget extends WidgetType {
|
|
constructor(
|
|
readonly pos: number,
|
|
readonly editorView: EditorView,
|
|
readonly t: ParseTree,
|
|
) {
|
|
super();
|
|
}
|
|
|
|
toDOM(): HTMLElement {
|
|
const dom = document.createElement("span");
|
|
dom.classList.add("sb-table-widget");
|
|
dom.addEventListener("click", (e) => {
|
|
// Pulling data-pos to put the cursor in the right place, falling back
|
|
// to the start of the table.
|
|
const dataAttributes = (e.target as any).dataset;
|
|
this.editorView.dispatch({
|
|
selection: {
|
|
anchor: dataAttributes.pos ? +dataAttributes.pos : this.pos,
|
|
},
|
|
});
|
|
});
|
|
|
|
dom.innerHTML = renderMarkdownToHtml(this.t, {
|
|
// Annotate every element with its position so we can use it to put
|
|
// the cursor there when the user clicks on the table.
|
|
annotationPositions: true,
|
|
});
|
|
return dom;
|
|
}
|
|
}
|
|
|
|
export function tablePlugin(editor: Editor) {
|
|
return decoratorStateField((state: EditorState) => {
|
|
const widgets: any[] = [];
|
|
syntaxTree(state).iterate({
|
|
enter: (node) => {
|
|
const { from, to, name } = node;
|
|
if (name !== "Table") return;
|
|
if (isCursorInRange(state, [from, to])) return;
|
|
|
|
const tableText = state.sliceDoc(from, to);
|
|
const lineStrings = tableText.split("\n");
|
|
|
|
const lines: { from: number; to: number }[] = [];
|
|
let fromIt = from;
|
|
for (const line of lineStrings) {
|
|
lines.push({
|
|
from: fromIt,
|
|
to: fromIt + line.length,
|
|
});
|
|
fromIt += line.length + 1;
|
|
}
|
|
|
|
const firstLine = lines[0], lastLine = lines[lines.length - 1];
|
|
|
|
// In case of doubt, back out
|
|
if (!firstLine || !lastLine) return;
|
|
|
|
widgets.push(invisibleDecoration.range(firstLine.from, firstLine.to));
|
|
widgets.push(invisibleDecoration.range(lastLine.from, lastLine.to));
|
|
|
|
lines.slice(1, lines.length - 1).forEach((line) => {
|
|
widgets.push(
|
|
Decoration.line({ class: "sb-line-table-outside" }).range(
|
|
line.from,
|
|
),
|
|
);
|
|
});
|
|
const text = state.sliceDoc(0, to);
|
|
widgets.push(
|
|
Decoration.widget({
|
|
widget: new TableViewWidget(
|
|
from,
|
|
editor.editorView!,
|
|
lezerToParseTree(text, node.node),
|
|
),
|
|
}).range(from),
|
|
);
|
|
},
|
|
});
|
|
return Decoration.set(widgets, true);
|
|
});
|
|
}
|