import {
  DispenserActivated,
  DispenserAllowedSwapperChanged,
  DispenserDeactivated,
  OwnerWithdrawed,
  TokensDispensed
} from '../@types/Dispenser/Dispenser'
import { DispenserCreated } from '../@types/ERC721Factory/ERC721Factory'
import { DispenserTransaction } from '../@types/schema'
import { ZERO_BD } from './utils/constants'
import { createDispenser, getDispenser } from './utils/dispenserUtils'
import { getUser } from './utils/userUtils'

export function handleNewDispenser(event: DispenserCreated): void {
  createDispenser(event.params.datatokenAddress.toHex())
}

export function handleActivate(event: DispenserActivated): void {
  const dispenser = getDispenser(event.params.datatokenAddress.toHex())
  dispenser.active = true
  dispenser.save()
}

export function handleDeactivate(event: DispenserDeactivated): void {
  const dispenser = getDispenser(event.params.datatokenAddress.toHex())
  dispenser.active = true
  dispenser.save()
}

export function handleAllowedSwapperChanged(
  event: DispenserAllowedSwapperChanged
): void {
  const dispenser = getDispenser(event.params.datatoken.toHex())
  dispenser.allowedSwapper = event.params.newAllowedSwapper.toHex()
  dispenser.save()
}

export function handleTokensDispensed(event: TokensDispensed): void {
  const id = event.transaction.hash
    .toHexString()
    .concat('-')
    .concat(event.params.datatokenAddress.toHexString())

  const dispenserTransaction = new DispenserTransaction(id)
  const dispenser = getDispenser(event.params.datatokenAddress.toHex())
  dispenser.balance = dispenser.balance.minus(
    event.params.amount.toBigDecimal()
  )
  dispenser.save()

  dispenserTransaction.dispenser = dispenser.id
  const user = getUser(event.params.userAddress.toHex())
  dispenserTransaction.user = user.id

  dispenserTransaction.createdTimestamp = event.block.timestamp.toI32()
  dispenserTransaction.tx = event.transaction.hash
  dispenserTransaction.block = event.block.number.toI32()
  dispenserTransaction.save()
}

export function handleOwnerWinthdraw(event: OwnerWithdrawed): void {
  const dispenser = getDispenser(event.params.datatoken.toHex())
  dispenser.balance = ZERO_BD
  dispenser.save()
}