mirror of
https://github.com/toeverything/AFFiNE.git
synced 2026-02-13 21:05:19 +00:00
353
blocksuite/framework/std/src/inline/range/range-binding.ts
Normal file
353
blocksuite/framework/std/src/inline/range/range-binding.ts
Normal file
@@ -0,0 +1,353 @@
|
||||
import type { BaseSelection, BlockModel } from '@blocksuite/store';
|
||||
import throttle from 'lodash-es/throttle';
|
||||
|
||||
import { TextSelection } from '../../selection/index.js';
|
||||
import type { BlockComponent } from '../../view/element/block-component.js';
|
||||
import { BLOCK_ID_ATTR } from '../../view/index.js';
|
||||
import { isActiveInEditor } from './active.js';
|
||||
import { RANGE_SYNC_EXCLUDE_ATTR } from './consts.js';
|
||||
import type { RangeManager } from './range-manager.js';
|
||||
/**
|
||||
* Two-way binding between native range and text selection
|
||||
*/
|
||||
export class RangeBinding {
|
||||
private _compositionStartCallback:
|
||||
| ((event: CompositionEvent) => Promise<void>)
|
||||
| null = null;
|
||||
|
||||
private readonly _computePath = (modelId: string) => {
|
||||
const block = this.host.std.store.getBlock(modelId)?.model;
|
||||
if (!block) return [];
|
||||
|
||||
const path: string[] = [];
|
||||
let parent: BlockModel | null = block;
|
||||
while (parent) {
|
||||
path.unshift(parent.id);
|
||||
parent = this.host.doc.getParent(parent);
|
||||
}
|
||||
|
||||
return path;
|
||||
};
|
||||
|
||||
private readonly _onBeforeInput = (event: InputEvent) => {
|
||||
const selection = this.selectionManager.find(TextSelection);
|
||||
if (!selection) return;
|
||||
|
||||
if (event.isComposing) return;
|
||||
|
||||
const { from, to } = selection;
|
||||
if (!to || from.blockId === to.blockId) return;
|
||||
|
||||
const range = this.rangeManager?.value;
|
||||
if (!range) return;
|
||||
|
||||
const blocks = this.rangeManager.getSelectedBlockComponentsByRange(range, {
|
||||
mode: 'flat',
|
||||
});
|
||||
|
||||
const start = blocks.at(0);
|
||||
const end = blocks.at(-1);
|
||||
if (!start || !end) return;
|
||||
|
||||
const startText = start.model.text;
|
||||
const endText = end.model.text;
|
||||
if (!startText || !endText) return;
|
||||
|
||||
event.preventDefault();
|
||||
|
||||
this.host.doc.transact(() => {
|
||||
startText.delete(from.index, from.length);
|
||||
startText.insert(event.data ?? '', from.index);
|
||||
endText.delete(0, to.length);
|
||||
startText.join(endText);
|
||||
|
||||
blocks
|
||||
.slice(1)
|
||||
// delete from lowest to highest
|
||||
.reverse()
|
||||
.forEach(block => {
|
||||
const parent = this.host.doc.getParent(block.model);
|
||||
if (!parent) return;
|
||||
this.host.doc.deleteBlock(block.model, {
|
||||
bringChildrenTo: parent,
|
||||
});
|
||||
});
|
||||
});
|
||||
|
||||
const newSelection = this.selectionManager.create(TextSelection, {
|
||||
from: {
|
||||
blockId: from.blockId,
|
||||
index: from.index + (event.data?.length ?? 0),
|
||||
length: 0,
|
||||
},
|
||||
to: null,
|
||||
});
|
||||
this.selectionManager.setGroup('note', [newSelection]);
|
||||
};
|
||||
|
||||
private readonly _onCompositionEnd = (event: CompositionEvent) => {
|
||||
if (this._compositionStartCallback) {
|
||||
event.preventDefault();
|
||||
event.stopPropagation();
|
||||
this._compositionStartCallback(event).catch(console.error);
|
||||
this._compositionStartCallback = null;
|
||||
}
|
||||
};
|
||||
|
||||
private readonly _onCompositionStart = () => {
|
||||
const selection = this.selectionManager.find(TextSelection);
|
||||
if (!selection) return;
|
||||
|
||||
const { from, to } = selection;
|
||||
if (!to) return;
|
||||
|
||||
this.isComposing = true;
|
||||
|
||||
const range = this.rangeManager?.value;
|
||||
if (!range) return;
|
||||
|
||||
const blocks = this.rangeManager.getSelectedBlockComponentsByRange(range, {
|
||||
mode: 'flat',
|
||||
});
|
||||
|
||||
const start = blocks.at(0);
|
||||
const end = blocks.at(-1);
|
||||
if (!start || !end) return;
|
||||
|
||||
const startText = start.model.text;
|
||||
const endText = end.model.text;
|
||||
if (!startText || !endText) return;
|
||||
|
||||
this._compositionStartCallback = async event => {
|
||||
this.isComposing = false;
|
||||
|
||||
this.host.renderRoot.replaceChildren();
|
||||
// Because we bypassed Lit and disrupted the DOM structure, this will cause an inconsistency in the original state of `ChildPart`.
|
||||
// Therefore, we need to remove the original `ChildPart`.
|
||||
// https://github.com/lit/lit/blob/a2cd76cfdea4ed717362bb1db32710d70550469d/packages/lit-html/src/lit-html.ts#L2248
|
||||
|
||||
delete (this.host.renderRoot as any)['_$litPart$'];
|
||||
this.host.requestUpdate();
|
||||
await this.host.updateComplete;
|
||||
|
||||
this.host.doc.captureSync();
|
||||
|
||||
this.host.doc.transact(() => {
|
||||
endText.delete(0, to.length);
|
||||
startText.delete(from.index, from.length);
|
||||
startText.insert(event.data, from.index);
|
||||
startText.join(endText);
|
||||
|
||||
blocks
|
||||
.slice(1)
|
||||
// delete from lowest to highest
|
||||
.reverse()
|
||||
.forEach(block => {
|
||||
const parent = this.host.doc.getParent(block.model);
|
||||
if (!parent) return;
|
||||
this.host.doc.deleteBlock(block.model, {
|
||||
bringChildrenTo: parent,
|
||||
});
|
||||
});
|
||||
});
|
||||
|
||||
await this.host.updateComplete;
|
||||
|
||||
const selection = this.selectionManager.create(TextSelection, {
|
||||
from: {
|
||||
blockId: from.blockId,
|
||||
index: from.index + (event.data?.length ?? 0),
|
||||
length: 0,
|
||||
},
|
||||
to: null,
|
||||
});
|
||||
this.host.selection.setGroup('note', [selection]);
|
||||
this.rangeManager?.syncTextSelectionToRange(selection);
|
||||
};
|
||||
};
|
||||
|
||||
private readonly _onNativeSelectionChanged = async () => {
|
||||
if (this.isComposing) return;
|
||||
if (!this.host) return; // Unstable when switching views, card <-> embed
|
||||
if (!isActiveInEditor(this.host)) {
|
||||
this._prevTextSelection = null;
|
||||
return;
|
||||
}
|
||||
|
||||
await this.host.updateComplete;
|
||||
|
||||
const selection = document.getSelection();
|
||||
if (!selection) {
|
||||
this.selectionManager.clear(['text']);
|
||||
return;
|
||||
}
|
||||
const range = selection.rangeCount > 0 ? selection.getRangeAt(0) : null;
|
||||
|
||||
if (!range) {
|
||||
this._prevTextSelection = null;
|
||||
this.selectionManager.clear(['text']);
|
||||
return;
|
||||
}
|
||||
|
||||
if (!this.host.contains(range.commonAncestorContainer)) {
|
||||
return;
|
||||
}
|
||||
|
||||
// range is in a non-editable element
|
||||
// ex. placeholder
|
||||
const isRangeOutNotEditable =
|
||||
range.startContainer instanceof HTMLElement &&
|
||||
range.startContainer.contentEditable === 'false' &&
|
||||
range.endContainer instanceof HTMLElement &&
|
||||
range.endContainer.contentEditable === 'false';
|
||||
if (isRangeOutNotEditable) {
|
||||
this._prevTextSelection = null;
|
||||
this.selectionManager.clear(['text']);
|
||||
|
||||
// force clear native selection to break inline editor input
|
||||
selection.removeRange(range);
|
||||
return;
|
||||
}
|
||||
|
||||
const el =
|
||||
range.commonAncestorContainer instanceof Element
|
||||
? range.commonAncestorContainer
|
||||
: range.commonAncestorContainer.parentElement;
|
||||
if (!el) return;
|
||||
const block = el.closest<BlockComponent>(`[${BLOCK_ID_ATTR}]`);
|
||||
if (block?.getAttribute(RANGE_SYNC_EXCLUDE_ATTR) === 'true') return;
|
||||
|
||||
const inlineEditor = this.rangeManager?.getClosestInlineEditor(
|
||||
range.commonAncestorContainer
|
||||
);
|
||||
if (inlineEditor?.isComposing) return;
|
||||
|
||||
const isRangeReversed =
|
||||
!!selection.anchorNode &&
|
||||
!!selection.focusNode &&
|
||||
(selection.anchorNode === selection.focusNode
|
||||
? selection.anchorOffset > selection.focusOffset
|
||||
: selection.anchorNode.compareDocumentPosition(selection.focusNode) ===
|
||||
Node.DOCUMENT_POSITION_PRECEDING);
|
||||
const textSelection = this.rangeManager?.rangeToTextSelection(
|
||||
range,
|
||||
isRangeReversed
|
||||
);
|
||||
if (!textSelection) {
|
||||
this._prevTextSelection = null;
|
||||
this.selectionManager.clear(['text']);
|
||||
return;
|
||||
}
|
||||
|
||||
const model = this.host.doc.getModelById(textSelection.blockId);
|
||||
// If the model is not found, the selection maybe in another editor
|
||||
if (!model) return;
|
||||
|
||||
this._prevTextSelection = {
|
||||
selection: textSelection,
|
||||
path: this._computePath(model.id),
|
||||
};
|
||||
this.rangeManager?.syncRangeToTextSelection(range, isRangeReversed);
|
||||
};
|
||||
|
||||
private readonly _onStdSelectionChanged = (selections: BaseSelection[]) => {
|
||||
// TODO(@mirone): this is a trade-off, we need to use separate awareness store for every store to make sure the selection is isolated.
|
||||
const closestHost = document.activeElement?.closest('editor-host');
|
||||
if (closestHost && closestHost !== this.host) return;
|
||||
const active = this.host.event.active;
|
||||
if (!active) return;
|
||||
|
||||
const text =
|
||||
selections.find((selection): selection is TextSelection =>
|
||||
selection.is(TextSelection)
|
||||
) ?? null;
|
||||
|
||||
if (text === this._prevTextSelection) {
|
||||
return;
|
||||
}
|
||||
// wait for lit updated
|
||||
this.host.updateComplete
|
||||
.then(() => {
|
||||
const id = text?.blockId;
|
||||
const path = id && this._computePath(id);
|
||||
|
||||
if (active) {
|
||||
const eq =
|
||||
text && this._prevTextSelection && path
|
||||
? text.equals(this._prevTextSelection.selection) &&
|
||||
path.join('') === this._prevTextSelection.path.join('')
|
||||
: false;
|
||||
|
||||
if (eq) return;
|
||||
}
|
||||
|
||||
this._prevTextSelection =
|
||||
text && path
|
||||
? {
|
||||
selection: text,
|
||||
path,
|
||||
}
|
||||
: null;
|
||||
if (text) {
|
||||
this.rangeManager?.syncTextSelectionToRange(text);
|
||||
} else {
|
||||
this.rangeManager?.clear();
|
||||
}
|
||||
})
|
||||
.catch(console.error);
|
||||
};
|
||||
|
||||
private _prevTextSelection: {
|
||||
selection: TextSelection;
|
||||
path: string[];
|
||||
} | null = null;
|
||||
|
||||
isComposing = false;
|
||||
|
||||
get host() {
|
||||
return this.manager.std.host;
|
||||
}
|
||||
|
||||
get rangeManager() {
|
||||
return this.host.range;
|
||||
}
|
||||
|
||||
get selectionManager() {
|
||||
return this.host.selection;
|
||||
}
|
||||
|
||||
constructor(public manager: RangeManager) {
|
||||
this.host.disposables.add(
|
||||
this.selectionManager.slots.changed.subscribe(this._onStdSelectionChanged)
|
||||
);
|
||||
|
||||
this.host.disposables.addFromEvent(
|
||||
document,
|
||||
'selectionchange',
|
||||
throttle(() => {
|
||||
this._onNativeSelectionChanged().catch(console.error);
|
||||
}, 10)
|
||||
);
|
||||
|
||||
this.host.disposables.add(
|
||||
this.host.event.add('beforeInput', ctx => {
|
||||
const event = ctx.get('defaultState').event as InputEvent;
|
||||
this._onBeforeInput(event);
|
||||
})
|
||||
);
|
||||
|
||||
this.host.disposables.addFromEvent(
|
||||
this.host,
|
||||
'compositionstart',
|
||||
this._onCompositionStart
|
||||
);
|
||||
this.host.disposables.addFromEvent(
|
||||
this.host,
|
||||
'compositionend',
|
||||
this._onCompositionEnd,
|
||||
{
|
||||
capture: true,
|
||||
}
|
||||
);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user