• Home
• io.github.CyberAgent
• compilerplugin_2.10
• 0.9.3
• source code
• package.scala

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

scalapb.package.scala Maven / Gradle / Ivy

import protocbridge.JvmGenerator
import scalapb.GeneratorOption._

package object scalapb {

  sealed trait GeneratorOption extends Product with Serializable

  object GeneratorOption {
    case object FlatPackage extends GeneratorOption

    case object JavaConversions extends GeneratorOption

    case object Grpc extends GeneratorOption

    case object SingleLineToProtoString extends GeneratorOption

    case object AsciiFormatToString extends GeneratorOption

    case object Lenses extends GeneratorOption

    case object RetainSourceCodeInfo extends GeneratorOption
  }

  def gen(options: Set[GeneratorOption]): (JvmGenerator, Seq[String]) =
    (
      JvmGenerator("scala", ScalaPbCodeGenerator),
      Seq(
        "flat_package"                -> options(FlatPackage),
        "java_conversions"            -> options(JavaConversions),
        "grpc"                        -> options(Grpc),
        "single_line_to_proto_string" -> options(SingleLineToProtoString),
        "ascii_format_to_string"      -> options(AsciiFormatToString),
        "no_lenses"                   -> !options(Lenses),
        "retain_source_code_info"     -> options(RetainSourceCodeInfo)
      ).collect { case (name, v) if v => name }
    )

  def gen(
      flatPackage: Boolean = false,
      javaConversions: Boolean = false,
      grpc: Boolean = true,
      singleLineToProtoString: Boolean = false,
      asciiFormatToString: Boolean = false,
      lenses: Boolean = true
  ): (JvmGenerator, Seq[String]) = {
    val optionsBuilder = Set.newBuilder[GeneratorOption]
    if (flatPackage) {
      optionsBuilder += FlatPackage
    }
    if (javaConversions) {
      optionsBuilder += JavaConversions
    }
    if (grpc) {
      optionsBuilder += Grpc
    }
    if (singleLineToProtoString) {
      optionsBuilder += SingleLineToProtoString
    }
    if (asciiFormatToString) {
      optionsBuilder += AsciiFormatToString
    }
    if (lenses) {
      optionsBuilder += Lenses
    }
    gen(optionsBuilder.result())
  }
}