allow to handle conflict with class loading and evaluate values lazily

Author: Lulu13022002

src/main/java/io/papermc/typewriter/preset/EnumRewriter.java

@@ -1,7 +1,6 @@
 package io.papermc.typewriter.preset;
 
 import com.google.common.base.Preconditions;
-import io.papermc.typewriter.SourceFile;
 import io.papermc.typewriter.preset.model.EnumValue;
 import io.papermc.typewriter.replace.SearchMetadata;
 import io.papermc.typewriter.replace.SearchReplaceRewriter;
@@ -37,28 +36,27 @@ protected void appendEnumValue(T item, StringBuilder builder, String indent, boo
         builder.append('\n');
     }
 
-    @Override
-    public boolean registerFor(SourceFile file) {
-        boolean canRegister = super.registerFor(file);
-        if (canRegister) {
-            this.values = this.getValues().iterator();
-        }
-        return canRegister;
-    }
-
     private boolean canReachEnd(SearchMetadata metadata) {
         // the default behavior might fail for simple enum or with trailing comments that's why a setting exists
         return Objects.requireNonNullElseGet(this.reachEnd, () -> metadata.replacedContent().stripTrailing().endsWith(";"));
     }
 
+    private void populateValues() {
+        if (this.values == null) {
+            this.values = this.getValues().iterator();
+        }
+    }
+
     @Override
     protected void replaceLine(SearchMetadata metadata, StringBuilder builder) {
+        this.populateValues();
         Preconditions.checkState(this.values.hasNext(), "Enum size doesn't match between generated values and replaced values.");
         appendEnumValue(this.values.next(), builder, metadata.indent(), this.canReachEnd(metadata));
     }
 
     @Override
     protected void insert(SearchMetadata metadata, StringBuilder builder) {
+        this.populateValues();
         boolean reachEnd = this.canReachEnd(metadata);
 
         while (this.values.hasNext()) {

src/main/java/io/papermc/typewriter/preset/SwitchCaseRewriter.java

@@ -14,13 +14,10 @@ public abstract class SwitchCaseRewriter extends SearchReplaceRewriter {
 
     protected abstract Iterable<String> getCases();
 
-    @Override
-    public boolean registerFor(SourceFile file) {
-        boolean canRegister = super.registerFor(file);
-        if (canRegister) {
+    private void populateCases() {
+        if (this.cases == null) {
             this.cases = this.getCases().iterator();
         }
-        return canRegister;
     }
 
     private void appendCase(StringBuilder builder, SearchMetadata metadata) {
@@ -30,12 +27,14 @@ private void appendCase(StringBuilder builder, SearchMetadata metadata) {
 
     @Override
     protected void replaceLine(SearchMetadata metadata, StringBuilder builder) {
+        this.populateCases();
         Preconditions.checkState(this.cases.hasNext(), "Switch case size doesn't match between generated values and replaced values.");
         appendCase(builder, metadata);
     }
 
     @Override
     protected void insert(SearchMetadata metadata, StringBuilder builder) {
+        this.populateCases();
         while (this.cases.hasNext()) {
             appendCase(builder, metadata);
         }

src/main/java/io/papermc/typewriter/util/ClassNamedView.java

@@ -71,8 +71,35 @@ public Optional<ClassNamed> tryFindFirst(String name) {
         return this.findFirstFile(name).map(SourceFile::mainClass);
     }
 
+    public Stream<ClassNamed> find(String name) {
+        int dotIndex = name.indexOf('.');
+        final String fileName;
+        @Nullable String nestedClassNames;
+        if (dotIndex != -1) { // take in account nested names
+            fileName = name.substring(0, dotIndex);
+            nestedClassNames = name.substring(dotIndex + 1);
+        } else {
+            fileName = name;
+            nestedClassNames = null;
+        }
+
+        return this.findFile(fileName)
+            .map(file -> {
+            if (nestedClassNames != null) {
+                ClassNamed clazz = file.mainClass();
+                return ClassNamed.of(
+                    clazz.packageName(),
+                    clazz.simpleName(),
+                    nestedClassNames
+                );
+            }
+
+            return file.mainClass();
+        });
+    }
+
     public Optional<SourceFile> findFirstFile(String name) {
-        try (Stream<SourceFile> stream = this.findFile(name)) { // handle conflict
+        try (Stream<SourceFile> stream = this.findFile(name)) {
             return stream.findFirst();
         }
     }