1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
| import {
| memo,
| useEffect,
| } from 'react'
| import {
| $insertNodes,
| COMMAND_PRIORITY_EDITOR,
| createCommand,
| } from 'lexical'
| import { mergeRegister } from '@lexical/utils'
| import { useLexicalComposerContext } from '@lexical/react/LexicalComposerContext'
| import type { HistoryBlockType } from '../../types'
| import {
| $createHistoryBlockNode,
| HistoryBlockNode,
| } from './node'
| import { noop } from 'lodash-es'
|
| export const INSERT_HISTORY_BLOCK_COMMAND = createCommand('INSERT_HISTORY_BLOCK_COMMAND')
| export const DELETE_HISTORY_BLOCK_COMMAND = createCommand('DELETE_HISTORY_BLOCK_COMMAND')
|
| export type RoleName = {
| user: string
| assistant: string
| }
|
| export type HistoryBlockProps = {
| roleName: RoleName
| onEditRole: () => void
| onInsert?: () => void
| onDelete?: () => void
| }
|
| const HistoryBlock = memo(({
| history = { user: '', assistant: '' },
| onEditRole = noop,
| onInsert,
| onDelete,
| }: HistoryBlockType) => {
| const [editor] = useLexicalComposerContext()
|
| useEffect(() => {
| if (!editor.hasNodes([HistoryBlockNode]))
| throw new Error('HistoryBlockPlugin: HistoryBlock not registered on editor')
|
| return mergeRegister(
| editor.registerCommand(
| INSERT_HISTORY_BLOCK_COMMAND,
| () => {
| const historyBlockNode = $createHistoryBlockNode(history, onEditRole)
|
| $insertNodes([historyBlockNode])
|
| if (onInsert)
| onInsert()
|
| return true
| },
| COMMAND_PRIORITY_EDITOR,
| ),
| editor.registerCommand(
| DELETE_HISTORY_BLOCK_COMMAND,
| () => {
| if (onDelete)
| onDelete()
|
| return true
| },
| COMMAND_PRIORITY_EDITOR,
| ),
| )
| }, [editor, history, onEditRole, onInsert, onDelete])
|
| return null
| })
| HistoryBlock.displayName = 'HistoryBlock'
|
| export { HistoryBlock }
| export { HistoryBlockNode } from './node'
| export { default as HistoryBlockReplacementBlock } from './history-block-replacement-block'
|
|
