Add TraceFile creation

Author: Chase Coalwell

smithy-typescript-codegen/build.gradle.kts

@@ -18,6 +18,7 @@ extra["displayName"] = "Smithy :: Typescript :: Codegen"
 extra["moduleName"] = "software.amazon.smithy.typescript.codegen"
 
 dependencies {
+    api("software.amazon.smithy:smithy-aws-traits:[1.7.2, 2.0[")
     api("software.amazon.smithy:smithy-codegen-core:[1.5.0, 2.0[")
     api("software.amazon.smithy:smithy-waiters:[1.5.0, 2.0[")
     implementation("software.amazon.smithy:smithy-protocol-test-traits:[1.5.0, 2.0[")

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

@@ -25,16 +25,22 @@
 import java.util.ServiceLoader;
 import java.util.Set;
 import java.util.TreeSet;
+import java.util.UUID;
 import java.util.logging.Logger;
+import software.amazon.smithy.aws.traits.ServiceTrait;
 import software.amazon.smithy.build.FileManifest;
 import software.amazon.smithy.build.PluginContext;
 import software.amazon.smithy.codegen.core.Symbol;
 import software.amazon.smithy.codegen.core.SymbolDependency;
 import software.amazon.smithy.codegen.core.SymbolProvider;
 import software.amazon.smithy.codegen.core.TopologicalIndex;
+import software.amazon.smithy.codegen.core.trace.ArtifactDefinitions;
+import software.amazon.smithy.codegen.core.trace.TraceMetadata;
+import software.amazon.smithy.codegen.core.trace.TracingSymbolProvider;
 import software.amazon.smithy.model.Model;
 import software.amazon.smithy.model.knowledge.TopDownIndex;
 import software.amazon.smithy.model.neighbor.Walker;
+import software.amazon.smithy.model.node.Node;
 import software.amazon.smithy.model.shapes.OperationShape;
 import software.amazon.smithy.model.shapes.ServiceShape;
 import software.amazon.smithy.model.shapes.Shape;
@@ -93,6 +99,7 @@ class CodegenVisitor extends ShapeVisitor.Default<Void> {
                         runtimePlugins.add(runtimePlugin);
                     });
                 });
+
         // Sort the integrations in specified order.
         integrations.sort(Comparator.comparingInt(TypeScriptIntegration::getOrder));
 
@@ -113,19 +120,55 @@ class CodegenVisitor extends ShapeVisitor.Default<Void> {
         fileManifest = context.getFileManifest();
         LOGGER.info(() -> String.format("Generating TypeScript %s for service %s",
                 settings.generateClient() ? "client" : "server", service.getId()));
+        // Resolve the nullable protocol generator and application protocol.
+        protocolGenerator = resolveProtocolGenerator(integrations, service, settings);
+        applicationProtocol = protocolGenerator == null
+                ? ApplicationProtocol.createDefaultHttpApplicationProtocol()
+                : protocolGenerator.getApplicationProtocol();
 
         // Decorate the symbol provider using integrations.
         SymbolProvider resolvedProvider = artifactType.createSymbolProvider(model, settings);
         for (TypeScriptIntegration integration : integrations) {
             resolvedProvider = integration.decorateSymbolProvider(settings, model, resolvedProvider);
         }
-        symbolProvider = SymbolProvider.cache(resolvedProvider);
 
-        // Resolve the nullable protocol generator and application protocol.
-        protocolGenerator = resolveProtocolGenerator(integrations, service, settings);
-        applicationProtocol = protocolGenerator == null
-                ? ApplicationProtocol.createDefaultHttpApplicationProtocol()
-                : protocolGenerator.getApplicationProtocol();
+        // Make the symbol provider a cachingSymbolProvider.
+        SymbolProvider cachedProvider = SymbolProvider.cache(resolvedProvider);
+        // Defining Definitions for TraceFile Generation.
+        ArtifactDefinitions artifactDefinitions = ArtifactDefinitions.builder()
+                .addType(TypeScriptShapeLinkProvider.FIELD_TYPE,
+                        "Field declaration (includes enum constants)")
+                .addType(TypeScriptShapeLinkProvider.METHOD_TYPE, "Method declaration")
+                .addType(TypeScriptShapeLinkProvider.TYPE_TYPE,
+                        "Class, interface (including annotation type), or enum declaration")
+                .addTag(TypeScriptShapeLinkProvider.SERVICE_TAG, "Service client")
+                .addTag(TypeScriptShapeLinkProvider.REQUEST_TAG, "AWS SDK request type")
+                .addTag(TypeScriptShapeLinkProvider.RESPONSE_TAG, "AWS SDK response type")
+                .addTag(TypeScriptShapeLinkProvider.SERIALIZER_TAG, "Command serializer")
+                .addTag(TypeScriptShapeLinkProvider.DESERIALIZER_TAG, "Command deserializer")
+                .build();
+
+        String serviceId;
+        if (service.hasTrait(ServiceTrait.class)) {
+            serviceId = service.getTrait(ServiceTrait.class).get().getSdkId();
+        } else {
+            serviceId = service.getId().getName();
+        }
+        TraceMetadata artifactMetadata = TraceMetadata.builder()
+                .setTimestampAsNow()
+                .id(serviceId)
+                .version(UUID.randomUUID().toString())
+                .type("TypeScript")
+                .build();
+
+
+        // Decorate the symbol provider using the trace file generator.
+        symbolProvider = TracingSymbolProvider.builder()
+                .symbolProvider(cachedProvider)
+                .metadata(artifactMetadata)
+                .artifactDefinitions(artifactDefinitions)
+                .shapeLinkCreator(new TypeScriptShapeLinkProvider())
+                .build();
 
         writers = new TypeScriptDelegator(settings, model, fileManifest, symbolProvider, integrations);
     }
@@ -212,6 +255,13 @@ void execute() {
                     settings, model, protocol, symbolProvider, writers, protocolGenerator).run();
         }
 
+        // Write the TraceFile.
+        TracingSymbolProvider traceProvider = (TracingSymbolProvider) symbolProvider;
+        String traceName = symbolProvider.toSymbol(service).getName().replace("Client", "")
+                .toLowerCase() + ".trace.json";
+        fileManifest.writeFile(traceName,
+                Node.prettyPrintJson(traceProvider.buildTraceFile().toNode()));
+
         // Write each pending writer.
         LOGGER.fine("Flushing TypeScript writers");
         List<SymbolDependency> dependencies = writers.getDependencies();
@@ -372,6 +422,10 @@ private void generateClient(ServiceShape shape) {
         boolean hasPaginatedOperation = false;
 
         for (OperationShape operation : containedOperations) {
+            OperationShape finalOperation = operation;
+            writers.useShapeWriter(operation, commandWriter -> new CommandGenerator(
+                    settings, model, finalOperation, symbolProvider, commandWriter,
+                    runtimePlugins, protocolGenerator, applicationProtocol).run());
             if (operation.hasTrait(PaginatedTrait.ID)) {
                 hasPaginatedOperation = true;
                 String outputFilename = PaginationGenerator.getOutputFilelocation(operation);
@@ -428,7 +482,7 @@ private void generateCommands(ServiceShape shape) {
         TopDownIndex topDownIndex = TopDownIndex.of(model);
         Set<OperationShape> containedOperations = new TreeSet<>(topDownIndex.getContainedOperations(shape));
         for (OperationShape operation : containedOperations) {
-            // Right now this only generates stubs
+            // Right now this only generates stubs.
             if (settings.generateClient()) {
                 writers.useShapeWriter(operation, commandWriter -> new CommandGenerator(
                         settings, model, operation, symbolProvider, commandWriter,

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

@@ -250,6 +250,7 @@ public Symbol operationShape(OperationShape shape) {
         // Add input and output type symbols (XCommandInput / XCommandOutput).
         builder.putProperty("inputType", intermediate.toBuilder().name(commandName + "Input").build());
         builder.putProperty("outputType", intermediate.toBuilder().name(commandName + "Output").build());
+
         return builder.build();
     }
 
@@ -327,14 +328,20 @@ private Symbol.Builder addSmithyUseImport(Symbol.Builder builder, String name, S
 
     @Override
     public Symbol unionShape(UnionShape shape) {
-        return createObjectSymbolBuilder(shape).build();
+        return createObjectSymbolBuilder(shape)
+                .putProperty("SymbolProvider", this)
+                .build();
     }
 
     @Override
     public Symbol memberShape(MemberShape shape) {
         Shape targetShape = model.getShape(shape.getTarget())
                 .orElseThrow(() -> new CodegenException("Shape not found: " + shape.getTarget()));
-        Symbol targetSymbol = toSymbol(targetShape);
+        Symbol targetSymbol = toSymbol(targetShape)
+                .toBuilder()
+                .putProperty("model", model)
+                .putProperty("SymbolProvider", this)
+                .build();
 
         if (targetSymbol.getProperties().containsKey(EnumTrait.class.getName())) {
             return createMemberSymbolWithEnumTarget(targetSymbol);
@@ -383,14 +390,20 @@ private Symbol.Builder createObjectSymbolBuilder(Shape shape) {
     }
 
     private Symbol.Builder createSymbolBuilder(Shape shape, String typeName) {
-        return Symbol.builder().putProperty("shape", shape).name(typeName);
+        return Symbol.builder()
+                .putProperty("shape", shape)
+                .putProperty("traceFileNamespace", moduleNameDelegator.formatModuleName(shape, null))
+                .putProperty("traceFileNamespaceDelimiter", "/")
+                .name(typeName);
     }
 
     private Symbol.Builder createSymbolBuilder(Shape shape, String typeName, String namespace) {
         return Symbol.builder()
                 .putProperty("shape", shape)
                 .name(typeName)
-                .namespace(namespace, "/");
+                .namespace(namespace, "/")
+                .putProperty("traceFileNamespace", moduleNameDelegator.formatModuleName(shape, null))
+                .putProperty("traceFileNamespaceDelimiter", "/");
     }
 
     private Symbol.Builder createGeneratedSymbolBuilder(Shape shape, String typeName, String namespace) {