Extract JavaGeneratorConfig.nativeTypeConfig()

2 files changed, 4 insertions, 1 deletions

diff --git a/src/main/java/org/zwobble/hobgoblin/compiler/output/generators/java/JavaGeneratorConfig.java b/src/main/java/org/zwobble/hobgoblin/compiler/output/generators/java/JavaGeneratorConfig.java
index 92fe951..b7bd6c5 100644
--- a/src/main/java/org/zwobble/hobgoblin/compiler/output/generators/java/JavaGeneratorConfig.java
+++ b/src/main/java/org/zwobble/hobgoblin/compiler/output/generators/java/JavaGeneratorConfig.java
@@ -74,4 +74,7 @@ public record JavaGeneratorConfig(
         Map.entry(NativeTypes.LIST_INNER, JavaNativeTypeConfig.of(JavaTypeRef.LIST, new JavaStaticMethodCall(JavaTypeRef.LIST, "of", List.of())))
     );
 
+    public Optional<JavaNativeTypeConfig> nativeTypeConfig(SimpleNativeType type) {
+        return Optional.ofNullable(this.nativeTypeConfigs.get(type));
+    }
 }
diff --git a/src/main/java/org/zwobble/hobgoblin/compiler/output/generators/javatypes/JavaTypesGenerator.java b/src/main/java/org/zwobble/hobgoblin/compiler/output/generators/javatypes/JavaTypesGenerator.java
index a3532be..ab156da 100644
--- a/src/main/java/org/zwobble/hobgoblin/compiler/output/generators/javatypes/JavaTypesGenerator.java
+++ b/src/main/java/org/zwobble/hobgoblin/compiler/output/generators/javatypes/JavaTypesGenerator.java
@@ -411,7 +411,7 @@ public class JavaTypesGenerator implements Generator {
     }
 
     private Optional<JavaNativeTypeConfig> nativeTypeConfig(SimpleNativeType type) {
-        return Optional.ofNullable(this.javaGeneratorConfig.nativeTypeConfigs().get(type));
+        return this.javaGeneratorConfig.nativeTypeConfig(type);
     }
 
     private static class Context {