dify/web/app/components/workflow/nodes/data-source/default.ts

import type { NodeDefault } from '../../types'
import type { DataSourceNodeType } from './types'
import { genNodeMetaData } from '@/app/components/workflow/utils'
import { BlockEnum } from '@/app/components/workflow/types'
import { CollectionType } from '@/app/components/tools/types'
import { OUTPUT_VARIABLES_MAP } from './constants'

const metaData = genNodeMetaData({
  sort: -1,
  type: BlockEnum.DataSource,
})
const nodeDefault: NodeDefault<DataSourceNodeType> = {
  metaData,
  defaultValue: {
    datasource_parameters: {},
    datasource_configurations: {},
  },
  checkValid() {
    return {
      isValid: true,
      errorMessage: '',
    }
  },
  getOutputVars(payload) {
    const { provider_id, provider_type } = payload
    const isLocalFile = provider_id === 'langgenius/file/file' && provider_type === CollectionType.datasource
    return [
      {
        variable: OUTPUT_VARIABLES_MAP.datasource_type.name,
        type: OUTPUT_VARIABLES_MAP.datasource_type.type,
      },
      ...(
        isLocalFile
          ? [
              {
                variable: OUTPUT_VARIABLES_MAP.file.name,
                type: OUTPUT_VARIABLES_MAP.file.type,
                children: OUTPUT_VARIABLES_MAP.file.subItems.map((item) => {
                  return {
                    variable: item.name,
                    type: item.type,
                  }
                }),
              },
          ]
          : []
      ),
    ]
  },
}

export default nodeDefault