kodadot/nft-gallery

import type { ComputedRef } from 'vue'
import { type Prefix } from '@kodadot1/static'
import { isEthereumAddress } from '@polkadot/util-crypto'
import { accountToPublicKey, getss58AddressByPrefix } from '@/utils/account'
import shortAddress from '@/utils/shortAddress'

export type IdentityFields = Record<string, string>

export const useIdentityQuery = (urlPrefix: Ref<Prefix>) => {
  const isDotAddress = computed(() => ['dot', 'ahp'].includes(urlPrefix.value))

  const getIdentityId = (address: string) => {
    if (isEthereumAddress(address)) {
      return address
    }
    else if (isDotAddress.value) {
      return accountToPublicKey(address)
    }
    else {
      return getss58AddressByPrefix(address, 'ahk')
    }
  }

  return { isDotAddress, getIdentityId }
}

export default function useIdentity({
  address = computed(() => ''),
  customNameOption = '',
}: {
  address: ComputedRef
  customNameOption?: string
}) {
  const { urlPrefix } = usePrefix()
  const { getIdentityId } = useIdentityQuery(urlPrefix)

  const id = computed(() => address.value && getIdentityId(address.value))

  const { profile, isPending: loading } = useFetchProfile(id)

  const shortenedAddress = computed(() => shortAddress(address.value))

  const getSocialHandle = platform =>
    computed(
      () =>
        profile?.value?.socials?.find(s => s.platform === platform)?.handle,
    )

  const twitter = getSocialHandle('Twitter')
  const web = getSocialHandle('Website')
  const farcaster = getSocialHandle('Farcaster')

  const display = computed(() => profile?.value?.name)
  const name = computed(() =>
    displayName({
      customNameOption,
      profileName: display.value,
      shortenedAddress,
    }),
  )

  const identity = computed(() =>
    profile.value
      ? {
          address: address.value,
          name: name.value,
          display: display.value,
          twitter: twitter.value,
          web: web.value,
          farcaster: farcaster.value,
        }
      : null,
  )

  const hasIdentity = computed(() => Boolean(profile.value))

  return {
    isFetchingIdentity: loading,
    shortenedAddress,
    twitter,
    display,
    name,
    hasIdentity,
    identity,
  }
}

const displayName = ({
  customNameOption,
  profileName,
  shortenedAddress,
}): string => {
  if (customNameOption) {
    return customNameOption
  }

  return profileName || shortenedAddress.value
}