dify/web/app/components/workflow/nodes/loop/use-single-run-form-params.helpers.ts

import type { InputVar, Node, ValueSelector, Variable } from '../../types'
import type { CaseItem, Condition, LoopVariable } from './types'
import { ValueType } from '@/app/components/workflow/types'
import { VALUE_SELECTOR_DELIMITER as DELIMITER } from '@/config'
import {
  getNodeInfoById,
  getNodeUsedVarPassToServerKey,
  getNodeUsedVars,
  isSystemVar,
} from '../_base/components/variable/utils'

export function getVarSelectorsFromCase(caseItem: CaseItem): ValueSelector[] {
  const vars: ValueSelector[] = []
  caseItem.conditions?.forEach((condition) => {
    vars.push(...getVarSelectorsFromCondition(condition))
  })
  return vars
}

export function getVarSelectorsFromCondition(condition: Condition): ValueSelector[] {
  const vars: ValueSelector[] = []
  if (condition.variable_selector) vars.push(condition.variable_selector)

  if (condition.sub_variable_condition?.conditions?.length)
    vars.push(...getVarSelectorsFromCase(condition.sub_variable_condition))

  return vars
}

export const createInputVarValues = (runInputData: Record<string, unknown>) => {
  const vars: Record<string, unknown> = {}
  Object.keys(runInputData).forEach((key) => {
    vars[key] = runInputData[key]
  })
  return vars
}

export const dedupeInputVars = (inputVars: InputVar[]) => {
  const seen: Record<string, boolean> = {}
  const uniqueInputVars: InputVar[] = []

  inputVars.forEach((input) => {
    if (!input || seen[input.variable]) return

    seen[input.variable] = true
    uniqueInputVars.push(input)
  })

  return uniqueInputVars
}

export const buildUsedOutVars = ({
  loopChildrenNodes,
  currentNodeId,
  canChooseVarNodes,
  isNodeInLoop,
  toVarInputs,
}: {
  loopChildrenNodes: Node[]
  currentNodeId: string
  canChooseVarNodes: Node[]
  isNodeInLoop: (nodeId: string) => boolean
  toVarInputs: (variables: Variable[]) => InputVar[]
}) => {
  const vars: ValueSelector[] = []
  const seenVarSelectors: Record<string, boolean> = {}
  const allVarObject: Record<string, { inSingleRunPassedKey: string }> = {}

  loopChildrenNodes.forEach((node) => {
    const nodeVars = getNodeUsedVars(node).filter((item) => item && item.length > 0)
    nodeVars.forEach((varSelector) => {
      if (varSelector[0] === currentNodeId) return
      if (isNodeInLoop(varSelector[0]!)) return

      const varSelectorStr = varSelector.join('.')
      if (!seenVarSelectors[varSelectorStr]) {
        seenVarSelectors[varSelectorStr] = true
        vars.push(varSelector)
      }

      let passToServerKeys = getNodeUsedVarPassToServerKey(node, varSelector)
      if (typeof passToServerKeys === 'string') passToServerKeys = [passToServerKeys]

      passToServerKeys.forEach((key: string, index: number) => {
        allVarObject[[varSelectorStr, node.id, index].join(DELIMITER)] = {
          inSingleRunPassedKey: key,
        }
      })
    })
  })

  const usedOutVars = toVarInputs(
    vars.map((valueSelector) => {
      const varInfo = getNodeInfoById(canChooseVarNodes, valueSelector[0]!)
      return {
        label: {
          nodeType: varInfo?.data.type,
          nodeName: varInfo?.data.title || canChooseVarNodes[0]?.data.title!,
          variable: isSystemVar(valueSelector)
            ? valueSelector.join('.')
            : valueSelector[valueSelector.length - 1]!,
        },
        variable: valueSelector.join('.'),
        value_selector: valueSelector,
      }
    }),
  )

  return { usedOutVars, allVarObject }
}

export const getDependentVarsFromLoopPayload = ({
  nodeId,
  usedOutVars,
  breakConditions,
  loopVariables,
}: {
  nodeId: string
  usedOutVars: InputVar[]
  breakConditions?: Condition[]
  loopVariables?: LoopVariable[]
}) => {
  const vars: ValueSelector[] = usedOutVars.map((item) => item.variable.split('.'))

  breakConditions?.forEach((condition) => {
    vars.push(...getVarSelectorsFromCondition(condition))
  })

  loopVariables?.forEach((loopVariable) => {
    if (loopVariable.value_type === ValueType.variable) vars.push(loopVariable.value)
  })

  return vars.filter((item) => item[0] !== nodeId)
}