chore: better handling

algolia · Jun 12, 2024 · 341fc70 · 341fc70
1 parent 3fd5ae6
commit 341fc70
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 6 deletions.
diff --git a/generators/src/main/java/com/algolia/codegen/AlgoliaGoGenerator.java b/generators/src/main/java/com/algolia/codegen/AlgoliaGoGenerator.java
@@ -1,8 +1,10 @@
 package com.algolia.codegen;
 
 import com.algolia.codegen.exceptions.*;
+import com.algolia.codegen.lambda.ScreamingSnakeCaseLambda;
 import com.algolia.codegen.utils.*;
 import com.algolia.codegen.utils.OneOf;
+import com.google.common.collect.ImmutableMap;
 import com.google.common.collect.Iterables;
 import com.samskivert.mustache.Mustache;
 import io.swagger.v3.oas.models.OpenAPI;
@@ -30,10 +32,6 @@ public void processOpts() {
     additionalProperties.put("packageName", client.equals("query-suggestions") ? "suggestions" : client);
     additionalProperties.put("enumClassPrefix", true);
     additionalProperties.put("isSearchClient", client.equals("search"));
-    additionalProperties.put(
-      "lambda.toUpperSnakeCase",
-      (Mustache.Lambda) (fragment, writer) -> writer.write(Helpers.toSnakeCase(fragment.execute()).toUpperCase(Locale.ROOT))
-    );
 
     String outputFolder = "algolia" + File.separator + client;
     setOutputDir(getOutputDir() + File.separator + outputFolder);
@@ -60,6 +58,11 @@ public void processOpts() {
     }
   }
 
+  @Override
+  protected ImmutableMap.Builder<String, Mustache.Lambda> addMustacheLambdas() {
+    return super.addMustacheLambdas().put("screamingSnakeCase", new ScreamingSnakeCaseLambda());
+  }
+
   @Override
   public void processOpenAPI(OpenAPI openAPI) {
     super.processOpenAPI(openAPI);

diff --git a/generators/src/main/java/com/algolia/codegen/lambda/ScreamingSnakeCaseLambda.java b/generators/src/main/java/com/algolia/codegen/lambda/ScreamingSnakeCaseLambda.java
@@ -0,0 +1,16 @@
+package com.algolia.codegen.lambda;
+
+import com.algolia.codegen.utils.Helpers;
+import com.samskivert.mustache.Mustache;
+import com.samskivert.mustache.Template;
+import java.io.IOException;
+import java.io.Writer;
+import java.util.Locale;
+
+public class ScreamingSnakeCaseLambda implements Mustache.Lambda {
+
+  @Override
+  public void execute(Template.Fragment fragment, Writer writer) throws IOException {
+    writer.write(Helpers.toSnakeCase(fragment.execute()).toUpperCase(Locale.ROOT));
+  }
+}
diff --git a/templates/go/model_enum.mustache b/templates/go/model_enum.mustache
@@ -7,7 +7,7 @@ const (
 	{{#enumVars}}
 	{{^-first}}
 	{{/-first}}
-	{{#enumClassPrefix}}{{#lambda.toUpperSnakeCase}}{{{classname}}}{{/lambda.toUpperSnakeCase}}_{{/enumClassPrefix}}{{name}} {{{classname}}} = {{{value}}}
+	{{#enumClassPrefix}}{{#lambda.screamingSnakeCase}}{{{classname}}}{{/lambda.screamingSnakeCase}}_{{/enumClassPrefix}}{{name}} {{{classname}}} = {{{value}}}
 	{{/enumVars}}
 	{{/allowableValues}}
 )
@@ -98,4 +98,4 @@ func (v Nullable{{{classname}}}) MarshalJSON() ([]byte, error) {
 func (v *Nullable{{{classname}}}) UnmarshalJSON(src []byte) error {
 	v.isSet = true
 	return json.Unmarshal(src, &v.value) //nolint:wrapcheck
-}
+}