Added Unstable information to manifest

aws · Apr 23, 2021 · 3bbd461 · 3bbd461
1 parent 7d6a773
commit 3bbd461
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 76 deletions.
diff --git a/...gen/smithy-go-codegen/src/main/java/software/amazon/smithy/go/codegen/CodegenVisitor.java b/...gen/smithy-go-codegen/src/main/java/software/amazon/smithy/go/codegen/CodegenVisitor.java
@@ -219,7 +219,7 @@ void execute() {
         writers.flushWriters();
 
         LOGGER.fine("Generating build manifest file");
-        ManifestWriter.writeManifest(settings, fileManifest, dependencies);
+        ManifestWriter.writeManifest(settings, model, fileManifest, dependencies);
     }
 
     @Override

diff --git a/...gen/smithy-go-codegen/src/main/java/software/amazon/smithy/go/codegen/GoModGenerator.java b/...gen/smithy-go-codegen/src/main/java/software/amazon/smithy/go/codegen/GoModGenerator.java
diff --git a/...gen/smithy-go-codegen/src/main/java/software/amazon/smithy/go/codegen/ManifestWriter.java b/...gen/smithy-go-codegen/src/main/java/software/amazon/smithy/go/codegen/ManifestWriter.java
@@ -30,10 +30,13 @@
 import software.amazon.smithy.build.FileManifest;
 import software.amazon.smithy.codegen.core.CodegenException;
 import software.amazon.smithy.codegen.core.SymbolDependency;
+import software.amazon.smithy.model.Model;
 import software.amazon.smithy.model.node.ArrayNode;
+import software.amazon.smithy.model.node.BooleanNode;
 import software.amazon.smithy.model.node.Node;
 import software.amazon.smithy.model.node.ObjectNode;
 import software.amazon.smithy.model.node.StringNode;
+import software.amazon.smithy.model.traits.UnstableTrait;
 
 /**
  * Generates a manifest description of the generated code, minimum go version, and minimum dependencies required.
@@ -48,11 +51,13 @@ private ManifestWriter() {
      * Write the manifest description of the generated code.
      *
      * @param settings     the go settings
+     * @param model        the smithy model
      * @param fileManifest the file manifest
      * @param dependencies the list of symbol dependencies
      */
     public static void writeManifest(
             GoSettings settings,
+            Model model,
             FileManifest fileManifest,
             List<SymbolDependency> dependencies
     ) {
@@ -67,15 +72,17 @@ public static void writeManifest(
         }
         fileManifest.addFile(manifestFile);
 
-        Node generatedJson = buildManifestFile(settings, fileManifest, dependencies);
+        Node generatedJson = buildManifestFile(settings, model, fileManifest, dependencies);
         fileManifest.writeFile(manifestFile.toString(), Node.prettyPrintJson(generatedJson) + "\n");
     }
 
     private static Node buildManifestFile(
             GoSettings settings,
+            Model model,
             FileManifest fileManifest,
             List<SymbolDependency> dependencies
     ) {
+
         List<SymbolDependency> nonStdLib = new ArrayList<>();
         Optional<SymbolDependency> minStandard = Optional.empty();
 
@@ -115,6 +122,8 @@ private static Node buildManifestFile(
                 manifestNodes.put(StringNode.from("go"), StringNode.from(symbolDependency.getVersion())));
         manifestNodes.put(StringNode.from("dependencies"), ObjectNode.objectNode(dependencyNodes));
         manifestNodes.put(StringNode.from("files"), ArrayNode.fromStrings(generatedFiles));
+        manifestNodes.put(StringNode.from("unstable"),
+                BooleanNode.from(settings.getService(model).getTrait(UnstableTrait.class).isPresent()));
 
         return ObjectNode.objectNode(manifestNodes).withDeepSortedKeys();
     }