sec.api.grpc.marshalling.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sec-core_3 Show documentation
sec-core
The newest version!
/*
 * Copyright 2020 Scala EventStoreDB Client
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package sec
package api
package grpc

import java.nio.charset.StandardCharsets
import java.util.Base64
import io.grpc.Metadata.AsciiMarshaller
import constants.Headers.BasicScheme

//======================================================================================================================

final private[grpc] case class InvalidInput(input: String, tpe: String)
  extends RuntimeException(s"Could not parse $input to $tpe")

//======================================================================================================================

private[grpc] object IntMarshaller extends NumericAsciiMarshaller[Int]("Int")
private[grpc] object LongMarshaller extends NumericAsciiMarshaller[Long]("Long")

sealed abstract private[grpc] class NumericAsciiMarshaller[T: Numeric](tpe: String) extends AsciiMarshaller[T]:
  final def toAsciiString(v: T): String    = v.toString
  final def parseAsciiString(s: String): T = Numeric[T].parseString(s).getOrElse(throw InvalidInput(s, tpe))

//======================================================================================================================

private[grpc] object StringMarshaller extends AsciiMarshaller[String]:
  def toAsciiString(value: String): String         = value
  def parseAsciiString(serialized: String): String = serialized

//======================================================================================================================

private[grpc] object BooleanMarshaller extends AsciiMarshaller[Boolean]:
  def toAsciiString(v: Boolean): String    = v.toString()
  def parseAsciiString(s: String): Boolean = s.toBooleanOption.getOrElse(throw InvalidInput(s, "Boolean"))

//======================================================================================================================

private[grpc] object UserCredentialsMarshaller extends AsciiMarshaller[UserCredentials]:

  val decodingNotSupported      = UserCredentials.unsafe("decoding-not-supported", "n/a")
  val encoder64: Base64.Encoder = Base64.getEncoder

  def toAsciiString(uc: UserCredentials): String =
    val encoded     = encoder64.encode(s"${uc.username}:${uc.password}".getBytes)
    val credentials = new String(encoded, StandardCharsets.US_ASCII)
    s"$BasicScheme $credentials"

  def parseAsciiString(serialized: String): UserCredentials = decodingNotSupported

//======================================================================================================================