dify/web/app/components/datasets/common/credential-icon.tsx

import cn from '@/utils/classnames'
import React, { useMemo } from 'react'

type CredentialIconProps = {
  avatar_url?: string
  name: string
  size?: number
  className?: string
}

const ICON_BG_COLORS = [
  'bg-components-icon-bg-orange-dark-solid',
  'bg-components-icon-bg-pink-solid',
  'bg-components-icon-bg-indigo-solid',
  'bg-components-icon-bg-teal-solid',
]

export const CredentialIcon: React.FC<CredentialIconProps> = ({
  avatar_url,
  name,
  size = 20,
  className = '',
}) => {
  const firstLetter = useMemo(() => name.charAt(0).toUpperCase(), [name])
  const bgColor = useMemo(() => ICON_BG_COLORS[firstLetter.charCodeAt(0) % ICON_BG_COLORS.length], [firstLetter])

  if (avatar_url && avatar_url !== 'default') {
    return (
      <img
        src={avatar_url}
        alt={`${name} logo`}
        width={size}
        height={size}
        className={cn('shrink-0 rounded-md border border-divider-regular object-contain', className)}
      />
    )
  }

  return (
    <div
      className={cn(
        'flex shrink-0 items-center justify-center rounded-md border border-divider-regular',
        bgColor,
        className,
      )}
      style={{ width: `${size}px`, height: `${size}px` }}
    >
      <span className='bg-gradient-to-b from-components-avatar-shape-fill-stop-0 to-components-avatar-shape-fill-stop-100 bg-clip-text text-[13px] font-semibold leading-[1.2] text-transparent opacity-90'>
        {firstLetter}
      </span>
    </div>
  )
}