scalapb.compiler.DescriptorImplicits.scala Maven / Gradle / Ivy
The newest version!
package scalapb.compiler
import com.google.protobuf.DescriptorProtos.{
DescriptorProto,
EnumDescriptorProto,
FileDescriptorProto,
ServiceDescriptorProto,
SourceCodeInfo
}
import com.google.protobuf.Descriptors._
import com.google.protobuf.WireFormat.FieldType
import scalapb.options.compiler.Scalapb
import scalapb.options.compiler.Scalapb.ScalaPbOptions.EnumValueNaming
import scalapb.options.compiler.Scalapb._
import scala.jdk.CollectionConverters._
import scala.collection.immutable.IndexedSeq
class DescriptorImplicits(params: GeneratorParams, files: Seq[FileDescriptor]) {
import DescriptorImplicits._
case class ScalaName(emptyPackage: Boolean, xs: Seq[String]) {
def name = xs.last
def nameSymbol = nameRelative(0)
def nameRelative(levels: Int) = xs.takeRight(levels + 1).map(_.asSymbol).mkString(".")
def fullName = xs.map(_.asSymbol).mkString(".")
def fullNameWithMaybeRoot: String = {
if (!emptyPackage)
s"_root_.${fullName}"
else fullName
}
def fullNameWithMaybeRoot(context: Descriptor): String = {
fullNameWithMaybeRoot(context.fields.map(_.scalaName))
}
def fullNameWithMaybeRoot(contextNames: Seq[String]): String = {
val topLevelPackage = xs.head
if (contextNames.contains(topLevelPackage) && !emptyPackage)
s"_root_.${fullName}"
else fullName
}
def /(name: String) = ScalaName(emptyPackage, xs :+ name)
def sibling(name: String) = ScalaName(emptyPackage, xs.dropRight(1) :+ name)
}
// Needs to be lazy since the input may be invalid... For example, if one of
// the cases is not a message, the call to getMessageType would fail.
private lazy val sealedOneofsCache: SealedOneofsCache = {
val sealedOneof = for {
file <- files
message <- file.allMessages if message.isSealedOneofType
} yield SealedOneof(
message,
message.getRealOneofs.get(0).getFields.asScala.map(_.getMessageType).toVector
)
new SealedOneofsCache(sealedOneof)
}
private lazy val fileOptionsCache = FileOptionsCache.buildCache(files)
implicit final class MethodDescriptorPimp(method: MethodDescriptor) {
class MethodTypeWrapper(descriptor: Descriptor) {
def customScalaType: Option[String] =
if (descriptor.isSealedOneofType)
Some(descriptor.sealedOneofScalaType)
else if (descriptor.messageOptions.hasType) Some(descriptor.messageOptions.getType)
else if (descriptor.getFile.usePrimitiveWrappers)
DescriptorImplicits.primitiveWrapperType(descriptor)
else None
def baseScalaType = descriptor.scalaType.fullNameWithMaybeRoot(Seq("build"))
def scalaType: String = customScalaType.getOrElse(baseScalaType)
}
def inputType = new MethodTypeWrapper(method.getInputType)
def outputType = new MethodTypeWrapper(method.getOutputType)
def isClientStreaming = method.toProto.getClientStreaming
def isServerStreaming = method.toProto.getServerStreaming
def streamType: StreamType = {
(isClientStreaming, isServerStreaming) match {
case (false, false) => StreamType.Unary
case (true, false) => StreamType.ClientStreaming
case (false, true) => StreamType.ServerStreaming
case (true, true) => StreamType.Bidirectional
}
}
def canBeBlocking = !method.toProto.getClientStreaming
private def name0: String = NameUtils.snakeCaseToCamelCase(method.getName)
def name: String = name0.asSymbol
def grpcDescriptor =
method.getService.companionObject / s"METHOD_${NameUtils.toAllCaps(method.getName)}"
@deprecated("Use grpcDescriptor instead to get the name or full name", "0.10.0")
def descriptorName = grpcDescriptor.name
def sourcePath: Seq[Int] = {
method.getService.sourcePath ++ Seq(
ServiceDescriptorProto.METHOD_FIELD_NUMBER,
method.getIndex
)
}
def comment: Option[String] = {
method.getFile
.findLocationByPath(sourcePath)
.map(t => t.getLeadingComments + t.getTrailingComments)
.map(Helper.escapeComment)
.filter(_.nonEmpty)
}
def deprecatedAnnotation: String = {
if (method.getOptions.getDeprecated) {
ProtobufGenerator.deprecatedAnnotation + " "
} else {
""
}
}
def javaDescriptorSource: String =
s"${method.getService.javaDescriptorSource}.getMethods.get(${method.getIndex})"
}
implicit final class ServiceDescriptorPimp(self: ServiceDescriptor) {
@deprecated("Use companionObject instead to get the name or the full name", "0.10.0")
def objectName = companionObject.name
def companionObject = self.getFile.scalaPackage / (self.getName + "Grpc")
def name = self.getName.asSymbol
def blockingClient = self.getName + "BlockingClient"
def blockingStub = self.getName + "BlockingStub"
def stub = self.getName + "Stub"
def methods = self.getMethods.asScala.toIndexedSeq
def grpcDescriptor = companionObject / "SERVICE"
@deprecated("Use grpcDescriptor to get the name of the full name", "0.10.0")
def descriptorName = "SERVICE"
def scalaDescriptorSource: String =
s"${self.getFile.fileDescriptorObject.fullName}.scalaDescriptor.services(${self.getIndex})"
def javaDescriptorSource: String =
s"${self.getFile.fileDescriptorObject.fullName}.javaDescriptor.getServices.get(${self.getIndex})"
def sourcePath: Seq[Int] = Seq(FileDescriptorProto.SERVICE_FIELD_NUMBER, self.getIndex)
def comment: Option[String] = {
self.getFile
.findLocationByPath(sourcePath)
.map(t => t.getLeadingComments + t.getTrailingComments)
.map(Helper.escapeComment)
.filter(_.nonEmpty)
}
def deprecatedAnnotation: String = {
if (self.getOptions.getDeprecated) {
ProtobufGenerator.deprecatedAnnotation + " "
} else {
""
}
}
}
implicit class FieldDescriptorPimp(val fd: FieldDescriptor) {
import NameUtils._
def containingOneOf: Option[OneofDescriptor] = Option(fd.getRealContainingOneof())
def isInOneof: Boolean = containingOneOf.isDefined
def isSealedOneofType: Boolean = fd.isMessage && fd.getMessageType.isSealedOneofType
def scalaName: String =
if (fieldOptions.getScalaName.nonEmpty) fieldOptions.getScalaName
else
fd.getName match {
case ("number" | "value") if fd.isInOneof => "_" + fd.getName
case "serialized_size" => "_serializedSize"
case x =>
getNameWithFallback(x, Case.CamelCase, Appendage.Prefix)
}
def upperScalaName: String =
if (fieldOptions.getScalaName.nonEmpty)
snakeCaseToCamelCase(fieldOptions.getScalaName, true)
else
fd.getName match {
case "serialized_size" => "_SerializedSize"
case "class" => "_Class"
case "empty" if fd.isInOneof => "_Empty"
case x => getNameWithFallback(x, Case.PascalCase, Appendage.Prefix)
}
private def getNameWithFallback(x: String, targetCase: Case, appendage: Appendage) = {
val candidate = snakeCaseToCamelCase(x, targetCase.isPascal)
if (ProtoValidation.ForbiddenFieldNames.contains(candidate)) {
if (appendage.isPrefix) "_" + candidate
else candidate + "_"
} else candidate
}
def upperJavaName: String = fd.getName match {
case "serialized_size" => "SerializedSize_"
case "class" => "Class_"
case x => getNameWithFallback(x, Case.PascalCase, Appendage.Postfix)
}
def fieldNumberConstantName: String = fd.getName.toUpperCase() + "_FIELD_NUMBER"
def oneOfTypeName: ScalaName = {
assert(isInOneof)
fd.getRealContainingOneof.scalaType / upperScalaName
}
def noBox =
if (fieldOptions.hasNoBox) fieldOptions.getNoBox
else if (fd.isMessage) fd.getMessageType.noBox
else false
// Is this field boxed inside an Option in Scala. Equivalent, does the Java API
// support hasX methods for this field.
def supportsPresence: Boolean =
fd.isOptional && !fd.isInOneof && (!fd.getFile.isProto3 || fd.isMessage || fd
.toProto()
.getProto3Optional()) &&
!noBox && !fd.isSealedOneofType
// Is the Scala representation of this field a singular type.
def isSingular =
fd.isRequired || (fd.getFile.isProto3 && !fd.isInOneof && fd.isOptional && !fd.isMessage && !fd
.toProto()
.getProto3Optional()) || (
fd.isOptional && (noBox || (fd.isSealedOneofType && !fd.isInOneof))
)
def enclosingType: EnclosingType =
if (isSingular) EnclosingType.None
else if (supportsPresence || fd.isInOneof) EnclosingType.ScalaOption
else {
EnclosingType.Collection(collectionType)
}
def fieldMapEnclosingType: EnclosingType =
if (isSingular) EnclosingType.None
else if (supportsPresence || fd.isInOneof) EnclosingType.ScalaOption
else if (!fd.isMapField) EnclosingType.Collection(collectionType)
else EnclosingType.Collection(ScalaSeq)
def isMapField = isMessage && fd.isRepeated && fd.getMessageType.isMapEntry
def mapType: MessageDescriptorPimp#MapType = {
assert(isMapField)
fd.getMessageType.mapType
}
def collectionBuilder: String = {
require(fd.isRepeated)
val t = if (collectionType == ScalaSeq) ScalaVector else collectionType
if (!fd.isMapField)
s"$t.newBuilder[$singleScalaTypeName]"
else {
s"$t.newBuilder[${fd.mapType.keyType}, ${fd.mapType.valueType}]"
}
}
def emptyCollection: String = {
s"${collectionType}.empty"
}
// In scalapb.proto, we separate between collection_type and map_type, but internally this is unified.
def collectionType: String = {
require(fd.isRepeated)
if (fd.isMapField) {
if (fd.fieldOptions.hasMapType) fd.fieldOptions.getMapType
else if (fd.getFile.scalaOptions.hasMapType) fd.getFile.scalaOptions.getMapType
else ScalaMap
} else {
if (fd.fieldOptions.hasCollectionType) fd.fieldOptions.getCollectionType
else if (fd.getFile.scalaOptions.hasCollectionType)
fd.getFile.scalaOptions.getCollectionType
else ScalaSeq
}
}
def fieldMapCollection(innerType: String) = {
if (supportsPresence) s"_root_.scala.Option[$innerType]"
else if (fd.isRepeated && !fd.isMapField) s"${collectionType}[$innerType]"
else if (fd.isRepeated && fd.isMapField) s"${ScalaSeq}[$innerType]"
else innerType
}
def fieldsMapEmptyCollection: String = {
require(fd.isRepeated)
if (fd.isMapField) s"$ScalaSeq.empty"
else emptyCollection
}
def scalaTypeName: String =
if (fd.isMapField) {
s"$collectionType[${mapType.keyType}, ${mapType.valueType}]"
} else if (fd.isRepeated) s"${collectionType}[$singleScalaTypeName]"
else if (supportsPresence) s"${ScalaOption}[$singleScalaTypeName]"
else singleScalaTypeName
def fieldOptions: FieldOptions = {
val localOptions = fd.getOptions.getExtension[FieldOptions](Scalapb.field)
fd.getFile.scalaOptions.getAuxFieldOptionsList.asScala
.find(_.getTarget == fd.getFullName())
.fold(localOptions)(aux =>
FieldOptions.newBuilder(aux.getOptions).mergeFrom(localOptions).build
)
}
def annotationList: Seq[String] = {
val deprecated = {
if (fd.getOptions.getDeprecated)
List(ProtobufGenerator.deprecatedAnnotation)
else
Nil
}
deprecated ++ fieldOptions.getAnnotationsList().asScala.toSeq
}
def customSingleScalaTypeName: Option[String] = {
// If the current message is within a MapEntry (that is a key, or a value), find the actual map
// field in the enclosing message. This is used to determine map level options when processing the
// key and value fields.
def fieldReferencingMap: FieldDescriptor = {
require(fd.getContainingType.isMapEntry)
val messageReferencingMap = fd.getContainingType.getContainingType
messageReferencingMap.getFields.asScala
.filter(_.isMapField)
.find(fd.getContainingType eq _.getMessageType)
.get
}
if (isMapField) Some(s"(${mapType.keyType}, ${mapType.valueType})")
else if (isSealedOneofType) Some(fd.getMessageType.sealedOneofScalaType)
else if (fieldOptions.hasType) Some(fieldOptions.getType)
else if (isMessage && fd.getMessageType.messageOptions.hasType)
Some(fd.getMessageType.messageOptions.getType)
else if (isEnum && fd.getEnumType.scalaOptions.hasType)
Some(fd.getEnumType.scalaOptions.getType)
else if (isBytes && fd.getFile.scalaOptions.hasBytesType)
Some(fd.getFile.scalaOptions.getBytesType)
else if (fd.getContainingType.isMapEntry && fd.getNumber == 1 && fieldReferencingMap.fieldOptions.hasKeyType)
Some(fieldReferencingMap.fieldOptions.getKeyType)
else if (fd.getContainingType.isMapEntry && fd.getNumber == 2 && fieldReferencingMap.fieldOptions.hasValueType)
Some(fieldReferencingMap.fieldOptions.getValueType)
else if (isMessage && fd.getFile.usePrimitiveWrappers)
DescriptorImplicits.primitiveWrapperType(fd.getMessageType)
else None
}
def baseSingleScalaTypeName: String = fd.getJavaType match {
case FieldDescriptor.JavaType.INT => "_root_.scala.Int"
case FieldDescriptor.JavaType.LONG => "_root_.scala.Long"
case FieldDescriptor.JavaType.FLOAT => "_root_.scala.Float"
case FieldDescriptor.JavaType.DOUBLE => "_root_.scala.Double"
case FieldDescriptor.JavaType.BOOLEAN => "_root_.scala.Boolean"
case FieldDescriptor.JavaType.BYTE_STRING => "_root_.com.google.protobuf.ByteString"
case FieldDescriptor.JavaType.STRING => "_root_.scala.Predef.String"
case FieldDescriptor.JavaType.MESSAGE =>
fd.getMessageType.scalaType
.fullNameWithMaybeRoot(fd.getContainingType.fields.map(_.scalaName))
case FieldDescriptor.JavaType.ENUM =>
fd.getEnumType.scalaType.fullNameWithMaybeRoot(fd.getContainingType.fields.map(_.scalaName))
}
def singleScalaTypeName = customSingleScalaTypeName.getOrElse(baseSingleScalaTypeName)
def getMethod = "get" + upperScalaName
def typeMapperValName = "_typemapper_" + scalaName
def typeMapper: ScalaName = {
if (!fd.isExtension)
fd.getContainingType.scalaType / typeMapperValName
else {
val c =
if (fd.getExtensionScope == null) fd.getFile.fileDescriptorObject
else fd.getExtensionScope.scalaType
c / typeMapperValName
}
}
def isEnum = fd.getType == FieldDescriptor.Type.ENUM
def isMessage = fd.getType == FieldDescriptor.Type.MESSAGE
def isBytes = fd.getType == FieldDescriptor.Type.BYTES
def javaExtensionFieldFullName = {
require(fd.isExtension)
val inClass =
if (fd.getExtensionScope == null) fd.getFile.javaFullOuterClassName
else fd.getExtensionScope.javaTypeName
s"$inClass.${fd.scalaName}"
}
def sourcePath: Seq[Int] = {
fd.getContainingType.sourcePath ++ Seq(DescriptorProto.FIELD_FIELD_NUMBER, fd.getIndex)
}
def comment: Option[String] = {
fd.getFile
.findLocationByPath(sourcePath)
.map(t => t.getLeadingComments + t.getTrailingComments)
.map(Helper.escapeComment)
.filter(_.nonEmpty)
}
}
implicit class OneofDescriptorPimp(val oneof: OneofDescriptor) {
def javaEnumName = {
val name = NameUtils.snakeCaseToCamelCase(oneof.getName, true)
s"get${name}Case"
}
def scalaName =
oneof.getContainingType.scalaType / NameUtils.snakeCaseToCamelCase(oneof.getName)
def scalaType = {
val name = oneof.getName match {
case "ValueType" | "value_type" => "ValueTypeOneof"
case n => NameUtils.snakeCaseToCamelCase(n, true)
}
val conflictsWithMessage =
oneof.getContainingType.getEnumTypes.asScala.exists(_.scalaType.name == name) ||
oneof.getContainingType.nestedTypes.exists(_.scalaType.name == name)
val n =
if (conflictsWithMessage) name + "Oneof"
else name
oneof.getContainingType.scalaType / n
}
def fields: IndexedSeq[FieldDescriptor] =
(0 until oneof.getFieldCount).map(oneof.getField).filter(_.getLiteType != FieldType.GROUP)
def empty = scalaType / "Empty"
def oneofOptions: OneofOptions = oneof.getOptions.getExtension[OneofOptions](Scalapb.oneof)
def baseClasses = "_root_.scalapb.GeneratedOneof" +: oneofOptions.getExtendsList.asScala.toSeq
}
private val OneofMessageSuffix = "Message"
implicit class MessageDescriptorPimp(val message: Descriptor) {
def fields = message.getFields.asScala.filter(_.getLiteType != FieldType.GROUP).toSeq
def fieldsWithoutOneofs = fields.filterNot(_.isInOneof)
def parent: Option[Descriptor] = Option(message.getContainingType)
def sealedOneofStyle: SealedOneofStyle = {
assert(isSealedOneofType)
if (message.getRealOneofs.asScala.exists(_.getName == "sealed_value"))
SealedOneofStyle.Default
else if (message.getRealOneofs.asScala.exists(_.getName == "sealed_value_optional"))
SealedOneofStyle.Optional
else throw new RuntimeException("Unexpected oneof style")
}
// every message that passes this filter must be a sealed oneof. The check that it actually
// obeys the rules is done in ProtoValidation.
def isSealedOneofType: Boolean = {
message.getRealOneofs.asScala
.exists(o => o.getName == "sealed_value" || o.getName == "sealed_value_optional")
}
def isSealedOneofCase: Boolean = sealedOneofsCache.getContainer(message).isDefined
def scalaType: ScalaName = {
val name = message.getName match {
case "Option" => "OptionProto"
case name =>
if (message.isSealedOneofType) name + OneofMessageSuffix
else name
}
parent.fold(message.getFile().scalaPackage)(_.scalaType) / name
}
@deprecated("Use scalaType.fullName instead", "0.10.0")
def scalaTypeName: String = scalaType.fullName
private[compiler] def hasConflictingJavaClassName(className: String): Boolean =
((message.getName == className) ||
(message.getEnumTypes.asScala.exists(_.getName == className)) ||
(message.nestedTypes.exists(_.hasConflictingJavaClassName(className))))
def javaTypeName = message.getFile.fullJavaName(message.getFullName)
def messageOptions: MessageOptions = {
val localOptions = message.getOptions.getExtension[MessageOptions](Scalapb.message)
message.getFile.scalaOptions.getAuxMessageOptionsList.asScala
.find(_.getTarget == message.getFullName())
.fold(localOptions)(aux =>
MessageOptions.newBuilder(aux.getOptions).mergeFrom(localOptions).build
)
}
def noBox = message.messageOptions.getNoBox
private[this] def deprecatedAnnotation: Seq[String] = {
if (message.getOptions.getDeprecated)
List(ProtobufGenerator.deprecatedAnnotation)
else
Nil
}
def annotationList: Seq[String] = {
deprecatedAnnotation ++ messageOptions.getAnnotationsList().asScala
}
def companionAnnotationList: Seq[String] = {
deprecatedAnnotation ++ message.messageOptions.getCompanionAnnotationsList().asScala
}
def extendsOption = messageOptions.getExtendsList.asScala.filterNot(valueClassNames).toSeq
def companionExtendsOption = messageOptions.getCompanionExtendsList.asScala.toSeq
def sealedOneofExtendsOption = messageOptions.getSealedOneofExtendsList.asScala.toSeq
def sealedOneOfExtendsCount = messageOptions.getSealedOneofExtendsCount
def sealedOneofTraitScalaType: ScalaName = {
require(isSealedOneofType)
val name = scalaType.name.stripSuffix(OneofMessageSuffix)
parent.fold(message.getFile().scalaPackage)(_.scalaType) / name
}
def sealedOneofScalaType = {
sealedOneofStyle match {
case SealedOneofStyle.Optional =>
s"_root_.scala.Option[${sealedOneofTraitScalaType.fullName}]"
case _ => sealedOneofTraitScalaType.fullName
}
}
def sealedOneofCaseBases: List[String] = {
sealedOneofStyle match {
case SealedOneofStyle.Optional => List(sealedOneofTraitScalaType.fullName)
case _ => List(sealedOneofNonEmptyScalaType.fullName)
}
}
def sealedOneofNonEmptyScalaType: ScalaName = sealedOneofStyle match {
case SealedOneofStyle.Default => sealedOneofTraitScalaType / "NonEmpty"
case SealedOneofStyle.Optional => ???
}
def sealedOneofTypeMapper =
sealedOneofTraitScalaType / (sealedOneofTraitScalaType.name + "TypeMapper")
private[this] val valueClassNames = Set("AnyVal", "scala.AnyVal", "_root_.scala.AnyVal")
def isValueClass: Boolean = messageOptions.getExtendsList.asScala.exists(valueClassNames)
// In protobuf 3.5.0 all messages preserve unknown fields. We make an exception for
// value classes since they must have an exactly one val.
def preservesUnknownFields =
(
message.isExtendable || message.getFile.scalaOptions.getPreserveUnknownFields
) && !isValueClass
def sealedOneofContainer: Option[Descriptor] =
sealedOneofsCache.getContainer(message)
def sealedOneofCases: Option[Seq[Descriptor]] =
sealedOneofsCache.getCases(message)
def generateLenses: Boolean =
if (message.getFile.scalaOptions.hasLenses)
(message.getFile.scalaOptions.getLenses)
else params.lenses
def baseClasses: Seq[String] = {
val specialMixins = message.getFullName match {
case "google.protobuf.Any" => Seq("_root_.scalapb.AnyMethods")
case _ => Seq()
}
val updatable =
if (message.generateLenses) Seq(s"scalapb.lenses.Updatable[${scalaType.nameSymbol}]")
else Nil
val extendable =
if (message.isExtendable) Seq(s"_root_.scalapb.ExtendableMessage[${scalaType.nameSymbol}]")
else Nil
val anyVal = if (isValueClass) Seq("AnyVal") else Nil
val sealedOneofTrait = sealedOneofContainer match {
case Some(parent) => parent.sealedOneofCaseBases
case _ => List()
}
anyVal ++ Seq(
"scalapb.GeneratedMessage"
) ++ sealedOneofTrait ++ updatable ++ extendable ++ extendsOption ++ specialMixins
}
def companionBaseClasses: Seq[String] = {
val mixins =
if (javaConversions)
Seq(s"scalapb.JavaProtoSupport[${scalaType.fullName}, $javaTypeName]")
else Nil
val specialMixins = message.getFullName match {
case "google.protobuf.Any" => Seq("scalapb.AnyCompanionMethods")
case "google.protobuf.Timestamp" => Seq("scalapb.TimestampCompanionMethods")
case "google.protobuf.Duration" => Seq("scalapb.DurationCompanionMethods")
case _ => Seq()
}
Seq(s"scalapb.GeneratedMessageCompanion[${scalaType.fullName}]") ++
mixins ++
companionExtendsOption ++
specialMixins
}
def sealedOneofBaseClasses: Seq[String] = sealedOneofStyle match {
case SealedOneofStyle.Default =>
messageOptions.getSealedOneofExtendsList.asScala.toSeq :+ "scalapb.GeneratedSealedOneof"
case SealedOneofStyle.Optional => messageOptions.getSealedOneofExtendsList.asScala.toSeq
}
def nestedTypes: Seq[Descriptor] = message.getNestedTypes.asScala.toSeq
def isMapEntry: Boolean = message.getOptions.getMapEntry
def javaConversions = message.getFile.javaConversions && !isMapEntry
def isTopLevel = message.getContainingType == null
def scalaFileName: String =
if (message.getFile().scalaOptions.getSingleFile())
message.getFile().scalaFileName
else if (message.isSealedOneofType)
message.getFile.scalaDirectory + "/" + message.sealedOneofTraitScalaType.name + ".scala"
else message.getFile.scalaDirectory + "/" + message.scalaType.name + ".scala"
def messageCompanionInsertionPoint: InsertionPoint =
InsertionPoint(scalaFileName, s"GeneratedMessageCompanion[${message.getFullName}]")
class MapType {
def keyField = message.findFieldByName("key")
def keyType = keyField.singleScalaTypeName
def valueField = message.findFieldByName("value")
def valueType = valueField.singleScalaTypeName
def pairType = s"($keyType, $valueType)"
}
def mapType: MapType = {
assert(message.isMapEntry)
new MapType
}
def javaDescriptorSource: String =
if (message.isTopLevel)
s"${message.getFile.fileDescriptorObject.name}.javaDescriptor.getMessageTypes().get(${message.getIndex})"
else
s"${message.getContainingType.scalaType.fullName}.javaDescriptor.getNestedTypes().get(${message.getIndex})"
def scalaDescriptorSource: String =
if (message.isTopLevel)
s"${message.getFile.fileDescriptorObject.name}.scalaDescriptor.messages(${message.getIndex})"
else
s"${message.getContainingType.scalaType.fullName}.scalaDescriptor.nestedMessages(${message.getIndex})"
def sourcePath: Seq[Int] = {
if (message.isTopLevel) Seq(FileDescriptorProto.MESSAGE_TYPE_FIELD_NUMBER, message.getIndex)
else
message.getContainingType.sourcePath ++ Seq(
DescriptorProto.NESTED_TYPE_FIELD_NUMBER,
message.getIndex
)
}
def comment: Option[String] = {
message.getFile
.findLocationByPath(sourcePath)
.map(t => t.getLeadingComments + t.getTrailingComments)
.map(Helper.escapeComment)
.filter(_.nonEmpty)
}
}
implicit class EnumDescriptorPimp(val enumDescriptor: EnumDescriptor) {
def parentMessage: Option[Descriptor] = Option(enumDescriptor.getContainingType)
def scalaOptions: EnumOptions = {
val localOptions = enumDescriptor.getOptions.getExtension[EnumOptions](Scalapb.enumOptions)
enumDescriptor.getFile.scalaOptions.getAuxEnumOptionsList.asScala
.find(_.getTarget == enumDescriptor.getFullName())
.fold(localOptions)(aux =>
EnumOptions.newBuilder(aux.getOptions).mergeFrom(localOptions).build
)
}
lazy val scalaType: ScalaName = {
val name: String = enumDescriptor.getName match {
case "Option" => "OptionEnum"
case "ValueType" =>
"ValueTypeEnum" // Issue 348, conflicts with "type ValueType" in GeneratedEnumCompanion.
case n => n
}
parentMessage.fold(enumDescriptor.getFile().scalaPackage)(_.scalaType) / name
}
def recognizedEnum: ScalaName = scalaType / "Recognized"
def isTopLevel = enumDescriptor.getContainingType == null
def javaTypeName = enumDescriptor.getFile.fullJavaName(enumDescriptor.getFullName)
def javaConversions = enumDescriptor.getFile.javaConversions
def valuesWithNoDuplicates =
enumDescriptor.getValues.asScala
.groupBy(_.getNumber)
.map { case (_, v) => v.head }
.toVector
.sortBy(_.getNumber)
def javaDescriptorSource: String =
if (enumDescriptor.isTopLevel)
s"${enumDescriptor.getFile.fileDescriptorObject.name}.javaDescriptor.getEnumTypes().get(${enumDescriptor.getIndex})"
else
s"${enumDescriptor.getContainingType.scalaType.fullName}.javaDescriptor.getEnumTypes().get(${enumDescriptor.getIndex})"
def scalaDescriptorSource: String =
if (enumDescriptor.isTopLevel)
s"${enumDescriptor.getFile.fileDescriptorObject.name}.scalaDescriptor.enums(${enumDescriptor.getIndex})"
else
s"${enumDescriptor.getContainingType.scalaType.fullName}.scalaDescriptor.enums(${enumDescriptor.getIndex})"
def baseTraitExtends: Seq[String] =
"_root_.scalapb.GeneratedEnum" +: scalaOptions.getExtendsList.asScala.toSeq
def companionExtends: Seq[String] =
s"_root_.scalapb.GeneratedEnumCompanion[${scalaType.nameSymbol}]" +: scalaOptions.getCompanionExtendsList.asScala.toSeq
def sourcePath: Seq[Int] = {
if (enumDescriptor.isTopLevel)
Seq(FileDescriptorProto.ENUM_TYPE_FIELD_NUMBER, enumDescriptor.getIndex)
else
enumDescriptor.getContainingType.sourcePath ++ Seq(
DescriptorProto.ENUM_TYPE_FIELD_NUMBER,
enumDescriptor.getIndex
)
}
def comment: Option[String] = {
enumDescriptor.getFile
.findLocationByPath(sourcePath)
.map(t => t.getLeadingComments + t.getTrailingComments)
.map(Helper.escapeComment)
.filter(_.nonEmpty)
}
}
implicit class EnumValueDescriptorPimp(val enumValue: EnumValueDescriptor) {
def scalaOptions: EnumValueOptions =
enumValue.getOptions.getExtension[EnumValueOptions](Scalapb.enumValue)
def valueExtends: Seq[String] =
s"${enumValue.getType.scalaType.nameSymbol}(${enumValue.getNumber})" +: enumValue.getType.recognizedEnum
.nameRelative(1) +: scalaOptions.getExtendsList.asScala.toSeq
def scalaName: String =
if (scalaOptions.hasScalaName) scalaOptions.getScalaName
else {
val enumValueName: String =
if (enumValue.getFile.scalaOptions.getEnumStripPrefix) {
val enumName = enumValue.getType.getName
val commonPrefix = NameUtils.toAllCaps(enumName) + "_"
enumValue.getName.stripPrefix(commonPrefix)
} else enumValue.getName
if (enumValue.getFile.scalaOptions.getEnumValueNaming == EnumValueNaming.CAMEL_CASE)
allCapsToCamelCase(enumValueName, true)
else enumValueName
}
def isName = {
Helper.makeUniqueNames(
enumValue.getType.getValues.asScala
.sortBy(v => (v.getNumber, v.scalaName))
.map { e => e -> ("is" + allCapsToCamelCase(e.scalaName, true)) }
.toSeq
)(enumValue)
}
def sourcePath: Seq[Int] = {
enumValue.getType.sourcePath ++ Seq(
EnumDescriptorProto.VALUE_FIELD_NUMBER,
enumValue.getIndex
)
}
def comment: Option[String] = {
enumValue.getFile
.findLocationByPath(sourcePath)
.map(t => t.getLeadingComments + t.getTrailingComments)
.map(Helper.escapeComment)
.filter(_.nonEmpty)
}
}
implicit class FileDescriptorPimp(val file: FileDescriptor) {
def scalaOptions: ScalaPbOptions =
fileOptionsCache(file)
def javaConversions = params.javaConversions && !scalaOptions.getTestOnlyNoJavaConversions
def javaPackage: String = {
if (file.getOptions.hasJavaPackage)
file.getOptions.getJavaPackage
else file.getPackage
}
def javaPackageAsSymbol: String =
javaPackage.split('.').map(_.asSymbol).mkString(".")
private def hasConflictingJavaClassName(className: String): Boolean =
(file.getEnumTypes.asScala.exists(_.getName == className) ||
file.getServices.asScala.exists(_.getName == className) ||
file.getMessageTypes.asScala.exists(_.hasConflictingJavaClassName(className)))
// This method does not scan recursively. Currently it is used to determine whether the file
// companion object would conflict with any top-level generated class.
private def hasConflictingScalaClassName(str: String): Boolean =
file.getMessageTypes.asScala.exists(_.getName.toLowerCase == str.toLowerCase) ||
file.getEnumTypes.asScala.exists(_.getName.toLowerCase == str.toLowerCase) ||
file.getServices.asScala.exists(_.getName.toLowerCase == str.toLowerCase)
def javaOuterClassName: String =
if (file.getOptions.hasJavaOuterClassname)
file.getOptions.getJavaOuterClassname
else {
val r = NameUtils.snakeCaseToCamelCase(baseName(file.getName), true)
if (!hasConflictingJavaClassName(r)) r
else r + "OuterClass"
}
private def isNonFlatDependency =
(file.getPackage == "google.protobuf") || (file.getPackage == "scalapb")
private def scalaPackageParts: Seq[String] = {
val requestedPackageName: Seq[String] =
(if (scalaOptions.hasPackageName) scalaOptions.getPackageName.split('.')
else javaPackage.split('.')).toIndexedSeq.filterNot(_.isEmpty)
if (scalaOptions.getFlatPackage || (params.flatPackage && !isNonFlatDependency))
requestedPackageName
else requestedPackageName ++ baseName(file.getName).replace('-', '_').split('.')
}
def scalaPackage = ScalaName(scalaPackageParts.isEmpty, scalaPackageParts)
@deprecated("Use scalaPackage.fullName", "0.10.0")
def scalaPackageName = scalaPackage.fullName
def scalaFileName = scalaDirectory + s"/${fileDescriptorObject.name}.scala"
def scalaDirectory = {
scalaPackageParts.mkString("/")
}
def javaFullOuterClassName = {
val pkg = javaPackageAsSymbol
if (pkg.isEmpty) javaOuterClassName
else pkg + "." + javaOuterClassName
}
private def stripPackageName(fullName: String): String =
if (file.getPackage.isEmpty) fullName
else {
assert(fullName.startsWith(file.getPackage + "."))
fullName.substring(file.getPackage.size + 1)
}
def fullJavaName(fullName: String) = {
val base =
if (!file.getOptions.getJavaMultipleFiles)
(javaFullOuterClassName + ".")
else {
val pkg = javaPackageAsSymbol
if (pkg.isEmpty) "" else (pkg + ".")
}
base + stripPackageName(fullName).split('.').map(_.asSymbol).mkString(".")
}
def fileDescriptorObject = {
def inner(s: String): String =
if (!hasConflictingJavaClassName(s) && !hasConflictingScalaClassName(s)) s
else (s + "Companion")
val objectName =
if (file.scalaOptions.hasObjectName) file.scalaOptions.getObjectName
else
inner(
NameUtils.snakeCaseToCamelCase(baseName(file.getName) + "Proto", upperInitial = true)
)
scalaPackage / objectName
}
def isProto2 = file.getSyntax == FileDescriptor.Syntax.PROTO2
def isProto3 = file.getSyntax == FileDescriptor.Syntax.PROTO3
def findLocationByPath(path: Seq[Int]): Option[SourceCodeInfo.Location] = {
file.toProto.getSourceCodeInfo.getLocationList.asScala.find(_.getPathList.asScala == path)
}
def usePrimitiveWrappers: Boolean = !scalaOptions.getNoPrimitiveWrappers
def retainSourceCodeInfo: Boolean = {
if (scalaOptions.hasRetainSourceCodeInfo) scalaOptions.getRetainSourceCodeInfo
else params.retainSourceCodeInfo
}
def noDefaultValuesInConstructor: Boolean = scalaOptions.getNoDefaultValuesInConstructor
/** Returns a vector with all messages (both top-level and nested) in the file. */
def allMessages: Vector[Descriptor] = {
val messages = Vector.newBuilder[Descriptor]
def visitMessage(d: Descriptor): Unit = {
messages += d
d.getNestedTypes.asScala.foreach(visitMessage)
}
file.getMessageTypes.asScala.foreach(visitMessage)
messages.result()
}
}
private def allCapsToCamelCase(name: String, upperInitial: Boolean): String = {
val b = new StringBuilder()
@annotation.tailrec
def inner(name: String, capNext: Boolean): Unit = if (name.nonEmpty) {
val (r, capNext2) = name.head match {
case c if c.isUpper =>
// capitalize according to capNext.
(Some(if (capNext) c else c.toLower), false)
case c if c.isLower =>
// Lower caps never get capitalized, but will force
// the next letter to be upper case.
(Some(c), true)
case c if c.isDigit => (Some(c), true)
case _ => (None, true)
}
r.foreach(b.append)
inner(name.tail, capNext2)
}
inner(name, upperInitial)
b.toString
}
def baseName(fileName: String) =
fileName.split("/").last.replaceAll(raw"[.]proto$$|[.]protodevel", "")
}
object DescriptorImplicits {
val ScalaSeq = "_root_.scala.Seq"
val ScalaMap = "_root_.scala.collection.immutable.Map"
val ScalaVector = "_root_.scala.collection.immutable.Vector"
val ScalaIterable = "_root_.scala.collection.immutable.Iterable"
val ScalaOption = "_root_.scala.Option"
implicit class AsSymbolPimp(val s: String) {
def asSymbol: String =
if (SCALA_RESERVED_WORDS.contains(s) || s(0).isDigit) s"`$s`"
else s
}
val SCALA_RESERVED_WORDS = Set(
"abstract",
"case",
"catch",
"class",
"def",
"do",
"else",
"enum",
"extends",
"false",
"final",
"finally",
"for",
"forSome",
"if",
"implicit",
"import",
"lazy",
"macro",
"match",
"new",
"null",
"object",
"override",
"package",
"private",
"protected",
"return",
"sealed",
"super",
"then",
"this",
"throw",
"trait",
"try",
"true",
"type",
"val",
"var",
"while",
"with",
"yield",
"ne"
)
def primitiveWrapperType(messageType: Descriptor): Option[String] =
messageType.getFullName match {
case "google.protobuf.Int32Value" => Some("_root_.scala.Int")
case "google.protobuf.Int64Value" => Some("_root_.scala.Long")
case "google.protobuf.UInt32Value" => Some("_root_.scala.Int")
case "google.protobuf.UInt64Value" => Some("_root_.scala.Long")
case "google.protobuf.DoubleValue" => Some("_root_.scala.Double")
case "google.protobuf.FloatValue" => Some("_root_.scala.Float")
case "google.protobuf.StringValue" => Some("_root_.scala.Predef.String")
case "google.protobuf.BoolValue" => Some("_root_.scala.Boolean")
case "google.protobuf.BytesValue" => Some("_root_.com.google.protobuf.ByteString")
case _ => None
}
}
object Helper {
def makeUniqueNames[T](values: Seq[(T, String)]): Map[T, String] = {
val newNameMap: Map[String, T] =
values.foldLeft(Map.empty[String, T]) {
case (nameMap, (t, name)) =>
var newName: String = name
var attempt: Int = 0
while (nameMap.contains(newName)) {
attempt += 1
newName = s"${name}_$attempt"
}
nameMap + (newName -> t)
}
newNameMap.map(_.swap)
}
def escapeComment(s: String): String = {
s.replace("&", "&")
.replace("/*", "/*")
.replace("*/", "*/")
.replace("@", "@")
.replace("<", "<")
.replace(">", ">")
.replace("\\", "&92;")
}
}