From 49f55b0d8e3a9531e3962bd831b74181ac71abd8 Mon Sep 17 00:00:00 2001
From: William Cheng <wing328hk@gmail.com>
Date: Fri, 18 Dec 2020 10:16:21 +0800
Subject: [PATCH] Add postProcess method to show donation message (#8142)

* add postProcess method to show donation message

* update comment
---
 .../java/org/openapitools/codegen/CodegenConfig.java   |  2 ++
 .../java/org/openapitools/codegen/DefaultCodegen.java  | 10 ++++++++++
 .../org/openapitools/codegen/DefaultGenerator.java     |  3 +++
 3 files changed, 15 insertions(+)

diff --git a/modules/openapi-generator/src/main/java/org/openapitools/codegen/CodegenConfig.java b/modules/openapi-generator/src/main/java/org/openapitools/codegen/CodegenConfig.java
index 069b2a2d100..d3b52507083 100644
--- a/modules/openapi-generator/src/main/java/org/openapitools/codegen/CodegenConfig.java
+++ b/modules/openapi-generator/src/main/java/org/openapitools/codegen/CodegenConfig.java
@@ -182,6 +182,8 @@ public interface CodegenConfig {
 
     Map<String, Object> updateAllModels(Map<String, Object> objs);
 
+    void postProcess();
+
     Map<String, Object> postProcessAllModels(Map<String, Object> objs);
 
     Map<String, Object> postProcessModels(Map<String, Object> objs);
diff --git a/modules/openapi-generator/src/main/java/org/openapitools/codegen/DefaultCodegen.java b/modules/openapi-generator/src/main/java/org/openapitools/codegen/DefaultCodegen.java
index f2d5b050d32..9738f9a9452 100644
--- a/modules/openapi-generator/src/main/java/org/openapitools/codegen/DefaultCodegen.java
+++ b/modules/openapi-generator/src/main/java/org/openapitools/codegen/DefaultCodegen.java
@@ -747,6 +747,16 @@ public class DefaultCodegen implements CodegenConfig {
         ModelUtils.setDisallowAdditionalPropertiesIfNotPresent(getDisallowAdditionalPropertiesIfNotPresent());
     }
 
+    // override with any message to be shown right before the process finishes
+    @SuppressWarnings("static-method")
+    public void postProcess() {
+        System.out.println("################################################################################");
+        System.out.println("# Thanks for using OpenAPI Generator.                                          #");
+        System.out.println("# Please consider donation to help us maintain this project \uD83D\uDE4F                 #");
+        System.out.println("# https://opencollective.com/openapi_generator/donate                          #");
+        System.out.println("################################################################################");
+    }
+
     // override with any special post-processing
     @SuppressWarnings("static-method")
     public Map<String, Object> postProcessOperationsWithModels(Map<String, Object> objs, List<Object> allModels) {
diff --git a/modules/openapi-generator/src/main/java/org/openapitools/codegen/DefaultGenerator.java b/modules/openapi-generator/src/main/java/org/openapitools/codegen/DefaultGenerator.java
index e7fd5a5522a..ba209679651 100644
--- a/modules/openapi-generator/src/main/java/org/openapitools/codegen/DefaultGenerator.java
+++ b/modules/openapi-generator/src/main/java/org/openapitools/codegen/DefaultGenerator.java
@@ -925,6 +925,9 @@ public class DefaultGenerator implements Generator {
             }
         }
 
+        // post-process
+        config.postProcess();
+
         // reset GlobalSettings, so that the running thread can be reused for another generator-run
         GlobalSettings.reset();