ComfyUI_frontend/src/extensions/core/cloudBadges.ts

import { computed } from 'vue'

import { remoteConfig } from '@/platform/remoteConfig/remoteConfig'
import { useExtensionService } from '@/services/extensionService'
import type { TopbarBadge } from '@/types/comfy'

const badges = computed<TopbarBadge[]>(() => {
  const result: TopbarBadge[] = []

  // Add server health alert first (if present)
  const alert = remoteConfig.value.server_health_alert
  if (alert) {
    result.push({
      text: alert.message,
      label: alert.badge,
      variant: alert.severity ?? 'error',
      tooltip: alert.tooltip
    })
  }

  // Always add cloud badge last (furthest right)
  result.push({
    icon: 'icon-[lucide--cloud]',
    text: 'Comfy Cloud'
  })

  return result
})

useExtensionService().registerExtension({
  name: 'Comfy.Cloud.Badges',
  get topbarBadges() {
    return badges.value
  }
})