From 37006f8a035fc9a7b8e7f8901a23e1424d79824c Mon Sep 17 00:00:00 2001
From: William Cheng <wing328hk@gmail.com>
Date: Fri, 22 Oct 2021 10:49:38 +0800
Subject: [PATCH] add snak case lambda (#10658)

---
 .../openapitools/codegen/DefaultCodegen.java  |  1 +
 .../templating/mustache/SnakecaseLambda.java  | 46 +++++++++++++++++++
 .../mustache/SnakecaseLambdaTest.java         | 33 +++++++++++++
 3 files changed, 80 insertions(+)
 create mode 100644 modules/openapi-generator/src/main/java/org/openapitools/codegen/templating/mustache/SnakecaseLambda.java
 create mode 100644 modules/openapi-generator/src/test/java/org/openapitools/codegen/templating/mustache/SnakecaseLambdaTest.java

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 8e76e62e56cc..d5920a2abc7c 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
@@ -394,6 +394,7 @@ protected ImmutableMap.Builder<String, Lambda> addMustacheLambdas() {
         return new ImmutableMap.Builder<String, Mustache.Lambda>()
                 .put("lowercase", new LowercaseLambda().generator(this))
                 .put("uppercase", new UppercaseLambda())
+                .put("snakecase", new SnakecaseLambda())
                 .put("titlecase", new TitlecaseLambda())
                 .put("camelcase", new CamelCaseLambda(true).generator(this))
                 .put("pascalcase", new CamelCaseLambda(false).generator(this))
diff --git a/modules/openapi-generator/src/main/java/org/openapitools/codegen/templating/mustache/SnakecaseLambda.java b/modules/openapi-generator/src/main/java/org/openapitools/codegen/templating/mustache/SnakecaseLambda.java
new file mode 100644
index 000000000000..55aeb1613827
--- /dev/null
+++ b/modules/openapi-generator/src/main/java/org/openapitools/codegen/templating/mustache/SnakecaseLambda.java
@@ -0,0 +1,46 @@
+/*
+ * Copyright 2018 OpenAPI-Generator Contributors (https://openapi-generator.tech)
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.openapitools.codegen.templating.mustache;
+
+import com.samskivert.mustache.Mustache;
+import com.samskivert.mustache.Template;
+
+import java.io.IOException;
+import java.io.Writer;
+import java.util.Locale;
+
+import static org.openapitools.codegen.utils.StringUtils.underscore;
+
+/**
+ * Converts text in a fragment to snake case.
+ *
+ * Register:
+ * <pre>
+ * additionalProperties.put("snakecase", new SnakecaseLambda());
+ * </pre>
+ *
+ * Use:
+ * <pre>
+ * {{#snakecase}}{{summary}}{{/snakecase}}
+ * </pre>
+ */
+public class SnakecaseLambda implements Mustache.Lambda {
+    @Override
+    public void execute(Template.Fragment fragment, Writer writer) throws IOException {
+        writer.write(underscore(fragment.execute()));
+    }
+}
diff --git a/modules/openapi-generator/src/test/java/org/openapitools/codegen/templating/mustache/SnakecaseLambdaTest.java b/modules/openapi-generator/src/test/java/org/openapitools/codegen/templating/mustache/SnakecaseLambdaTest.java
new file mode 100644
index 000000000000..9dc0d57a5066
--- /dev/null
+++ b/modules/openapi-generator/src/test/java/org/openapitools/codegen/templating/mustache/SnakecaseLambdaTest.java
@@ -0,0 +1,33 @@
+/*
+ * Copyright 2018 OpenAPI-Generator Contributors (https://openapi-generator.tech)
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.openapitools.codegen.templating.mustache;
+
+import java.util.Map;
+
+import org.testng.annotations.Test;
+
+public class SnakecaseLambdaTest extends LambdaTest {
+
+    @Test
+    public void snakecaseTest() {
+        // Given
+        Map<String, Object> ctx = context("snakecase", new SnakecaseLambda());
+
+        // When & Then
+        test("access_code", "{{#snakecase}}accessCode{{/snakecase}}", ctx);
+    }
+
+}