chore(go): update enum casing

algolia · Jun 12, 2024 · a8045fd · a8045fd
1 parent d67050d
commit a8045fd
Show file tree

Hide file tree

Showing 7 changed files with 38 additions and 13 deletions.
diff --git a/generators/src/main/java/com/algolia/codegen/AlgoliaGoGenerator.java b/generators/src/main/java/com/algolia/codegen/AlgoliaGoGenerator.java
@@ -1,9 +1,12 @@
 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;
 import io.swagger.v3.oas.models.Operation;
 import io.swagger.v3.oas.models.servers.Server;
@@ -55,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/playground/go/go.mod b/playground/go/go.mod
@@ -13,7 +13,7 @@ require (
 	github.com/gabriel-vasile/mimetype v1.4.3 // indirect
 	github.com/go-playground/locales v0.14.1 // indirect
 	github.com/go-playground/universal-translator v0.18.1 // indirect
-	github.com/go-playground/validator/v10 v10.19.0 // indirect
+	github.com/go-playground/validator/v10 v10.21.0 // indirect
 	github.com/leodido/go-urn v1.4.0 // indirect
 	golang.org/x/crypto v0.22.0 // indirect
 	golang.org/x/net v0.24.0 // indirect

diff --git a/playground/go/go.sum b/playground/go/go.sum
@@ -10,6 +10,7 @@ github.com/go-playground/universal-translator v0.18.1 h1:Bcnm0ZwsGyWbCzImXv+pAJn
 github.com/go-playground/universal-translator v0.18.1/go.mod h1:xekY+UJKNuX9WP91TpwSH2VMlDf28Uj24BCp08ZFTUY=
 github.com/go-playground/validator/v10 v10.19.0 h1:ol+5Fu+cSq9JD7SoSqe04GMI92cbn0+wvQ3bZ8b/AU4=
 github.com/go-playground/validator/v10 v10.19.0/go.mod h1:dbuPbCMFw/DrkbEynArYaCwl3amGuJotoKCe95atGMM=
+github.com/go-playground/validator/v10 v10.21.0/go.mod h1:dbuPbCMFw/DrkbEynArYaCwl3amGuJotoKCe95atGMM=
 github.com/joho/godotenv v1.5.1 h1:7eLL/+HRGLY0ldzfGMeQkb7vMd0as4CfYvUVzLqw0N0=
 github.com/joho/godotenv v1.5.1/go.mod h1:f4LDr5Voq0i2e/R5DDNOoa2zzDfwtkZa6DnEwAbqwq4=
 github.com/leodido/go-urn v1.4.0 h1:WT9HwE9SGECu3lg4d/dIA+jxlljEa1/ffXKmRjqdmIQ=

diff --git a/templates/go/model_enum.mustache b/templates/go/model_enum.mustache
@@ -7,7 +7,7 @@ const (
 	{{#enumVars}}
 	{{^-first}}
 	{{/-first}}
-	{{#enumClassPrefix}}{{{classname.toUpperCase}}}_{{/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
-}
+}
diff --git a/templates/go/search_helpers.mustache b/templates/go/search_helpers.mustache
@@ -73,7 +73,7 @@ func (c *APIClient) WaitForTaskWithContext(
         return false
       }
 
-      return response.Status == TASKSTATUS_PUBLISHED
+      return response.Status == TASK_STATUS_PUBLISHED
     },
     maxRetries,
     initialDelay,
@@ -284,11 +284,11 @@ func (c *APIClient) WaitForApiKeyWithContext(
   maxDelay *time.Duration,
   opts ...Option,
 ) (*GetApiKeyResponse, error) {
-  if operation != APIKEYOPERATION_ADD && operation != APIKEYOPERATION_DELETE && operation != APIKEYOPERATION_UPDATE {
+  if operation != API_KEY_OPERATION_ADD && operation != API_KEY_OPERATION_DELETE && operation != API_KEY_OPERATION_UPDATE {
     return nil, &errs.WaitKeyOperationError{}
   }
 
-  if operation == APIKEYOPERATION_UPDATE {
+  if operation == API_KEY_OPERATION_UPDATE {
     if apiKey == nil {
       return nil, &errs.WaitKeyUpdateError{}
     }
@@ -353,9 +353,9 @@ func (c *APIClient) WaitForApiKeyWithContext(
     },
     func(response *GetApiKeyResponse, err error) bool {
       switch operation {
-      case APIKEYOPERATION_ADD:
+      case API_KEY_OPERATION_ADD:
         return err == nil && response != nil && response.CreatedAt > 0
-      case APIKEYOPERATION_DELETE:
+      case API_KEY_OPERATION_DELETE:
         if _, ok := err.(*APIError); ok {
           apiErr := err.(*APIError)
 
@@ -520,7 +520,7 @@ func (c *APIClient) ChunkedBatch(indexName string, objects []map[string]any, act
 func (c *APIClient) ReplaceAllObjects(indexName string, objects []map[string]any, batchSize *int) (*ReplaceAllObjectsResponse, error) {
 	tmpIndexName := fmt.Sprintf("%s_tmp_%d", indexName, time.Now().UnixNano())
 
-	copyResp, err := c.OperationIndex(c.NewApiOperationIndexRequest(indexName, NewOperationIndexParams(OPERATIONTYPE_COPY, tmpIndexName, WithOperationIndexParamsScope([]ScopeType{SCOPETYPE_RULES, SCOPETYPE_SETTINGS, SCOPETYPE_SYNONYMS}))))
+	copyResp, err := c.OperationIndex(c.NewApiOperationIndexRequest(indexName, NewOperationIndexParams(OPERATION_TYPE_COPY, tmpIndexName, WithOperationIndexParamsScope([]ScopeType{SCOPE_TYPE_RULES, SCOPE_TYPE_SETTINGS, SCOPE_TYPE_SYNONYMS}))))
 	if err != nil {
 		return nil, err
 	}
@@ -547,7 +547,7 @@ func (c *APIClient) ReplaceAllObjects(indexName string, objects []map[string]any
 		return nil, err
 	}
 
-	moveResp, err := c.OperationIndex(c.NewApiOperationIndexRequest(tmpIndexName, NewOperationIndexParams(OPERATIONTYPE_MOVE, indexName)))
+	moveResp, err := c.OperationIndex(c.NewApiOperationIndexRequest(tmpIndexName, NewOperationIndexParams(OPERATION_TYPE_MOVE, indexName)))
 	if err != nil {
 		return nil, err
 	}

diff --git a/website/src/snippets.js b/website/src/snippets.js
@@ -66,15 +66,15 @@ export function waitForApiKeySnippet(language, operation) {
       'delete': ``,
     },
     'go': {
-      'add': `waitResponse, err := client.WaitForApiKey(search.APIKEYOPERATION_ADD, response.Key, nil)
+      'add': `waitResponse, err := client.WaitForApiKey(search.API_KEY_OPERATION_ADD, response.Key, nil)
 if err != nil {
   panic(err)
 }`,
-      'update': `waitResponse, err := client.WaitForApiKey(search.APIKEYOPERATION_UPDATE, response.Key, search.NewEmptyApiKey().SetAcl([]search.Acl{search.Acl("search"), search.Acl("addObject")}).SetValidity(300).SetMaxQueriesPerIPPerHour(100).SetMaxHitsPerQuery(20))
+      'update': `waitResponse, err := client.WaitForApiKey(search.API_KEY_OPERATION_UPDATE, response.Key, search.NewEmptyApiKey().SetAcl([]search.Acl{search.Acl("search"), search.Acl("addObject")}).SetValidity(300).SetMaxQueriesPerIPPerHour(100).SetMaxHitsPerQuery(20))
 if err != nil {
   panic(err)
 }`,
-      'delete': `waitResponse, err := client.WaitForApiKey(search.APIKEYOPERATION_DELETE, response.Key, nil)
+      'delete': `waitResponse, err := client.WaitForApiKey(search.API_KEY_OPERATION_DELETE, response.Key, nil)
 if err != nil {
   panic(err)
 }`,