ocean-subgraph/src/mappings/helpers.ts

import {
  BigDecimal,
  BigInt,
  Bytes,
  dataSource,
  Address,
  ethereum
} from '@graphprotocol/graph-ts'

import {
  Pool as PoolEntity,
  User,
  PoolToken,
  PoolShare,
  PoolTransaction,
  PoolFactory,
  Datatoken,
  TokenBalance,
  TokenTransaction,
  PoolTransactionTokenValues
} from '../types/schema'
import { log } from '@graphprotocol/graph-ts'
import { Pool } from '../types/templates/Pool/Pool'

export let ZERO_BD = BigDecimal.fromString('0.0')
export let MINUS_1_BD = BigDecimal.fromString('-1.0')
export let ONE_BD = BigDecimal.fromString('1.0')

export let ONE_BASE_18 = BigInt.fromI32(10).pow(18 as u8)
export let BONE = BigDecimal.fromString('1000000000000000000')

export let ENABLE_DEBUG = false

let network = dataSource.network()

export let OCEAN: string = (network == 'mainnet')
  ? '0x967da4048cd07ab37855c090aaf366e4ce1b9f48'
  : '0x8967BCF84170c91B0d24D4302C2376283b0B3a07'

export function _debuglog(message: string, event: ethereum.Event, args: Array<string>): void {
  if (event != null) {
    args.push(event.transaction.hash.toHex())
    args.push(event.address.toHex())
  }
  for (let i=0; i < args.length; i++) {
    message = message.concat(' {}')
  }
  log.info('@@@@@@ ' + message, args)
}

export function debuglog(message: string, event: ethereum.Event, args: Array<string>): void {
  if (!ENABLE_DEBUG) return
  _debuglog(message, event, args)
}

export function hexToDecimal(hexString: String, decimals: i32): BigDecimal {
  let bytes = Bytes.fromHexString(hexString.toString()).reverse() as Bytes
  let bi = BigInt.fromUnsignedBytes(bytes)
  let scale = BigInt.fromI32(10).pow(decimals as u8).toBigDecimal()
  return bi.divDecimal(scale)
}

export function bigIntToDecimal(amount: BigInt, decimals: i32): BigDecimal {
  let scale = BigInt.fromI32(10).pow(decimals as u8).toBigDecimal()
  return amount.toBigDecimal().div(scale)
}

export function tokenToDecimal(amount: BigDecimal, decimals: i32): BigDecimal {
  let scale = BigInt.fromI32(10).pow(decimals as u8).toBigDecimal()
  return amount.div(scale)
}

export function decimalToBigInt(value: BigDecimal): BigInt {
  value.truncate(18)
  let scale = BigInt.fromI32(10).pow((value.exp.toI32() + 18) as u8)
  return value.digits.times(scale)

}

export function updatePoolTokenBalance(poolToken: PoolToken, balance: BigDecimal, source: string): void {
  debuglog('########## updating poolToken balance (source, oldBalance, newBalance, poolId) ', null,
    [source, poolToken.balance.toString(), balance.toString(), poolToken.poolId])
  poolToken.balance = balance
}

export function createPoolShareEntity(id: string, pool: string, user: string): void {
  let poolShare = new PoolShare(id)

  createUserEntity(user)

  poolShare.userAddress = user
  poolShare.poolId = pool
  poolShare.balance = ZERO_BD
  poolShare.save()
}

export function createPoolTokenEntity(id: string, pool: string, address: string): void {
  let datatoken = Datatoken.load(address)

  let poolToken = new PoolToken(id)
  poolToken.poolId = pool
  poolToken.tokenId = datatoken ? datatoken.id: ''
  poolToken.tokenAddress = address
  poolToken.balance = ZERO_BD
  poolToken.denormWeight = ZERO_BD
  poolToken.save()
}

export function updatePoolTransactionToken(
  poolTx: string, poolTokenId: string, amount: BigDecimal,
  balance: BigDecimal, feeValue: BigDecimal
): void {

  let ptx = PoolTransaction.load(poolTx)
  let poolToken = PoolToken.load(poolTokenId)
  let pool = PoolEntity.load(poolToken.poolId)
  let ptxTokenValuesId = poolTx.concat('-').concat(poolToken.tokenAddress)
  let ptxTokenValues = PoolTransactionTokenValues.load(ptxTokenValuesId)
  if (ptxTokenValues == null) {
    ptxTokenValues = new PoolTransactionTokenValues(ptxTokenValuesId)
  }
  ptxTokenValues.txId = poolTx
  ptxTokenValues.poolToken = poolTokenId
  ptxTokenValues.poolAddress = poolToken.poolId
  ptxTokenValues.userAddress = ptx.userAddress
  ptxTokenValues.tokenAddress = PoolToken.load(poolTokenId).tokenAddress

  ptxTokenValues.value = amount
  ptxTokenValues.tokenReserve = balance
  ptxTokenValues.feeValue = feeValue
  if (amount.lt(ZERO_BD)) {
    ptxTokenValues.type = 'out'
  } else {
    ptxTokenValues.type = 'in'
  }

  ptxTokenValues.save()

  if (ptxTokenValues.tokenAddress == OCEAN) {
    ptx.oceanReserve = ptxTokenValues.tokenReserve
    pool.oceanReserve = ptxTokenValues.tokenReserve
  } else {
    ptx.datatokenReserve = ptxTokenValues.tokenReserve
    pool.datatokenReserve = ptxTokenValues.tokenReserve
  }
  debuglog('########## updatePoolTransactionToken: ', null,
    [
      BigInt.fromI32(ptx.block).toString(),
      BigInt.fromI32(ptx.timestamp).toString(),
      ptxTokenValues.type,
      ptxTokenValues.value.toString(),
      ptxTokenValues.tokenReserve.toString(),
      poolToken.poolId,
    ])

  ptx.save()
  pool.save()
}

export function createPoolTransaction(event: ethereum.Event, event_type: string, userAddress: string): void {
  let poolId = event.address.toHex()
  let pool = PoolEntity.load(poolId)
  let ptx = event.transaction.hash.toHexString()

  let ocnToken = PoolToken.load(poolId.concat('-').concat(OCEAN))
  let dtToken = PoolToken.load(poolId.concat('-').concat(pool.datatokenAddress))

  if (ocnToken == null || dtToken == null) {
    return
  }

  let poolTx = PoolTransaction.load(ptx)
  if (poolTx != null) {
    return
  }
  poolTx = new PoolTransaction(ptx)

  poolTx.poolAddress = poolId
  poolTx.userAddress = userAddress
  poolTx.poolAddressStr = poolId
  poolTx.userAddressStr = userAddress

  poolTx.sharesTransferAmount = ZERO_BD
  poolTx.sharesBalance = ZERO_BD

  // pool.datatokenReserve = dtToken.balance
  // pool.oceanReserve = ocnToken.balance
  // Initial reserve values, will be updated in `updatePoolTransactionToken`
  poolTx.datatokenReserve = dtToken.balance
  poolTx.oceanReserve = ocnToken.balance

  let p = Pool.bind(Address.fromString(poolId))
  let priceResult = p.try_calcInGivenOut(
    decimalToBigInt(ocnToken.balance), decimalToBigInt(ocnToken.denormWeight),
    decimalToBigInt(dtToken.balance), decimalToBigInt(dtToken.denormWeight),
    ONE_BASE_18, decimalToBigInt(pool.swapFee)
  )
  debuglog(
    'args to calcInGivenOut (ocnBalance, ocnWeight, dtBalance, dtWeight, dtAmount, swapFee, result)', null,
    [
      decimalToBigInt(ocnToken.balance).toString(),
      decimalToBigInt(ocnToken.denormWeight).toString(),
      decimalToBigInt(dtToken.balance).toString(),
      decimalToBigInt(dtToken.denormWeight).toString(),
      ONE_BASE_18.toString(),
      decimalToBigInt(pool.swapFee).toString(),
      priceResult.reverted ? 'failed' : priceResult.value.toString()
    ]
  )
  poolTx.consumePrice = priceResult.reverted ? MINUS_1_BD : bigIntToDecimal(priceResult.value, 18)
  poolTx.spotPrice = calcSpotPrice(
    ocnToken.balance, ocnToken.denormWeight,
    dtToken.balance, dtToken.denormWeight,
    pool.swapFee
  )

  pool.consumePrice = poolTx.consumePrice
  pool.spotPrice = poolTx.spotPrice

  pool.transactionCount = pool.transactionCount.plus(BigInt.fromI32(1))

  pool.save()

  debuglog('updated pool reserves (source, dtBalance, ocnBalance, dtReserve, ocnReserve): ',
    event,
    ['createPoolTransaction', dtToken.balance.toString(), ocnToken.balance.toString(),
      pool.datatokenReserve.toString(), pool.oceanReserve.toString()])

  poolTx.tx = event.transaction.hash
  poolTx.event = event_type
  poolTx.block = event.block.number.toI32()
  poolTx.timestamp = event.block.timestamp.toI32()
  poolTx.gasUsed = event.transaction.gasUsed.toBigDecimal()
  poolTx.gasPrice = event.transaction.gasPrice.toBigDecimal()


  debuglog('####################### poolTransaction: ', event,
    [
      BigInt.fromI32(poolTx.block).toString(),
      BigInt.fromI32(poolTx.timestamp).toString(),
      pool.oceanReserve.toString()
    ])

  poolTx.save()
}

export function calcSpotPrice(
  balanceIn: BigDecimal, wIn: BigDecimal,
  balanceOut: BigDecimal, wOut: BigDecimal,
  swapFee: BigDecimal
): BigDecimal {
  if (balanceIn <= ZERO_BD || balanceOut <= ZERO_BD) return MINUS_1_BD
  debuglog('################ calcSpotPrice', null,
    [balanceIn.toString(), wIn.toString(),
     balanceOut.toString(), wOut.toString(),
     swapFee.toString()])

  let numer = balanceIn.div(wIn)
  let denom = balanceOut.div(wOut)
  if (denom <= ZERO_BD) return MINUS_1_BD

  let ratio = numer.div(denom)
  let scale = ONE_BD.div(ONE_BD.minus(swapFee))
  let price = ratio.times(scale)
  price.truncate(18)
  debuglog('################ calcSpotPrice values:', null,
    [numer.toString(), denom.toString(), ratio.toString(), scale.toString(), price.toString()])
  return price
}

export function decrPoolCount(finalized: boolean): void {
  let factory = PoolFactory.load('1')
  factory.poolCount -= 1
  if (finalized) factory.finalizedPoolCount -= 1
  factory.save()
}

export function saveTokenTransaction(event: ethereum.Event, eventName: string): void {
  let tx = event.transaction.hash.toHexString().concat('-').concat(event.logIndex.toString())
  let userAddress = event.transaction.from.toHex()
  let transaction = TokenTransaction.load(tx)
  if (transaction == null) {
    transaction = new TokenTransaction(tx)
  }
  transaction.event = eventName
  transaction.datatokenAddress = event.address.toHex()
  transaction.userAddress = userAddress
  transaction.gasUsed = event.transaction.gasUsed.toBigDecimal()
  transaction.gasPrice = event.transaction.gasPrice.toBigDecimal()
  transaction.tx = event.transaction.hash
  transaction.timestamp = event.block.timestamp.toI32()
  transaction.block = event.block.number.toI32()
  transaction.save()

  createUserEntity(userAddress)
}

export function createUserEntity(address: string): void {
  if (User.load(address) == null) {
    let user = new User(address)
    user.save()
  }
}

export function updateTokenBalance(id: string, token: string, user: string, amount: BigDecimal): void {
  let tokenBalance = TokenBalance.load(id)
  if (tokenBalance == null) {
    tokenBalance = new TokenBalance(id)
    createUserEntity(user)
    tokenBalance.userAddress = user
    tokenBalance.datatokenId = token
    tokenBalance.balance = ZERO_BD
  }

  tokenBalance.balance = tokenBalance.balance.plus(amount)
  tokenBalance.save()
}