Fix module naming

Author: glbrntt

Sources/protoc-gen-grpc-swift-2/Options.swift

@@ -62,6 +62,8 @@ struct GeneratorOptions {
   }
 
   init(pairs: [(key: String, value: String)]) throws {
+    var moduleMapPath: String?
+
     for pair in pairs {
       switch pair.key {
       case "Visibility":
@@ -87,14 +89,7 @@ struct GeneratorOptions {
 
       case "ProtoPathModuleMappings":
         if !pair.value.isEmpty {
-          do {
-            self.protoToModuleMappings = try ProtoFileToModuleMappings(path: pair.value)
-          } catch let e {
-            throw GenerationError.wrappedError(
-              message: "Parameter 'ProtoPathModuleMappings=\(pair.value)'",
-              error: e
-            )
-          }
+            moduleMapPath = pair.value
         }
 
       case "FileNaming":
@@ -164,6 +159,24 @@ struct GeneratorOptions {
         throw GenerationError.unknownParameter(name: pair.key)
       }
     }
+
+    if let moduleMapPath = moduleMapPath {
+      do {
+        self.protoToModuleMappings = try ProtoFileToModuleMappings(
+          path: moduleMapPath,
+          swiftProtobufModuleName: self.config.moduleNames.swiftProtobuf
+        )
+      } catch let e {
+        throw GenerationError.wrappedError(
+          message: "Parameter 'ProtoPathModuleMappings=\(moduleMapPath)'",
+          error: e
+        )
+      }
+    } else {
+      self.protoToModuleMappings = ProtoFileToModuleMappings(
+        swiftProtobufModuleName: self.config.moduleNames.swiftProtobuf
+      )
+    }
   }
 
   static func parseParameter(string: String?) -> [(key: String, value: String)] {

Tests/GRPCProtobufCodeGenTests/ProtobufCodeGenParserTests.swift

@@ -265,5 +265,37 @@ struct ProtobufCodeGenParserTests {
       ]
       #expect(codeGen.dependencies == expected)
     }
+
+    @Test("Generate with different module names")
+    @available(gRPCSwiftProtobuf 2.0, *)
+    func generateWithDifferentModuleNames() throws {
+      var config = ProtobufCodeGenerator.Config.defaults
+      let defaultNames = config.moduleNames
+
+      config.accessLevel = .public
+      config.indentation = 2
+      config.moduleNames.grpcCore = String(config.moduleNames.grpcCore.reversed())
+      config.moduleNames.grpcProtobuf = String(config.moduleNames.grpcProtobuf.reversed())
+      config.moduleNames.swiftProtobuf = String(config.moduleNames.swiftProtobuf.reversed())
+
+      let generator = ProtobufCodeGenerator(config: config)
+      let generated = try generator.generateCode(
+        fileDescriptor: Self.fileDescriptor,
+        protoFileModuleMappings: ProtoFileToModuleMappings(
+          swiftProtobufModuleName: config.moduleNames.swiftProtobuf
+        ),
+        extraModuleImports: []
+      )
+
+      // Mustn't contain the default names.
+      #expect(!generated.contains(defaultNames.grpcCore))
+      #expect(!generated.contains(defaultNames.grpcProtobuf))
+      #expect(!generated.contains(defaultNames.swiftProtobuf))
+
+      // Must contain the configured names.
+      #expect(generated.contains(config.moduleNames.grpcCore))
+      #expect(generated.contains(config.moduleNames.grpcProtobuf))
+      #expect(generated.contains(config.moduleNames.swiftProtobuf))
+    }
   }
 }

Tests/GRPCProtobufCodeGenTests/ProtobufCodeGeneratorTests.swift

@@ -99,7 +99,6 @@ struct ProtobufCodeGeneratorTests {
 
         import GRPCCore
         import GRPCProtobuf
-        import SwiftProtobuf
 
         // MARK: - test.TestService
 
@@ -1103,35 +1102,6 @@ struct ProtobufCodeGeneratorTests {
       #expect(generated == expected)
     }
 
-    @Test("Generate with different module names")
-    @available(gRPCSwiftProtobuf 2.0, *)
-    func generateWithDifferentModuleNames() throws {
-      var config = ProtobufCodeGenerator.Config.defaults
-      let defaultNames = config.moduleNames
-
-      config.accessLevel = .public
-      config.indentation = 2
-      config.moduleNames.grpcCore = String(config.moduleNames.grpcCore.reversed())
-      config.moduleNames.grpcProtobuf = String(config.moduleNames.grpcProtobuf.reversed())
-      config.moduleNames.swiftProtobuf = String(config.moduleNames.swiftProtobuf.reversed())
-
-      let generator = ProtobufCodeGenerator(config: config)
-      let generated = try generator.generateCode(
-        fileDescriptor: Self.fileDescriptor,
-        protoFileModuleMappings: ProtoFileToModuleMappings(),
-        extraModuleImports: []
-      )
-
-      // Mustn't contain the default names.
-      #expect(!generated.contains(defaultNames.grpcCore))
-      #expect(!generated.contains(defaultNames.grpcProtobuf))
-      #expect(!generated.contains(defaultNames.swiftProtobuf))
-
-      // Must contain the configured names.
-      #expect(generated.contains(config.moduleNames.grpcCore))
-      #expect(generated.contains(config.moduleNames.grpcProtobuf))
-      #expect(generated.contains(config.moduleNames.swiftProtobuf))
-    }
   }
 
   @Suite("File-without-services (foo-messages.proto)")