Briven/apps/studio/components/interfaces/Storage/StorageExplorer/useCopyUrl.tsx

import { useCallback } from 'react'
import { toast } from 'sonner'
import { copyToClipboard } from 'ui'

import { URL_EXPIRY_DURATION } from '../Storage.constants'
import { fetchFileUrl } from './useFetchFileUrlQuery'
import { useProjectApiUrl } from '@/data/config/project-endpoint-query'
import { useStorageExplorerStateSnapshot } from '@/state/storage-explorer'

export const useCopyUrl = () => {
  const { projectRef, selectedBucket } = useStorageExplorerStateSnapshot()

  const { hostEndpoint, customEndpoint } = useProjectApiUrl({ projectRef })
  const isCustomDomainActive = !!customEndpoint

  const getFileUrl = useCallback(
    (filePath: string, expiresIn?: URL_EXPIRY_DURATION) => {
      return fetchFileUrl(filePath, projectRef, selectedBucket.id, selectedBucket.public, expiresIn)
    },
    [projectRef, selectedBucket]
  )

  const onCopyUrl = useCallback(
    (filePath: string, expiresIn?: URL_EXPIRY_DURATION) => {
      const formattedUrl = getFileUrl(filePath, expiresIn).then((url) => {
        return isCustomDomainActive && hostEndpoint
          ? url.replace(hostEndpoint, customEndpoint)
          : url
      })

      return copyToClipboard(formattedUrl, () => {
        toast.success(`Copied URL for ${filePath} to clipboard.`)
      })
    },
    [customEndpoint, getFileUrl, hostEndpoint, isCustomDomainActive]
  )

  return { onCopyUrl }
}