embeddable-nfts/src/utils.ts

import { CollectionTraits, Trait, TraitType } from './types'
// @ts-ignore ts error TS7016
import Web3 from 'web3'
import { Network } from 'opensea-js/lib'

export const formatTraitType = (traitType: string) =>
  traitType.replace(/_/g, ' ')

const isBoost = (trait: Trait) =>
  trait.display_type && trait.display_type.includes('boost')

const isRanking = (trait: Trait, collectionTraits: CollectionTraits) =>
  trait.display_type === null &&
  trait.trait_type in collectionTraits &&
  'max' in collectionTraits[trait.trait_type]

/**
 * IsStat - Checks to see if the given trait is a 'Stat'
 * A 'Stat' is defined as any trait that has a `display_type` of 'number'
 *
 * @param trait - The object containing an asset's trait
 * @return true if the trait is a 'Stat' and false otherwise
 */
const isStat = (trait: Trait) => trait.display_type === 'number'

/**
 * IsProperty - Checks to see if the given trait is a 'Property'.
 * A 'Property' is defined as any trait that has a `display_type` of null
 * and does not have a min/max value
 *
 * @param trait - The object containing an asset's trait
 * @param collectionTraits - List of collection traits
 * @return true if the trait is a 'Property' and false otherwise
 */
const isProperty = (trait: Trait, collectionTraits: CollectionTraits) =>
  (trait.display_type === null &&
    trait.trait_type in collectionTraits &&
    !('max' in collectionTraits[trait.trait_type])) ||
  !(trait.trait_type in collectionTraits)

export const toBaseDenomination = (value: number, decimals: number) =>
  +value.toFixed() / Math.pow(10, decimals)

export const getTraitType = (
  trait: Trait,
  collectionTraits: CollectionTraits
) => {
  if (isProperty(trait, collectionTraits)) {
    return TraitType.Property
  }
  if (isRanking(trait, collectionTraits)) {
    return TraitType.Ranking
  }
  if (isStat(trait)) {
    return TraitType.Stat
  }
  if (isBoost(trait)) {
    return TraitType.Boost
  }
  return null // Default return statement
}

export const getProvider = () =>
  (Web3.givenProvider || new Web3.providers.HttpProvider('https://mainnet.infura.io'))

export const networkFromString = (name: string) => {
  switch (name) {
    case 'rinkeby':
    case 'testnet':
    case 'testnets':
      return Network.Rinkeby
    case 'mainnet':
    case 'main':
    default:
      return Network.Main
  }
}