panel.tsx 6.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197
  1. import type {
  2. FC,
  3. ReactElement,
  4. } from 'react'
  5. import {
  6. cloneElement,
  7. memo,
  8. useCallback,
  9. } from 'react'
  10. import {
  11. RiCloseLine,
  12. RiPlayLargeLine,
  13. } from '@remixicon/react'
  14. import { useShallow } from 'zustand/react/shallow'
  15. import { useTranslation } from 'react-i18next'
  16. import NextStep from './components/next-step'
  17. import PanelOperator from './components/panel-operator'
  18. import HelpLink from './components/help-link'
  19. import {
  20. DescriptionInput,
  21. TitleInput,
  22. } from './components/title-description-input'
  23. import ErrorHandleOnPanel from './components/error-handle/error-handle-on-panel'
  24. import { useResizePanel } from './hooks/use-resize-panel'
  25. import cn from '@/utils/classnames'
  26. import BlockIcon from '@/app/components/workflow/block-icon'
  27. import {
  28. WorkflowHistoryEvent,
  29. useAvailableBlocks,
  30. useNodeDataUpdate,
  31. useNodesInteractions,
  32. useNodesReadOnly,
  33. useNodesSyncDraft,
  34. useToolIcon,
  35. useWorkflow,
  36. useWorkflowHistory,
  37. } from '@/app/components/workflow/hooks'
  38. import {
  39. canRunBySingle,
  40. hasErrorHandleNode,
  41. } from '@/app/components/workflow/utils'
  42. import Tooltip from '@/app/components/base/tooltip'
  43. import type { Node } from '@/app/components/workflow/types'
  44. import { useStore as useAppStore } from '@/app/components/app/store'
  45. import { useStore } from '@/app/components/workflow/store'
  46. type BasePanelProps = {
  47. children: ReactElement
  48. } & Node
  49. const BasePanel: FC<BasePanelProps> = ({
  50. id,
  51. data,
  52. children,
  53. }) => {
  54. const { t } = useTranslation()
  55. const { showMessageLogModal } = useAppStore(useShallow(state => ({
  56. showMessageLogModal: state.showMessageLogModal,
  57. })))
  58. const showSingleRunPanel = useStore(s => s.showSingleRunPanel)
  59. const panelWidth = localStorage.getItem('workflow-node-panel-width') ? parseFloat(localStorage.getItem('workflow-node-panel-width')!) : 420
  60. const {
  61. setPanelWidth,
  62. } = useWorkflow()
  63. const { handleNodeSelect } = useNodesInteractions()
  64. const { handleSyncWorkflowDraft } = useNodesSyncDraft()
  65. const { nodesReadOnly } = useNodesReadOnly()
  66. const { availableNextBlocks } = useAvailableBlocks(data.type, data.isInIteration)
  67. const toolIcon = useToolIcon(data)
  68. const handleResize = useCallback((width: number) => {
  69. setPanelWidth(width)
  70. }, [setPanelWidth])
  71. const {
  72. triggerRef,
  73. containerRef,
  74. } = useResizePanel({
  75. direction: 'horizontal',
  76. triggerDirection: 'left',
  77. minWidth: 420,
  78. maxWidth: 720,
  79. onResize: handleResize,
  80. })
  81. const { saveStateToHistory } = useWorkflowHistory()
  82. const {
  83. handleNodeDataUpdate,
  84. handleNodeDataUpdateWithSyncDraft,
  85. } = useNodeDataUpdate()
  86. const handleTitleBlur = useCallback((title: string) => {
  87. handleNodeDataUpdateWithSyncDraft({ id, data: { title } })
  88. saveStateToHistory(WorkflowHistoryEvent.NodeTitleChange)
  89. }, [handleNodeDataUpdateWithSyncDraft, id, saveStateToHistory])
  90. const handleDescriptionChange = useCallback((desc: string) => {
  91. handleNodeDataUpdateWithSyncDraft({ id, data: { desc } })
  92. saveStateToHistory(WorkflowHistoryEvent.NodeDescriptionChange)
  93. }, [handleNodeDataUpdateWithSyncDraft, id, saveStateToHistory])
  94. return (
  95. <div className={cn(
  96. 'relative mr-2 h-full',
  97. showMessageLogModal && '!absolute !mr-0 w-[384px] overflow-hidden -top-[5px] right-[416px] z-0 shadow-lg border-[0.5px] border-components-panel-border rounded-2xl transition-all',
  98. )}>
  99. <div
  100. ref={triggerRef}
  101. className='absolute top-1/2 -translate-y-1/2 -left-2 w-3 h-6 cursor-col-resize resize-x'>
  102. <div className='w-1 h-6 bg-divider-regular rounded-sm'></div>
  103. </div>
  104. <div
  105. ref={containerRef}
  106. className={cn('h-full bg-components-panel-bg shadow-lg border-[0.5px] border-components-panel-border rounded-2xl', showSingleRunPanel ? 'overflow-hidden' : 'overflow-y-auto')}
  107. style={{
  108. width: `${panelWidth}px`,
  109. }}
  110. >
  111. <div className='sticky top-0 bg-components-panel-bg border-b-[0.5px] border-black/5 z-10'>
  112. <div className='flex items-center px-4 pt-4 pb-1'>
  113. <BlockIcon
  114. className='shrink-0 mr-1'
  115. type={data.type}
  116. toolIcon={toolIcon}
  117. size='md'
  118. />
  119. <TitleInput
  120. value={data.title || ''}
  121. onBlur={handleTitleBlur}
  122. />
  123. <div className='shrink-0 flex items-center text-gray-500'>
  124. {
  125. canRunBySingle(data.type) && !nodesReadOnly && (
  126. <Tooltip
  127. popupContent={t('workflow.panel.runThisStep')}
  128. popupClassName='mr-1'
  129. >
  130. <div
  131. className='flex items-center justify-center mr-1 w-6 h-6 rounded-md hover:bg-black/5 cursor-pointer'
  132. onClick={() => {
  133. handleNodeDataUpdate({ id, data: { _isSingleRun: true } })
  134. handleSyncWorkflowDraft(true)
  135. }}
  136. >
  137. <RiPlayLargeLine className='w-4 h-4 text-text-tertiary' />
  138. </div>
  139. </Tooltip>
  140. )
  141. }
  142. <HelpLink nodeType={data.type} />
  143. <PanelOperator id={id} data={data} showHelpLink={false} />
  144. <div className='mx-3 w-[1px] h-3.5 bg-divider-regular' />
  145. <div
  146. className='flex items-center justify-center w-6 h-6 cursor-pointer'
  147. onClick={() => handleNodeSelect(id, true)}
  148. >
  149. <RiCloseLine className='w-4 h-4 text-text-tertiary' />
  150. </div>
  151. </div>
  152. </div>
  153. <div className='p-2'>
  154. <DescriptionInput
  155. value={data.desc || ''}
  156. onChange={handleDescriptionChange}
  157. />
  158. </div>
  159. </div>
  160. <div>
  161. {cloneElement(children, { id, data })}
  162. </div>
  163. {
  164. hasErrorHandleNode(data.type) && (
  165. <ErrorHandleOnPanel
  166. id={id}
  167. data={data}
  168. />
  169. )
  170. }
  171. {
  172. !!availableNextBlocks.length && (
  173. <div className='p-4 border-t-[0.5px] border-t-black/5'>
  174. <div className='flex items-center mb-1 system-sm-semibold-uppercase text-text-secondary'>
  175. {t('workflow.panel.nextStep').toLocaleUpperCase()}
  176. </div>
  177. <div className='mb-2 system-xs-regular text-text-tertiary'>
  178. {t('workflow.panel.addNextStep')}
  179. </div>
  180. <NextStep selectedNode={{ id, data } as Node} />
  181. </div>
  182. )
  183. }
  184. </div>
  185. </div>
  186. )
  187. }
  188. export default memo(BasePanel)