wasm.httpwasm.utils.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of otoroshi_2.12 Show documentation
Lightweight api management on top of a modern http reverse proxy
The newest version!
package otoroshi.wasm.httpwasm.api

sealed trait HeaderKind {
  def value: Int
}

object HeaderKind {
  case object HeaderKindRequest          extends HeaderKind {
    def value: Int = 0
  }
  case object HeaderKindResponse         extends HeaderKind {
    def value: Int = 1
  }
  case object HeaderKindRequestTrailers  extends HeaderKind {
    def value: Int = 2
  }
  case object HeaderKindResponseTrailers extends HeaderKind {
    def value: Int = 3
  }

  def fromValue(value: Int): HeaderKind = {
    value match {
      case 0 => HeaderKindRequest
      case 1 => HeaderKindResponse
      case 2 => HeaderKindRequestTrailers
      case 3 => HeaderKindResponseTrailers
      case _ => throw new Exception("invalid header kind")
    }
  }
}

sealed trait BodyKind {
  def value: Int
}

object BodyKind {
  case object BodyKindRequest extends BodyKind {
    def value: Int = 0
  }

  case object BodyKindResponse extends BodyKind {
    def value: Int = 1
  }

  def fromValue(value: Int): BodyKind = {
    value match {
      case 0 => BodyKindRequest
      case 1 => BodyKindResponse
    }
  }

  def toString(value: BodyKind): String = {
    value match {
      case BodyKindRequest  => "BodyKindRequest"
      case BodyKindResponse => "BodyKindResponse"
      case _                => throw new Exception("invalid body kind")
    }
  }
}

sealed trait LogLevel {
  def value: Int
}

object LogLevel {
  case object LogLevelDebug extends LogLevel {
    def value: Int = -1
  }

  case object LogLevelInfo extends LogLevel {
    def value: Int = 0
  }

  case object LogLevelWarn extends LogLevel {
    def value: Int = 1
  }

  case object LogLevelError extends LogLevel {
    def value: Int = 2
  }

  case object LogLevelNone extends LogLevel {
    def value: Int = 3
  }

  def fromValue(value: Int): LogLevel = {
    value match {
      case -1 => LogLevelDebug
      case 0  => LogLevelInfo
      case 1  => LogLevelWarn
      case 2  => LogLevelError
      case 3  => LogLevelNone
      case _  => throw new Exception("invalid log level")
    }
  }
}

sealed trait Feature {
  def value: Int
}

object Feature {
  case object FeatureBufferRequest extends Feature {
    def value: Int = 1 << 0
  }

  case object FeatureBufferResponse extends Feature {
    def value: Int = 1 << 1
  }

  case object FeatureTrailers extends Feature {
    def value: Int = 1 << 2
  }

  def toString(feature: Feature): String = {
    feature match {
      case FeatureBufferRequest  => "FeatureBufferRequest"
      case FeatureBufferResponse => "FeatureBufferResponse"
      case FeatureTrailers       => "FeatureTrailers"
      case _                     => throw new Exception("invalid feature")
    }
  }
}

case class Features(f: Int) {
  def withEnabled(feature: Int): Features = {
    Features(f | feature)
  }

  // returns true if the feature (or group of features) is enabled.
  def isEnabled(feature: Feature): Boolean = {
    (f & feature.value) != 0
  }
}