From 87ce208dc0a987ed4e9ee831bac97dee13b9cab9 Mon Sep 17 00:00:00 2001
From: George Fu <kuhe@users.noreply.github.com>
Date: Wed, 1 Mar 2023 19:54:20 +0000
Subject: [PATCH] docs: WIP add doc examples to operations

---
 .../typescript/codegen/CommandGenerator.java  | 31 +++++++++++++++++--
 1 file changed, 28 insertions(+), 3 deletions(-)

diff --git a/smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/CommandGenerator.java b/smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/CommandGenerator.java
index b4cb6030916..a913db2027e 100644
--- a/smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/CommandGenerator.java
+++ b/smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/CommandGenerator.java
@@ -47,6 +47,7 @@
 import software.amazon.smithy.model.traits.DocumentationTrait;
 import software.amazon.smithy.model.traits.ErrorTrait;
 import software.amazon.smithy.model.traits.InternalTrait;
+import software.amazon.smithy.model.traits.ExamplesTrait;
 import software.amazon.smithy.rulesengine.traits.EndpointRuleSetTrait;
 import software.amazon.smithy.typescript.codegen.documentation.StructureExampleGenerator;
 import software.amazon.smithy.typescript.codegen.endpointsV2.EndpointsParamNameMap;
@@ -200,10 +201,12 @@ private void generateClientCommand() {
                 });
     }
 
-    private String getCommandExample(String serviceName, String configName, String commandName, String commandInput,
-            String commandOutput) {
+    private String getCommandExample(
+            String serviceName, String configName, String commandName,
+            String commandInput, String commandOutput
+    ) {
         String packageName = settings.getPackageName();
-        return "@example\n"
+        String exampleDoc = "@example\n"
                 + "Use a bare-bones client and the command you need to make an API call.\n"
                 + "```javascript\n"
                 + String.format("import { %s, %s } from \"%s\"; // ES Modules import%n", serviceName, commandName,
@@ -226,6 +229,13 @@ private String getCommandExample(String serviceName, String configName, String c
                 + String.format("@see {@link %s} for command's `input` shape.%n", commandInput)
                 + String.format("@see {@link %s} for command's `response` shape.%n", commandOutput)
                 + String.format("@see {@link %s | config} for %s's `config` shape.%n", configName, serviceName);
+
+        if (operation.getTrait(ExamplesTrait.class).isPresent()) {
+            List<ExamplesTrait.Example> examples = operation.getTrait(ExamplesTrait.class).get().getExamples();
+            exampleDoc += getOperationExample(examples);
+        }
+
+        return exampleDoc;
     }
 
     private String getThrownExceptions() {
@@ -253,6 +263,21 @@ private String getThrownExceptions() {
         return buffer.toString();
     }
 
+    private String getOperationExample(List<ExamplesTrait.Example> examples) {
+        StringBuilder buffer = new StringBuilder();
+        for (ExamplesTrait.Example example : examples) {
+            buffer
+                .append("\n")
+                .append(String.format("@example %s", example.getTitle()))
+                .append("```javascript")
+                .append(String.format("/* %s */", example.getDocumentation()))
+                .append("some code here")
+                .append("```")
+                .append("\n");
+        }
+        return buffer.toString();
+    }
+
     private void generateCommandConstructor() {
         writer.writeDocs("@public")
                 .openBlock("constructor(readonly input: $T) {", "}", inputType, () -> {