dify/web/app/components/goto-anything/command-selector.tsx

import type { FC } from 'react'
import { useEffect, useMemo } from 'react'
import { Command } from 'cmdk'
import { useTranslation } from 'react-i18next'
import type { ActionItem } from './actions/types'
import { slashCommandRegistry } from './actions/commands/registry'

type Props = {
  actions: Record<string, ActionItem>
  onCommandSelect: (commandKey: string) => void
  searchFilter?: string
  commandValue?: string
  onCommandValueChange?: (value: string) => void
  originalQuery?: string
}

const CommandSelector: FC<Props> = ({ actions, onCommandSelect, searchFilter, commandValue, onCommandValueChange, originalQuery }) => {
  const { t } = useTranslation()

  // Check if we're in slash command mode
  const isSlashMode = originalQuery?.trim().startsWith('/') || false

  // Get slash commands from registry
  const slashCommands = useMemo(() => {
    if (!isSlashMode) return []

    const allCommands = slashCommandRegistry.getAllCommands()
    const filter = searchFilter?.toLowerCase() || '' // searchFilter already has '/' removed

    return allCommands.filter((cmd) => {
      if (!filter) return true
      return cmd.name.toLowerCase().includes(filter)
    }).map(cmd => ({
      key: `/${cmd.name}`,
      shortcut: `/${cmd.name}`,
      title: cmd.name,
      description: cmd.description,
    }))
  }, [isSlashMode, searchFilter])

  const filteredActions = useMemo(() => {
    if (isSlashMode) return []

    return Object.values(actions).filter((action) => {
      // Exclude slash action when in @ mode
      if (action.key === '/') return false
      if (!searchFilter)
        return true
      const filterLower = searchFilter.toLowerCase()
      return action.shortcut.toLowerCase().includes(filterLower)
    })
  }, [actions, searchFilter, isSlashMode])

  const allItems = isSlashMode ? slashCommands : filteredActions

  useEffect(() => {
    if (allItems.length > 0 && onCommandValueChange) {
      const currentValueExists = allItems.some(item => item.shortcut === commandValue)
      if (!currentValueExists)
        onCommandValueChange(allItems[0].shortcut)
    }
  }, [searchFilter, allItems.length])

  if (allItems.length === 0) {
    return (
      <div className="p-4">
        <div className="flex items-center justify-center py-8 text-center text-text-tertiary">
          <div>
            <div className="text-sm font-medium text-text-tertiary">
              {t('app.gotoAnything.noMatchingCommands')}
            </div>
            <div className="mt-1 text-xs text-text-quaternary">
              {t('app.gotoAnything.tryDifferentSearch')}
            </div>
          </div>
        </div>
      </div>
    )
  }

  return (
    <div className="p-4">
      <div className="mb-3 text-left text-sm font-medium text-text-secondary">
        {isSlashMode ? t('app.gotoAnything.groups.commands') : t('app.gotoAnything.selectSearchType')}
      </div>
      <Command.Group className="space-y-1">
        {allItems.map(item => (
          <Command.Item
            key={item.key}
            value={item.shortcut}
            className="flex cursor-pointer items-center rounded-md
                     p-2.5
                     transition-all
                     duration-150 hover:bg-state-base-hover aria-[selected=true]:bg-state-base-hover-alt"
            onSelect={() => onCommandSelect(item.shortcut)}
          >
            <span className="min-w-[4.5rem] text-left font-mono text-xs text-text-tertiary">
              {item.shortcut}
            </span>
            <span className="ml-3 text-sm text-text-secondary">
              {isSlashMode ? (
                (() => {
                  const slashKeyMap: Record<string, string> = {
                    '/theme': 'app.gotoAnything.actions.themeCategoryDesc',
                    '/language': 'app.gotoAnything.actions.languageChangeDesc',
                    '/account': 'app.gotoAnything.actions.accountDesc',
                    '/feedback': 'app.gotoAnything.actions.feedbackDesc',
                    '/doc': 'app.gotoAnything.actions.docDesc',
                    '/community': 'app.gotoAnything.actions.communityDesc',
                  }
                  return t(slashKeyMap[item.key] || item.description)
                })()
              ) : (
                (() => {
                  const keyMap: Record<string, string> = {
                    '@app': 'app.gotoAnything.actions.searchApplicationsDesc',
                    '@plugin': 'app.gotoAnything.actions.searchPluginsDesc',
                    '@knowledge': 'app.gotoAnything.actions.searchKnowledgeBasesDesc',
                    '@node': 'app.gotoAnything.actions.searchWorkflowNodesDesc',
                  }
                  return t(keyMap[item.key])
                })()
              )}
            </span>
          </Command.Item>
        ))}
      </Command.Group>
    </div>
  )
}

export default CommandSelector