Skip to content

Commit 644d1a8

Browse files
committed
get the default region from the spec
1 parent 6b615ac commit 644d1a8

File tree

1 file changed

+13
-4
lines changed

1 file changed

+13
-4
lines changed

generators/src/main/java/com/algolia/codegen/cts/AlgoliaCTSGenerator.java

Lines changed: 13 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -88,9 +88,16 @@ public Map<String, Object> postProcessSupportingFileData(Map<String, Object> obj
8888

8989
Object lambda = objs.get("lambda");
9090
List<CodegenServer> servers = (List<CodegenServer>) objs.get("servers");
91-
boolean hasRegionalHost = servers
92-
.stream()
93-
.anyMatch(server -> server.variables.stream().anyMatch(variable -> variable.name.equals("region")));
91+
CodegenServerVariable regionVariable = null;
92+
outerLoop:for (CodegenServer server : servers) {
93+
for (CodegenServerVariable var : server.variables) {
94+
if (var.name.equals("region")) {
95+
regionVariable = var;
96+
break outerLoop;
97+
}
98+
}
99+
}
100+
boolean hasRegionalHost = regionVariable != null;
94101

95102
Map<String, Object> bundle = objs;
96103
bundle.clear();
@@ -106,7 +113,9 @@ public Map<String, Object> postProcessSupportingFileData(Map<String, Object> obj
106113
bundle.put("client", Utils.createClientName(importClientName, language) + "Client");
107114
bundle.put("clientPrefix", Utils.createClientName(importClientName, language));
108115
bundle.put("hasRegionalHost", hasRegionalHost);
109-
bundle.put("defaultRegion", client.equals("predict") ? "eu" : "us");
116+
if (hasRegionalHost) {
117+
bundle.put("defaultRegion", regionVariable.defaultValue);
118+
}
110119
bundle.put("lambda", lambda);
111120

112121
ctsManager.addDataToBundle(bundle);

0 commit comments

Comments
 (0)