forked from GeyserMC/Floodgate
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
14 changed files
with
403 additions
and
16 deletions.
There are no files selected for viewing
Empty file.
38 changes: 38 additions & 0 deletions
38
ap/src/main/java/org/geysermc/floodgate/ap/AutoBindProcessor.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,38 @@ | ||
| /* | ||
| * Copyright (c) 2019-2022 GeyserMC. http://geysermc.org | ||
| * | ||
| * Permission is hereby granted, free of charge, to any person obtaining a copy | ||
| * of this software and associated documentation files (the "Software"), to deal | ||
| * in the Software without restriction, including without limitation the rights | ||
| * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell | ||
| * copies of the Software, and to permit persons to whom the Software is | ||
| * furnished to do so, subject to the following conditions: | ||
| * | ||
| * The above copyright notice and this permission notice shall be included in | ||
| * all copies or substantial portions of the Software. | ||
| * | ||
| * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR | ||
| * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, | ||
| * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE | ||
| * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER | ||
| * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, | ||
| * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN | ||
| * THE SOFTWARE. | ||
| * | ||
| * @author GeyserMC | ||
| * @link https://github.com/GeyserMC/Floodgate | ||
| */ | ||
|
|
||
| package org.geysermc.floodgate.ap; | ||
|
|
||
| import javax.annotation.processing.SupportedAnnotationTypes; | ||
| import javax.annotation.processing.SupportedSourceVersion; | ||
| import javax.lang.model.SourceVersion; | ||
|
|
||
| @SupportedAnnotationTypes("*") | ||
| @SupportedSourceVersion(SourceVersion.RELEASE_8) | ||
| public class AutoBindProcessor extends ClassProcessor { | ||
| public AutoBindProcessor() { | ||
| super("org.geysermc.floodgate.util.AutoBind"); | ||
| } | ||
| } |
221 changes: 221 additions & 0 deletions
221
ap/src/main/java/org/geysermc/floodgate/ap/ClassProcessor.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,221 @@ | ||
| /* | ||
| * Copyright (c) 2019-2022 GeyserMC. http://geysermc.org | ||
| * | ||
| * Permission is hereby granted, free of charge, to any person obtaining a copy | ||
| * of this software and associated documentation files (the "Software"), to deal | ||
| * in the Software without restriction, including without limitation the rights | ||
| * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell | ||
| * copies of the Software, and to permit persons to whom the Software is | ||
| * furnished to do so, subject to the following conditions: | ||
| * | ||
| * The above copyright notice and this permission notice shall be included in | ||
| * all copies or substantial portions of the Software. | ||
| * | ||
| * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR | ||
| * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, | ||
| * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE | ||
| * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER | ||
| * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, | ||
| * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN | ||
| * THE SOFTWARE. | ||
| * | ||
| * @author GeyserMC | ||
| * @link https://github.com/GeyserMC/Floodgate | ||
| */ | ||
|
|
||
| package org.geysermc.floodgate.ap; | ||
|
|
||
| import java.io.BufferedReader; | ||
| import java.io.BufferedWriter; | ||
| import java.io.IOException; | ||
| import java.nio.file.Files; | ||
| import java.nio.file.NoSuchFileException; | ||
| import java.nio.file.Path; | ||
| import java.nio.file.Paths; | ||
| import java.util.Collection; | ||
| import java.util.HashSet; | ||
| import java.util.List; | ||
| import java.util.Set; | ||
| import javax.annotation.processing.AbstractProcessor; | ||
| import javax.annotation.processing.ProcessingEnvironment; | ||
| import javax.annotation.processing.RoundEnvironment; | ||
| import javax.lang.model.element.AnnotationMirror; | ||
| import javax.lang.model.element.Element; | ||
| import javax.lang.model.element.ElementKind; | ||
| import javax.lang.model.element.TypeElement; | ||
| import javax.tools.Diagnostic.Kind; | ||
| import javax.tools.FileObject; | ||
| import javax.tools.StandardLocation; | ||
|
|
||
| /* | ||
| * Copied from Geyser | ||
| */ | ||
| public class ClassProcessor extends AbstractProcessor { | ||
| private final String annotationClassName; | ||
|
|
||
| private Path outputPath; | ||
|
|
||
| private final Set<String> locations = new HashSet<>(); | ||
|
|
||
| public ClassProcessor(String annotationClassName) { | ||
| this.annotationClassName = annotationClassName; | ||
| } | ||
|
|
||
| @Override | ||
| public synchronized void init(ProcessingEnvironment processingEnv) { | ||
| super.init(processingEnv); | ||
|
|
||
| processingEnv.getMessager().printMessage( | ||
| Kind.NOTE, "Initializing processor " + annotationClassName | ||
| ); | ||
|
|
||
| String outputFile = processingEnv.getOptions().get("metadataOutputFile"); | ||
| if (outputFile != null && !outputFile.isEmpty()) { | ||
| outputPath = Paths.get(outputFile); | ||
| } | ||
| } | ||
|
|
||
| @Override | ||
| public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment roundEnv) { | ||
| if (roundEnv.processingOver()) { | ||
| if (!roundEnv.errorRaised()) { | ||
| complete(); | ||
| } | ||
|
|
||
| return false; | ||
| } | ||
|
|
||
| if (!contains(annotations, annotationClassName)) { | ||
| return false; | ||
| } | ||
|
|
||
| for (Element element : roundEnv.getRootElements()) { | ||
| if (element.getKind() != ElementKind.CLASS) { | ||
| continue; | ||
| } | ||
|
|
||
| if (!contains(element.getAnnotationMirrors(), annotationClassName)) { | ||
| continue; | ||
| } | ||
|
|
||
| TypeElement typeElement = (TypeElement) element; | ||
| locations.add(typeElement.getQualifiedName().toString()); | ||
| } | ||
| return false; | ||
| } | ||
|
|
||
| public boolean contains(Collection<? extends TypeElement> elements, String className) { | ||
| if (elements.isEmpty()) { | ||
| return false; | ||
| } | ||
|
|
||
| for (TypeElement element : elements) { | ||
| if (element.getQualifiedName().contentEquals(className)) { | ||
| return true; | ||
| } | ||
| } | ||
|
|
||
| return false; | ||
| } | ||
|
|
||
| public boolean contains(List<? extends AnnotationMirror> elements, String className) { | ||
| if (elements.isEmpty()) { | ||
| return false; | ||
| } | ||
|
|
||
| for (AnnotationMirror element : elements) { | ||
| if (element.getAnnotationType().toString().equals(className)) { | ||
| return true; | ||
| } | ||
| } | ||
|
|
||
| return false; | ||
| } | ||
|
|
||
| public void complete() { | ||
| // Read existing annotation list and verify each class still has this annotation | ||
| try (BufferedReader reader = createReader()) { | ||
| if (reader != null) { | ||
| reader.lines().forEach(canonicalName -> { | ||
| if (!locations.contains(canonicalName)) { | ||
|
|
||
| TypeElement element = | ||
| processingEnv.getElementUtils().getTypeElement(canonicalName); | ||
|
|
||
| if (element != null && element.getKind() == ElementKind.CLASS && | ||
| contains(element.getAnnotationMirrors(), annotationClassName)) { | ||
| locations.add(canonicalName); | ||
| } | ||
| } | ||
| }); | ||
| } | ||
| } catch (IOException e) { | ||
| e.printStackTrace(); | ||
| } | ||
|
|
||
| if (!locations.isEmpty()) { | ||
| try (BufferedWriter writer = createWriter()) { | ||
| for (String location : locations) { | ||
| writer.write(location); | ||
| writer.newLine(); | ||
| } | ||
| } catch (IOException ex) { | ||
| ex.printStackTrace(); | ||
| } | ||
| } else { | ||
| processingEnv.getMessager().printMessage(Kind.NOTE, | ||
| "Did not find any classes annotated with " + annotationClassName | ||
| ); | ||
| } | ||
|
|
||
| processingEnv.getMessager().printMessage( | ||
| Kind.NOTE, "Completed processing for " + annotationClassName | ||
| ); | ||
| } | ||
|
|
||
| private BufferedReader createReader() throws IOException { | ||
| if (outputPath != null) { | ||
| processingEnv.getMessager().printMessage(Kind.NOTE, | ||
| "Reading existing " + annotationClassName + " list from " + outputPath | ||
| ); | ||
|
|
||
| return Files.newBufferedReader(outputPath); | ||
| } | ||
|
|
||
| FileObject obj = processingEnv.getFiler().getResource( | ||
| StandardLocation.CLASS_OUTPUT, "", annotationClassName | ||
| ); | ||
|
|
||
| if (obj != null) { | ||
| processingEnv.getMessager().printMessage( | ||
| Kind.NOTE, | ||
| "Reading existing " + annotationClassName + " list from " + obj.toUri() | ||
| ); | ||
|
|
||
| try { | ||
| return new BufferedReader(obj.openReader(false)); | ||
| } catch (NoSuchFileException ignored) {} | ||
| } | ||
| return null; | ||
| } | ||
|
|
||
| private BufferedWriter createWriter() throws IOException { | ||
| if (outputPath != null) { | ||
| processingEnv.getMessager().printMessage( | ||
| Kind.NOTE, "Writing " + annotationClassName + " to " + outputPath | ||
| ); | ||
|
|
||
| return Files.newBufferedWriter(outputPath); | ||
| } | ||
|
|
||
| FileObject obj = processingEnv.getFiler().createResource( | ||
| StandardLocation.CLASS_OUTPUT, "", annotationClassName | ||
| ); | ||
|
|
||
| processingEnv.getMessager().printMessage( | ||
| Kind.NOTE, "Writing " + annotationClassName + " to " + obj.toUri() | ||
| ); | ||
|
|
||
| return new BufferedWriter(obj.openWriter()); | ||
| } | ||
| } |
1 change: 1 addition & 0 deletions
1
ap/src/main/resources/META-INF/services/javax.annotation.processing.Processor
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1 @@ | ||
| org.geysermc.floodgate.ap.AutoBindProcessor |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
39 changes: 39 additions & 0 deletions
39
core/src/main/java/org/geysermc/floodgate/module/AutoBindModule.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,39 @@ | ||
| /* | ||
| * Copyright (c) 2019-2022 GeyserMC. http://geysermc.org | ||
| * | ||
| * Permission is hereby granted, free of charge, to any person obtaining a copy | ||
| * of this software and associated documentation files (the "Software"), to deal | ||
| * in the Software without restriction, including without limitation the rights | ||
| * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell | ||
| * copies of the Software, and to permit persons to whom the Software is | ||
| * furnished to do so, subject to the following conditions: | ||
| * | ||
| * The above copyright notice and this permission notice shall be included in | ||
| * all copies or substantial portions of the Software. | ||
| * | ||
| * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR | ||
| * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, | ||
| * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE | ||
| * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER | ||
| * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, | ||
| * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN | ||
| * THE SOFTWARE. | ||
| * | ||
| * @author GeyserMC | ||
| * @link https://github.com/GeyserMC/Floodgate | ||
| */ | ||
|
|
||
| package org.geysermc.floodgate.module; | ||
|
|
||
| import com.google.inject.AbstractModule; | ||
| import org.geysermc.floodgate.util.AutoBind; | ||
| import org.geysermc.floodgate.util.Utils; | ||
|
|
||
| public class AutoBindModule extends AbstractModule { | ||
| @Override | ||
| protected void configure() { | ||
| for (Class<?> clazz : Utils.getGeneratedClassesForAnnotation(AutoBind.class)) { | ||
| bind(clazz).asEagerSingleton(); | ||
| } | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.