com.wavesplatform.state.diffs.invoke.CallArgumentPolicy.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of node Show documentation

node

The newest version!

package com.wavesplatform.state.diffs.invoke

import com.wavesplatform.lang.v1.compiler.ContractCompiler
import com.wavesplatform.lang.v1.compiler.Terms.*

trait CallArgumentPolicy {
  def check(e: EXPR): Boolean
  val expectedTypes: Set[String]
}

object CallArgumentPolicy {
  case object OnlyPrimitives extends CallArgumentPolicy {
    override def check(e: EXPR): Boolean =
      e match {
        case _: CONST_LONG | _: CONST_BYTESTR | _: CONST_STRING | _: CONST_BOOLEAN => true
        case _                                                                     => false
      }
    override val expectedTypes: Set[String] = ContractCompiler.primitiveCallableTypes
  }

  case object PrimitivesAndLists extends CallArgumentPolicy {
    override def check(e: EXPR): Boolean    = OnlyPrimitives.check(e) || e.isInstanceOf[ARR]
    override val expectedTypes: Set[String] = ContractCompiler.allowedCallableTypesV4
  }

  case object PrimitivesAndListsOfPrimitives extends CallArgumentPolicy {
    override def check(e: EXPR): Boolean    = OnlyPrimitives.check(e) || (e.isInstanceOf[ARR] && e.asInstanceOf[ARR].xs.forall(OnlyPrimitives.check))
    override val expectedTypes: Set[String] = ContractCompiler.allowedCallableTypesV4
  }

  implicit class CallCheck(fc: FUNCTION_CALL) {
    def check(c: CallArgumentPolicy): Either[String, Unit] =
      Either.cond(
        fc.args.forall(c.check),
        (),
        s"All arguments of InvokeScript must be one of the types: ${c.expectedTypes.mkString(", ")}"
      )
  }
}