dify/web/app/components/base/prompt-editor/plugins/request-url-block/request-url-block-replacement-block.tsx

import type { RequestURLBlockType } from '../../types'
import { useLexicalComposerContext } from '@lexical/react/LexicalComposerContext'
import { mergeRegister } from '@lexical/utils'
import { $applyNodeReplacement } from 'lexical'
import { memo, useCallback, useEffect } from 'react'
import { REQUEST_URL_PLACEHOLDER_TEXT } from '../../constants'
import { decoratorTransform } from '../../utils'
import { CustomTextNode } from '../custom-text/node'
import { $createRequestURLBlockNode, RequestURLBlockNode } from '../request-url-block/node'

const REGEX = new RegExp(REQUEST_URL_PLACEHOLDER_TEXT)

const RequestURLBlockReplacementBlock = ({ onInsert }: RequestURLBlockType) => {
  const [editor] = useLexicalComposerContext()

  useEffect(() => {
    if (!editor.hasNodes([RequestURLBlockNode]))
      throw new Error('RequestURLBlockNodePlugin: RequestURLBlockNode not registered on editor')
  }, [editor])

  const createRequestURLBlockNode = useCallback((): RequestURLBlockNode => {
    if (onInsert) onInsert()
    return $applyNodeReplacement($createRequestURLBlockNode())
  }, [onInsert])

  const getMatch = useCallback((text: string) => {
    const matchArr = REGEX.exec(text)

    if (matchArr === null) return null

    const startOffset = matchArr.index
    const endOffset = startOffset + REQUEST_URL_PLACEHOLDER_TEXT.length
    return {
      end: endOffset,
      start: startOffset,
    }
  }, [])

  useEffect(() => {
    REGEX.lastIndex = 0
    return mergeRegister(
      editor.registerNodeTransform(CustomTextNode, (textNode) =>
        decoratorTransform(textNode, getMatch, createRequestURLBlockNode),
      ),
    )
  }, [])

  return null
}

export default memo(RequestURLBlockReplacementBlock)