From 0b30d8422696e737787a30c6b9ac1aa4e28eeb37 Mon Sep 17 00:00:00 2001
From: xil <fridalu66@gmail.com>
Date: Fri, 21 Mar 2025 20:52:30 +0000
Subject: [PATCH] protobuf import format

---
 bin/configs/protobuf-schema-config.yaml                     | 6 +++++-
 bin/configs/protobuf-schema.yaml                            | 4 ++++
 .../codegen/languages/ProtobufSchemaCodegen.java            | 6 +++++-
 .../src/main/resources/protobuf-schema/api.mustache         | 2 +-
 .../src/main/resources/protobuf-schema/model.mustache       | 2 +-
 .../src/test/resources/3_0/protobuf/petstore.yaml           | 2 ++
 .../petstore/protobuf-schema-config/models/order.proto      | 3 +++
 samples/config/petstore/protobuf-schema/models/order.proto  | 3 +++
 8 files changed, 24 insertions(+), 4 deletions(-)

diff --git a/bin/configs/protobuf-schema-config.yaml b/bin/configs/protobuf-schema-config.yaml
index 03df6bce30cf..44ae7540fd8a 100644
--- a/bin/configs/protobuf-schema-config.yaml
+++ b/bin/configs/protobuf-schema-config.yaml
@@ -7,4 +7,8 @@ additionalProperties:
   addJsonNameAnnotation: true
   numberedFieldNumberList: true
   startEnumsWithUnspecified: true
-  wrapComplexType: false
\ No newline at end of file
+  wrapComplexType: false
+typeMappings:
+  object: "google.protobuf.Struct"
+importMappings:
+  google.protobuf.Struct: "google/protobuf/struct"
\ No newline at end of file
diff --git a/bin/configs/protobuf-schema.yaml b/bin/configs/protobuf-schema.yaml
index 30f606c915f3..c3947e075dca 100644
--- a/bin/configs/protobuf-schema.yaml
+++ b/bin/configs/protobuf-schema.yaml
@@ -4,3 +4,7 @@ inputSpec: modules/openapi-generator/src/test/resources/3_0/protobuf/petstore.ya
 templateDir: modules/openapi-generator/src/main/resources/protobuf-schema
 additionalProperties:
   packageName: petstore
+typeMappings:
+  object: "google.protobuf.Struct"
+importMappings:
+  google.protobuf.Struct: "google/protobuf/struct"
diff --git a/modules/openapi-generator/src/main/java/org/openapitools/codegen/languages/ProtobufSchemaCodegen.java b/modules/openapi-generator/src/main/java/org/openapitools/codegen/languages/ProtobufSchemaCodegen.java
index 052d1aa8321b..78938f1cacf8 100644
--- a/modules/openapi-generator/src/main/java/org/openapitools/codegen/languages/ProtobufSchemaCodegen.java
+++ b/modules/openapi-generator/src/main/java/org/openapitools/codegen/languages/ProtobufSchemaCodegen.java
@@ -912,7 +912,11 @@ public OperationsMap postProcessOperationsWithModels(OperationsMap objs, List<Mo
 
     @Override
     public String toModelImport(String name) {
-        return underscore(name);
+        if ("".equals(modelPackage())) {
+            return name;
+        } else {
+            return modelPackage() + "/" + underscore(name);
+        }
     }
 
     @Override
diff --git a/modules/openapi-generator/src/main/resources/protobuf-schema/api.mustache b/modules/openapi-generator/src/main/resources/protobuf-schema/api.mustache
index 4e4518615ac9..cab1a8ac794b 100644
--- a/modules/openapi-generator/src/main/resources/protobuf-schema/api.mustache
+++ b/modules/openapi-generator/src/main/resources/protobuf-schema/api.mustache
@@ -6,7 +6,7 @@ package {{#lambda.lowercase}}{{{packageName}}}.{{{apiPackage}}}.{{{classname}}};
 import "google/protobuf/empty.proto";
 {{#imports}}
 {{#import}}
-import public "{{{modelPackage}}}/{{{.}}}.proto";
+import public "{{{import}}}.proto";
 {{/import}}
 {{/imports}}
 
diff --git a/modules/openapi-generator/src/main/resources/protobuf-schema/model.mustache b/modules/openapi-generator/src/main/resources/protobuf-schema/model.mustache
index 6872f57f18d7..df6b8af644c6 100644
--- a/modules/openapi-generator/src/main/resources/protobuf-schema/model.mustache
+++ b/modules/openapi-generator/src/main/resources/protobuf-schema/model.mustache
@@ -5,7 +5,7 @@ package {{#lambda.lowercase}}{{{packageName}}};{{/lambda.lowercase}}
 
 {{#imports}}
 {{#import}}
-import public "{{{modelPackage}}}/{{{import}}}.proto";
+import public "{{{.}}}.proto";
 {{/import}}
 {{/imports}}
 
diff --git a/modules/openapi-generator/src/test/resources/3_0/protobuf/petstore.yaml b/modules/openapi-generator/src/test/resources/3_0/protobuf/petstore.yaml
index 2c4ee2647cd2..01b109bdad16 100644
--- a/modules/openapi-generator/src/test/resources/3_0/protobuf/petstore.yaml
+++ b/modules/openapi-generator/src/test/resources/3_0/protobuf/petstore.yaml
@@ -669,6 +669,8 @@ components:
         complete:
           type: boolean
           default: false
+        meta:
+          type: object
       xml:
         name: Order
     Category:
diff --git a/samples/config/petstore/protobuf-schema-config/models/order.proto b/samples/config/petstore/protobuf-schema-config/models/order.proto
index cc93af2330e8..7f5283263ed1 100644
--- a/samples/config/petstore/protobuf-schema-config/models/order.proto
+++ b/samples/config/petstore/protobuf-schema-config/models/order.proto
@@ -12,6 +12,7 @@ syntax = "proto3";
 
 package petstore;
 
+import public "google/protobuf/struct.proto";
 
 message Order {
 
@@ -35,4 +36,6 @@ message Order {
 
   bool complete = 6;
 
+  google.protobuf.Struct meta = 7;
+
 }
diff --git a/samples/config/petstore/protobuf-schema/models/order.proto b/samples/config/petstore/protobuf-schema/models/order.proto
index cdd08cc15f3b..c8df611f89b0 100644
--- a/samples/config/petstore/protobuf-schema/models/order.proto
+++ b/samples/config/petstore/protobuf-schema/models/order.proto
@@ -12,6 +12,7 @@ syntax = "proto3";
 
 package petstore;
 
+import public "google/protobuf/struct.proto";
 
 message Order {
 
@@ -34,4 +35,6 @@ message Order {
 
   bool complete = 62574280;
 
+  google.protobuf.Struct meta = 3347973;
+
 }