add custom options

Author: andrewwilsonnew

bin/configs/protobuf-schema-config.yaml

@@ -9,6 +9,14 @@ additionalProperties:
   startEnumsWithUnspecified: true
   wrapComplexType: false
   aggregateModelsName: data
+  customOptionsApi: |
+    option java_multiple_files = true;
+    option java_package = "com.example.tutorial.protos.api";
+    option java_outer_classname = "ExampleProtos";
+  customOptionsModel: |
+    option java_multiple_files = false;
+    option java_package = "com.example.tutorial.protos.model";
+    option java_outer_classname = "ExampleProtos";
 typeMappings:
   object: "google.protobuf.Struct"
 importMappings:

modules/openapi-generator/src/main/java/org/openapitools/codegen/languages/ProtobufSchemaCodegen.java

@@ -68,12 +68,22 @@ public class ProtobufSchemaCodegen extends DefaultCodegen implements CodegenConf
 
     public static final String AGGREGATE_MODELS_NAME = "aggregateModelsName";
 
+    public static final String CUSTOM_OPTIONS_API = "customOptionsApi";
+
+    public static final String CUSTOM_OPTIONS_MODEL = "customOptionsModel";
+
     private final Logger LOGGER = LoggerFactory.getLogger(ProtobufSchemaCodegen.class);
 
     @Setter protected String packageName = "openapitools";
 
     @Setter protected String aggregateModelsName = null;
 
+    @SuppressWarnings("unused")
+    @Setter protected String customOptionsApi = null;
+
+    @SuppressWarnings("unused")
+    @Setter protected String customOptionsModel = null;
+
     private boolean numberedFieldNumberList = false;
 
     private boolean startEnumsWithUnspecified = false;
@@ -193,6 +203,8 @@ public ProtobufSchemaCodegen() {
         addSwitch(ADD_JSON_NAME_ANNOTATION, "Append \"json_name\" annotation to message field when the specification name differs from the protobuf field name", addJsonNameAnnotation);
         addSwitch(WRAP_COMPLEX_TYPE, "Generate Additional message for complex type", wrapComplexType);
         addOption(AGGREGATE_MODELS_NAME, "Aggregated model filename. If set, all generated models will be combined into this single file.", null);
+        addOption(CUSTOM_OPTIONS_API, "Custom options for the api files.", null);
+        addOption(CUSTOM_OPTIONS_MODEL, "Custom options for the model files.", null);
     }
 
     @Override
@@ -239,6 +251,14 @@ public void processOpts() {
             this.setAggregateModelsName((String) additionalProperties.get(AGGREGATE_MODELS_NAME));
         }
 
+        if (additionalProperties.containsKey(CUSTOM_OPTIONS_API)) {
+            this.setCustomOptionsApi((String) additionalProperties.get(CUSTOM_OPTIONS_API));
+        }
+
+        if (additionalProperties.containsKey(CUSTOM_OPTIONS_MODEL)) {
+            this.setCustomOptionsModel((String) additionalProperties.get(CUSTOM_OPTIONS_MODEL));
+        }
+
         supportingFiles.add(new SupportingFile("README.mustache", "", "README.md"));
     }
 

modules/openapi-generator/src/main/resources/protobuf-schema/api.mustache

@@ -3,6 +3,9 @@ syntax = "proto3";
 
 package {{#lambda.lowercase}}{{{packageName}}}.{{{apiPackage}}}.{{{classname}}};{{/lambda.lowercase}}
 
+{{#customOptionsApi}}
+{{{.}}}
+{{/customOptionsApi}}
 import "google/protobuf/empty.proto";
 {{#imports}}
 {{#import}}

modules/openapi-generator/src/main/resources/protobuf-schema/model.mustache

@@ -3,6 +3,9 @@ syntax = "proto3";
 
 package {{#lambda.lowercase}}{{{packageName}}};{{/lambda.lowercase}}
 
+{{#customOptionsModel}}
+{{{.}}}
+{{/customOptionsModel}}
 {{#imports}}
 {{#import}}
 import public "{{{.}}}.proto";

samples/config/petstore/protobuf-schema-config/models/data.proto

@@ -12,6 +12,10 @@ syntax = "proto3";
 
 package petstore;
 
+option java_multiple_files = false;
+option java_package = "com.example.tutorial.protos.model";
+option java_outer_classname = "ExampleProtos";
+
 import public "google/protobuf/struct.proto";
 
 message ApiResponse {

samples/config/petstore/protobuf-schema-config/services/default_service.proto

@@ -12,6 +12,10 @@ syntax = "proto3";
 
 package petstore.services.defaultservice;
 
+option java_multiple_files = true;
+option java_package = "com.example.tutorial.protos.api";
+option java_outer_classname = "ExampleProtos";
+
 import "google/protobuf/empty.proto";
 import public "models/data.proto";
 

samples/config/petstore/protobuf-schema-config/services/pet_service.proto

@@ -12,6 +12,10 @@ syntax = "proto3";
 
 package petstore.services.petservice;
 
+option java_multiple_files = true;
+option java_package = "com.example.tutorial.protos.api";
+option java_outer_classname = "ExampleProtos";
+
 import "google/protobuf/empty.proto";
 import public "models/data.proto";
 

samples/config/petstore/protobuf-schema-config/services/store_service.proto

@@ -12,6 +12,10 @@ syntax = "proto3";
 
 package petstore.services.storeservice;
 
+option java_multiple_files = true;
+option java_package = "com.example.tutorial.protos.api";
+option java_outer_classname = "ExampleProtos";
+
 import "google/protobuf/empty.proto";
 import public "models/data.proto";
 

samples/config/petstore/protobuf-schema-config/services/user_service.proto

@@ -12,6 +12,10 @@ syntax = "proto3";
 
 package petstore.services.userservice;
 
+option java_multiple_files = true;
+option java_package = "com.example.tutorial.protos.api";
+option java_outer_classname = "ExampleProtos";
+
 import "google/protobuf/empty.proto";
 import public "models/data.proto";