Only collect annotations with a name starting with KotlinScript

This commit is contained in:
Rodrigo B. de Oliveira
2018-06-14 11:44:58 +02:00
committed by Ilya Chernikov
parent 0adfa7b038
commit e24a9d08f2
@@ -19,12 +19,13 @@ private class TemplateAnnotationVisitor(val anns: ArrayList<BinAnnData> = arrayL
} }
private class TemplateClassVisitor(val annVisitor: TemplateAnnotationVisitor) : ClassVisitor(Opcodes.ASM5) { private class TemplateClassVisitor(val annVisitor: TemplateAnnotationVisitor) : ClassVisitor(Opcodes.ASM5) {
override fun visitAnnotation(desc: String, visible: Boolean): AnnotationVisitor { override fun visitAnnotation(desc: String, visible: Boolean): AnnotationVisitor? {
val shortName = Type.getType(desc).internalName.substringAfterLast("/") val shortName = Type.getType(desc).internalName.substringAfterLast("/")
if (shortName.startsWith("KotlinScript")) { if (shortName.startsWith("KotlinScript")) {
annVisitor.anns.add(BinAnnData(shortName)) annVisitor.anns.add(BinAnnData(shortName))
return annVisitor
} }
return annVisitor return null
} }
} }