org.jetbrains.kotlin.serialization.AnnotationSerializer.kt Maven / Gradle / Ivy
/*
* Copyright 2010-2015 JetBrains s.r.o.
*
* 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 org.jetbrains.kotlin.serialization
import org.jetbrains.kotlin.descriptors.ClassDescriptor
import org.jetbrains.kotlin.descriptors.annotations.AnnotationArgumentVisitor
import org.jetbrains.kotlin.descriptors.annotations.AnnotationDescriptor
import org.jetbrains.kotlin.resolve.constants.*
import org.jetbrains.kotlin.resolve.descriptorUtil.annotationClass
import org.jetbrains.kotlin.serialization.ProtoBuf.Annotation.Argument.Value
import org.jetbrains.kotlin.serialization.ProtoBuf.Annotation.Argument.Value.Type
import org.jetbrains.kotlin.types.ErrorUtils
class AnnotationSerializer(private val stringTable: StringTable) {
fun serializeAnnotation(annotation: AnnotationDescriptor): ProtoBuf.Annotation = ProtoBuf.Annotation.newBuilder().apply {
val annotationClass = annotation.annotationClass ?: error("Annotation type is not a class: ${annotation.type}")
if (ErrorUtils.isError(annotationClass)) {
error("Unresolved annotation type: ${annotation.type} at ${annotation.source.containingFile}")
}
id = stringTable.getFqNameIndex(annotationClass)
for ((parameter, value) in annotation.allValueArguments) {
val argument = ProtoBuf.Annotation.Argument.newBuilder()
argument.nameId = stringTable.getStringIndex(parameter.name.asString())
argument.setValue(valueProto(value))
addArgument(argument)
}
}.build()
fun valueProto(constant: ConstantValue<*>): Value.Builder = Value.newBuilder().apply {
constant.accept(object : AnnotationArgumentVisitor {
override fun visitAnnotationValue(value: AnnotationValue, data: Unit) {
type = Type.ANNOTATION
annotation = serializeAnnotation(value.value)
}
override fun visitArrayValue(value: ArrayValue, data: Unit) {
type = Type.ARRAY
for (element in value.value) {
addArrayElement(valueProto(element).build())
}
}
override fun visitBooleanValue(value: BooleanValue, data: Unit) {
type = Type.BOOLEAN
setIntValue(if (value.value) 1 else 0)
}
override fun visitByteValue(value: ByteValue, data: Unit) {
type = Type.BYTE
intValue = value.value.toLong()
}
override fun visitCharValue(value: CharValue, data: Unit) {
type = Type.CHAR
intValue = value.value.toLong()
}
override fun visitDoubleValue(value: DoubleValue, data: Unit) {
type = Type.DOUBLE
doubleValue = value.value
}
override fun visitEnumValue(value: EnumValue, data: Unit) {
type = Type.ENUM
val enumEntry = value.value
classId = stringTable.getFqNameIndex(enumEntry.containingDeclaration as ClassDescriptor)
enumValueId = stringTable.getStringIndex(enumEntry.name.asString())
}
override fun visitErrorValue(value: ErrorValue, data: Unit) {
throw UnsupportedOperationException("Error value: $value")
}
override fun visitFloatValue(value: FloatValue, data: Unit) {
type = Type.FLOAT
floatValue = value.value
}
override fun visitIntValue(value: IntValue, data: Unit) {
type = Type.INT
intValue = value.value.toLong()
}
override fun visitKClassValue(value: KClassValue?, data: Unit?) {
// TODO: support class literals
throw UnsupportedOperationException("Class literal annotation arguments are not yet supported: $value")
}
override fun visitLongValue(value: LongValue, data: Unit) {
type = Type.LONG
intValue = value.value
}
override fun visitNullValue(value: NullValue, data: Unit) {
throw UnsupportedOperationException("Null should not appear in annotation arguments")
}
override fun visitShortValue(value: ShortValue, data: Unit) {
type = Type.SHORT
intValue = value.value.toLong()
}
override fun visitStringValue(value: StringValue, data: Unit) {
type = Type.STRING
stringValue = stringTable.getStringIndex(value.value)
}
}, Unit)
}
}