Handle subnamespaces when mapping namespaces to Java packages

2 files changed, 23 insertions, 7 deletions

diff --git a/src/main/java/org/zwobble/hobgoblin/compiler/output/generators/java/JavaGenerator.java b/src/main/java/org/zwobble/hobgoblin/compiler/output/generators/java/JavaGenerator.java
index 617d438..e68acaa 100644
--- a/src/main/java/org/zwobble/hobgoblin/compiler/output/generators/java/JavaGenerator.java
+++ b/src/main/java/org/zwobble/hobgoblin/compiler/output/generators/java/JavaGenerator.java
@@ -34,14 +34,22 @@ public class JavaGenerator {
     }
 
     public List<String> namespaceToJavaPackageParts(NamespaceName namespaceName) {
-        // TODO: handle subpackages
-        return this.config.namespaceConfig(namespaceName)
-            .map(namespaceConfig -> namespaceConfig.packageName())
-            .orElseGet(() -> {
-                var packageParts = new ArrayList<>(this.config.packageName());
-                packageParts.addAll(namespaceName.parts());
+        var ancestorNamespaceName = namespaceName;
+        var remainingNamespaceNameParts = new ArrayList<String>();
+        while (!ancestorNamespaceName.isRoot()) {
+            var namespaceConfig = this.config.namespaceConfig(ancestorNamespaceName);
+            if (namespaceConfig.isPresent()) {
+                var packageParts = new ArrayList<>(namespaceConfig.get().packageName());
+                packageParts.addAll(remainingNamespaceNameParts.reversed());
                 return packageParts;
-            });
+            }
+            remainingNamespaceNameParts.add(ancestorNamespaceName.parts().getLast());
+            ancestorNamespaceName = ancestorNamespaceName.parent();
+        }
+
+        var packageParts = new ArrayList<>(this.config.packageName());
+        packageParts.addAll(namespaceName.parts());
+        return packageParts;
     }
 
     public JavaTypeRef generateTypeRef(Type type) {
diff --git a/src/main/java/org/zwobble/hobgoblin/compiler/types/NamespaceName.java b/src/main/java/org/zwobble/hobgoblin/compiler/types/NamespaceName.java
index e0fb2d3..2cb3497 100644
--- a/src/main/java/org/zwobble/hobgoblin/compiler/types/NamespaceName.java
+++ b/src/main/java/org/zwobble/hobgoblin/compiler/types/NamespaceName.java
@@ -8,6 +8,14 @@ public record NamespaceName(List<String> parts) {
         return new NamespaceName(Arrays.asList(parts));
     }
 
+    public NamespaceName parent() {
+        return new NamespaceName(parts.subList(0, parts.size() - 1));
+    }
+
+    public boolean isRoot() {
+        return parts.isEmpty();
+    }
+
     @Override
     public String toString() {
         return String.join("/", parts);