Skip to content

Commit 0ab146e

Browse files
committed
Add Endpoints v2 logic to DirectedCodegen
It was in CodegenVisitor, which has been repalced by DirectedCodegen in this branch.
1 parent 695ca39 commit 0ab146e

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/DirectedTypeScriptCodegen.java

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,8 @@
4646
import software.amazon.smithy.model.shapes.ShapeId;
4747
import software.amazon.smithy.model.traits.PaginatedTrait;
4848
import software.amazon.smithy.model.validation.ValidationEvent;
49+
import software.amazon.smithy.rulesengine.traits.EndpointRuleSetTrait;
50+
import software.amazon.smithy.typescript.codegen.endpointsV2.EndpointsV2Generator;
4951
import software.amazon.smithy.typescript.codegen.integration.ProtocolGenerator;
5052
import software.amazon.smithy.typescript.codegen.integration.RuntimeClientPlugin;
5153
import software.amazon.smithy.typescript.codegen.integration.TypeScriptIntegration;
@@ -160,6 +162,7 @@ public void generateService(GenerateServiceDirective<TypeScriptCodegenContext, T
160162
}
161163
if (settings.generateClient() || settings.generateServerSdk()) {
162164
generateCommands(directive);
165+
generateEndpointV2(directive);
163166
}
164167

165168
if (settings.generateServerSdk()) {
@@ -168,7 +171,6 @@ public void generateService(GenerateServiceDirective<TypeScriptCodegenContext, T
168171

169172
ProtocolGenerator protocolGenerator = directive.context().protocolGenerator();
170173
SymbolProvider symbolProvider = directive.symbolProvider();
171-
List<TypeScriptIntegration> integrations = directive.context().integrations();
172174
if (protocolGenerator != null) {
173175
LOGGER.info("Generating serde for protocol " + protocolGenerator.getName() + " on " + service.getId());
174176
String fileName = Paths.get(CodegenUtils.SOURCE_FOLDER, ProtocolGenerator.PROTOCOLS_FOLDER,
@@ -297,6 +299,14 @@ private void generateCommands(GenerateServiceDirective<TypeScriptCodegenContext,
297299
}
298300
}
299301

302+
private void generateEndpointV2(GenerateServiceDirective<TypeScriptCodegenContext, TypeScriptSettings> directive) {
303+
if (!directive.shape().hasTrait(EndpointRuleSetTrait.class)) {
304+
return;
305+
}
306+
307+
new EndpointsV2Generator(directive.settings(), directive.model(), directive.fileManifest()).run();
308+
}
309+
300310
private void generateServiceInterface(GenerateServiceDirective<TypeScriptCodegenContext,
301311
TypeScriptSettings> directive) {
302312
ServiceShape service = directive.shape();

0 commit comments

Comments
 (0)