dify/web/app/components/workflow/nodes/question-classifier/components/class-label-utils.ts

'use client'
import type { TFunction } from 'i18next'

const i18nPrefix = 'nodes.questionClassifiers'
const LEGACY_DEFAULT_LABEL_PREFIX = 'CLASS'
const DEFAULT_EQUIVALENT_PREFIXES = ['CLASS', '分类', '分類', 'クラス']

const getCanonicalDefaultClassLabel = (index: number) => `${LEGACY_DEFAULT_LABEL_PREFIX} ${index}`

const getTranslatedDefaultClassLabel = (t: TFunction, index: number) => {
  const translated = t(`${i18nPrefix}.defaultLabel`, { ns: 'workflow', index })
  if (typeof translated !== 'string')
    return undefined

  const resolvedLabel = translated.replace('{{index}}', String(index))
  const rawWorkflowKey = `workflow.${i18nPrefix}.defaultLabel`
  const rawKey = `${i18nPrefix}.defaultLabel`
  if (
    resolvedLabel === rawWorkflowKey
    || resolvedLabel === rawKey
    || resolvedLabel.startsWith(`${rawWorkflowKey}:`)
    || resolvedLabel.startsWith(`${rawKey}:`)
  ) {
    return undefined
  }

  return resolvedLabel
}

const normalizeClassLabel = (label?: string | null) => label?.trim() ?? ''

export const getDefaultClassLabel = (_t: TFunction, index: number) => getCanonicalDefaultClassLabel(index)

export const getDisplayClassLabel = (
  label: string | undefined,
  index: number,
  t: TFunction,
) => normalizeClassLabel(label) || getTranslatedDefaultClassLabel(t, index) || getCanonicalDefaultClassLabel(index)

export const isDefaultClassLabel = (
  label: string | undefined,
  index: number,
  t: TFunction,
) => {
  const normalizedLabel = normalizeClassLabel(label)
  if (!normalizedLabel)
    return true

  return DEFAULT_EQUIVALENT_PREFIXES.some(prefix => normalizedLabel === `${prefix} ${index}`)
    || normalizedLabel === getTranslatedDefaultClassLabel(t, index)
}

export const getCanonicalClassLabel = (
  label: string | undefined,
  index: number,
  t: TFunction,
) => {
  const normalizedLabel = normalizeClassLabel(label)
  if (!normalizedLabel)
    return getCanonicalDefaultClassLabel(index)

  if (isDefaultClassLabel(normalizedLabel, index, t))
    return getCanonicalDefaultClassLabel(index)

  return normalizedLabel
}