dify/web/features/agent-v2/roster/components/agent-form.ts

import type { AppIconSelection } from '@/app/components/base/app-icon-picker'

export type AgentFormValues = {
  description?: string
  name?: string
  role?: string
}

export type AgentIconSelection =
  | AppIconSelection
  | {
      type: 'link'
      icon: string
      url: string
    }

export const defaultAgentIcon = {
  type: 'emoji',
  icon: '🧸',
  background: '#F5F3FF',
} satisfies AppIconSelection

type AgentIconSource = {
  icon?: string | null
  icon_background?: string | null
  icon_type?: string | null
}

export const createAgentIconSelection = (agent: AgentIconSource): AgentIconSelection => {
  if (agent.icon_type === 'image' && agent.icon) {
    return {
      type: 'image',
      fileId: agent.icon,
      url: agent.icon,
    }
  }

  if (agent.icon_type === 'link' && agent.icon) {
    return {
      type: 'link',
      icon: agent.icon,
      url: agent.icon,
    }
  }

  return {
    type: 'emoji',
    icon: agent.icon || defaultAgentIcon.icon,
    background: agent.icon_background || defaultAgentIcon.background,
  }
}

export const getAgentIconKey = (icon: AgentIconSelection) => {
  if (icon.type === 'emoji') return `${icon.type}:${icon.icon}:${icon.background}`

  if (icon.type === 'image') return `${icon.type}:${icon.fileId}`

  return `${icon.type}:${icon.icon}`
}