mirror of https://github.com/langgenius/dify.git
42 lines
1.4 KiB
TypeScript
42 lines
1.4 KiB
TypeScript
import { useMemo } from 'react'
|
|
import { useTranslation } from 'react-i18next'
|
|
import produce from 'immer'
|
|
import type { BlockEnum } from '../types'
|
|
import {
|
|
NODES_EXTRA_DATA,
|
|
NODES_INITIAL_DATA,
|
|
} from '../constants'
|
|
import { useStore } from '../store'
|
|
import { useIsChatMode } from './use-workflow'
|
|
|
|
export const useNodesInitialData = () => {
|
|
const { t } = useTranslation()
|
|
const nodesDefaultConfigs = useStore(s => s.nodesDefaultConfigs)
|
|
|
|
return useMemo(() => produce(NODES_INITIAL_DATA, (draft) => {
|
|
Object.keys(draft).forEach((key) => {
|
|
draft[key as BlockEnum].title = t(`workflow.blocks.${key}`)
|
|
draft[key as BlockEnum]._isReady = true
|
|
if (nodesDefaultConfigs[key as BlockEnum]) {
|
|
draft[key as BlockEnum] = {
|
|
...draft[key as BlockEnum],
|
|
...nodesDefaultConfigs[key as BlockEnum],
|
|
}
|
|
}
|
|
})
|
|
}), [t, nodesDefaultConfigs])
|
|
}
|
|
|
|
export const useNodesExtraData = () => {
|
|
const { t } = useTranslation()
|
|
const isChatMode = useIsChatMode()
|
|
|
|
return useMemo(() => produce(NODES_EXTRA_DATA, (draft) => {
|
|
Object.keys(draft).forEach((key) => {
|
|
draft[key as BlockEnum].about = t(`workflow.blocksAbout.${key}`)
|
|
draft[key as BlockEnum].availablePrevNodes = draft[key as BlockEnum].getAvailablePrevNodes(isChatMode)
|
|
draft[key as BlockEnum].availableNextNodes = draft[key as BlockEnum].getAvailableNextNodes(isChatMode)
|
|
})
|
|
}), [t, isChatMode])
|
|
}
|