import { useCallback } from 'react' import produce from 'immer' import { useBoolean } from 'ahooks' import useVarList from '../_base/hooks/use-var-list' import { VarType } from '../../types' import type { Var } from '../../types' import type { Authorization, Body, HttpNodeType, Method } from './types' import useKeyValueList from './hooks/use-key-value-list' import useNodeCrud from '@/app/components/workflow/nodes/_base/hooks/use-node-crud' import useOneStepRun from '@/app/components/workflow/nodes/_base/hooks/use-one-step-run' const useConfig = (id: string, payload: HttpNodeType) => { const { inputs, setInputs } = useNodeCrud(id, payload) // console.log(inputs) const { handleVarListChange, handleAddVariable } = useVarList({ inputs, setInputs, }) const handleMethodChange = useCallback((method: Method) => { const newInputs = produce(inputs, (draft: HttpNodeType) => { draft.method = method }) setInputs(newInputs) }, [inputs, setInputs]) const handleUrlChange = useCallback((url: string) => { const newInputs = produce(inputs, (draft: HttpNodeType) => { draft.url = url }) setInputs(newInputs) }, [inputs, setInputs]) const { list: headers, setList: setHeaders, addItem: addHeader, isKeyValueEdit: isHeaderKeyValueEdit, toggleIsKeyValueEdit: toggleIsHeaderKeyValueEdit, } = useKeyValueList(inputs.headers) const { list: params, setList: setParams, addItem: addParam, isKeyValueEdit: isParamKeyValueEdit, toggleIsKeyValueEdit: toggleIsParamKeyValueEdit, } = useKeyValueList(inputs.params) const setBody = useCallback((data: Body) => { const newInputs = produce(inputs, (draft: HttpNodeType) => { draft.body = data }) setInputs(newInputs) }, [inputs, setInputs]) // authorization const [isShowAuthorization, { setTrue: showAuthorization, setFalse: hideAuthorization, }] = useBoolean(false) const setAuthorization = useCallback((authorization: Authorization) => { const newInputs = produce(inputs, (draft: HttpNodeType) => { draft.authorization = authorization }) setInputs(newInputs) }, [inputs, setInputs]) const filterVar = useCallback((varPayload: Var) => { return [VarType.string, VarType.number].includes(varPayload.type) }, []) // single run const { isShowSingleRun, hideSingleRun, toVarInputs, runningStatus, handleRun, handleStop, runInputData, setRunInputData, runResult, } = useOneStepRun({ id, data: inputs, defaultRunInputData: {}, }) const varInputs = toVarInputs(inputs.variables) const inputVarValues = (() => { const vars: Record = {} Object.keys(runInputData) .forEach((key) => { vars[key] = runInputData[key] }) return vars })() const setInputVarValues = useCallback((newPayload: Record) => { setRunInputData(newPayload) }, [setRunInputData]) return { inputs, handleVarListChange, handleAddVariable, filterVar, handleMethodChange, handleUrlChange, // headers headers, setHeaders, addHeader, isHeaderKeyValueEdit, toggleIsHeaderKeyValueEdit, // params params, setParams, addParam, isParamKeyValueEdit, toggleIsParamKeyValueEdit, // body setBody, // authorization isShowAuthorization, showAuthorization, hideAuthorization, setAuthorization, // single run isShowSingleRun, hideSingleRun, runningStatus, handleRun, handleStop, varInputs, inputVarValues, setInputVarValues, runResult, } } export default useConfig