96 lines
2.9 KiB
Java
96 lines
2.9 KiB
Java
|
//: annotations/ifx/IfaceExtractorProcessor.java
|
||
|
|
// javac-based annotation processing.
|
||
|
|
package annotations.ifx;
|
||
|
|
import javax.annotation.processing.*;
|
||
|
|
import javax.lang.model.SourceVersion;
|
||
|
|
import javax.lang.model.element.*;
|
||
|
|
import javax.lang.model.util.*;
|
||
|
|
import java.util.*;
|
||
|
|
import java.io.*;
|
||
|
|
|
||
|
|
@SupportedAnnotationTypes(
|
||
|
|
"annotations.ifx.ExtractInterface")
|
||
|
|
@SupportedSourceVersion(SourceVersion.RELEASE_8)
|
||
|
|
public class IfaceExtractorProcessor
|
||
|
|
extends AbstractProcessor {
|
||
|
|
private ArrayList<Element>
|
||
|
|
interfaceMethods = new ArrayList<>();
|
||
|
|
Types typeUtils;
|
||
|
|
Elements elementUtils;
|
||
|
|
|
||
|
|
private ProcessingEnvironment processingEnv;
|
||
|
|
@Override public void
|
||
|
|
init(ProcessingEnvironment processingEnv) {
|
||
|
|
this.processingEnv = processingEnv;
|
||
|
|
typeUtils = processingEnv.getTypeUtils();
|
||
|
|
elementUtils = processingEnv.getElementUtils();
|
||
|
|
}
|
||
|
|
@Override public boolean
|
||
|
|
process(Set<? extends TypeElement> annotations,
|
||
|
|
RoundEnvironment env) {
|
||
|
|
for(Element elem:env.getElementsAnnotatedWith(
|
||
|
|
ExtractInterface.class)) {
|
||
|
|
String interfaceName = elem.getAnnotation(
|
||
|
|
ExtractInterface.class).interfaceName();
|
||
|
|
for(Element enclosed :
|
||
|
|
elem.getEnclosedElements()) {
|
||
|
|
if(enclosed.getKind()
|
||
|
|
.equals(ElementKind.METHOD) &&
|
||
|
|
enclosed.getModifiers()
|
||
|
|
.contains(Modifier.PUBLIC) &&
|
||
|
|
!enclosed.getModifiers()
|
||
|
|
.contains(Modifier.STATIC)) {
|
||
|
|
interfaceMethods.add(enclosed);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if(interfaceMethods.size() > 0) {
|
||
|
|
writeInterfaceFile(interfaceName);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
private String createArgList(
|
||
|
|
List<? extends VariableElement> parameters) {
|
||
|
|
if(parameters.size() == 0)
|
||
|
|
return "()";
|
||
|
|
String args = "(";
|
||
|
|
for(VariableElement p : parameters) {
|
||
|
|
args += p.asType() + " ";
|
||
|
|
args += p.getSimpleName() + ", ";
|
||
|
|
}
|
||
|
|
args = args.substring(0, args.length() - 2);
|
||
|
|
args += ")";
|
||
|
|
return args;
|
||
|
|
}
|
||
|
|
private void
|
||
|
|
writeInterfaceFile(String interfaceName) {
|
||
|
|
try {
|
||
|
|
try(Writer writer = processingEnv.getFiler()
|
||
|
|
.createSourceFile(interfaceName)
|
||
|
|
.openWriter()) {
|
||
|
|
String packageName = elementUtils
|
||
|
|
.getPackageOf(interfaceMethods
|
||
|
|
.get(0)).toString();
|
||
|
|
writer.write(
|
||
|
|
"package " + packageName + ";\n");
|
||
|
|
writer.write("public interface " +
|
||
|
|
interfaceName + " {\n");
|
||
|
|
for(Element elem : interfaceMethods) {
|
||
|
|
ExecutableElement method =
|
||
|
|
(ExecutableElement)elem;
|
||
|
|
String signature = " public ";
|
||
|
|
signature += method.getReturnType()+" ";
|
||
|
|
signature += method.getSimpleName();
|
||
|
|
signature += createArgList(
|
||
|
|
method.getParameters());
|
||
|
|
System.out.println(signature);
|
||
|
|
writer.write(signature + ";\n");
|
||
|
|
}
|
||
|
|
writer.write("}");
|
||
|
|
}
|
||
|
|
} catch(Exception e) {
|
||
|
|
throw new RuntimeException(e);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
} ///:~
|