Smithy generator for paginator

generated/src/aws-cpp-sdk-s3/include/aws/s3/S3ClientPagination.h

@@ -0,0 +1,26 @@
+/**
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0.
+ */
+
+#pragma once
+#include <aws/core/utils/pagination/Paginator.h>
+#include <aws/s3/S3Client.h>
+#include <aws/s3/model/pagination/ListBucketsPaginationTraits.h>
+#include <aws/s3/model/pagination/ListDirectoryBucketsPaginationTraits.h>
+#include <aws/s3/model/pagination/ListObjectsV2PaginationTraits.h>
+#include <aws/s3/model/pagination/ListPartsPaginationTraits.h>
+
+namespace Aws {
+namespace S3 {
+
+using ListBucketsPaginator =
+    Aws::Utils::Pagination::PagePaginator<S3Client, Model::ListBucketsRequest, Pagination::ListBucketsPaginationTraits>;
+using ListDirectoryBucketsPaginator =
+    Aws::Utils::Pagination::PagePaginator<S3Client, Model::ListDirectoryBucketsRequest, Pagination::ListDirectoryBucketsPaginationTraits>;
+using ListObjectsV2Paginator =
+    Aws::Utils::Pagination::PagePaginator<S3Client, Model::ListObjectsV2Request, Pagination::ListObjectsV2PaginationTraits>;
+using ListPartsPaginator = Aws::Utils::Pagination::PagePaginator<S3Client, Model::ListPartsRequest, Pagination::ListPartsPaginationTraits>;
+
+}  // namespace S3
+}  // namespace Aws

generated/src/aws-cpp-sdk-s3/include/aws/s3/model/pagination/ListBucketsPaginationTraits.h

@@ -0,0 +1,33 @@
+/**
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0.
+ */
+
+#pragma once
+#include <aws/s3/S3Client.h>
+#include <aws/s3/S3_EXPORTS.h>
+#include <aws/s3/model/ListBucketsRequest.h>
+#include <aws/s3/model/ListBucketsResult.h>
+
+namespace Aws {
+namespace S3 {
+namespace Pagination {
+
+struct ListBucketsPaginationTraits {
+  using RequestType = Model::ListBucketsRequest;
+  using ResultType = Model::ListBucketsResult;
+  using OutcomeType = Model::ListBucketsOutcome;
+  using ClientType = S3Client;
+
+  static OutcomeType Invoke(ClientType& client, const RequestType& request) { return client.ListBuckets(request); }
+
+  static bool HasMoreResults(const ResultType& result) { return !result.GetContinuationToken().empty(); }
+
+  static void SetNextRequest(const ResultType& result, RequestType& request) {
+    request.SetContinuationToken(result.GetContinuationToken());
+  }
+};
+
+}  // namespace Pagination
+}  // namespace S3
+}  // namespace Aws

generated/src/aws-cpp-sdk-s3/include/aws/s3/model/pagination/ListDirectoryBucketsPaginationTraits.h

@@ -0,0 +1,33 @@
+/**
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0.
+ */
+
+#pragma once
+#include <aws/s3/S3Client.h>
+#include <aws/s3/S3_EXPORTS.h>
+#include <aws/s3/model/ListDirectoryBucketsRequest.h>
+#include <aws/s3/model/ListDirectoryBucketsResult.h>
+
+namespace Aws {
+namespace S3 {
+namespace Pagination {
+
+struct ListDirectoryBucketsPaginationTraits {
+  using RequestType = Model::ListDirectoryBucketsRequest;
+  using ResultType = Model::ListDirectoryBucketsResult;
+  using OutcomeType = Model::ListDirectoryBucketsOutcome;
+  using ClientType = S3Client;
+
+  static OutcomeType Invoke(ClientType& client, const RequestType& request) { return client.ListDirectoryBuckets(request); }
+
+  static bool HasMoreResults(const ResultType& result) { return !result.GetContinuationToken().empty(); }
+
+  static void SetNextRequest(const ResultType& result, RequestType& request) {
+    request.SetContinuationToken(result.GetContinuationToken());
+  }
+};
+
+}  // namespace Pagination
+}  // namespace S3
+}  // namespace Aws

generated/src/aws-cpp-sdk-s3/include/aws/s3/model/pagination/ListObjectsV2PaginationTraits.h

@@ -0,0 +1,33 @@
+/**
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0.
+ */
+
+#pragma once
+#include <aws/s3/S3Client.h>
+#include <aws/s3/S3_EXPORTS.h>
+#include <aws/s3/model/ListObjectsV2Request.h>
+#include <aws/s3/model/ListObjectsV2Result.h>
+
+namespace Aws {
+namespace S3 {
+namespace Pagination {
+
+struct ListObjectsV2PaginationTraits {
+  using RequestType = Model::ListObjectsV2Request;
+  using ResultType = Model::ListObjectsV2Result;
+  using OutcomeType = Model::ListObjectsV2Outcome;
+  using ClientType = S3Client;
+
+  static OutcomeType Invoke(ClientType& client, const RequestType& request) { return client.ListObjectsV2(request); }
+
+  static bool HasMoreResults(const ResultType& result) { return !result.GetNextContinuationToken().empty(); }
+
+  static void SetNextRequest(const ResultType& result, RequestType& request) {
+    request.SetContinuationToken(result.GetNextContinuationToken());
+  }
+};
+
+}  // namespace Pagination
+}  // namespace S3
+}  // namespace Aws

generated/src/aws-cpp-sdk-s3/include/aws/s3/model/pagination/ListPartsPaginationTraits.h

@@ -0,0 +1,33 @@
+/**
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0.
+ */
+
+#pragma once
+#include <aws/s3/S3Client.h>
+#include <aws/s3/S3_EXPORTS.h>
+#include <aws/s3/model/ListPartsRequest.h>
+#include <aws/s3/model/ListPartsResult.h>
+
+namespace Aws {
+namespace S3 {
+namespace Pagination {
+
+struct ListPartsPaginationTraits {
+  using RequestType = Model::ListPartsRequest;
+  using ResultType = Model::ListPartsResult;
+  using OutcomeType = Model::ListPartsOutcome;
+  using ClientType = S3Client;
+
+  static OutcomeType Invoke(ClientType& client, const RequestType& request) { return client.ListParts(request); }
+
+  static bool HasMoreResults(const ResultType& result) { return !result.GetNextPartNumberMarker().empty(); }
+
+  static void SetNextRequest(const ResultType& result, RequestType& request) {
+    request.SetPartNumberMarker(result.GetNextPartNumberMarker());
+  }
+};
+
+}  // namespace Pagination
+}  // namespace S3
+}  // namespace Aws

generated/tests/s3-gen-tests/S3PaginationCompilationTests.cpp

@@ -0,0 +1,25 @@
+/**
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0.
+ */
+
+// Header compilation test for S3 pagination headers
+// This test ensures all generated pagination headers compile successfully
+
+#include <aws/s3/S3ClientPagination.h>
+#include <aws/s3/model/pagination/ListBucketsPaginationTraits.h>
+#include <aws/s3/model/pagination/ListDirectoryBucketsPaginationTraits.h>
+#include <aws/s3/model/pagination/ListPartsPaginationTraits.h>
+#include <aws/s3/model/pagination/ListObjectsV2PaginationTraits.h>
+
+#include <aws/testing/AwsCppSdkGTestSuite.h>
+
+class S3PaginationCompilationTest : public Aws::Testing::AwsCppSdkGTestSuite
+{
+};
+
+TEST_F(S3PaginationCompilationTest, S3PaginationHeadersCompile)
+{
+        // Test passes if compilation succeeds
+        SUCCEED();
+}

tools/code-generation/smithy/codegen/cpp-pagination-codegen/build.gradle.kts

@@ -0,0 +1,18 @@
+plugins {
+    `java-library`
+}
+
+repositories {
+    mavenLocal()
+    mavenCentral()
+}
+
+dependencies {
+    implementation("software.amazon.smithy:smithy-model:1.51.0")
+    implementation("software.amazon.smithy:smithy-codegen-core:1.51.0")
+    implementation("software.amazon.smithy:smithy-aws-traits:1.51.0")
+    implementation("software.amazon.smithy:smithy-waiters:1.51.0")
+    implementation("software.amazon.smithy:smithy-rules-engine:1.51.0")
+    implementation("software.amazon.smithy:smithy-aws-endpoints:1.51.0")
+    implementation("software.amazon.smithy:smithy-aws-iam-traits:1.51.0")
+}

tools/code-generation/smithy/codegen/cpp-pagination-codegen/src/main/java/com/amazonaws/util/awsclientsmithygenerator/generators/ClientCodegenSettings.java

@@ -0,0 +1,20 @@
+/**
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0.
+ */
+package com.amazonaws.util.awsclientsmithygenerator.generators;
+
+import software.amazon.smithy.model.node.ObjectNode;
+import software.amazon.smithy.model.shapes.ShapeId;
+
+public class ClientCodegenSettings {
+    private final ShapeId service;
+
+    public ClientCodegenSettings(ObjectNode settings) {
+        this.service = ShapeId.from(settings.expectStringMember("service").getValue());
+    }
+
+    public ShapeId getService() {
+        return service;
+    }
+}

tools/code-generation/smithy/codegen/cpp-pagination-codegen/src/main/java/com/amazonaws/util/awsclientsmithygenerator/generators/CompilationTestParser.java

@@ -0,0 +1,45 @@
+/**
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0.
+ */
+package com.amazonaws.util.awsclientsmithygenerator.generators;
+
+import software.amazon.smithy.build.PluginContext;
+import software.amazon.smithy.model.shapes.*;
+import software.amazon.smithy.aws.traits.ServiceTrait;
+import com.amazonaws.util.awsclientsmithygenerator.generators.CppWriter;
+import com.amazonaws.util.awsclientsmithygenerator.generators.ServiceNameUtil;
+import java.util.*;
+import java.util.function.Consumer;
+
+public class CompilationTestParser<T> {
+    private final ServiceShape service;
+    private final CppWriterDelegator writerDelegator;
+    private final String testType;
+    private final Consumer<CppWriter> renderFunction;
+    private final Map<String, String> c2jMap;
+
+    public CompilationTestParser(PluginContext context, ServiceShape service, List<T> operations, 
+                                String testType, Consumer<CppWriter> renderFunction, Map<String, String> c2jMap) {
+        this.service = service;
+        this.writerDelegator = new CppWriterDelegator(context.getFileManifest());
+        this.testType = testType;
+        this.renderFunction = renderFunction;
+        this.c2jMap = c2jMap;
+    }
+
+    public void run() {
+        generateCompilationTest();
+        writerDelegator.flushWriters();
+    }
+
+    private void generateCompilationTest() {
+        String serviceName = ServiceNameUtil.getServiceName(service);
+        String c2jServiceName = ServiceNameUtil.getC2jServiceName(service, c2jMap);
+
+        writerDelegator.useFileWriter(
+            "generated/tests/" + c2jServiceName + "-gen-tests/" + serviceName + testType + "CompilationTests.cpp",
+            renderFunction
+        );
+    }
+}

tools/code-generation/smithy/codegen/cpp-pagination-codegen/src/main/java/com/amazonaws/util/awsclientsmithygenerator/generators/CppImportContainer.java

@@ -0,0 +1,36 @@
+/**
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0.
+ */
+package com.amazonaws.util.awsclientsmithygenerator.generators;
+
+import software.amazon.smithy.codegen.core.ImportContainer;
+import software.amazon.smithy.codegen.core.Symbol;
+import java.util.HashSet;
+import java.util.Set;
+
+/**
+ * CppImportContainer tracks symbols used in code generation.
+ * For C++, we handle includes manually via writeInclude method.
+ */
+public class CppImportContainer implements ImportContainer {
+    private final Set<String> imports = new HashSet<>();
+
+    @Override
+    public void importSymbol(Symbol symbol, String alias) {
+        // Track the symbol's namespace for potential future use
+        if (symbol.getNamespace() != null && !symbol.getNamespace().isEmpty()) {
+            imports.add(symbol.getNamespace());
+        }
+    }
+
+    public Set<String> getImports() {
+        return imports;
+    }
+
+    @Override
+    public String toString() {
+        // Return empty string since we handle includes manually
+        return "";
+    }
+}

tools/code-generation/smithy/codegen/cpp-pagination-codegen/src/main/java/com/amazonaws/util/awsclientsmithygenerator/generators/CppWriter.java

@@ -0,0 +1,45 @@
+/**
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0.
+ */
+package com.amazonaws.util.awsclientsmithygenerator.generators;
+
+import software.amazon.smithy.codegen.core.SymbolWriter;
+
+public class CppWriter extends SymbolWriter<CppWriter, CppImportContainer> {
+
+    public CppWriter() {
+        super(new CppImportContainer());
+
+        // Add custom formatter for symbols
+        putFormatter('T', (arg, indent) -> {
+            if (!(arg instanceof software.amazon.smithy.codegen.core.Symbol symbol)) {
+                throw new software.amazon.smithy.codegen.core.CodegenException("Expected a symbol but got " + arg);
+            }
+
+            // Record our symbol so we can generate appropriate includes later if needed
+            getImportContainer().importSymbol(symbol, null);
+
+            // For C++, use namespace::name syntax (e.g. Aws::S3::Model::ListBucketsRequest)
+            if (symbol.getNamespace() != null && !symbol.getNamespace().isEmpty()) {
+                return symbol.getNamespace() + "::" + symbol.getName();
+            }
+            return symbol.getName();
+        });
+    }
+
+    public CppWriter writeInclude(String header) {
+        write("#include <$L>", header);
+        return this;
+    }
+
+    public CppWriter writeNamespaceOpen(String namespace) {
+        openBlock("namespace $L\n{", namespace);
+        return this;
+    }
+
+    public CppWriter writeNamespaceClose(String namespace) {
+        closeBlock("} // namespace $L", namespace);
+        return this;
+    }
+}