2024-04-08 18:51:46 +08:00
|
|
|
import { useCallback, useEffect, useState } from 'react'
|
|
|
|
import produce from 'immer'
|
|
|
|
import useVarList from '../_base/hooks/use-var-list'
|
|
|
|
import useOutputVarList from '../_base/hooks/use-output-var-list'
|
|
|
|
import { BlockEnum, VarType } from '../../types'
|
2024-11-01 12:39:32 +09:00
|
|
|
import type { Var, Variable } from '../../types'
|
2024-04-08 18:51:46 +08:00
|
|
|
import { useStore } from '../../store'
|
2024-08-21 16:57:14 +08:00
|
|
|
import type { CodeNodeType, OutputVar } from './types'
|
2024-04-08 18:51:46 +08:00
|
|
|
import { CodeLanguage } from './types'
|
|
|
|
import useNodeCrud from '@/app/components/workflow/nodes/_base/hooks/use-node-crud'
|
2025-06-13 16:38:35 +08:00
|
|
|
import {
|
|
|
|
fetchNodeDefault,
|
|
|
|
fetchPipelineNodeDefault,
|
|
|
|
} from '@/service/workflow'
|
2024-04-08 18:51:46 +08:00
|
|
|
import {
|
|
|
|
useNodesReadOnly,
|
|
|
|
} from '@/app/components/workflow/hooks'
|
|
|
|
|
|
|
|
const useConfig = (id: string, payload: CodeNodeType) => {
|
|
|
|
const { nodesReadOnly: readOnly } = useNodesReadOnly()
|
|
|
|
|
2025-06-13 14:42:27 +08:00
|
|
|
const appId = useStore(s => s.appId)
|
2025-06-13 16:38:35 +08:00
|
|
|
const pipelineId = useStore(s => s.pipelineId)
|
2024-04-08 18:51:46 +08:00
|
|
|
|
|
|
|
const [allLanguageDefault, setAllLanguageDefault] = useState<Record<CodeLanguage, CodeNodeType> | null>(null)
|
|
|
|
useEffect(() => {
|
|
|
|
if (appId) {
|
|
|
|
(async () => {
|
|
|
|
const { config: javaScriptConfig } = await fetchNodeDefault(appId, BlockEnum.Code, { code_language: CodeLanguage.javascript }) as any
|
2024-08-21 16:57:14 +08:00
|
|
|
const { config: pythonConfig } = await fetchNodeDefault(appId, BlockEnum.Code, { code_language: CodeLanguage.python3 }) as any
|
2024-04-08 18:51:46 +08:00
|
|
|
setAllLanguageDefault({
|
|
|
|
[CodeLanguage.javascript]: javaScriptConfig as CodeNodeType,
|
|
|
|
[CodeLanguage.python3]: pythonConfig as CodeNodeType,
|
|
|
|
} as any)
|
|
|
|
})()
|
|
|
|
}
|
|
|
|
}, [appId])
|
|
|
|
|
2025-06-13 16:38:35 +08:00
|
|
|
useEffect(() => {
|
|
|
|
if (pipelineId) {
|
|
|
|
(async () => {
|
|
|
|
const { config: javaScriptConfig } = await fetchPipelineNodeDefault(pipelineId, BlockEnum.Code, { code_language: CodeLanguage.javascript }) as any
|
|
|
|
const { config: pythonConfig } = await fetchPipelineNodeDefault(pipelineId, BlockEnum.Code, { code_language: CodeLanguage.python3 }) as any
|
|
|
|
setAllLanguageDefault({
|
|
|
|
[CodeLanguage.javascript]: javaScriptConfig as CodeNodeType,
|
|
|
|
[CodeLanguage.python3]: pythonConfig as CodeNodeType,
|
|
|
|
} as any)
|
|
|
|
})()
|
|
|
|
}
|
|
|
|
}, [pipelineId])
|
|
|
|
|
2025-06-13 14:42:27 +08:00
|
|
|
const defaultConfig = useStore(s => s.nodesDefaultConfigs)?.[payload.type]
|
2024-04-08 18:51:46 +08:00
|
|
|
const { inputs, setInputs } = useNodeCrud<CodeNodeType>(id, payload)
|
|
|
|
const { handleVarListChange, handleAddVariable } = useVarList<CodeNodeType>({
|
|
|
|
inputs,
|
|
|
|
setInputs,
|
|
|
|
})
|
|
|
|
|
|
|
|
const [outputKeyOrders, setOutputKeyOrders] = useState<string[]>([])
|
|
|
|
const syncOutputKeyOrders = useCallback((outputs: OutputVar) => {
|
|
|
|
setOutputKeyOrders(Object.keys(outputs))
|
|
|
|
}, [])
|
|
|
|
useEffect(() => {
|
|
|
|
if (inputs.code) {
|
|
|
|
if (inputs.outputs && Object.keys(inputs.outputs).length > 0)
|
|
|
|
syncOutputKeyOrders(inputs.outputs)
|
|
|
|
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
const isReady = defaultConfig && Object.keys(defaultConfig).length > 0
|
|
|
|
if (isReady) {
|
|
|
|
setInputs({
|
|
|
|
...inputs,
|
|
|
|
...defaultConfig,
|
|
|
|
})
|
|
|
|
syncOutputKeyOrders(defaultConfig.outputs)
|
|
|
|
}
|
2025-06-24 09:10:30 +08:00
|
|
|
// eslint-disable-next-line react-hooks/exhaustive-deps
|
2024-04-08 18:51:46 +08:00
|
|
|
}, [defaultConfig])
|
|
|
|
|
|
|
|
const handleCodeChange = useCallback((code: string) => {
|
|
|
|
const newInputs = produce(inputs, (draft) => {
|
|
|
|
draft.code = code
|
|
|
|
})
|
|
|
|
setInputs(newInputs)
|
|
|
|
}, [inputs, setInputs])
|
|
|
|
|
|
|
|
const handleCodeLanguageChange = useCallback((codeLanguage: CodeLanguage) => {
|
|
|
|
const currDefaultConfig = allLanguageDefault?.[codeLanguage]
|
|
|
|
|
|
|
|
const newInputs = produce(inputs, (draft) => {
|
|
|
|
draft.code_language = codeLanguage
|
|
|
|
if (!currDefaultConfig)
|
|
|
|
return
|
|
|
|
draft.code = currDefaultConfig.code
|
|
|
|
draft.variables = currDefaultConfig.variables
|
|
|
|
draft.outputs = currDefaultConfig.outputs
|
|
|
|
})
|
|
|
|
setInputs(newInputs)
|
|
|
|
}, [allLanguageDefault, inputs, setInputs])
|
|
|
|
|
2025-07-01 10:57:07 +08:00
|
|
|
const handleSyncFunctionSignature = useCallback(() => {
|
|
|
|
const generateSyncSignatureCode = (code: string) => {
|
|
|
|
let mainDefRe
|
|
|
|
let newMainDef
|
|
|
|
if (inputs.code_language === CodeLanguage.javascript) {
|
|
|
|
mainDefRe = /function\s+main\b\s*\([\s\S]*?\)/g
|
|
|
|
newMainDef = 'function main({{var_list}})'
|
|
|
|
let param_list = inputs.variables?.map(item => item.variable).join(', ') || ''
|
|
|
|
param_list = param_list ? `{${param_list}}` : ''
|
|
|
|
newMainDef = newMainDef.replace('{{var_list}}', param_list)
|
|
|
|
}
|
|
|
|
|
|
|
|
else if (inputs.code_language === CodeLanguage.python3) {
|
|
|
|
mainDefRe = /def\s+main\b\s*\([\s\S]*?\)/g
|
|
|
|
const param_list = []
|
|
|
|
for (const item of inputs.variables) {
|
|
|
|
let param = item.variable
|
|
|
|
let param_type = ''
|
|
|
|
switch (item.value_type) {
|
|
|
|
case VarType.string:
|
|
|
|
param_type = ': str'
|
|
|
|
break
|
|
|
|
case VarType.number:
|
|
|
|
param_type = ': float'
|
|
|
|
break
|
|
|
|
case VarType.object:
|
|
|
|
param_type = ': dict'
|
|
|
|
break
|
|
|
|
case VarType.array:
|
|
|
|
param_type = ': list'
|
|
|
|
break
|
|
|
|
case VarType.arrayNumber:
|
|
|
|
param_type = ': list[float]'
|
|
|
|
break
|
|
|
|
case VarType.arrayString:
|
|
|
|
param_type = ': list[str]'
|
|
|
|
break
|
|
|
|
case VarType.arrayObject:
|
|
|
|
param_type = ': list[dict]'
|
|
|
|
break
|
|
|
|
}
|
|
|
|
param += param_type
|
|
|
|
param_list.push(`${param}`)
|
|
|
|
}
|
|
|
|
|
|
|
|
newMainDef = `def main(${param_list.join(', ')})`
|
|
|
|
}
|
|
|
|
else { return code }
|
|
|
|
|
|
|
|
const newCode = code.replace(mainDefRe, newMainDef)
|
|
|
|
return newCode
|
|
|
|
}
|
|
|
|
|
|
|
|
const newInputs = produce(inputs, (draft) => {
|
|
|
|
draft.code = generateSyncSignatureCode(draft.code)
|
|
|
|
})
|
|
|
|
setInputs(newInputs)
|
|
|
|
}, [inputs, setInputs])
|
|
|
|
|
2024-04-08 18:51:46 +08:00
|
|
|
const {
|
|
|
|
handleVarsChange,
|
|
|
|
handleAddVariable: handleAddOutputVariable,
|
|
|
|
handleRemoveVariable,
|
|
|
|
isShowRemoveVarConfirm,
|
|
|
|
hideRemoveVarConfirm,
|
|
|
|
onRemoveVarConfirm,
|
|
|
|
} = useOutputVarList<CodeNodeType>({
|
|
|
|
id,
|
|
|
|
inputs,
|
|
|
|
setInputs,
|
|
|
|
outputKeyOrders,
|
|
|
|
onOutputKeyOrdersChange: setOutputKeyOrders,
|
|
|
|
})
|
|
|
|
|
|
|
|
const filterVar = useCallback((varPayload: Var) => {
|
2025-04-03 16:23:19 +08:00
|
|
|
return [VarType.string, VarType.number, VarType.secret, VarType.object, VarType.array, VarType.arrayNumber, VarType.arrayString, VarType.arrayObject, VarType.file, VarType.arrayFile].includes(varPayload.type)
|
2024-04-08 18:51:46 +08:00
|
|
|
}, [])
|
|
|
|
|
2024-11-01 12:39:32 +09:00
|
|
|
const handleCodeAndVarsChange = useCallback((code: string, inputVariables: Variable[], outputVariables: OutputVar) => {
|
|
|
|
const newInputs = produce(inputs, (draft) => {
|
|
|
|
draft.code = code
|
|
|
|
draft.variables = inputVariables
|
|
|
|
draft.outputs = outputVariables
|
|
|
|
})
|
|
|
|
setInputs(newInputs)
|
|
|
|
syncOutputKeyOrders(outputVariables)
|
|
|
|
}, [inputs, setInputs, syncOutputKeyOrders])
|
2024-04-08 18:51:46 +08:00
|
|
|
return {
|
|
|
|
readOnly,
|
|
|
|
inputs,
|
|
|
|
outputKeyOrders,
|
|
|
|
handleVarListChange,
|
|
|
|
handleAddVariable,
|
|
|
|
handleRemoveVariable,
|
2025-07-01 10:57:07 +08:00
|
|
|
handleSyncFunctionSignature,
|
2024-04-08 18:51:46 +08:00
|
|
|
handleCodeChange,
|
|
|
|
handleCodeLanguageChange,
|
|
|
|
handleVarsChange,
|
|
|
|
filterVar,
|
|
|
|
handleAddOutputVariable,
|
|
|
|
isShowRemoveVarConfirm,
|
|
|
|
hideRemoveVarConfirm,
|
|
|
|
onRemoveVarConfirm,
|
2024-11-01 12:39:32 +09:00
|
|
|
handleCodeAndVarsChange,
|
2024-04-08 18:51:46 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
export default useConfig
|