feat(web): variable select support key operate

This commit is contained in:
zhaoying
2026-04-17 11:51:21 +08:00
parent e87529876c
commit 5a17b7fd0d
3 changed files with 542 additions and 318 deletions

View File

@@ -5,6 +5,7 @@
* @Last Modified time: 2026-04-13 14:00:07 * @Last Modified time: 2026-04-13 14:00:07
*/ */
import { useEffect, useLayoutEffect, useState, useRef, type FC } from 'react'; import { useEffect, useLayoutEffect, useState, useRef, type FC } from 'react';
import { createPortal } from 'react-dom';
import { useLexicalComposerContext } from '@lexical/react/LexicalComposerContext'; import { useLexicalComposerContext } from '@lexical/react/LexicalComposerContext';
import { $getSelection, $isRangeSelection, COMMAND_PRIORITY_HIGH, KEY_ENTER_COMMAND, KEY_ARROW_DOWN_COMMAND, KEY_ARROW_UP_COMMAND, KEY_ESCAPE_COMMAND } from 'lexical'; import { $getSelection, $isRangeSelection, COMMAND_PRIORITY_HIGH, KEY_ENTER_COMMAND, KEY_ARROW_DOWN_COMMAND, KEY_ARROW_UP_COMMAND, KEY_ESCAPE_COMMAND } from 'lexical';
import { Space, Flex } from 'antd'; import { Space, Flex } from 'antd';
@@ -35,61 +36,62 @@ const AutocompletePlugin: FC<{ options: Suggestion[] }> = ({ options }) => {
const [selectedIndex, setSelectedIndex] = useState(0); const [selectedIndex, setSelectedIndex] = useState(0);
const [popupPosition, setPopupPosition] = useState({ top: 0, left: 0, anchorBottom: 0 }); const [popupPosition, setPopupPosition] = useState({ top: 0, left: 0, anchorBottom: 0 });
const [expandedParent, setExpandedParent] = useState<Suggestion | null>(null); const [expandedParent, setExpandedParent] = useState<Suggestion | null>(null);
const [childPanelTop, setChildPanelTop] = useState(0); const [childPanelPos, setChildPanelPos] = useState({ top: 0, right: 0 });
const [activePanel, setActivePanel] = useState<'main' | 'child'>('main');
const [childActiveIndex, setChildActiveIndex] = useState(-1);
const popupRef = useRef<HTMLDivElement>(null); const popupRef = useRef<HTMLDivElement>(null);
const itemRefs = useRef<Map<string, HTMLElement>>(new Map()); const itemRefs = useRef<Map<string, HTMLElement>>(new Map());
const childItemRefs = useRef<Map<string, HTMLElement>>(new Map());
// Adjust popup position after render based on actual height // Adjust popup position after render based on actual size
useLayoutEffect(() => { useLayoutEffect(() => {
if (!popupRef.current || !showSuggestions) return; if (!popupRef.current || !showSuggestions) return;
const { top, anchorBottom } = popupPosition; const { top, left, anchorBottom } = popupPosition;
const popupHeight = popupRef.current.offsetHeight; const popupHeight = popupRef.current.offsetHeight;
const popupWidth = popupRef.current.offsetWidth;
const viewportHeight = window.innerHeight; const viewportHeight = window.innerHeight;
const viewportWidth = window.innerWidth;
const MARGIN = 10; const MARGIN = 10;
let finalTop: number; let finalTop: number;
if (top - popupHeight - MARGIN >= 0) { if (top - popupHeight - MARGIN >= 0) {
// Enough space above: show above cursor
finalTop = top - popupHeight - MARGIN; finalTop = top - popupHeight - MARGIN;
} else { } else {
// Not enough space above: show below cursor
finalTop = anchorBottom + MARGIN; finalTop = anchorBottom + MARGIN;
if (finalTop + popupHeight > viewportHeight - MARGIN) { if (finalTop + popupHeight > viewportHeight - MARGIN) {
finalTop = viewportHeight - popupHeight - MARGIN; finalTop = viewportHeight - popupHeight - MARGIN;
} }
} }
if (finalTop !== top) { let finalLeft = left;
setPopupPosition(prev => ({ ...prev, top: finalTop })); if (finalLeft + popupWidth > viewportWidth - MARGIN) {
finalLeft = viewportWidth - popupWidth - MARGIN;
}
if (finalLeft < MARGIN) finalLeft = MARGIN;
if (finalTop !== top || finalLeft !== left) {
setPopupPosition(prev => ({ ...prev, top: finalTop, left: finalLeft }));
} }
}, [showSuggestions, popupPosition.anchorBottom]); }, [showSuggestions, popupPosition.anchorBottom]);
const CHILD_PANEL_HEIGHT = 280; // max-h-60 (240) + header (~40) const CHILD_PANEL_HEIGHT = 280;
const calcChildPanelTop = (elRect: DOMRect, popupRect: DOMRect) => { const calcChildPanelPos = (key: string) => {
const relativeTop = elRect.top - popupRect.top; const el = itemRefs.current.get(key);
const absoluteBottom = popupRect.top + relativeTop + CHILD_PANEL_HEIGHT; if (!el || !popupRef.current) return;
const overflow = absoluteBottom - (window.innerHeight - 10); const elRect = el.getBoundingClientRect();
return overflow > 0 ? relativeTop - overflow : relativeTop; const popupRect = popupRef.current.getBoundingClientRect();
const actualChildHeight = Math.min(CHILD_PANEL_HEIGHT, popupRect.height);
const top = Math.max(10, popupRect.bottom - actualChildHeight);
setChildPanelPos({ top, right: window.innerWidth - elRect.left + 8 });
}; };
const scrollSelectedIntoView = () => { const resetState = () => {
if (!popupRef.current) return; setShowSuggestions(false);
setExpandedParent(null);
const selectedElement = popupRef.current.querySelector('[data-selected="true"]'); setChildPanelPos({ top: 0, right: 0 });
if (!selectedElement) return; setActivePanel('main');
setChildActiveIndex(-1);
const container = popupRef.current;
const element = selectedElement as HTMLElement;
const containerRect = container.getBoundingClientRect();
const elementRect = element.getBoundingClientRect();
if (elementRect.bottom > containerRect.bottom) {
container.scrollTop += elementRect.bottom - containerRect.bottom;
} else if (elementRect.top < containerRect.top) {
container.scrollTop -= containerRect.top - elementRect.top;
}
}; };
// Listen to editor updates and show suggestions when '/' is typed // Listen to editor updates and show suggestions when '/' is typed
@@ -105,11 +107,7 @@ const AutocompletePlugin: FC<{ options: Suggestion[] }> = ({ options }) => {
const anchorNode = selection.anchor.getNode(); const anchorNode = selection.anchor.getNode();
const anchorOffset = selection.anchor.offset; const anchorOffset = selection.anchor.offset;
// Get the text content of the current node
const nodeText = anchorNode.getTextContent(); const nodeText = anchorNode.getTextContent();
// Check if we have a '/' at the current position or after line break
const textBeforeCursor = nodeText.substring(0, anchorOffset); const textBeforeCursor = nodeText.substring(0, anchorOffset);
const shouldShow = textBeforeCursor.endsWith('/') || const shouldShow = textBeforeCursor.endsWith('/') ||
(textBeforeCursor === '/' && anchorOffset === 1); (textBeforeCursor === '/' && anchorOffset === 1);
@@ -118,10 +116,11 @@ const AutocompletePlugin: FC<{ options: Suggestion[] }> = ({ options }) => {
if (!shouldShow) { if (!shouldShow) {
setSelectedIndex(0); setSelectedIndex(0);
setExpandedParent(null); setExpandedParent(null);
setChildPanelTop(0); setChildPanelPos({ top: 0, right: 0 });
setActivePanel('main');
setChildActiveIndex(-1);
} }
// Calculate popup position to keep it within viewport bounds
if (shouldShow) { if (shouldShow) {
const domSelection = window.getSelection(); const domSelection = window.getSelection();
if (domSelection && domSelection.rangeCount > 0) { if (domSelection && domSelection.rangeCount > 0) {
@@ -149,9 +148,7 @@ const AutocompletePlugin: FC<{ options: Suggestion[] }> = ({ options }) => {
return editor.registerCommand( return editor.registerCommand(
CLOSE_AUTOCOMPLETE_COMMAND, CLOSE_AUTOCOMPLETE_COMMAND,
() => { () => {
setShowSuggestions(false); resetState();
setExpandedParent(null);
setChildPanelTop(0);
return true; return true;
}, },
COMMAND_PRIORITY_HIGH COMMAND_PRIORITY_HIGH
@@ -161,9 +158,7 @@ const AutocompletePlugin: FC<{ options: Suggestion[] }> = ({ options }) => {
// Insert selected suggestion into editor // Insert selected suggestion into editor
const insertMention = (suggestion: Suggestion) => { const insertMention = (suggestion: Suggestion) => {
editor.dispatchCommand(INSERT_VARIABLE_COMMAND, { data: suggestion }); editor.dispatchCommand(INSERT_VARIABLE_COMMAND, { data: suggestion });
setShowSuggestions(false); resetState();
setExpandedParent(null);
setChildPanelTop(0);
}; };
// Group suggestions by node ID // Group suggestions by node ID
@@ -177,13 +172,28 @@ const AutocompletePlugin: FC<{ options: Suggestion[] }> = ({ options }) => {
return groups; return groups;
}, {}); }, {});
// Flat list for keyboard navigation // Flat list of main-panel items for keyboard navigation
const flatOptions = Object.values(groupedSuggestions).flat().flatMap(option => { const flatOptions = Object.values(groupedSuggestions).flat();
if (option.key === expandedParent?.key && option.children?.length) {
return [option, ...option.children]; // Sync child panel position when keyboard navigates to a parent with children
useEffect(() => {
if (selectedIndex < 0 || selectedIndex >= flatOptions.length) return;
const s = flatOptions[selectedIndex];
if (s.children?.length) {
calcChildPanelPos(s.key);
setExpandedParent(s);
} else {
setExpandedParent(null);
} }
return [option]; // eslint-disable-next-line react-hooks/exhaustive-deps
}); }, [selectedIndex]);
// Scroll child active item into view
useEffect(() => {
if (!expandedParent?.children?.length || childActiveIndex < 0) return;
const child = expandedParent.children[childActiveIndex];
if (child) childItemRefs.current.get(child.key)?.scrollIntoView({ block: 'nearest' });
}, [childActiveIndex, expandedParent]);
// Handle Enter key to select suggestion // Handle Enter key to select suggestion
useEffect(() => { useEffect(() => {
@@ -192,7 +202,15 @@ const AutocompletePlugin: FC<{ options: Suggestion[] }> = ({ options }) => {
return editor.registerCommand( return editor.registerCommand(
KEY_ENTER_COMMAND, KEY_ENTER_COMMAND,
(event) => { (event) => {
if (showSuggestions && flatOptions.length > 0) { if (!showSuggestions) return false;
if (activePanel === 'child' && expandedParent?.children?.length) {
const child = expandedParent.children[childActiveIndex];
if (child && !child.disabled) {
event?.preventDefault();
insertMention(child);
return true;
}
} else if (flatOptions.length > 0) {
const selectedOption = flatOptions[selectedIndex]; const selectedOption = flatOptions[selectedIndex];
if (selectedOption && !selectedOption.disabled) { if (selectedOption && !selectedOption.disabled) {
event?.preventDefault(); event?.preventDefault();
@@ -204,57 +222,56 @@ const AutocompletePlugin: FC<{ options: Suggestion[] }> = ({ options }) => {
}, },
COMMAND_PRIORITY_HIGH COMMAND_PRIORITY_HIGH
); );
}, [showSuggestions, selectedIndex, flatOptions, insertMention, editor]); }, [showSuggestions, selectedIndex, flatOptions, insertMention, editor, activePanel, childActiveIndex, expandedParent]);
// Handle keyboard navigation (Arrow Up/Down, Escape) // Handle keyboard navigation (Arrow Up/Down/Left/Right, Escape)
useEffect(() => { useEffect(() => {
if (!showSuggestions) return; if (!showSuggestions) return;
// Navigate down through suggestions, skip disabled items
const unregisterArrowDown = editor.registerCommand( const unregisterArrowDown = editor.registerCommand(
KEY_ARROW_DOWN_COMMAND, KEY_ARROW_DOWN_COMMAND,
(event) => { (event) => {
if (showSuggestions && flatOptions.length > 0) { if (!showSuggestions) return false;
event?.preventDefault(); event?.preventDefault();
if (activePanel === 'child' && expandedParent?.children) {
setChildActiveIndex(i => Math.min(i + 1, expandedParent.children!.length - 1));
} else {
setSelectedIndex(prev => { setSelectedIndex(prev => {
let nextIndex = prev + 1; let next = prev + 1;
while (nextIndex < flatOptions.length && flatOptions[nextIndex].disabled) { // skip items that are disabled AND have no children
nextIndex++; while (next < flatOptions.length && flatOptions[next].disabled && !flatOptions[next].children?.length) next++;
} const newIndex = next >= flatOptions.length ? prev : next;
const newIndex = nextIndex >= flatOptions.length ? prev : nextIndex; setTimeout(() => itemRefs.current.get(flatOptions[newIndex]?.key)?.scrollIntoView({ block: 'nearest' }), 0);
setTimeout(() => scrollSelectedIntoView(), 0);
return newIndex; return newIndex;
}); });
return true;
} }
return false; return true;
}, },
COMMAND_PRIORITY_HIGH COMMAND_PRIORITY_HIGH
); );
// Navigate up through suggestions, skip disabled items
const unregisterArrowUp = editor.registerCommand( const unregisterArrowUp = editor.registerCommand(
KEY_ARROW_UP_COMMAND, KEY_ARROW_UP_COMMAND,
(event) => { (event) => {
if (showSuggestions && flatOptions.length > 0) { if (!showSuggestions) return false;
event?.preventDefault(); event?.preventDefault();
if (activePanel === 'child' && expandedParent?.children) {
setChildActiveIndex(i => Math.max(i - 1, 0));
} else {
setSelectedIndex(prev => { setSelectedIndex(prev => {
let prevIndex = prev - 1; let prevIdx = prev - 1;
while (prevIndex >= 0 && flatOptions[prevIndex].disabled) { // skip items that are disabled AND have no children
prevIndex--; while (prevIdx >= 0 && flatOptions[prevIdx].disabled && !flatOptions[prevIdx].children?.length) prevIdx--;
} const newIndex = prevIdx < 0 ? prev : prevIdx;
const newIndex = prevIndex < 0 ? prev : prevIndex; setTimeout(() => itemRefs.current.get(flatOptions[newIndex]?.key)?.scrollIntoView({ block: 'nearest' }), 0);
setTimeout(() => scrollSelectedIntoView(), 0);
return newIndex; return newIndex;
}); });
return true;
} }
return false; return true;
}, },
COMMAND_PRIORITY_HIGH COMMAND_PRIORITY_HIGH
); );
// Close suggestions on Escape key
const unregisterEscape = editor.registerCommand( const unregisterEscape = editor.registerCommand(
KEY_ESCAPE_COMMAND, KEY_ESCAPE_COMMAND,
(event) => { (event) => {
@@ -273,99 +290,122 @@ const AutocompletePlugin: FC<{ options: Suggestion[] }> = ({ options }) => {
unregisterArrowUp(); unregisterArrowUp();
unregisterEscape(); unregisterEscape();
}; };
}, [showSuggestions, selectedIndex, flatOptions, editor]); }, [showSuggestions, selectedIndex, flatOptions, editor, activePanel, childActiveIndex, expandedParent]);
// Handle ArrowLeft/Right for panel switching via native keydown (lexical doesn't expose these commands)
useEffect(() => {
if (!showSuggestions) return;
const handler = (e: KeyboardEvent) => {
if (e.key === 'ArrowLeft') {
const current = flatOptions[selectedIndex];
if (activePanel === 'main' && current?.children?.length) {
e.preventDefault();
setActivePanel('child');
setChildActiveIndex(0);
}
} else if (e.key === 'ArrowRight') {
if (activePanel === 'child') {
e.preventDefault();
setActivePanel('main');
setChildActiveIndex(-1);
}
}
};
document.addEventListener('keydown', handler, true);
return () => document.removeEventListener('keydown', handler, true);
}, [showSuggestions, activePanel, selectedIndex, flatOptions]);
if (!showSuggestions) return null; if (!showSuggestions) return null;
if (Object.entries(groupedSuggestions).length === 0) return null;
if (Object.entries(groupedSuggestions).length === 0) {
return null
}
return ( return (
<div <>
ref={popupRef} <div
data-autocomplete-popup="true" ref={popupRef}
onMouseDown={(e) => e.preventDefault()} data-autocomplete-popup="true"
className="rb:fixed rb:z-1000 rb:bg-white rb:rounded-lg rb:border-[0.5px] rb:border-[#EBEBEB] rb:shadow-[0px_2px_6px_0px_rgba(0,0,0,0.1)] rb:py-3 rb:px-2" onMouseDown={(e) => e.preventDefault()}
style={{ className="rb:fixed rb:z-1000 rb:bg-white rb:rounded-lg rb:border-[0.5px] rb:border-[#EBEBEB] rb:shadow-[0px_2px_6px_0px_rgba(0,0,0,0.1)] rb:py-3 rb:px-2"
top: popupPosition.top, style={{
left: popupPosition.left, top: popupPosition.top,
}} left: popupPosition.left,
> }}
<div className="rb:min-w-70 rb:max-h-57.5 rb:overflow-y-auto"> >
<Flex vertical gap={12}> <div className="rb:min-w-70 rb:max-h-57.5 rb:overflow-y-auto">
{Object.entries(groupedSuggestions).map(([nodeId, nodeOptions]) => { <Flex vertical gap={12}>
const nodeName = nodeOptions[0]?.nodeData?.name || nodeId; {Object.entries(groupedSuggestions).map(([nodeId, nodeOptions]) => {
return ( const nodeName = nodeOptions[0]?.nodeData?.name || nodeId;
<div key={nodeId} className="rb:text-[12px]"> return (
{nodeName !== 'undefined' && <div key={nodeId} className="rb:text-[12px]">
<div className="rb:px-2 rb:leading-4.25 rb:mb-1.25 rb:font-medium rb:text-[#5B6167]"> {nodeName !== 'undefined' &&
{nodeName} <div className="rb:px-2 rb:leading-4.25 rb:mb-1.25 rb:font-medium rb:text-[#5B6167]">
</div> {nodeName}
} </div>
<Flex vertical gap={2}> }
{nodeOptions.map((option) => { <Flex vertical gap={2}>
const globalIndex = flatOptions.indexOf(option); {nodeOptions.map((option) => {
const isExpanded = expandedParent?.key === option.key; const globalIndex = flatOptions.indexOf(option);
const hasChildren = !!option.children?.length; const isExpanded = expandedParent?.key === option.key;
return ( const hasChildren = !!option.children?.length;
<Flex const isActive = activePanel === 'main' && selectedIndex === globalIndex;
key={option.key} return (
ref={(el) => { if (el) itemRefs.current.set(option.key, el); }} <Flex
data-selected={selectedIndex === globalIndex} key={option.key}
className={clsx("rb:px-2! rb:py-0.75! rb:rounded-sm rb:leading-4.5 rb:text-[#5B6167] rb:hover:bg-[#F6F6F6]", { ref={(el) => { if (el) itemRefs.current.set(option.key, el); }}
'rb:bg-[#F6F6F6]': selectedIndex === globalIndex || isExpanded, className={clsx("rb:px-2! rb:py-0.75! rb:rounded-sm rb:leading-4.5 rb:text-[#5B6167] rb:hover:bg-[#F6F6F6]", {
'rb:cursor-not-allowed rb:opacity-65': option.disabled, 'rb:bg-[#F6F6F6]': isActive || isExpanded,
'rb:cursor-pointer': !option.disabled, 'rb:cursor-not-allowed rb:opacity-65': option.disabled,
})} 'rb:cursor-pointer': !option.disabled,
align="center" })}
justify="space-between" align="center"
onClick={() => { justify="space-between"
if (option.disabled) return; onClick={() => {
insertMention(option); if (option.disabled && !hasChildren) return;
}} if (!option.disabled) insertMention(option);
onMouseEnter={() => { if (hasChildren) {
setSelectedIndex(globalIndex); calcChildPanelPos(option.key);
if (hasChildren) { setExpandedParent(option);
const el = itemRefs.current.get(option.key);
if (el && popupRef.current) {
const elRect = el.getBoundingClientRect();
const popupRect = popupRef.current.getBoundingClientRect();
setChildPanelTop(calcChildPanelTop(elRect, popupRect));
} }
setExpandedParent(option); }}
} else { onMouseEnter={() => {
setExpandedParent(null); setSelectedIndex(globalIndex);
setActivePanel('main');
setChildActiveIndex(-1);
if (hasChildren) {
calcChildPanelPos(option.key);
setExpandedParent(option);
} else {
setExpandedParent(null);
}
}}
>
{option.label &&
<div className="rb:font-medium">
<span className="rb:text-[#155EEF]">{`{x}`}</span> {option.label}
</div>
} }
}} <Space size={2}>
> {option.dataType && <span>{option.dataType}</span>}
{option.label && {hasChildren && <div className="rb:size-3 rb:bg-cover rb:bg-[url('@/assets/images/common/arrow_up.svg')] rb:rotate-90"></div>}
<div className="rb:font-medium"> </Space>
<span className="rb:text-[#155EEF]">{`{x}`}</span> {option.label} </Flex>
</div> );
} })}
<Space size={2}> </Flex>
{option.dataType && <span>{option.dataType}</span>} </div>
{hasChildren && <div className="rb:size-3 rb:bg-cover rb:bg-[url('@/assets/images/common/arrow_up.svg')] rb:rotate-90"></div>} );
</Space> })}
</Flex> </Flex>
); </div>
})}
</Flex>
</div>
);
})}
</Flex>
</div> </div>
{/* Child variables panel - floats to the left */}
{expandedParent?.children?.length && ( {/* Child variables panel - fixed positioned via portal to avoid clipping */}
{expandedParent?.children?.length && createPortal(
<div <div
className="rb:absolute rb:min-w-70 rb:max-h-57.5 rb:overflow-y-auto rb:text-[12px] rb:z-1000 rb:bg-white rb:rounded-lg rb:border-[0.5px] rb:border-[#EBEBEB] rb:shadow-[0px_2px_6px_0px_rgba(0,0,0,0.1)] rb:py-3 rb:px-2" onMouseDown={(e) => e.preventDefault()}
style={{ className="rb:min-w-70 rb:max-h-57.5 rb:overflow-y-auto rb:text-[12px] rb:fixed rb:z-1000 rb:bg-white rb:rounded-lg rb:border-[0.5px] rb:border-[#EBEBEB] rb:shadow-[0px_2px_6px_0px_rgba(0,0,0,0.1)] rb:py-3 rb:px-2"
top: childPanelTop, style={{ top: childPanelPos.top, right: childPanelPos.right }}
right: 'calc(100% + 8px)', onMouseEnter={() => setActivePanel('child')}
transform: 'translateY(-8px)', onMouseLeave={() => { setActivePanel('main'); setChildActiveIndex(-1); }}
}}
onMouseEnter={() => setExpandedParent(expandedParent)}
> >
<div className="rb:pb-2 rb:mb-1 rb:font-medium rb:text-[#5B6167] rb-border-b"> <div className="rb:pb-2 rb:mb-1 rb:font-medium rb:text-[#5B6167] rb-border-b">
<Flex justify="space-between" align="center" gap={8}> <Flex justify="space-between" align="center" gap={8}>
@@ -373,21 +413,21 @@ const AutocompletePlugin: FC<{ options: Suggestion[] }> = ({ options }) => {
<span>{expandedParent.dataType}</span> <span>{expandedParent.dataType}</span>
</Flex> </Flex>
</div> </div>
{expandedParent.children.map((child) => { {expandedParent.children.map((child, ci) => {
const childIndex = flatOptions.indexOf(child); const isChildActive = activePanel === 'child' && ci === childActiveIndex;
return ( return (
<Flex <Flex
key={child.key} key={child.key}
data-selected={selectedIndex === childIndex} ref={(el) => { if (el) childItemRefs.current.set(child.key, el); }}
className={clsx("rb:px-2! rb:py-0.75! rb:rounded-sm rb:leading-4.5 rb:text-[#5B6167] rb:hover:bg-[#F6F6F6]", { className={clsx("rb:px-2! rb:py-0.75! rb:rounded-sm rb:leading-4.5 rb:text-[#5B6167] rb:hover:bg-[#F6F6F6]", {
'rb:bg-[#F6F6F6]': selectedIndex === childIndex, 'rb:bg-[#F6F6F6]': isChildActive,
'rb:cursor-not-allowed rb:opacity-65': child.disabled, 'rb:cursor-not-allowed rb:opacity-65': child.disabled,
'rb:cursor-pointer': !child.disabled, 'rb:cursor-pointer': !child.disabled,
})} })}
align="center" align="center"
justify="space-between" justify="space-between"
onClick={() => !child.disabled && insertMention(child)} onClick={() => !child.disabled && insertMention(child)}
onMouseEnter={() => setSelectedIndex(childIndex)} onMouseEnter={() => { setActivePanel('child'); setChildActiveIndex(ci); }}
> >
<span className="rb:font-medium"> <span className="rb:font-medium">
<span className="rb:text-[#155EEF]">{`{x}`}</span> {child.label} <span className="rb:text-[#155EEF]">{`{x}`}</span> {child.label}
@@ -396,9 +436,10 @@ const AutocompletePlugin: FC<{ options: Suggestion[] }> = ({ options }) => {
</Flex> </Flex>
); );
})} })}
</div> </div>,
document.body
)} )}
</div> </>
); );
} }
export default AutocompletePlugin export default AutocompletePlugin

View File

@@ -5,6 +5,7 @@
* @Last Modified time: 2026-04-07 14:50:14 * @Last Modified time: 2026-04-07 14:50:14
*/ */
import { useEffect, useLayoutEffect, useState, useRef, type FC } from 'react'; import { useEffect, useLayoutEffect, useState, useRef, type FC } from 'react';
import { createPortal } from 'react-dom';
import { useLexicalComposerContext } from '@lexical/react/LexicalComposerContext'; import { useLexicalComposerContext } from '@lexical/react/LexicalComposerContext';
import { import {
$getSelection, $isRangeSelection, $isTextNode, $getSelection, $isRangeSelection, $isTextNode,
@@ -12,6 +13,7 @@ import {
KEY_ARROW_UP_COMMAND, KEY_ESCAPE_COMMAND, KEY_ARROW_UP_COMMAND, KEY_ESCAPE_COMMAND,
} from 'lexical'; } from 'lexical';
import { Space, Flex } from 'antd'; import { Space, Flex } from 'antd';
import clsx from 'clsx';
import { CLOSE_AUTOCOMPLETE_COMMAND } from '../commands'; import { CLOSE_AUTOCOMPLETE_COMMAND } from '../commands';
import type { Suggestion } from './AutocompletePlugin'; import type { Suggestion } from './AutocompletePlugin';
@@ -22,17 +24,22 @@ const Jinja2AutocompletePlugin: FC<{ options: Suggestion[] }> = ({ options }) =>
const [selectedIndex, setSelectedIndex] = useState(0); const [selectedIndex, setSelectedIndex] = useState(0);
const [popupPosition, setPopupPosition] = useState({ top: 0, left: 0, anchorBottom: 0 }); const [popupPosition, setPopupPosition] = useState({ top: 0, left: 0, anchorBottom: 0 });
const [expandedParent, setExpandedParent] = useState<Suggestion | null>(null); const [expandedParent, setExpandedParent] = useState<Suggestion | null>(null);
const [childPanelTop, setChildPanelTop] = useState(0); const [childPanelPos, setChildPanelPos] = useState({ top: 0, right: 0 });
const [activePanel, setActivePanel] = useState<'main' | 'child'>('main');
const [childActiveIndex, setChildActiveIndex] = useState(-1);
const popupRef = useRef<HTMLDivElement>(null); const popupRef = useRef<HTMLDivElement>(null);
const itemRefs = useRef<Map<string, HTMLElement>>(new Map()); const itemRefs = useRef<Map<string, HTMLElement>>(new Map());
const childItemRefs = useRef<Map<string, HTMLElement>>(new Map());
const CHILD_PANEL_HEIGHT = 280; const CHILD_PANEL_HEIGHT = 280;
useLayoutEffect(() => { useLayoutEffect(() => {
if (!popupRef.current || !showSuggestions) return; if (!popupRef.current || !showSuggestions) return;
const { top, anchorBottom } = popupPosition; const { top, left, anchorBottom } = popupPosition;
const popupHeight = popupRef.current.offsetHeight; const popupHeight = popupRef.current.offsetHeight;
const popupWidth = popupRef.current.offsetWidth;
const MARGIN = 10; const MARGIN = 10;
let finalTop: number; let finalTop: number;
if (top - popupHeight - MARGIN >= 0) { if (top - popupHeight - MARGIN >= 0) {
finalTop = top - popupHeight - MARGIN; finalTop = top - popupHeight - MARGIN;
@@ -41,51 +48,57 @@ const Jinja2AutocompletePlugin: FC<{ options: Suggestion[] }> = ({ options }) =>
if (finalTop + popupHeight > window.innerHeight - MARGIN) if (finalTop + popupHeight > window.innerHeight - MARGIN)
finalTop = window.innerHeight - popupHeight - MARGIN; finalTop = window.innerHeight - popupHeight - MARGIN;
} }
if (finalTop !== top) setPopupPosition(prev => ({ ...prev, top: finalTop }));
let finalLeft = left;
if (finalLeft + popupWidth > window.innerWidth - MARGIN)
finalLeft = window.innerWidth - popupWidth - MARGIN;
if (finalLeft < MARGIN) finalLeft = MARGIN;
if (finalTop !== top || finalLeft !== left)
setPopupPosition(prev => ({ ...prev, top: finalTop, left: finalLeft }));
}, [showSuggestions, popupPosition.anchorBottom]); }, [showSuggestions, popupPosition.anchorBottom]);
const calcChildPanelTop = (elRect: DOMRect, popupRect: DOMRect) => { const calcChildPanelPos = (key: string) => {
const relativeTop = elRect.top - popupRect.top; const el = itemRefs.current.get(key);
const overflow = popupRect.top + relativeTop + CHILD_PANEL_HEIGHT - (window.innerHeight - 10); if (!el || !popupRef.current) return;
return overflow > 0 ? relativeTop - overflow : relativeTop; const elRect = el.getBoundingClientRect();
const popupRect = popupRef.current.getBoundingClientRect();
const actualChildHeight = Math.min(CHILD_PANEL_HEIGHT, popupRect.height);
const top = Math.max(10, popupRect.bottom - actualChildHeight);
setChildPanelPos({ top, right: window.innerWidth - elRect.left + 8 });
}; };
const scrollSelectedIntoView = () => { const resetState = () => {
if (!popupRef.current) return; setShowSuggestions(false);
const selectedElement = popupRef.current.querySelector('[data-selected="true"]'); setExpandedParent(null);
if (!selectedElement) return; setChildPanelPos({ top: 0, right: 0 });
const container = popupRef.current; setActivePanel('main');
const element = selectedElement as HTMLElement; setChildActiveIndex(-1);
const containerRect = container.getBoundingClientRect();
const elementRect = element.getBoundingClientRect();
if (elementRect.bottom > containerRect.bottom) {
container.scrollTop += elementRect.bottom - containerRect.bottom;
} else if (elementRect.top < containerRect.top) {
container.scrollTop -= containerRect.top - elementRect.top;
}
}; };
useEffect(() => { useEffect(() => {
return editor.registerUpdateListener(({ editorState }) => { return editor.registerUpdateListener(({ editorState }) => {
editorState.read(() => { editorState.read(() => {
const selection = $getSelection(); const selection = $getSelection();
if (!selection || !$isRangeSelection(selection)) { if (!selection || !$isRangeSelection(selection)) { setShowSuggestions(false); return; }
setShowSuggestions(false);
return;
}
const anchorNode = selection.anchor.getNode(); const anchorNode = selection.anchor.getNode();
const anchorOffset = selection.anchor.offset; const anchorOffset = selection.anchor.offset;
const textBeforeCursor = anchorNode.getTextContent().substring(0, anchorOffset); const textBeforeCursor = anchorNode.getTextContent().substring(0, anchorOffset);
const shouldShow = textBeforeCursor.endsWith('/'); const shouldShow = textBeforeCursor.endsWith('/');
setShowSuggestions(shouldShow); setShowSuggestions(shouldShow);
if (!shouldShow) { setSelectedIndex(0); setExpandedParent(null); setChildPanelTop(0); return; } if (!shouldShow) {
setSelectedIndex(0);
setExpandedParent(null);
setChildPanelPos({ top: 0, right: 0 });
setActivePanel('main');
setChildActiveIndex(-1);
return;
}
const domSelection = window.getSelection(); const domSelection = window.getSelection();
if (domSelection && domSelection.rangeCount > 0) { if (domSelection && domSelection.rangeCount > 0) {
const rect = domSelection.getRangeAt(0).getBoundingClientRect(); const rect = domSelection.getRangeAt(0).getBoundingClientRect();
const popupWidth = 280;
let left = rect.left; let left = rect.left;
if (left + popupWidth > window.innerWidth) left = window.innerWidth - popupWidth - 10; if (left + 280 > window.innerWidth) left = window.innerWidth - 280 - 10;
if (left < 10) left = 10; if (left < 10) left = 10;
setPopupPosition({ top: rect.top, left, anchorBottom: rect.bottom }); setPopupPosition({ top: rect.top, left, anchorBottom: rect.bottom });
} }
@@ -96,7 +109,7 @@ const Jinja2AutocompletePlugin: FC<{ options: Suggestion[] }> = ({ options }) =>
useEffect(() => { useEffect(() => {
return editor.registerCommand( return editor.registerCommand(
CLOSE_AUTOCOMPLETE_COMMAND, CLOSE_AUTOCOMPLETE_COMMAND,
() => { setShowSuggestions(false); setExpandedParent(null); setChildPanelTop(0); return true; }, () => { resetState(); return true; },
COMMAND_PRIORITY_HIGH, COMMAND_PRIORITY_HIGH,
); );
}, [editor]); }, [editor]);
@@ -119,9 +132,7 @@ const Jinja2AutocompletePlugin: FC<{ options: Suggestion[] }> = ({ options }) =>
} }
}); });
document.dispatchEvent(new CustomEvent('jinja2-variable-inserted', { detail: { value: suggestion.value } })); document.dispatchEvent(new CustomEvent('jinja2-variable-inserted', { detail: { value: suggestion.value } }));
setShowSuggestions(false); resetState();
setExpandedParent(null);
setChildPanelTop(0);
}; };
const groupedSuggestions = options.reduce((groups: Record<string, Suggestion[]>, s) => { const groupedSuggestions = options.reduce((groups: Record<string, Suggestion[]>, s) => {
@@ -131,152 +142,227 @@ const Jinja2AutocompletePlugin: FC<{ options: Suggestion[] }> = ({ options }) =>
return groups; return groups;
}, {}); }, {});
const allOptions = Object.values(groupedSuggestions).flat().flatMap(o => // Flat list of main-panel items for keyboard navigation
o.key === expandedParent?.key && o.children?.length ? [o, ...o.children] : [o] const flatOptions = Object.values(groupedSuggestions).flat();
);
// Sync child panel position when keyboard navigates to a parent with children
useEffect(() => {
if (selectedIndex < 0 || selectedIndex >= flatOptions.length) return;
const s = flatOptions[selectedIndex];
if (s.children?.length) {
calcChildPanelPos(s.key);
setExpandedParent(s);
} else {
setExpandedParent(null);
}
// eslint-disable-next-line react-hooks/exhaustive-deps
}, [selectedIndex]);
// Scroll child active item into view
useEffect(() => {
if (!expandedParent?.children?.length || childActiveIndex < 0) return;
const child = expandedParent.children[childActiveIndex];
if (child) childItemRefs.current.get(child.key)?.scrollIntoView({ block: 'nearest' });
}, [childActiveIndex, expandedParent]);
useEffect(() => { useEffect(() => {
if (!showSuggestions) return; if (!showSuggestions) return;
return editor.registerCommand( return editor.registerCommand(
KEY_ENTER_COMMAND, KEY_ENTER_COMMAND,
(event) => { (event) => {
const opt = allOptions[selectedIndex]; if (!showSuggestions) return false;
if (opt && !opt.disabled) { event?.preventDefault(); insertMention(opt); return true; } if (activePanel === 'child' && expandedParent?.children?.length) {
const child = expandedParent.children[childActiveIndex];
if (child && !child.disabled) { event?.preventDefault(); insertMention(child); return true; }
} else if (flatOptions.length > 0) {
const opt = flatOptions[selectedIndex];
if (opt && !opt.disabled) { event?.preventDefault(); insertMention(opt); return true; }
}
return false; return false;
}, },
COMMAND_PRIORITY_HIGH, COMMAND_PRIORITY_HIGH,
); );
}, [showSuggestions, selectedIndex, allOptions]); }, [showSuggestions, selectedIndex, flatOptions, activePanel, childActiveIndex, expandedParent]);
useEffect(() => { useEffect(() => {
if (!showSuggestions) return; if (!showSuggestions) return;
const down = editor.registerCommand(KEY_ARROW_DOWN_COMMAND, (e) => { const down = editor.registerCommand(KEY_ARROW_DOWN_COMMAND, (e) => {
if (!showSuggestions) return false;
e?.preventDefault(); e?.preventDefault();
setSelectedIndex(prev => { if (activePanel === 'child' && expandedParent?.children) {
let next = prev + 1; setChildActiveIndex(i => Math.min(i + 1, expandedParent.children!.length - 1));
while (next < allOptions.length && allOptions[next].disabled) next++; } else {
setTimeout(scrollSelectedIntoView, 0); setSelectedIndex(prev => {
return next >= allOptions.length ? prev : next; let next = prev + 1;
}); while (next < flatOptions.length && flatOptions[next].disabled && !flatOptions[next].children?.length) next++;
const newIndex = next >= flatOptions.length ? prev : next;
setTimeout(() => itemRefs.current.get(flatOptions[newIndex]?.key)?.scrollIntoView({ block: 'nearest' }), 0);
return newIndex;
});
}
return true; return true;
}, COMMAND_PRIORITY_HIGH); }, COMMAND_PRIORITY_HIGH);
const up = editor.registerCommand(KEY_ARROW_UP_COMMAND, (e) => { const up = editor.registerCommand(KEY_ARROW_UP_COMMAND, (e) => {
if (!showSuggestions) return false;
e?.preventDefault(); e?.preventDefault();
setSelectedIndex(prev => { if (activePanel === 'child' && expandedParent?.children) {
let p = prev - 1; setChildActiveIndex(i => Math.max(i - 1, 0));
while (p >= 0 && allOptions[p].disabled) p--; } else {
setTimeout(scrollSelectedIntoView, 0); setSelectedIndex(prev => {
return p < 0 ? prev : p; let p = prev - 1;
}); while (p >= 0 && flatOptions[p].disabled && !flatOptions[p].children?.length) p--;
const newIndex = p < 0 ? prev : p;
setTimeout(() => itemRefs.current.get(flatOptions[newIndex]?.key)?.scrollIntoView({ block: 'nearest' }), 0);
return newIndex;
});
}
return true; return true;
}, COMMAND_PRIORITY_HIGH); }, COMMAND_PRIORITY_HIGH);
const esc = editor.registerCommand(KEY_ESCAPE_COMMAND, (e) => { const esc = editor.registerCommand(KEY_ESCAPE_COMMAND, (e) => {
e?.preventDefault(); setShowSuggestions(false); return true; e?.preventDefault(); setShowSuggestions(false); return true;
}, COMMAND_PRIORITY_HIGH); }, COMMAND_PRIORITY_HIGH);
return () => { down(); up(); esc(); }; return () => { down(); up(); esc(); };
}, [showSuggestions, selectedIndex, allOptions, editor]); }, [showSuggestions, selectedIndex, flatOptions, editor, activePanel, childActiveIndex, expandedParent]);
// ArrowLeft/Right for panel switching
useEffect(() => {
if (!showSuggestions) return;
const handler = (e: KeyboardEvent) => {
if (e.key === 'ArrowLeft') {
const current = flatOptions[selectedIndex];
if (activePanel === 'main' && current?.children?.length) {
e.preventDefault();
setActivePanel('child');
setChildActiveIndex(0);
}
} else if (e.key === 'ArrowRight') {
if (activePanel === 'child') {
e.preventDefault();
setActivePanel('main');
setChildActiveIndex(-1);
}
}
};
document.addEventListener('keydown', handler, true);
return () => document.removeEventListener('keydown', handler, true);
}, [showSuggestions, activePanel, selectedIndex, flatOptions]);
if (!showSuggestions || Object.keys(groupedSuggestions).length === 0) return null; if (!showSuggestions || Object.keys(groupedSuggestions).length === 0) return null;
return ( return (
<div <>
ref={popupRef} <div
data-autocomplete-popup="true" ref={popupRef}
onMouseDown={(e) => e.preventDefault()} data-autocomplete-popup="true"
className="rb:fixed rb:z-1000 rb:bg-white rb:rounded-xl rb:shadow-[0px_2px_12px_0px_rgba(23,23,25,0.12)]" onMouseDown={(e) => e.preventDefault()}
style={{ top: popupPosition.top, left: popupPosition.left }} className="rb:fixed rb:z-1000 rb:bg-white rb:rounded-lg rb:border-[0.5px] rb:border-[#EBEBEB] rb:shadow-[0px_2px_6px_0px_rgba(0,0,0,0.1)] rb:py-3 rb:px-2"
> style={{ top: popupPosition.top, left: popupPosition.left }}
<div className="rb:py-1 rb:min-w-70 rb:max-h-50 rb:overflow-y-auto"> >
<Flex vertical gap={12}> <div className="rb:min-w-70 rb:max-h-57.5 rb:overflow-y-auto">
{Object.entries(groupedSuggestions).map(([nodeId, nodeOptions]) => ( <Flex vertical gap={12}>
<div key={nodeId}> {Object.entries(groupedSuggestions).map(([nodeId, nodeOptions]) => {
<Flex align="center" gap={4} className="rb:px-3! rb:text-[12px] rb:py-1.25! rb:font-medium rb:text-[#5B6167]"> const nodeName = nodeOptions[0]?.nodeData?.name || nodeId;
{nodeOptions[0]?.nodeData?.icon && <div className={`rb:size-3 rb:bg-cover ${nodeOptions[0].nodeData.icon}`} />}
{nodeOptions[0]?.nodeData?.name || nodeId}
</Flex>
{nodeOptions.map((option) => {
const globalIndex = allOptions.indexOf(option);
const hasChildren = !!option.children?.length;
const isExpanded = expandedParent?.key === option.key;
return ( return (
<Flex <div key={nodeId} className="rb:text-[12px]">
key={option.key} {nodeName !== 'undefined' &&
ref={(el) => { if (el) itemRefs.current.set(option.key, el); }} <div className="rb:px-2 rb:leading-4.25 rb:mb-1.25 rb:font-medium rb:text-[#5B6167]">
data-selected={selectedIndex === globalIndex} {nodeName}
className="rb:pl-6! rb:pr-3! rb:py-2!" </div>
align="center" }
justify="space-between" <Flex vertical gap={2}>
style={{ {nodeOptions.map((option) => {
cursor: option.disabled ? 'not-allowed' : 'pointer', const globalIndex = flatOptions.indexOf(option);
background: (selectedIndex === globalIndex || isExpanded) ? '#f0f8ff' : 'white', const hasChildren = !!option.children?.length;
opacity: option.disabled ? 0.5 : 1, const isExpanded = expandedParent?.key === option.key;
}} const isActive = activePanel === 'main' && selectedIndex === globalIndex;
onClick={() => { if (option.disabled || hasChildren) return; insertMention(option); }} return (
onMouseEnter={() => { <Flex
setSelectedIndex(globalIndex); key={option.key}
if (hasChildren) { ref={(el) => { if (el) itemRefs.current.set(option.key, el); }}
const el = itemRefs.current.get(option.key); className={clsx('rb:px-2! rb:py-0.75! rb:rounded-sm rb:leading-4.5 rb:text-[#5B6167] rb:hover:bg-[#F6F6F6]', {
if (el && popupRef.current) { 'rb:bg-[#F6F6F6]': isActive || isExpanded,
setChildPanelTop(calcChildPanelTop(el.getBoundingClientRect(), popupRef.current.getBoundingClientRect())); 'rb:cursor-not-allowed rb:opacity-65': option.disabled && !hasChildren,
} 'rb:cursor-pointer': !option.disabled || hasChildren,
setExpandedParent(option); })}
} else { align="center"
setExpandedParent(null); justify="space-between"
} onClick={() => {
}} if (option.disabled && !hasChildren) return;
> if (!option.disabled) insertMention(option);
<Space size={4}> if (hasChildren) { calcChildPanelPos(option.key); setExpandedParent(option); }
<span className="rb:text-[#155EEF]">{option.isContext ? '📄' : '{x}'}</span> }}
<span>{option.label}</span> onMouseEnter={() => {
</Space> setSelectedIndex(globalIndex);
<Space size={4}> setActivePanel('main');
{option.dataType && <span className="rb:text-[#5B6167]">{option.dataType}</span>} setChildActiveIndex(-1);
{hasChildren && <span className="rb:text-[#5B6167] rb:ml-1"></span>} if (hasChildren) { calcChildPanelPos(option.key); setExpandedParent(option); }
</Space> else setExpandedParent(null);
</Flex> }}
>
{option.label &&
<div className="rb:font-medium">
<span className="rb:text-[#155EEF]">{`{x}`}</span> {option.label}
</div>
}
<Space size={2}>
{option.dataType && <span>{option.dataType}</span>}
{hasChildren && <div className="rb:size-3 rb:bg-cover rb:bg-[url('@/assets/images/common/arrow_up.svg')] rb:rotate-90"></div>}
</Space>
</Flex>
);
})}
</Flex>
</div>
); );
})} })}
</div> </Flex>
))} </div>
</Flex>
</div> </div>
{expandedParent?.children?.length && (
{expandedParent?.children?.length && createPortal(
<div <div
className="rb:absolute rb:bg-white rb:rounded-xl rb:py-1 rb:min-w-60 rb:max-h-60 rb:overflow-y-auto rb:shadow-[0px_2px_12px_0px_rgba(23,23,25,0.12)]" onMouseDown={(e) => e.preventDefault()}
style={{ top: childPanelTop, right: 'calc(100% + 8px)', transform: 'translateY(-8px)' }} className="rb:min-w-70 rb:max-h-57.5 rb:overflow-y-auto rb:text-[12px] rb:fixed rb:z-1000 rb:bg-white rb:rounded-lg rb:border-[0.5px] rb:border-[#EBEBEB] rb:shadow-[0px_2px_6px_0px_rgba(0,0,0,0.1)] rb:py-3 rb:px-2"
onMouseEnter={() => setExpandedParent(expandedParent)} style={{ top: childPanelPos.top, right: childPanelPos.right }}
onMouseEnter={() => setActivePanel('child')}
onMouseLeave={() => { setActivePanel('main'); setChildActiveIndex(-1); }}
> >
<div className="rb:px-3 rb:py-2 rb:text-[12px] rb:font-medium rb:text-[#5B6167] rb:border-b rb:border-[#F0F0F0]"> <div className="rb:pb-2 rb:mb-1 rb:font-medium rb:text-[#5B6167] rb-border-b">
<Flex justify="space-between" align="center"> <Flex justify="space-between" align="center" gap={8}>
<span>{expandedParent.nodeData.name}.{expandedParent.label}</span> <span>{expandedParent.nodeData.name}.{expandedParent.label}</span>
<span>{expandedParent.dataType}</span> <span>{expandedParent.dataType}</span>
</Flex> </Flex>
</div> </div>
{expandedParent.children.map((child) => { {expandedParent.children.map((child, ci) => {
const childIndex = allOptions.indexOf(child); const isChildActive = activePanel === 'child' && ci === childActiveIndex;
return ( return (
<Flex <Flex
key={child.key} key={child.key}
data-selected={selectedIndex === childIndex} ref={(el) => { if (el) childItemRefs.current.set(child.key, el); }}
className="rb:px-3! rb:py-2!" className={clsx('rb:px-2! rb:py-0.75! rb:rounded-sm rb:leading-4.5 rb:text-[#5B6167] rb:hover:bg-[#F6F6F6]', {
'rb:bg-[#F6F6F6]': isChildActive,
'rb:cursor-not-allowed rb:opacity-65': child.disabled,
'rb:cursor-pointer': !child.disabled,
})}
align="center" align="center"
justify="space-between" justify="space-between"
style={{
cursor: child.disabled ? 'not-allowed' : 'pointer',
background: selectedIndex === childIndex ? '#f0f8ff' : 'white',
opacity: child.disabled ? 0.5 : 1,
}}
onClick={() => !child.disabled && insertMention(child)} onClick={() => !child.disabled && insertMention(child)}
onMouseEnter={() => setSelectedIndex(childIndex)} onMouseEnter={() => { setActivePanel('child'); setChildActiveIndex(ci); }}
> >
<span>{child.label}</span> <span className="rb:font-medium">
{child.dataType && <span className="rb:text-[#5B6167]">{child.dataType}</span>} <span className="rb:text-[#155EEF]">{`{x}`}</span> {child.label}
</span>
{child.dataType && <span>{child.dataType}</span>}
</Flex> </Flex>
); );
})} })}
</div> </div>,
document.body
)} )}
</div> </>
); );
}; };

View File

@@ -2,7 +2,7 @@
* @Author: ZhaoYing * @Author: ZhaoYing
* @Date: 2026-02-03 15:40:13 * @Date: 2026-02-03 15:40:13
* @Last Modified by: ZhaoYing * @Last Modified by: ZhaoYing
* @Last Modified time: 2026-04-13 11:25:40 * @Last Modified time: 2026-04-16 13:57:30
*/ */
import { useState, useRef, useEffect, useLayoutEffect, type FC } from 'react' import { useState, useRef, useEffect, useLayoutEffect, type FC } from 'react'
import { createPortal } from 'react-dom' import { createPortal } from 'react-dom'
@@ -41,14 +41,33 @@ const VariableSelect: FC<VariableSelectProps> = ({
const [open, setOpen] = useState(false); const [open, setOpen] = useState(false);
const [search, setSearch] = useState(''); const [search, setSearch] = useState('');
const [expandedParentKey, setExpandedParentKey] = useState<string | null>(null); const [expandedParentKey, setExpandedParentKey] = useState<string | null>(null);
const [activeIndex, setActiveIndex] = useState<number>(-1);
const [activePanel, setActivePanel] = useState<'main' | 'child'>('main');
const [childActiveIndex, setChildActiveIndex] = useState<number>(-1);
const [dropdownPos, setDropdownPos] = useState({ top: 0, left: 0, width: 0 }); const [dropdownPos, setDropdownPos] = useState({ top: 0, left: 0, width: 0 });
const [childPanelPos, setChildPanelPos] = useState({ top: 0, right: 0 }); const [childPanelPos, setChildPanelPos] = useState({ top: 0, right: 0 });
const containerRef = useRef<HTMLDivElement>(null); const containerRef = useRef<HTMLDivElement>(null);
const dropdownRef = useRef<HTMLDivElement>(null); const dropdownRef = useRef<HTMLDivElement>(null);
const itemRefs = useRef<Map<string, HTMLElement>>(new Map()); const itemRefs = useRef<Map<string, HTMLElement>>(new Map());
const childItemRefs = useRef<Map<string, HTMLElement>>(new Map());
const activeKeyRef = useRef<string | null>(null);
const CHILD_PANEL_HEIGHT = 280; // max-h-60 (240) + header (~40) const CHILD_PANEL_HEIGHT = 280; // max-h-60 (240) + header (~40)
const calcChildPos = (key: string) => {
const el = itemRefs.current.get(key);
if (!el) return;
const rect = el.getBoundingClientRect();
const dropdownEl = dropdownRef.current;
if (!dropdownEl) return;
const dropdownRect = dropdownEl.getBoundingClientRect();
const dropdownBottom = dropdownRect.bottom;
const actualChildHeight = Math.min(CHILD_PANEL_HEIGHT, dropdownRect.height);
// Bottom-align child panel with main panel
const top = Math.max(10, dropdownBottom - actualChildHeight);
setChildPanelPos({ top, right: window.innerWidth - rect.left + 8 });
};
// Calculate dropdown position (runs synchronously after DOM paint to avoid flicker) // Calculate dropdown position (runs synchronously after DOM paint to avoid flicker)
useLayoutEffect(() => { useLayoutEffect(() => {
if (!open || !containerRef.current) return; if (!open || !containerRef.current) return;
@@ -69,7 +88,9 @@ const VariableSelect: FC<VariableSelectProps> = ({
? triggerRect.bottom + MARGIN ? triggerRect.bottom + MARGIN
: Math.max(MARGIN, triggerRect.top - dropdownHeight - MARGIN); : Math.max(MARGIN, triggerRect.top - dropdownHeight - MARGIN);
setDropdownPos({ top, left, width }); setDropdownPos({ top, left, width });
}, [open, search, Array.isArray(value) ? value.length : 0]); // Re-calculate child panel position if expanded
if (expandedParentKey) calcChildPos(expandedParentKey);
}, [open, search, Array.isArray(value) ? value.length : 0, options.length, expandedParentKey]);
const filteredOptions = filterBooleanType const filteredOptions = filterBooleanType
? options.filter(o => o.dataType !== 'boolean') ? options.filter(o => o.dataType !== 'boolean')
@@ -107,6 +128,12 @@ const VariableSelect: FC<VariableSelectProps> = ({
}, {}) }, {})
: groupedSuggestions; : groupedSuggestions;
useEffect(() => {
if (!expandedParentKey) return;
calcChildPos(expandedParentKey);
// eslint-disable-next-line react-hooks/exhaustive-deps
}, [dropdownPos, expandedParentKey]);
useEffect(() => { useEffect(() => {
if (!open) return; if (!open) return;
const updatePos = () => { const updatePos = () => {
@@ -151,6 +178,87 @@ const VariableSelect: FC<VariableSelectProps> = ({
return () => document.removeEventListener('mousedown', handler); return () => document.removeEventListener('mousedown', handler);
}, [open]); }, [open]);
// Flat list of all visible selectable items (main panel only, no children expanded inline)
const flatItems = Object.values(filteredGroups).flat();
useEffect(() => {
setActiveIndex(-1);
setActivePanel('main');
setChildActiveIndex(-1);
}, [open, search]);
useEffect(() => {
if (activeIndex < 0 || activeIndex >= flatItems.length) {
setExpandedParentKey(null);
return;
}
const s = flatItems[activeIndex];
activeKeyRef.current = s.key;
itemRefs.current.get(s.key)?.scrollIntoView({ block: 'nearest' });
if (s.children?.length) {
calcChildPos(s.key);
setExpandedParentKey(s.key);
} else {
setExpandedParentKey(null);
}
// eslint-disable-next-line react-hooks/exhaustive-deps
}, [activeIndex]);
useEffect(() => {
if (!expandedParent?.children?.length || childActiveIndex < 0) return;
const child = expandedParent.children[childActiveIndex];
if (child) childItemRefs.current.get(child.key)?.scrollIntoView({ block: 'nearest' });
}, [childActiveIndex, expandedParent]);
useEffect(() => {
if (!open) return;
const handler = (e: KeyboardEvent) => {
const children = expandedParent?.children ?? [];
if (activePanel === 'child') {
if (e.key === 'ArrowDown') {
e.preventDefault();
setChildActiveIndex(i => Math.min(i + 1, children.length - 1));
} else if (e.key === 'ArrowUp') {
e.preventDefault();
setChildActiveIndex(i => Math.max(i - 1, 0));
} else if (e.key === 'ArrowRight') {
e.preventDefault();
setActivePanel('main');
setChildActiveIndex(-1);
} else if (e.key === 'Enter' && childActiveIndex >= 0 && childActiveIndex < children.length) {
e.preventDefault();
const child = children[childActiveIndex];
if (!child.disabled) handleSelect(child);
} else if (e.key === 'Escape') {
setOpen(false);
}
} else {
if (e.key === 'ArrowDown') {
e.preventDefault();
setActiveIndex(i => Math.min(i + 1, flatItems.length - 1));
} else if (e.key === 'ArrowUp') {
e.preventDefault();
setActiveIndex(i => Math.max(i - 1, 0));
} else if (e.key === 'ArrowLeft') {
e.preventDefault();
if (expandedParent?.children?.length) {
setActivePanel('child');
setChildActiveIndex(0);
}
} else if (e.key === 'Enter' && activeIndex >= 0 && activeIndex < flatItems.length) {
e.preventDefault();
const s = flatItems[activeIndex];
if (!s.disabled) handleSelect(s);
} else if (e.key === 'Escape') {
setOpen(false);
}
}
};
document.addEventListener('keydown', handler);
return () => document.removeEventListener('keydown', handler);
// eslint-disable-next-line react-hooks/exhaustive-deps
}, [open, activeIndex, activePanel, childActiveIndex, flatItems, expandedParent]);
const handleSelect = (suggestion: Suggestion) => { const handleSelect = (suggestion: Suggestion) => {
if (multiple) { if (multiple) {
const key = `{{${suggestion.value}}}`; const key = `{{${suggestion.value}}}`;
@@ -171,19 +279,6 @@ const VariableSelect: FC<VariableSelectProps> = ({
e.stopPropagation(); e.stopPropagation();
onChange?.(multiple ? [] : '', multiple ? [] : undefined); onChange?.(multiple ? [] : '', multiple ? [] : undefined);
}; };
const updateChildPos = (key: string) => {
const el = itemRefs.current.get(key);
if (el) {
const rect = el.getBoundingClientRect();
const spaceBelow = window.innerHeight - rect.top - 10;
const top = spaceBelow >= CHILD_PANEL_HEIGHT
? rect.top
: Math.max(10, window.innerHeight - CHILD_PANEL_HEIGHT - 10);
setChildPanelPos({ top, right: window.innerWidth - rect.left + 8 });
}
};
const sep = <span className="rb:text-[#DFE4ED] rb:mx-0.5">/</span>; const sep = <span className="rb:text-[#DFE4ED] rb:mx-0.5">/</span>;
const isConversation = (parentOfSelected ?? selectedSuggestion)?.group === 'CONVERSATION' || const isConversation = (parentOfSelected ?? selectedSuggestion)?.group === 'CONVERSATION' ||
(selectedSuggestion ? filteredOptions.some(o => o.group === 'CONVERSATION' && o.children?.some(c => `{{${c.value}}}` === value)) : false); (selectedSuggestion ? filteredOptions.some(o => o.group === 'CONVERSATION' && o.children?.some(c => `{{${c.value}}}` === value)) : false);
@@ -197,7 +292,7 @@ const VariableSelect: FC<VariableSelectProps> = ({
'rb:w-full rb:flex rb:items-center rb:justify-between rb:cursor-pointer rb:rounded-lg rb:px-2 rb:transition-colors', { 'rb:w-full rb:flex rb:items-center rb:justify-between rb:cursor-pointer rb:rounded-lg rb:px-2 rb:transition-colors', {
'rb:bg-[#F6F6F6] rb:border-none rb:shadow-none': variant === 'filled', 'rb:bg-[#F6F6F6] rb:border-none rb:shadow-none': variant === 'filled',
'rb:border rb:border-[#d9d9d9] hover:rb:border-[#4096ff] rb:bg-white': variant === 'outlined', 'rb:border rb:border-[#d9d9d9] hover:rb:border-[#4096ff] rb:bg-white': variant === 'outlined',
'rb:border-[#4096ff] rb:shadow-[0_0_0_2px_rgba(5,145,255,0.1)]': variant === 'outlined' && open, 'rb:border-[#171719]!': variant === 'outlined' && open,
'rb:border-none rb:shadow-none rb:bg-transparent': variant === 'borderless', 'rb:border-none rb:shadow-none rb:bg-transparent': variant === 'borderless',
'rb:text-[12px]': size === 'small', 'rb:text-[12px]': size === 'small',
'rb:text-[14px]': size !== 'small', 'rb:text-[14px]': size !== 'small',
@@ -244,7 +339,7 @@ const VariableSelect: FC<VariableSelectProps> = ({
})} })}
</Flex> </Flex>
) : ( ) : (
<span className="rb:text-ellipsis rb:overflow-hidden rb:whitespace-nowrap rb:flex-1">{placeholder}</span> <span className="rb:text-[rgba(23,23,25,0.25)] rb:text-ellipsis rb:overflow-hidden rb:whitespace-nowrap rb:flex-1">{placeholder}</span>
) )
) : selectedSuggestion ? ( ) : selectedSuggestion ? (
<div className="rb:flex rb:flex-1 rb:min-w-0 rb:max-w-full"> <div className="rb:flex rb:flex-1 rb:min-w-0 rb:max-w-full">
@@ -260,7 +355,7 @@ const VariableSelect: FC<VariableSelectProps> = ({
</span> </span>
</div> </div>
) : ( ) : (
<span className="rb:text-[#bfbfbf] rb:flex-1">{placeholder}</span> <span className="rb:text-[rgba(23,23,25,0.25)] rb:flex-1">{placeholder}</span>
)} )}
<Space size={4} className="rb:shrink-0 rb:ml-1"> <Space size={4} className="rb:shrink-0 rb:ml-1">
{allowClear && ( {allowClear && (
@@ -306,7 +401,7 @@ const VariableSelect: FC<VariableSelectProps> = ({
key={s.key} key={s.key}
ref={(el) => { if (el) itemRefs.current.set(s.key, el); }} ref={(el) => { if (el) itemRefs.current.set(s.key, el); }}
className={clsx("rb:px-2! rb:py-0.75! rb:rounded-sm rb:leading-4.5 rb:text-[#5B6167] rb:hover:bg-[#F6F6F6]", { className={clsx("rb:px-2! rb:py-0.75! rb:rounded-sm rb:leading-4.5 rb:text-[#5B6167] rb:hover:bg-[#F6F6F6]", {
'rb:bg-[#F6F6F6]': isSelected || isExpanded, 'rb:bg-[#F6F6F6]': isSelected || isExpanded || flatItems.indexOf(s) === activeIndex,
'rb:cursor-not-allowed rb:opacity-65': s.disabled, 'rb:cursor-not-allowed rb:opacity-65': s.disabled,
'rb:cursor-pointer': !s.disabled, 'rb:cursor-pointer': !s.disabled,
})} })}
@@ -315,14 +410,14 @@ const VariableSelect: FC<VariableSelectProps> = ({
onClick={() => { onClick={() => {
if (s.disabled) return; if (s.disabled) return;
if (hasChildren) { if (hasChildren) {
updateChildPos(s.key); calcChildPos(s.key);
setExpandedParentKey(prev => prev === s.key ? null : s.key); setExpandedParentKey(prev => prev === s.key ? null : s.key);
} }
handleSelect(s); handleSelect(s);
}} }}
onMouseEnter={() => { onMouseEnter={() => {
if (hasChildren) { if (hasChildren) {
updateChildPos(s.key); calcChildPos(s.key);
setExpandedParentKey(s.key); setExpandedParentKey(s.key);
} else { } else {
setExpandedParentKey(null); setExpandedParentKey(null);
@@ -370,15 +465,17 @@ const VariableSelect: FC<VariableSelectProps> = ({
<span>{expandedParent.dataType}</span> <span>{expandedParent.dataType}</span>
</Flex> </Flex>
</div> </div>
{expandedParent.children.map(child => { {expandedParent.children.map((child, ci) => {
const isSelected = multiple const isSelected = multiple
? selectedValues.includes(`{{${child.value}}}`) ? selectedValues.includes(`{{${child.value}}}`)
: `{{${child.value}}}` === value; : `{{${child.value}}}` === value;
const isChildActive = activePanel === 'child' && ci === childActiveIndex;
return ( return (
<Flex <Flex
key={child.key} key={child.key}
ref={(el) => { if (el) childItemRefs.current.set(child.key, el); }}
className={clsx("rb:px-2! rb:py-0.75! rb:rounded-sm rb:leading-4.5 rb:text-[#5B6167] rb:hover:bg-[#F6F6F6]", { className={clsx("rb:px-2! rb:py-0.75! rb:rounded-sm rb:leading-4.5 rb:text-[#5B6167] rb:hover:bg-[#F6F6F6]", {
'rb:bg-[#F6F6F6]': isSelected, 'rb:bg-[#F6F6F6]': isSelected || isChildActive,
'rb:cursor-not-allowed rb:opacity-65': child.disabled, 'rb:cursor-not-allowed rb:opacity-65': child.disabled,
'rb:cursor-pointer': !child.disabled, 'rb:cursor-pointer': !child.disabled,
})} })}