2025-02-05 00:07:38 +05:30

74 lines
2.9 KiB
TypeScript

import { DotbaseNodesEnum } from '@/components/dashboard/nodes/types/nodeTypes';
import { DefaultContent, MethodHeaderSkeleton, ToolbarSkeleton } from '@/components/dashboard/nodes/common/ToolbarSkeleton';
import { ValidatorContext } from '@/contexts/ValidatorContext';
import useDnDStore from '@/stores/useDnDStore';
import { InformationCircleIcon } from '@heroicons/react/24/outline';
import React from 'react';
import { Handle, NodeToolbar, Position, NodeProps as ReactFlowNodeProps } from 'reactflow';
import { highlight, languages } from 'prismjs';
import 'prismjs/components/prism-clike';
import 'prismjs/components/prism-python';
import 'prismjs/themes/prism.css';
import Editor from 'react-simple-code-editor';
const Spark: React.FC<ReactFlowNodeProps> = (props) => {
const { errors } = React.useContext(ValidatorContext);
const { updateNode } = useDnDStore();
const [toolbarVisible, setToolbarVisible] = React.useState(false);
const [code, setCode] = React.useState('');
const onCustomFuncChange = (code: string) => {
updateNode(props.id, { func: code });
setCode(code);
};
return (
<div className="nowheel rounded-sm bg-slate-500/10 text-white w-[266px] border-none">
<div
className={`${DotbaseNodesEnum.SPARK} flex justify-between items-center py-2`}
>
<div className="font-bold ml-2">SPARK</div>
<InformationCircleIcon
width={20}
className="text-gray-300 mr-2"
onMouseEnter={() => setToolbarVisible(true)}
onMouseLeave={() => setToolbarVisible(false)}
/>
<NodeToolbar isVisible={toolbarVisible} position={Position.Top}>
<ToolbarSkeleton
header={<MethodHeaderSkeleton name="Spark" />}
content={
<DefaultContent
name="Spark"
description="is a method for your agents to consume when it's necessary."
/>
}
/>
</NodeToolbar>
</div>
<div className="pb-2 px-2 bg-slate-500/10 text-sm">
<div className="flex justify-between items-center">
<Editor
value={code}
onValueChange={onCustomFuncChange}
placeholder="def my_action_cen(arg1, arg2):"
highlight={(code) => highlight(code || '', languages.python!, 'py')}
padding={10}
className="max-w-96 max-h-96 min-h-16 overflow-y-auto bg-[#1a1a1a] text-gray-400 w-full rounded-sm mt-2"
style={{
fontFamily: '"Fira code", "Fira Mono", monospace',
fontSize: 12,
}}
textareaClassName="outline-none w-80"
/>
</div>
{errors?.[props.id]?.func && <span className="text-red-500 text-xs">{errors?.[props.id]?.func}</span>}
</div>
<Handle type="source" position={Position.Bottom} className="rounded-none border-none w-16 h-1" />
</div>
);
};
export default Spark;