@ensdomains/ensjs 2.0.0 → 3.0.0-alpha.1

package/src/functions/getProfile.ts

@@ -0,0 +1,479 @@
+import { formatsByName } from '@ensdomains/address-encoder'
+import { ethers } from 'ethers'
+import { ENSArgs } from '..'
+import { decodeContenthash, DecodedContentHash } from '../utils/contentHash'
+import { hexEncodeName } from '../utils/hexEncodedName'
+import { parseInputType } from '../utils/validation'
+
+type InternalProfileOptions = {
+  contentHash?: boolean | string | DecodedContentHash
+  texts?: string[]
+  coinTypes?: string[]
+}
+
+type ProfileResponse = {
+  contentHash?: string | DecodedContentHash
+  texts?: string[]
+  coinTypes?: string[]
+}
+
+type DataItem = {
+  key: string | number
+  type: 'addr' | 'text' | 'contentHash'
+  coin?: string
+  value: string
+}
+
+type ResolvedProfile = {
+  isMigrated: boolean | null
+  createdAt: string | null
+  address?: string
+  name?: string | null
+  match?: boolean
+  message?: string
+  records?: {
+    contentHash?: DecodedContentHash | string | null
+    texts?: DataItem[]
+    coinTypes?: DataItem[]
+  }
+  resolverAddress?: string
+  reverseResolverAddress?: string
+}
+
+const makeMulticallData = async (
+  {
+    _getAddr,
+    _getContentHash,
+    _getText,
+    resolverMulticallWrapper,
+  }: ENSArgs<
+    '_getText' | '_getAddr' | '_getContentHash' | 'resolverMulticallWrapper'
+  >,
+  name: string,
+  options: InternalProfileOptions,
+) => {
+  let calls: any[] = []
+  options.texts &&
+    (calls = [
+      ...calls,
+      ...(await Promise.all(
+        options.texts.map(async (x) => ({
+          key: x,
+          data: await _getText.raw(name, x),
+          type: 'text',
+        })),
+      )),
+    ])
+  options.coinTypes &&
+    (calls = [
+      ...calls,
+      ...(await Promise.all(
+        options.coinTypes.map(async (x) => ({
+          key: x,
+          data: await _getAddr.raw(name, x, true),
+          type: 'addr',
+        })),
+      )),
+    ])
+  if (typeof options.contentHash === 'boolean' && options.contentHash) {
+    calls.push({
+      key: 'contentHash',
+      data: await _getContentHash.raw(name),
+      type: 'contenthash',
+    })
+  }
+
+  if (!calls.find((x) => x.key === '60')) {
+    calls.push({
+      key: '60',
+      data: await _getAddr.raw(name, '60', true),
+      type: 'addr',
+    })
+  }
+
+  const prRawData = await resolverMulticallWrapper.raw(calls.map((x) => x.data))
+
+  return { data: prRawData.data, calls }
+}
+
+const makeHashIndexes = (data: string, name: string) =>
+  [...data.matchAll(ethers.utils.namehash(name).substring(2) as any)].map(
+    (x: any) => x.index / 2 - 1,
+  )
+
+const getDataForName = async (
+  {
+    contracts,
+    _getAddr,
+    _getContentHash,
+    _getText,
+    resolverMulticallWrapper,
+  }: ENSArgs<
+    | 'contracts'
+    | '_getText'
+    | '_getAddr'
+    | '_getContentHash'
+    | 'resolverMulticallWrapper'
+  >,
+  name: string,
+  options: InternalProfileOptions,
+) => {
+  const universalResolver = await contracts?.getUniversalResolver()
+
+  const { data, calls } = await makeMulticallData(
+    { _getAddr, _getContentHash, _getText, resolverMulticallWrapper },
+    name,
+    options,
+  )
+
+  let resolver: any
+  try {
+    resolver = await universalResolver?.resolve(hexEncodeName(name), data)
+  } catch {
+    return
+  }
+
+  const [recordData] = await resolverMulticallWrapper.decode(resolver['0'])
+
+  const matchAddress = recordData[calls.findIndex((x) => x.key === '60')]
+
+  return {
+    address: matchAddress && (await _getAddr.decode(matchAddress)),
+    records: await formatRecords(
+      { _getAddr, _getContentHash, _getText },
+      recordData,
+      calls,
+      options,
+    ),
+    resolverAddress: resolver['1'],
+  }
+}
+
+const formatRecords = async (
+  {
+    _getText,
+    _getAddr,
+    _getContentHash,
+  }: ENSArgs<'_getText' | '_getAddr' | '_getContentHash'>,
+  data: any[],
+  calls: any[],
+  options: InternalProfileOptions,
+) => {
+  let returnedRecords: DataItem[] = (
+    await Promise.all(
+      data.map(async (item: string, i: number) => {
+        let decodedFromAbi: any
+        let itemRet: Record<string, any> = {
+          key: calls[i].key,
+          type: calls[i].type,
+        }
+        if (itemRet.type === 'addr' || itemRet.type === 'contenthash') {
+          decodedFromAbi = ethers.utils.defaultAbiCoder.decode(
+            ['bytes'],
+            item,
+          )[0]
+          if (ethers.utils.hexStripZeros(decodedFromAbi) === '0x') {
+            return
+          }
+        }
+        switch (calls[i].type) {
+          case 'text':
+            itemRet = {
+              ...itemRet,
+              value: await _getText.decode(item),
+            }
+            if (itemRet.value === '' || itemRet.value === undefined) return
+            break
+          case 'addr':
+            try {
+              const addr = await _getAddr.decode(item, '', calls[i].key)
+              if (addr) {
+                itemRet = {
+                  ...itemRet,
+                  ...addr,
+                }
+                break
+              } else {
+                return
+              }
+            } catch {
+              return
+            }
+          case 'contenthash':
+            try {
+              itemRet = {
+                ...itemRet,
+                value: await _getContentHash.decode(item),
+              }
+            } catch {
+              return
+            }
+        }
+        return itemRet
+      }),
+    )
+  )
+    .filter((x): x is DataItem => {
+      return typeof x === 'object'
+    })
+    .filter((x) => x)
+
+  let returnedResponse: {
+    contentHash?: string | null | DecodedContentHash
+    coinTypes?: DataItem[]
+    texts?: DataItem[]
+  } = {}
+
+  if (
+    typeof options.contentHash === 'string' ||
+    typeof options.contentHash === 'object'
+  ) {
+    if (
+      typeof options.contentHash === 'string' &&
+      ethers.utils.hexStripZeros(options.contentHash) === '0x'
+    ) {
+      returnedResponse.contentHash = null
+    } else if (
+      ethers.utils.isBytesLike((options.contentHash as any).decoded) &&
+      ethers.utils.hexStripZeros((options.contentHash as any).decoded) === '0x'
+    ) {
+      returnedResponse.contentHash = null
+    } else {
+      returnedResponse.contentHash = options.contentHash
+    }
+  } else if (options.contentHash) {
+    const foundRecord = returnedRecords.find(
+      (item: any) => item.type === 'contenthash',
+    )
+    returnedResponse.contentHash = foundRecord ? foundRecord.value : null
+  }
+  if (options.texts) {
+    returnedResponse.texts = returnedRecords.filter(
+      (x: any) => x.type === 'text',
+    )
+  }
+  if (options.coinTypes) {
+    returnedResponse.coinTypes = returnedRecords.filter(
+      (x: any) => x.type === 'addr',
+    )
+  }
+  return returnedResponse
+}
+
+const graphFetch = async (
+  { gqlInstance }: ENSArgs<'gqlInstance'>,
+  name: string,
+  wantedRecords?: ProfileOptions,
+) => {
+  const query = gqlInstance.gql`
+    query getRecords($name: String!) {
+      domains(where: { name: $name }) {
+        isMigrated
+        createdAt
+        resolver {
+          texts
+          coinTypes
+          contentHash
+          addr {
+            id
+          }
+        }
+      }
+    }
+  `
+
+  const client = gqlInstance.client
+
+  const { domains } = await client.request(query, { name })
+
+  if (!domains || domains.length === 0) return
+
+  const [{ resolver: resolverResponse, isMigrated, createdAt }] = domains
+
+  let returnedRecords: ProfileResponse = {}
+
+  if (!wantedRecords) return { isMigrated, createdAt }
+
+  Object.keys(wantedRecords).forEach((key: string) => {
+    const data = wantedRecords[key as keyof ProfileOptions]
+    if (typeof data === 'boolean' && data) {
+      if (key === 'contentHash') {
+        returnedRecords[key] = decodeContenthash(resolverResponse.contentHash)
+      } else {
+        returnedRecords[key as keyof ProfileOptions] = resolverResponse[key]
+      }
+    }
+  })
+
+  return { ...returnedRecords, isMigrated, createdAt }
+}
+
+type ProfileOptions = {
+  contentHash?: boolean
+  texts?: boolean | string[]
+  coinTypes?: boolean | string[]
+}
+
+const getProfileFromName = async (
+  {
+    contracts,
+    gqlInstance,
+    _getAddr,
+    _getContentHash,
+    _getText,
+    resolverMulticallWrapper,
+  }: ENSArgs<
+    | 'contracts'
+    | 'gqlInstance'
+    | '_getText'
+    | '_getAddr'
+    | '_getContentHash'
+    | 'resolverMulticallWrapper'
+  >,
+  name: string,
+  options?: ProfileOptions,
+) => {
+  const usingOptions =
+    !options || options?.texts === true || options?.coinTypes === true
+      ? options || { contentHash: true, texts: true, coinTypes: true }
+      : undefined
+  const graphResult = await graphFetch({ gqlInstance }, name, usingOptions)
+  if (!graphResult) return
+  const {
+    isMigrated,
+    createdAt,
+    ...wantedRecords
+  }: { isMigrated: boolean; createdAt: string } & InternalProfileOptions =
+    graphResult
+  const result = await getDataForName(
+    {
+      contracts,
+      _getAddr,
+      _getContentHash,
+      _getText,
+      resolverMulticallWrapper,
+    },
+    name,
+    usingOptions ? wantedRecords : (options as InternalProfileOptions),
+  )
+  if (!result)
+    return { isMigrated, createdAt, message: "Records fetch didn't complete" }
+  return { ...result, isMigrated, createdAt, message: undefined }
+}
+
+const getProfileFromAddress = async (
+  {
+    contracts,
+    gqlInstance,
+    getName,
+    _getAddr,
+    _getContentHash,
+    _getText,
+    resolverMulticallWrapper,
+  }: ENSArgs<
+    | 'contracts'
+    | 'gqlInstance'
+    | 'getName'
+    | '_getText'
+    | '_getAddr'
+    | '_getContentHash'
+    | 'resolverMulticallWrapper'
+  >,
+  address: string,
+  options?: ProfileOptions,
+) => {
+  let name
+  try {
+    name = await getName(address)
+  } catch (e) {
+    return
+  }
+  if (!name || !name.name || name.name === '') return
+  if (!name.match) return { ...name, isMigrated: null, createdAt: null }
+  const result = await getProfileFromName(
+    {
+      contracts,
+      gqlInstance,
+      _getAddr,
+      _getContentHash,
+      _getText,
+      resolverMulticallWrapper,
+    },
+    name.name,
+    options,
+  )
+  if (!result || result.message) return
+  delete (result as any).address
+  return {
+    ...result,
+    ...name,
+    message: undefined,
+  }
+}
+
+export default async function (
+  {
+    contracts,
+    gqlInstance,
+    getName,
+    _getAddr,
+    _getContentHash,
+    _getText,
+    resolverMulticallWrapper,
+  }: ENSArgs<
+    | 'contracts'
+    | 'gqlInstance'
+    | 'getName'
+    | '_getText'
+    | '_getAddr'
+    | '_getContentHash'
+    | 'resolverMulticallWrapper'
+  >,
+  nameOrAddress: string,
+  options?: ProfileOptions,
+): Promise<ResolvedProfile | undefined> {
+  if (options && options.coinTypes && typeof options.coinTypes !== 'boolean') {
+    options.coinTypes = options.coinTypes.map((coin: string) => {
+      if (!isNaN(parseInt(coin))) {
+        return coin
+      } else {
+        return `${formatsByName[coin.toUpperCase()].coinType}`
+      }
+    })
+  }
+
+  const inputType = parseInputType(nameOrAddress)
+
+  if (inputType.type === 'unknown' || inputType.info === 'unsupported') {
+    throw new Error('Invalid input type')
+  }
+
+  if (inputType.type === 'address') {
+    return getProfileFromAddress(
+      {
+        contracts,
+        gqlInstance,
+        getName,
+        _getAddr,
+        _getContentHash,
+        _getText,
+        resolverMulticallWrapper,
+      },
+      nameOrAddress,
+      options,
+    )
+  }
+
+  return getProfileFromName(
+    {
+      contracts,
+      gqlInstance,
+      _getAddr,
+      _getContentHash,
+      _getText,
+      resolverMulticallWrapper,
+    },
+    nameOrAddress,
+    options,
+  )
+}

package/src/functions/getRecords.ts

@@ -0,0 +1,22 @@
+import { ENSArgs } from '..'
+import { parseInputType } from '../utils/validation'
+
+type ProfileOptions = {
+  contentHash?: boolean
+  texts?: boolean | string[]
+  coinTypes?: boolean | string[]
+}
+
+export default async function (
+  { getProfile }: ENSArgs<'getProfile'>,
+  name: string,
+  options?: ProfileOptions,
+) {
+  const inputType = parseInputType(name)
+
+  if (inputType.type !== 'name' && inputType.type !== 'label') {
+    throw new Error('Input must be an ENS name')
+  }
+
+  return await getProfile(name, options)
+}

package/src/functions/getResolver.ts

@@ -0,0 +1,32 @@
+import { ethers } from 'ethers'
+import { ENSArgs } from '..'
+import { hexEncodeName } from '../utils/hexEncodedName'
+
+const raw = async ({ contracts }: ENSArgs<'contracts'>, name: string) => {
+  const universalResolver = await contracts?.getUniversalResolver()!
+  return {
+    to: universalResolver.address,
+    data: universalResolver.interface.encodeFunctionData('findResolver', [
+      hexEncodeName(name),
+    ]),
+  }
+}
+
+const decode = async ({ contracts }: ENSArgs<'contracts'>, data: string) => {
+  const universalResolver = await contracts?.getUniversalResolver()!
+  const response = universalResolver.interface.decodeFunctionResult(
+    'findResolver',
+    data,
+  )
+
+  if (
+    !response ||
+    !response[0] ||
+    ethers.utils.hexStripZeros(response[0]) === '0x'
+  ) {
+    return
+  }
+  return response[0]
+}
+
+export default { raw, decode }