dify/web/app/components/workflow/block-selector/hooks.ts

import {
  useCallback,
  useEffect,
  useMemo,
  useState,
} from 'react'
import { useTranslation } from 'react-i18next'
import { BLOCKS, START_BLOCKS } from './constants'
import {
  TabsEnum,
  ToolTypeEnum,
} from './types'

export const useBlocks = () => {
  const { t } = useTranslation()

  return BLOCKS.map((block) => {
    return {
      ...block,
      title: t(`workflow.blocks.${block.type}`),
    }
  })
}

export const useStartBlocks = () => {
  const { t } = useTranslation()

  return START_BLOCKS.map((block) => {
    return {
      ...block,
      title: t(`workflow.blocks.${block.type}`),
    }
  })
}

export const useTabs = ({
  noBlocks,
  noSources,
  noTools,
  noStart = true,
  defaultActiveTab,
  hasUserInputNode = false,
  forceEnableStartTab = false, // When true, Start tab remains enabled even if trigger/user input nodes already exist.
}: {
  noBlocks?: boolean
  noSources?: boolean
  noTools?: boolean
  noStart?: boolean
  defaultActiveTab?: TabsEnum
  hasUserInputNode?: boolean
  forceEnableStartTab?: boolean
}) => {
  const { t } = useTranslation()
  const shouldShowStartTab = !noStart
  const shouldDisableStartTab = !forceEnableStartTab && hasUserInputNode
  const tabs = useMemo(() => {
    const tabConfigs = [{
      key: TabsEnum.Blocks,
      name: t('workflow.tabs.blocks'),
      show: !noBlocks,
    }, {
      key: TabsEnum.Sources,
      name: t('workflow.tabs.sources'),
      show: !noSources,
    }, {
      key: TabsEnum.Tools,
      name: t('workflow.tabs.tools'),
      show: !noTools,
    },
    {
      key: TabsEnum.Start,
      name: t('workflow.tabs.start'),
      show: shouldShowStartTab,
      disabled: shouldDisableStartTab,
    }]

    return tabConfigs.filter(tab => tab.show)
  }, [t, noBlocks, noSources, noTools, shouldShowStartTab, shouldDisableStartTab])

  const getValidTabKey = useCallback((targetKey?: TabsEnum) => {
    if (!targetKey)
      return undefined
    const tab = tabs.find(tabItem => tabItem.key === targetKey)
    if (!tab || tab.disabled)
      return undefined
    return tab.key
  }, [tabs])

  const initialTab = useMemo(() => {
    const fallbackTab = tabs.find(tab => !tab.disabled)?.key ?? TabsEnum.Blocks
    const preferredDefault = getValidTabKey(defaultActiveTab)
    if (preferredDefault)
      return preferredDefault

    const preferredOrder: TabsEnum[] = []
    if (!noBlocks)
      preferredOrder.push(TabsEnum.Blocks)
    if (!noTools)
      preferredOrder.push(TabsEnum.Tools)
    if (!noSources)
      preferredOrder.push(TabsEnum.Sources)
    if (!noStart)
      preferredOrder.push(TabsEnum.Start)

    for (const tabKey of preferredOrder) {
      const validKey = getValidTabKey(tabKey)
      if (validKey)
        return validKey
    }

    return fallbackTab
  }, [defaultActiveTab, noBlocks, noSources, noTools, noStart, tabs, getValidTabKey])
  const [activeTab, setActiveTab] = useState(initialTab)

  useEffect(() => {
    const currentTab = tabs.find(tab => tab.key === activeTab)
    if (!currentTab || currentTab.disabled)
      setActiveTab(initialTab)
  }, [tabs, activeTab, initialTab])

  return {
    tabs,
    activeTab,
    setActiveTab,
  }
}

export const useToolTabs = (isHideMCPTools?: boolean) => {
  const { t } = useTranslation()
  const tabs = [
    {
      key: ToolTypeEnum.All,
      name: t('workflow.tabs.allTool'),
    },
    {
      key: ToolTypeEnum.BuiltIn,
      name: t('workflow.tabs.plugin'),
    },
    {
      key: ToolTypeEnum.Custom,
      name: t('workflow.tabs.customTool'),
    },
    {
      key: ToolTypeEnum.Workflow,
      name: t('workflow.tabs.workflowTool'),
    },
  ]
  if (!isHideMCPTools) {
    tabs.push({
      key: ToolTypeEnum.MCP,
      name: 'MCP',
    })
  }

  return tabs
}