org.alephium.api.model.TestInputAsset.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of alephium-api_2.13 Show documentation

alephium-api

The newest version!

// Copyright 2018 The Alephium Authors
// This file is part of the alephium project.
//
// The library is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// The library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with the library. If not, see .

package org.alephium.api.model

import akka.util.ByteString

import org.alephium.protocol.model.{Address, AssetOutput}
import org.alephium.protocol.vm
import org.alephium.protocol.vm._
import org.alephium.util.{AVector, TimeStamp, U256}

final case class TestInputAsset(address: Address.Asset, asset: AssetState) {
  def toAssetOutput: AssetOutput =
    AssetOutput(
      asset.attoAlphAmount,
      address.lockupScript,
      TimeStamp.zero,
      asset.flatTokens.map(token => (token.id, token.amount)),
      ByteString.empty
    )

  def approveAll(gasFeeOpt: Option[U256]): AVector[Instr[StatefulContext]] = {
    val addressConst = AddressConst(vm.Val.Address(address.lockupScript))
    val attoAlphAmount = gasFeeOpt match {
      case Some(gasFee) => asset.attoAlphAmount.subUnsafe(gasFee)
      case None         => asset.attoAlphAmount
    }
    val alphInstrs = AVector[Instr[StatefulContext]](
      addressConst,
      U256Const(vm.Val.U256(attoAlphAmount)),
      ApproveAlph
    )
    val tokenInstrs = asset.flatTokens.flatMap[Instr[StatefulContext]] { token =>
      AVector(
        addressConst,
        BytesConst(vm.Val.ByteVec(token.id.bytes)),
        U256Const(vm.Val.U256(token.amount)),
        ApproveToken
      )
    }
    alphInstrs ++ tokenInstrs
  }
}