dify/web/app/components/workflow/nodes/assigner/use-config.helpers.ts

import type { ValueSelector, Var } from '../../types'
import type { AssignerNodeOperation, AssignerNodeType } from './types'
import { produce } from 'immer'
import { VarType } from '../../types'
import { WriteMode } from './types'
import { normalizeOperationItems } from './utils'

export const filterVarByType = (varType: VarType) => {
  return (variable: Var) => {
    if (varType === VarType.any || variable.type === VarType.any) return true

    return variable.type === varType
  }
}

export const normalizeAssignedVarType = (assignedVarType: VarType, writeMode: WriteMode) => {
  if (
    writeMode === WriteMode.overwrite ||
    writeMode === WriteMode.increment ||
    writeMode === WriteMode.decrement ||
    writeMode === WriteMode.multiply ||
    writeMode === WriteMode.divide ||
    writeMode === WriteMode.extend
  ) {
    return assignedVarType
  }

  if (writeMode === WriteMode.append) {
    switch (assignedVarType) {
      case VarType.arrayString:
        return VarType.string
      case VarType.arrayNumber:
        return VarType.number
      case VarType.arrayObject:
        return VarType.object
      default:
        return VarType.string
    }
  }

  return VarType.string
}

export const canAssignVar = (_varPayload: Var, selector: ValueSelector) => {
  return selector.join('.').startsWith('conversation')
}

export const canAssignToVar = (varPayload: Var, assignedVarType: VarType, writeMode: WriteMode) => {
  if (
    writeMode === WriteMode.overwrite ||
    writeMode === WriteMode.extend ||
    writeMode === WriteMode.increment ||
    writeMode === WriteMode.decrement ||
    writeMode === WriteMode.multiply ||
    writeMode === WriteMode.divide
  ) {
    return varPayload.type === assignedVarType
  }

  if (writeMode === WriteMode.append) {
    switch (assignedVarType) {
      case VarType.arrayString:
        return varPayload.type === VarType.string
      case VarType.arrayNumber:
        return varPayload.type === VarType.number
      case VarType.arrayObject:
        return varPayload.type === VarType.object
      default:
        return false
    }
  }

  return true
}

export const ensureAssignerVersion = (newInputs: AssignerNodeType) =>
  produce(newInputs, (draft) => {
    if (draft.version !== '2') draft.version = '2'
  })

export const updateOperationItems = (inputs: AssignerNodeType, items: AssignerNodeOperation[]) =>
  produce(inputs, (draft) => {
    draft.items = normalizeOperationItems(items)
  })