From 351509441845aef45d10171e6ce8b7f919b70a58 Mon Sep 17 00:00:00 2001
From: Konrad 'ktoso' Malawski <ktoso@apple.com>
Date: Wed, 2 Jul 2025 12:08:20 +0900
Subject: [PATCH] wrap-java: add option to match java package style directory
 nesting

---
 Sources/SwiftJavaTool/Commands/WrapJavaCommand.swift | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/Sources/SwiftJavaTool/Commands/WrapJavaCommand.swift b/Sources/SwiftJavaTool/Commands/WrapJavaCommand.swift
index 349144ab..05ac7651 100644
--- a/Sources/SwiftJavaTool/Commands/WrapJavaCommand.swift
+++ b/Sources/SwiftJavaTool/Commands/WrapJavaCommand.swift
@@ -52,6 +52,9 @@ extension SwiftJava {
     @Option(help: "Cache directory for intermediate results and other outputs between runs")
     var cacheDirectory: String?
 
+    @Option(help: "Match java package directory structure with generated Swift files")
+    var swiftMatchPackageDirectoryStructure: Bool = false
+
     @Argument(help: "Path to .jar file whose Java classes should be wrapped using Swift bindings")
     var input: String
   }
@@ -229,11 +232,16 @@ extension SwiftJava.WrapJavaCommand {
 
                           """
 
+      var generatedFileOutputDir = self.actualOutputDirectory
+      if self.swiftMatchPackageDirectoryStructure {
+        generatedFileOutputDir?.append(path: javaClass.getPackageName().replacing(".", with: "/"))
+      }
+
       let swiftFileName = try! translator.getSwiftTypeName(javaClass, preferValueTypes: false)
         .swiftName.replacing(".", with: "+") + ".swift"
       try writeContents(
         swiftFileText,
-        outputDirectory: self.actualOutputDirectory,
+        outputDirectory: generatedFileOutputDir,
         to: swiftFileName,
         description: "Java class '\(javaClass.getName())' translation"
       )