feat: schema code generation

Author: kuhe

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

@@ -27,13 +27,16 @@
 import java.util.HashSet;
 import java.util.List;
 import java.util.Map;
+import java.util.Objects;
 import java.util.Optional;
 import java.util.Set;
 import java.util.TreeSet;
 import java.util.function.Consumer;
 import java.util.function.Function;
 import java.util.stream.Collectors;
 import software.amazon.smithy.build.FileManifest;
+import software.amazon.smithy.codegen.core.ReservedWords;
+import software.amazon.smithy.codegen.core.ReservedWordsBuilder;
 import software.amazon.smithy.codegen.core.Symbol;
 import software.amazon.smithy.codegen.core.SymbolProvider;
 import software.amazon.smithy.model.Model;
@@ -57,6 +60,7 @@
 import software.amazon.smithy.typescript.codegen.endpointsV2.RuleSetParameterFinder;
 import software.amazon.smithy.typescript.codegen.integration.ProtocolGenerator;
 import software.amazon.smithy.typescript.codegen.integration.RuntimeClientPlugin;
+import software.amazon.smithy.typescript.codegen.schema.SchemaGenerationAllowlist;
 import software.amazon.smithy.typescript.codegen.sections.CommandBodyExtraCodeSection;
 import software.amazon.smithy.typescript.codegen.sections.CommandConstructorCodeSection;
 import software.amazon.smithy.typescript.codegen.sections.CommandPropertiesCodeSection;
@@ -74,6 +78,7 @@
 final class CommandGenerator implements Runnable {
 
     static final String COMMANDS_FOLDER = "commands";
+    static final String SCHEMAS_FOLDER = "schemas";
 
     private final TypeScriptSettings settings;
     private final Model model;
@@ -89,6 +94,9 @@ final class CommandGenerator implements Runnable {
     private final ProtocolGenerator protocolGenerator;
     private final ApplicationProtocol applicationProtocol;
     private final SensitiveDataFinder sensitiveDataFinder;
+    private final ReservedWords reservedWords = new ReservedWordsBuilder()
+        .loadWords(Objects.requireNonNull(TypeScriptClientCodegenPlugin.class.getResource("reserved-words.txt")))
+        .build();
 
     CommandGenerator(
             TypeScriptSettings settings,
@@ -488,7 +496,10 @@ private void generateCommandMiddlewareResolver(String configType) {
         );
         {
             // Add serialization and deserialization plugin.
-            writer.write("$T(config, this.serialize, this.deserialize),", serde);
+            if (!SchemaGenerationAllowlist.contains(service.getId())) {
+                writer.write("$T(config, this.serialize, this.deserialize),", serde);
+            }
+
             // EndpointsV2
             writer.addImport(
                 "getEndpointPlugin",
@@ -654,22 +665,37 @@ private void addCommandSpecificPlugins() {
         }
     }
 
+    private void writeSchemaSerde() {
+        String operationSchema = reservedWords.escape(operation.getId().getName());
+        writer.addRelativeImport(operationSchema, null, Paths.get(
+            ".", CodegenUtils.SOURCE_FOLDER, SCHEMAS_FOLDER, "schemas"
+        ));
+        writer.write("""
+            .sc($L)""",
+            operationSchema
+        );
+    }
+
     private void writeSerde() {
-        writer
-            .write(".ser($L)", getSerdeDispatcher(true))
-            .write(".de($L)", getSerdeDispatcher(false));
+        if (SchemaGenerationAllowlist.contains(service.getId())) {
+            writeSchemaSerde();
+        } else {
+            writer
+                .write(".ser($L)", getSerdeDispatcher(true))
+                .write(".de($L)", getSerdeDispatcher(false));
+        }
     }
 
     private String getSerdeDispatcher(boolean isInput) {
         if (protocolGenerator == null) {
             return "() => { throw new Error(\"No supported protocol was found\"); }";
         } else {
             String serdeFunctionName = isInput
-                    ? ProtocolGenerator.getSerFunctionShortName(symbol)
-                    : ProtocolGenerator.getDeserFunctionShortName(symbol);
+                ? ProtocolGenerator.getSerFunctionShortName(symbol)
+                : ProtocolGenerator.getDeserFunctionShortName(symbol);
             writer.addRelativeImport(serdeFunctionName, null,
-                    Paths.get(".", CodegenUtils.SOURCE_FOLDER, ProtocolGenerator.PROTOCOLS_FOLDER,
-                            ProtocolGenerator.getSanitizedName(protocolGenerator.getName())));
+                Paths.get(".", CodegenUtils.SOURCE_FOLDER, ProtocolGenerator.PROTOCOLS_FOLDER,
+                    ProtocolGenerator.getSanitizedName(protocolGenerator.getName())));
             return serdeFunctionName;
         }
     }

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

@@ -54,6 +54,7 @@
 import software.amazon.smithy.typescript.codegen.integration.ProtocolGenerator;
 import software.amazon.smithy.typescript.codegen.integration.RuntimeClientPlugin;
 import software.amazon.smithy.typescript.codegen.integration.TypeScriptIntegration;
+import software.amazon.smithy.typescript.codegen.schema.SchemaGenerator;
 import software.amazon.smithy.typescript.codegen.validation.LongValidator;
 import software.amazon.smithy.typescript.codegen.validation.ReplaceLast;
 import software.amazon.smithy.utils.MapUtils;
@@ -111,10 +112,15 @@ public TypeScriptCodegenContext createContext(CreateContextDirective<TypeScriptS
                 directive.model(),
                 directive.service(),
                 directive.settings());
+
         ApplicationProtocol applicationProtocol = protocolGenerator == null
                 ? ApplicationProtocol.createDefaultHttpApplicationProtocol()
                 : protocolGenerator.getApplicationProtocol();
 
+        if (null != protocolGenerator) {
+            directive.settings().setProtocol(protocolGenerator.getProtocol());
+        }
+
         return TypeScriptCodegenContext.builder()
                 .model(directive.model())
                 .settings(directive.settings())
@@ -281,6 +287,8 @@ private void generateClient(GenerateServiceDirective<TypeScriptCodegenContext, T
             }
         }
 
+        new SchemaGenerator(model, fileManifest, settings, symbolProvider).run();
+
         if (containedOperations.stream().anyMatch(operation -> operation.hasTrait(PaginatedTrait.ID))) {
             PaginationGenerator.writeIndex(model, service, fileManifest);
             delegator.useFileWriter(PaginationGenerator.PAGINATION_INTERFACE_FILE, paginationWriter ->