Fix incorrect links for paginated trait in docgen

Author: joewyz

smithy-docgen/src/main/java/software/amazon/smithy/docgen/DocSymbolProvider.java

@@ -209,6 +209,8 @@ public Symbol structureShape(StructureShape shape) {
             var operation = ioToOperation.get(shape.getId());
             builder.definitionFile(getDefinitionFile(serviceShape, operation));
             builder.putProperty(OPERATION_PROPERTY, operation);
+            // Input and output structures should use the operation as its link_id
+            builder.putProperty(LINK_ID_PROPERTY, getLinkId(getShapeName(serviceShape, operation)));
         }
         return builder.build();
     }
@@ -230,16 +232,22 @@ public Symbol unionShape(UnionShape shape) {
 
     @Override
     public Symbol memberShape(MemberShape shape) {
-        var builder = getSymbolBuilder(shape)
-                .definitionFile(getDefinitionFile(serviceShape, model.expectShape(shape.getId().withoutMember())));
+        var builder = getSymbolBuilder(shape);
 
-        Optional<String> containerLinkId = model.expectShape(shape.getContainer())
+        ShapeId containerId = shape.getContainer();
+        Optional<String> containerLinkId = model.expectShape(containerId)
                 .accept(this)
                 .getProperty(LINK_ID_PROPERTY, String.class);
         if (containerLinkId.isPresent()) {
             var linkId = containerLinkId.get() + "-" + getLinkId(getShapeName(serviceShape, shape));
             builder.putProperty(LINK_ID_PROPERTY, linkId);
         }
+        // Definition file of input / output structure members should be the operation file.
+        if (ioToOperation.containsKey(containerId)) {
+            builder.definitionFile(getDefinitionFile(serviceShape, ioToOperation.get(containerId)));
+        } else {
+            builder.definitionFile(getDefinitionFile(serviceShape, model.expectShape(shape.getId().withoutMember())));
+        }
         return builder.build();
     }
 

smithy-docgen/src/test/java/software/amazon/smithy/docgen/generators/OperationGeneratorTest.java

@@ -142,4 +142,50 @@ public void testGeneratesSnippetsFromDiscoveredConfig(@TempDir Path tempDir) {
                 ```
                 :::"""));
     }
+
+    @Test
+    public void testPaginatedOperation(@TempDir Path tempDir) {
+        MockManifest manifest = new MockManifest();
+        FileManifest sharedManifest = FileManifest.create(tempDir);
+        ObjectNode settings = settings().toBuilder()
+                .withoutMember("snippetConfigs")
+                .build();
+        sharedManifest.writeFile("snippets/snippets.json", IoUtils.readUtf8Url(SNIPPETS_FILE));
+        execute(manifest, sharedManifest, settings);
+        var operationDocs = manifest.expectFileString("/content/operations/PaginatedOperation.md");
+        System.out.println(operationDocs);
+        assertThat(operationDocs,
+                containsString(
+                        """
+                                (paginatedoperation)=
+                                # PaginatedOperation
+
+                                Placeholder documentation for `smithy.example#PaginatedOperation`
+
+                                :::{important}
+                                This operation returns partial results in pages, whose maximum size may be
+                                configured with [pageSize](./PaginatedOperation.md#paginatedoperation-pagesize). Each request may return an [output token](./PaginatedOperation.md#paginatedoperation-nexttoken) that may be used as an [input token](./PaginatedOperation.md#paginatedoperation-nexttoken) in subsequent requests to fetch the next page of results. If the operation does not return an [output token](./PaginatedOperation.md#paginatedoperation-nexttoken), that means that there are no more results. If the operation returns a repeated [output token](./PaginatedOperation.md#paginatedoperation-nexttoken), there MAY be more results later.
+                                :::
+
+                                (paginatedoperation-request-members)=
+                                ## Request Members
+
+                                **nextToken (String)**
+                                : Placeholder documentation for `smithy.example#PaginatedOperationInput$nextToken`
+
+                                **pageSize (Integer)**
+                                : Placeholder documentation for `smithy.example#PaginatedOperationInput$pageSize`
+
+
+                                (paginatedoperation-response-members)=
+                                ## Response Members
+
+                                **items (List\\<String\\>)**
+                                : Placeholder documentation for `smithy.example#PaginatedOperationOutput$items`
+
+                                **nextToken (String)**
+                                : Placeholder documentation for `smithy.example#PaginatedOperationOutput$nextToken`
+                                """));
+
+    }
 }

smithy-docgen/src/test/resources/software/amazon/smithy/docgen/generators/operation-generator.smithy

@@ -6,6 +6,7 @@ service TestService {
     operations: [
         NoSnippets
         BasicOperation
+        PaginatedOperation
     ]
 }
 
@@ -83,3 +84,25 @@ operation NoSnippets {
 structure BasicError {
     message: String
 }
+
+@paginated(inputToken: "nextToken", outputToken: "nextToken", pageSize: "pageSize", items: "items")
+operation PaginatedOperation {
+    input: PaginatedOperationInput
+
+    output := {
+        items: Items
+
+        nextToken: String
+    }
+}
+
+@input
+structure PaginatedOperationInput {
+    nextToken: String
+
+    pageSize: Integer
+}
+
+list Items {
+    member: String
+}