Skip to content

Commit b0a50d7

Browse files
kuhekuhe
committed
feat(endpoint): endpoints v2 codegen
1 parent d32963f commit b0a50d7

12 files changed

+705
-3
lines changed

‎smithy-typescript-codegen/build.gradle.kts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@ buildscript {
2929

3030
dependencies {
3131
api("software.amazon.smithy:smithy-codegen-core:${rootProject.extra["smithyVersion"]}")
32+
api("software.amazon.smithy:smithy-rules-engine:${rootProject.extra["smithyVersion"]}")
3233
api("software.amazon.smithy:smithy-waiters:${rootProject.extra["smithyVersion"]}")
3334
implementation("software.amazon.smithy:smithy-protocol-test-traits:${rootProject.extra["smithyVersion"]}")
3435
}

‎smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/CodegenVisitor.java

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,9 @@
5151
import software.amazon.smithy.model.traits.EnumTrait;
5252
import software.amazon.smithy.model.traits.PaginatedTrait;
5353
import software.amazon.smithy.model.validation.ValidationEvent;
54+
import software.amazon.smithy.rulesengine.traits.EndpointRuleSetTrait;
5455
import software.amazon.smithy.typescript.codegen.TypeScriptSettings.ArtifactType;
56+
import software.amazon.smithy.typescript.codegen.endpointsV2.EndpointsV2Generator;
5557
import software.amazon.smithy.typescript.codegen.integration.ProtocolGenerator;
5658
import software.amazon.smithy.typescript.codegen.integration.RuntimeClientPlugin;
5759
import software.amazon.smithy.typescript.codegen.integration.TypeScriptIntegration;
@@ -323,6 +325,7 @@ public Void serviceShape(ServiceShape shape) {
323325
}
324326
if (settings.generateClient() || settings.generateServerSdk()) {
325327
generateCommands(shape);
328+
generateEndpointV2(shape);
326329
}
327330

328331
if (settings.generateServerSdk()) {
@@ -444,4 +447,14 @@ private void generateCommands(ServiceShape shape) {
444447
}
445448
}
446449
}
450+
451+
private void generateEndpointV2(ServiceShape shape) {
452+
if (!shape.hasTrait(EndpointRuleSetTrait.class)) {
453+
return;
454+
}
455+
456+
new EndpointsV2Generator(
457+
settings, model, fileManifest
458+
).run();
459+
}
447460
}

‎smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/CommandGenerator.java

Lines changed: 47 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,9 @@
3535
import software.amazon.smithy.model.shapes.OperationShape;
3636
import software.amazon.smithy.model.shapes.ServiceShape;
3737
import software.amazon.smithy.model.shapes.StructureShape;
38+
import software.amazon.smithy.rulesengine.traits.EndpointRuleSetTrait;
39+
import software.amazon.smithy.typescript.codegen.endpointsV2.EndpointsParamNameMap;
40+
import software.amazon.smithy.typescript.codegen.endpointsV2.RuleSetParameterFinder;
3841
import software.amazon.smithy.typescript.codegen.integration.ProtocolGenerator;
3942
import software.amazon.smithy.typescript.codegen.integration.RuntimeClientPlugin;
4043
import software.amazon.smithy.utils.OptionalUtils;
@@ -64,6 +67,7 @@ final class CommandGenerator implements Runnable {
6467
private final Symbol outputType;
6568
private final ProtocolGenerator protocolGenerator;
6669
private final ApplicationProtocol applicationProtocol;
70+
private final boolean hasEndpointRuleSetTrait;
6771

6872
CommandGenerator(
6973
TypeScriptSettings settings,
@@ -91,6 +95,9 @@ final class CommandGenerator implements Runnable {
9195
operationIndex = OperationIndex.of(model);
9296
inputType = symbol.expectProperty("inputType", Symbol.class);
9397
outputType = symbol.expectProperty("outputType", Symbol.class);
98+
hasEndpointRuleSetTrait = service.hasTrait(
99+
EndpointRuleSetTrait.class
100+
);
94101
}
95102

96103
@Override
@@ -185,6 +192,46 @@ private void generateCommandMiddlewareResolver(String configType) {
185192
// Add customizations.
186193
addCommandSpecificPlugins();
187194

195+
// EndpointsV2
196+
if (hasEndpointRuleSetTrait) {
197+
writer.addImport(
198+
"getEndpointPlugin",
199+
"getEndpointPlugin",
200+
"@aws-sdk/middleware-endpoint"
201+
);
202+
writer.openBlock(
203+
"this.middlewareStack.use(getEndpointPlugin(configuration, {",
204+
"}));",
205+
() -> {
206+
RuleSetParameterFinder parameterFinder = new RuleSetParameterFinder(service);
207+
parameterFinder.getBuiltInParams().forEach((name, type) -> {
208+
writer.write(
209+
"$L: { type: \"builtInParams\", name: \"$L\" },",
210+
name, EndpointsParamNameMap.getLocalName(name)
211+
);
212+
});
213+
parameterFinder.getClientContextParams().forEach((name, type) -> {
214+
writer.write(
215+
"$L: { type: \"clientContextParams\", name: \"$L\" },",
216+
name, EndpointsParamNameMap.getLocalName(name)
217+
);
218+
});
219+
parameterFinder.getStaticContextParamValues(operation).forEach((name, value) -> {
220+
writer.write(
221+
"$L: { type: \"staticContextParams\", value: $L },",
222+
name, value
223+
);
224+
});
225+
parameterFinder.getContextParams(operation).forEach((name, type) -> {
226+
writer.write(
227+
"$L: { type: \"contextParams\", name: \"$L\" },",
228+
name, EndpointsParamNameMap.getLocalName(name)
229+
);
230+
});
231+
}
232+
);
233+
}
234+
188235
// Resolve the middleware stack.
189236
writer.write("\nconst stack = clientStack.concat(this.middlewareStack);\n");
190237
writer.write("const { logger } = configuration;");

‎smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/HttpProtocolTestGenerator.java

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -389,7 +389,6 @@ private void generateMalformedRequestTest(OperationShape operation, HttpMalforme
389389
});
390390
writer.write("const r = await handler.handle(request, {});").write("");
391391

392-
// Assert that the function has been called exactly once.
393392
writer.write("expect(testFunction.mock.calls.length).toBe(0);");
394393

395394
writeHttpResponseAssertions(testCase.getResponse());

‎smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/ServiceBareBonesClientGenerator.java

Lines changed: 29 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,7 @@
3131
import software.amazon.smithy.model.knowledge.TopDownIndex;
3232
import software.amazon.smithy.model.shapes.OperationShape;
3333
import software.amazon.smithy.model.shapes.ServiceShape;
34+
import software.amazon.smithy.rulesengine.traits.EndpointRuleSetTrait;
3435
import software.amazon.smithy.typescript.codegen.integration.RuntimeClientPlugin;
3536
import software.amazon.smithy.typescript.codegen.integration.TypeScriptIntegration;
3637
import software.amazon.smithy.utils.OptionalUtils;
@@ -169,7 +170,15 @@ private void generateConfig() {
169170
if (!inputTypes.isEmpty()) {
170171
writer.indent();
171172
for (SymbolReference symbolReference : inputTypes) {
172-
writer.write("& $T", symbolReference);
173+
if (service.hasTrait(EndpointRuleSetTrait.class)
174+
&& symbolReference.getAlias().equals("EndpointInputConfig")) {
175+
writer.write("& $T<$L>", symbolReference, "EndpointParameters");
176+
} else {
177+
writer.write("& $T", symbolReference);
178+
}
179+
}
180+
if (service.hasTrait(EndpointRuleSetTrait.class)) {
181+
writer.write("& ClientInputEndpointParameters");
173182
}
174183
writer.dedent();
175184
}
@@ -189,7 +198,17 @@ private void generateConfig() {
189198
writer.indent();
190199
runtimePlugins.stream()
191200
.flatMap(p -> OptionalUtils.stream(p.getResolvedConfig()))
192-
.forEach(symbol -> writer.write("& $T", symbol));
201+
.forEach(symbol -> {
202+
if (service.hasTrait(EndpointRuleSetTrait.class)
203+
&& symbol.getAlias().equals("EndpointResolvedConfig")) {
204+
writer.write("& $T<$L>", symbol, "EndpointParameters");
205+
} else {
206+
writer.write("& $T", symbol);
207+
}
208+
});
209+
if (service.hasTrait(EndpointRuleSetTrait.class)) {
210+
writer.write("& ClientResolvedEndpointParameters");
211+
}
193212
writer.dedent();
194213
}
195214

@@ -331,6 +350,14 @@ private void generateConstructor() {
331350
}
332351
}
333352

353+
if (service.hasTrait(EndpointRuleSetTrait.class)) {
354+
configVariable++;
355+
writer.write("let $L = $L($L);",
356+
generateConfigVariable(configVariable),
357+
"resolveClientEndpointParameters",
358+
generateConfigVariable(configVariable - 1));
359+
}
360+
334361
writer.write("super($L);", generateConfigVariable(configVariable));
335362
writer.write("this.config = $L;", generateConfigVariable(configVariable));
336363

‎smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/TypeScriptDependency.java

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -51,6 +51,7 @@ public enum TypeScriptDependency implements SymbolDependencyContainer {
5151
MIDDLEWARE_SERDE("dependencies", "@aws-sdk/middleware-serde", true),
5252
MIDDLEWARE_RETRY("dependencies", "@aws-sdk/middleware-retry", true),
5353
MIDDLEWARE_STACK("dependencies", "@aws-sdk/middleware-stack", true),
54+
MIDDLEWARE_ENDPOINTS_V2("dependencies", "@aws-sdk/middleware-endpoint", true),
5455

5556
AWS_CRYPTO_SHA256_BROWSER("dependencies", "@aws-crypto/sha256-browser", "2.0.0", true),
5657
AWS_CRYPTO_SHA256_JS("dependencies", "@aws-crypto/sha256-js", "2.0.0", true),

‎smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/endpointsV2/EndpointsParamNameMap.java

Lines changed: 41 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,41 @@
1+
/*
2+
* Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved.
3+
*
4+
* Licensed under the Apache License, Version 2.0 (the "License").
5+
* You may not use this file except in compliance with the License.
6+
* A copy of the License is located at
7+
*
8+
* http://aws.amazon.com/apache2.0
9+
*
10+
* or in the "license" file accompanying this file. This file is distributed
11+
* on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
12+
* express or implied. See the License for the specific language governing
13+
* permissions and limitations under the License.
14+
*/
15+
16+
package software.amazon.smithy.typescript.codegen.endpointsV2;
17+
18+
import java.util.Map;
19+
import software.amazon.smithy.utils.MapUtils;
20+
21+
22+
public final class EndpointsParamNameMap {
23+
/**
24+
* Map of EndpointsV2 ruleset param name to existing JSv3 config param names.
25+
*/
26+
private static final Map<String, String> MAPPING = MapUtils.of(
27+
"Region", "region",
28+
"UseFIPS", "useFipsEndpoint",
29+
"UseDualStack", "useDualstackEndpoint",
30+
"ForcePathStyle", "forcePathStyle",
31+
"Accelerate", "useAccelerateEndpoint",
32+
"DisableMRAP", "disableMultiregionAccessPoints",
33+
"UseArnRegion", "useArnRegion"
34+
);
35+
36+
private EndpointsParamNameMap() {}
37+
38+
public static String getLocalName(String endpointsV2ParamName) {
39+
return MAPPING.getOrDefault(endpointsV2ParamName, endpointsV2ParamName);
40+
}
41+
}

0 commit comments

Comments
 (0)