io.micronaut.inject.writer.ArrayAwareSignatureWriter Maven / Gradle / Ivy
/*
* Copyright 2017-2021 original authors
*
* 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
*
* https://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 io.micronaut.inject.writer;
import io.micronaut.core.annotation.Internal;
import org.objectweb.asm.Opcodes;
import org.objectweb.asm.signature.SignatureVisitor;
/**
* Fork of {@link org.objectweb.asm.signature.SignatureWriter} which doesn't handle primitive array generics correctly.
*
* @author graemerocher
* @since 3.1.0
*/
@Internal
final class ArrayAwareSignatureWriter extends SignatureVisitor {
/** The builder used to construct the visited signature. */
private final StringBuilder stringBuilder = new StringBuilder();
/** Whether the visited signature contains formal type parameters. */
private boolean hasFormals;
/** Whether the visited signature contains method parameter types. */
private boolean hasParameters;
/**
* The stack used to keep track of class types that have arguments. Each element of this stack is
* a boolean encoded in one bit. The top of the stack is the least significant bit. Pushing false
* = *2, pushing true = *2+1, popping = /2.
*
* Class type arguments must be surrounded with '<' and '>' and, because
*
*
* - class types can be nested (because type arguments can themselves be class types),
*
- SignatureWriter always returns 'this' in each visit* method (to avoid allocating new
* SignatureWriter instances),
*
*
* we need a stack to properly balance these 'parentheses'. A new element is pushed on this
* stack for each new visited type, and popped when the visit of this type ends (either is
* visitEnd, or because visitInnerClassType is called).
*/
private int argumentStack;
/** Constructs a new {@link org.objectweb.asm.signature.SignatureWriter}. */
public ArrayAwareSignatureWriter() {
super(Opcodes.ASM7);
}
// -----------------------------------------------------------------------------------------------
// Implementation of the SignatureVisitor interface
// -----------------------------------------------------------------------------------------------
@Override
public void visitFormalTypeParameter(final String name) {
if (!hasFormals) {
hasFormals = true;
stringBuilder.append('<');
}
stringBuilder.append(name);
stringBuilder.append(':');
}
@Override
public SignatureVisitor visitClassBound() {
return this;
}
@Override
public SignatureVisitor visitInterfaceBound() {
stringBuilder.append(':');
return this;
}
@Override
public SignatureVisitor visitSuperclass() {
endFormals();
return this;
}
@Override
public SignatureVisitor visitInterface() {
return this;
}
@Override
public SignatureVisitor visitParameterType() {
endFormals();
if (!hasParameters) {
hasParameters = true;
stringBuilder.append('(');
}
return this;
}
@Override
public SignatureVisitor visitReturnType() {
endFormals();
if (!hasParameters) {
stringBuilder.append('(');
}
stringBuilder.append(')');
return this;
}
@Override
public SignatureVisitor visitExceptionType() {
stringBuilder.append('^');
return this;
}
@Override
public void visitBaseType(final char descriptor) {
stringBuilder.append(descriptor);
// Pushes 'false' on the stack, meaning that this type does not have type arguments (as far as
// we can tell at this point).
argumentStack *= 2;
}
@Override
public void visitTypeVariable(final String name) {
stringBuilder.append('T');
stringBuilder.append(name);
stringBuilder.append(';');
}
@Override
public SignatureVisitor visitArrayType() {
stringBuilder.append('[');
return this;
}
@Override
public void visitClassType(final String name) {
stringBuilder.append('L');
stringBuilder.append(name);
// Pushes 'false' on the stack, meaning that this type does not have type arguments (as far as
// we can tell at this point).
argumentStack *= 2;
}
@Override
public void visitInnerClassType(final String name) {
endArguments();
stringBuilder.append('.');
stringBuilder.append(name);
// Pushes 'false' on the stack, meaning that this type does not have type arguments (as far as
// we can tell at this point).
argumentStack *= 2;
}
@Override
public void visitTypeArgument() {
// If the top of the stack is 'false', this means we are visiting the first type argument of the
// currently visited type. We therefore need to append a '<', and to replace the top stack
// element with 'true' (meaning that the current type does have type arguments).
if (argumentStack % 2 == 0) {
argumentStack |= 1;
stringBuilder.append('<');
}
stringBuilder.append('*');
}
@Override
public ArrayAwareSignatureWriter visitTypeArgument(final char wildcard) {
// If the top of the stack is 'false', this means we are visiting the first type argument of the
// currently visited type. We therefore need to append a '<', and to replace the top stack
// element with 'true' (meaning that the current type does have type arguments).
if (argumentStack % 2 == 0) {
argumentStack |= 1;
stringBuilder.append('<');
}
if (wildcard != '=') {
stringBuilder.append(wildcard);
}
return this;
}
/**
* Ends an array.
*/
public void visitEndArray() {
endArguments();
}
@Override
public void visitEnd() {
endArguments();
stringBuilder.append(';');
}
/**
* Returns the signature that was built by this signature writer.
*
* @return the signature that was built by this signature writer.
*/
@Override
public String toString() {
return stringBuilder.toString();
}
// -----------------------------------------------------------------------------------------------
// Utility methods
// -----------------------------------------------------------------------------------------------
/** Ends the formal type parameters section of the signature. */
private void endFormals() {
if (hasFormals) {
hasFormals = false;
stringBuilder.append('>');
}
}
/** Ends the type arguments of a class or inner class type. */
private void endArguments() {
// If the top of the stack is 'true', this means that some type arguments have been visited for
// the type whose visit is now ending. We therefore need to append a '>', and to pop one element
// from the stack.
if (argumentStack % 2 == 1) {
stringBuilder.append('>');
}
argumentStack /= 2;
}
}