Feat: Ignore marked resources (#4)

* feat: add filters to skip the generation of marked models and apis
* feat: accept more than one skipTag using globalSettings

Signed-off-by: Florin Mihalache <[email protected]>

Author: mimihalescu

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

@@ -53,6 +53,11 @@ public class CodegenOperation {
     public List<Map<String, String>> examples;
     public List<Map<String, String>> requestBodyExamples;
     public ExternalDocumentation externalDocs;
+
+    public Map<String, Object> getVendorExtensions() {
+        return vendorExtensions;
+    }
+
     public Map<String, Object> vendorExtensions = new HashMap<String, Object>();
     public String nickname; // legacy support
     public String operationIdOriginal; // for plug-in

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

@@ -63,6 +63,7 @@
 import java.time.ZonedDateTime;
 import java.util.*;
 import java.util.function.Function;
+import java.util.function.Predicate;
 import java.util.stream.Collectors;
 
 import static org.apache.commons.lang3.StringUtils.removeStart;
@@ -88,6 +89,7 @@ public class DefaultGenerator implements Generator {
     private String basePath;
     private String basePathWithoutHost;
     private String contextPath;
+    private List<String> skipTags = new ArrayList<>();
     private Map<String, String> generatorPropertyDefaults = new HashMap<>();
     protected TemplateProcessor templateProcessor = null;
 
@@ -101,6 +103,11 @@ public DefaultGenerator() {
     public DefaultGenerator(Boolean dryRun) {
         this.dryRun = Boolean.TRUE.equals(dryRun);
         LOGGER.info("Generating with dryRun={}", this.dryRun);
+        String skipTagsProperty = GlobalSettings.getProperty("skipTags");
+        if(skipTagsProperty != null) {
+            skipTags = Arrays.asList(skipTagsProperty.split(","));
+        }
+        LOGGER.info("Generating with skipTags={}", skipTags);
     }
 
     @SuppressWarnings("deprecation")
@@ -527,6 +534,10 @@ void generateModels(List<File> files, List<Object> allModels, List<String> unuse
                             // for PythonClientCodegen, all aliases are generated as models
                             continue;  // Don't create user-defined classes for aliases
                         }
+                        if(hasAnySkipTag(skipTags, m)) {
+                            LOGGER.info("Skipped generating model {}", m.name);
+                            continue;
+                        }
                     }
                     allModels.add(modelTemplate);
                 }
@@ -551,6 +562,20 @@ void generateModels(List<File> files, List<Object> allModels, List<String> unuse
 
     }
 
+    private boolean hasAnySkipTag(List<String> skipTags, CodegenModel m) {
+        Map<String, Object> vendorExtensions = m.vendorExtensions;
+        return skipTags.stream().anyMatch(hasSkipTag(vendorExtensions));
+    }
+
+    private boolean hasAnySkipTag(List<String> skipTags, CodegenOperation op) {
+        Map<String, Object> vendorExtensions = op.vendorExtensions;
+        return skipTags.stream().anyMatch(hasSkipTag(vendorExtensions));
+    }
+
+    private Predicate<String> hasSkipTag(Map<String, Object> vendorExtensions) {
+        return skipTag -> vendorExtensions.containsKey(skipTag) && (boolean) vendorExtensions.get(skipTag);
+    }
+
     @SuppressWarnings("unchecked")
     void generateApis(List<File> files, List<Object> allOperations, List<Object> allModels) {
         if (!generateApis) {
@@ -577,6 +602,11 @@ void generateApis(List<File> files, List<Object> allOperations, List<Object> all
             try {
                 List<CodegenOperation> ops = paths.get(tag);
                 ops.sort((one, another) -> ObjectUtils.compare(one.operationId, another.operationId));
+                ops.removeIf(op -> hasInProgressExtension(tag, op));
+                if(ops.isEmpty()) {
+                    LOGGER.info("The {} client will not be generated", tag);
+                    continue;
+                }
                 Map<String, Object> operation = processOperations(config, tag, ops, allModels);
                 URL url = URLPathUtils.getServerURL(openAPI, config.serverVariableOverrides());
                 operation.put("basePath", basePath);
@@ -683,6 +713,13 @@ void generateApis(List<File> files, List<Object> allOperations, List<Object> all
         }
 
     }
+    private boolean hasInProgressExtension(String tag, CodegenOperation op) {
+        if (hasAnySkipTag(skipTags, op)) {
+            LOGGER.info("Operation {} of {} will not be generated", op.operationId, tag);
+            return true;
+        }
+        return false;
+    }
 
     private void generateSupportingFiles(List<File> files, Map<String, Object> bundle) {
         if (!generateSupportingFiles) {