chore: try add namespace sort

LiteLDev · Oct 9, 2024 · a2c6e56 · a2c6e56
1 parent 5f155c5
commit a2c6e56
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/main/kotlin/com/liteldev/headeroutput/entity/BaseType.kt b/src/main/kotlin/com/liteldev/headeroutput/entity/BaseType.kt
@@ -26,6 +26,7 @@ abstract class BaseType(
     val forwardDeclareList: MutableSet<String> = mutableSetOf()
 
     val simpleName = name.substringAfterLast("::")
+    val namespace = name.substringBeforeLast("::")
     val fullEscapeName = name.replace("::", "_")
     val fullUpperEscapeName = fullEscapeName.uppercase(Locale.getDefault())
 
@@ -56,6 +57,17 @@ abstract class BaseType(
                     ?.let {
                         return@run "$root/${it.dst}/${this.simpleName}.$HEADER_SUFFIX"
                     }
+                val parentRules = GeneratorConfig.getSortRules().namespace
+                parentRules.find { this.namespace.startsWith(it.namespace) }
+                    ?.let {
+                        return@run "$root/${it.dst}/${this.simpleName}.$HEADER_SUFFIX"
+                    }
+            } else if (this is NamespaceType) {
+                val parentRules = GeneratorConfig.getSortRules().namespace
+                parentRules.find { this.name.startsWith(it.namespace) }
+                    ?.let {
+                        return@run "$root/${it.dst}/${this.simpleName}.$HEADER_SUFFIX"
+                    }
             }
             regexRules.filter { !it.override }.find { this.name.matches(it.regex.toRegex()) }?.let {
                 return@run "$root/${it.dst}/${this.simpleName}.$HEADER_SUFFIX"