clash-verge-rev/src/hooks/use-icon-cache.ts

import { convertFileSrc } from '@tauri-apps/api/core'
import { useMemo } from 'react'
import useSWR from 'swr'

import { downloadIconCache } from '@/services/cmds'
import { SWR_DEFAULTS } from '@/services/config'

export interface UseIconCacheOptions {
  icon?: string | null
  cacheKey?: string
  enabled?: boolean
}

const getFileNameFromUrl = (url: string) => {
  const lastSlashIndex = url.lastIndexOf('/')
  return lastSlashIndex >= 0 ? url.slice(lastSlashIndex + 1) : url
}

export const useIconCache = ({
  icon,
  cacheKey,
  enabled = true,
}: UseIconCacheOptions) => {
  const iconValue = icon?.trim() ?? ''
  const cacheKeyValue = cacheKey?.trim() ?? ''

  const swrKey = useMemo(() => {
    if (!enabled || !iconValue.startsWith('http') || cacheKeyValue === '') {
      return null
    }

    return ['icon-cache', iconValue, cacheKeyValue] as const
  }, [enabled, iconValue, cacheKeyValue])

  const { data } = useSWR(
    swrKey,
    async () => {
      try {
        const fileName = `${cacheKeyValue}-${getFileNameFromUrl(iconValue)}`
        const iconPath = await downloadIconCache(iconValue, fileName)
        return convertFileSrc(iconPath)
      } catch {
        return ''
      }
    },
    SWR_DEFAULTS,
  )

  if (!swrKey) {
    return ''
  }

  return data ?? ''
}