zaaarf/lillero-processor
1
0
Fork 0
mirror of https://github.com/zaaarf/lillero-processor.git synced 2024-11-22 16:04:50 +01:00
Code Issues Projects Releases Packages Wiki Activity

feat: implemented generation of service provider

Browse source
This commit is contained in:
zaaarf 2023-02-24 21:54:39 +01:00
parent 211142cd59
commit aab3d58afb
No known key found for this signature in database
GPG key ID: AD8563472FD43386
1 changed files with 15 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
src/main/java/ftbsc/lll/processor/LilleroProcessor.java
View file

@ -6,6 +6,7 @@ import ftbsc.lll.processor.annotations.Injector;
import ftbsc.lll.processor.annotations.Patch; import ftbsc.lll.processor.annotations.Patch;
import ftbsc.lll.tools.DescriptorBuilder; import ftbsc.lll.tools.DescriptorBuilder;
import ftbsc.lll.tools.SrgMapper; import ftbsc.lll.tools.SrgMapper;
import org.gradle.internal.impldep.org.objectweb.asm.Type;
import org.objectweb.asm.tree.ClassNode; import org.objectweb.asm.tree.ClassNode;
import org.objectweb.asm.tree.MethodNode; import org.objectweb.asm.tree.MethodNode;
@ -17,7 +18,9 @@ import javax.lang.model.element.TypeElement;
import javax.lang.model.type.ExecutableType; import javax.lang.model.type.ExecutableType;
import javax.lang.model.type.TypeMirror; import javax.lang.model.type.TypeMirror;
import javax.tools.Diagnostic; import javax.tools.Diagnostic;
import javax.tools.FileObject;
import javax.tools.JavaFileObject; import javax.tools.JavaFileObject;
import javax.tools.StandardLocation;
import java.io.IOException; import java.io.IOException;
import java.io.PrintWriter; import java.io.PrintWriter;
import java.lang.annotation.Annotation; import java.lang.annotation.Annotation;
@ -109,7 +112,7 @@ public class LilleroProcessor extends AbstractProcessor {
return sb.toString(); return sb.toString();
} }
private String getSrgElementName(MethodSpec m) { private String getSrgMethodName(MethodSpec m) {
return m.name; //TODO; return m.name; //TODO;
} }
@ -141,13 +144,13 @@ public class LilleroProcessor extends AbstractProcessor {
MethodSpec targetClass = MethodSpec.methodBuilder("targetClass") MethodSpec targetClass = MethodSpec.methodBuilder("targetClass")
.addModifiers(Modifier.PUBLIC) .addModifiers(Modifier.PUBLIC)
.returns(String.class) .returns(String.class)
.addStatement("return $S", ann.value().getName()) .addStatement("return $S", mapper.getMcpClass(Type.getInternalName(ann.value())))
.build(); .build();
MethodSpec methodName = MethodSpec.methodBuilder("methodName") MethodSpec methodName = MethodSpec.methodBuilder("methodName")
.addModifiers(Modifier.PUBLIC) .addModifiers(Modifier.PUBLIC)
.returns(String.class) .returns(String.class)
.addStatement("return $S", getSrgElementName(targetMethod)) .addStatement("return $S", getSrgMethodName(targetMethod))
.build(); .build();
MethodSpec methodDesc = MethodSpec.methodBuilder("methodDesc") MethodSpec methodDesc = MethodSpec.methodBuilder("methodDesc")
@ -176,15 +179,20 @@ public class LilleroProcessor extends AbstractProcessor {
.build(); .build();
try { try {
JavaFileObject builderFile = processingEnv.getFiler().createSourceFile(injectorClassName); JavaFileObject injectorFile = processingEnv.getFiler().createSourceFile(injectorClassName);
PrintWriter out = new PrintWriter(builderFile.openWriter()); PrintWriter out = new PrintWriter(injectorFile.openWriter());
JavaFile javaFile = JavaFile.builder(packageName, injectorClass).build(); JavaFile javaFile = JavaFile.builder(packageName, injectorClass).build();
javaFile.writeTo(out); javaFile.writeTo(out);
out.close();
} catch(IOException e) {} } catch(IOException e) {}
} }
private void generateServiceProvider(Set<TypeElement> inj) { private void generateServiceProvider(Set<TypeElement> inj) {
//todo generate the services file try {
FileObject serviceProvider = processingEnv.getFiler().createResource(StandardLocation.SOURCE_OUTPUT, "", "ftbsc.lll.IInjector");
PrintWriter out = new PrintWriter(serviceProvider.openWriter());
inj.forEach(i -> out.println(i.getQualifiedName() + "Injector"));
out.close();
} catch(IOException e) {}
} }
} }