ocean-subgraph/src/mappings/erc721Factory.ts

import {
  NFTCreated,
  TokenCreated,
  ERC721Factory
} from '../@types/ERC721Factory/ERC721Factory'
import { decimal } from './utils/constants'
import { weiToDecimal } from './utils/generic'

import { getUser } from './utils/userUtils'
import { getToken, getNftToken } from './utils/tokenUtils'
import { addDatatoken } from './utils/globalUtils'
import { BigInt } from '@graphprotocol/graph-ts'

export function handleNftCreated(event: NFTCreated): void {
  // const nft = new Nft(event.params.newTokenAddress.toHexString())
  const nft = getNftToken(event.params.newTokenAddress)
  const user = getUser(event.params.admin.toHexString())
  nft.owner = user.id
  const creator = getUser(event.params.creator.toHexString())
  nft.creator = creator.id
  nft.address = event.params.newTokenAddress.toHexString()
  nft.name = event.params.tokenName
  nft.symbol = event.params.symbol.toString()
  nft.tokenUri = event.params.tokenURI.toString()
  nft.createdTimestamp = event.block.timestamp.toI32()
  nft.tx = event.transaction.hash.toHex()
  nft.block = event.block.number.toI32()
  nft.transferable = event.params.transferable

  nft.save()
}

export function handleNewToken(event: TokenCreated): void {
  const token = getToken(event.params.newTokenAddress, true)
  // const token = new Token(event.params.newTokenAddress.toHexString())

  token.isDatatoken = true
  token.address = event.params.newTokenAddress.toHexString()
  token.createdTimestamp = event.block.timestamp.toI32()
  token.tx = event.transaction.hash.toHex()
  token.block = event.block.number.toI32()

  token.nft = event.params.creator.toHexString()

  token.name = event.params.name
  token.symbol = event.params.symbol
  token.decimals = 18
  token.supply = decimal.ZERO
  token.cap = weiToDecimal(event.params.cap.toBigDecimal(), 18)
  const eventTemplateAddress = event.params.templateAddress
    .toHexString()
    .toLowerCase()
  const contract = ERC721Factory.bind(event.address)
  const templateCount = contract.try_getCurrentTemplateCount()
  if (templateCount.reverted) return
  const templateCountNum = templateCount.value.toI32()

  for (let i = 0; i < templateCountNum; i++) {
    const template = contract.try_getTokenTemplate(BigInt.fromI32(1 + i))
    if (template.reverted) return
    const templateAddress = template.value.templateAddress
      .toHexString()
      .toLowerCase()
    if (templateAddress == eventTemplateAddress) {
      token.templateId = 1 + i
    }
  }

  token.save()
  addDatatoken()
}