From a813b26628a2d459c9d3fe268d94e320b38efc90 Mon Sep 17 00:00:00 2001
From: Sichan Yoo <chanyoo@amazon.com>
Date: Fri, 28 Jun 2024 14:35:29 -0700
Subject: [PATCH] Add AWS copyrightNotice to smithy-build.json codegen & to
 testUtil.

---
 codegen/protocol-test-codegen-local/build.gradle.kts           | 3 ++-
 codegen/protocol-test-codegen/build.gradle.kts                 | 3 ++-
 codegen/sdk-codegen/build.gradle.kts                           | 3 ++-
 .../software/amazon/smithy/aws/swift/codegen/TestUtils.kt      | 1 +
 4 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/codegen/protocol-test-codegen-local/build.gradle.kts b/codegen/protocol-test-codegen-local/build.gradle.kts
index 8ad3486ef2e..9968e624432 100644
--- a/codegen/protocol-test-codegen-local/build.gradle.kts
+++ b/codegen/protocol-test-codegen-local/build.gradle.kts
@@ -66,7 +66,8 @@ fun generateSmithyBuild(tests: List<CodegenTest>): String {
                       "build": {
                         "rootProject": true
                       },
-                      "mergeModels": false
+                      "mergeModels": false,
+                      "copyrightNotice": "//\n// Copyright Amazon.com Inc. or its affiliates.\n// All Rights Reserved.\n//\n// SPDX-License-Identifier: Apache-2.0\n//\n\n// Code generated by smithy-swift-codegen. DO NOT EDIT!\n\n"
                       ${it.extraConfig ?: ""}
                  }
                }
diff --git a/codegen/protocol-test-codegen/build.gradle.kts b/codegen/protocol-test-codegen/build.gradle.kts
index ccd88947832..712a4ef8540 100644
--- a/codegen/protocol-test-codegen/build.gradle.kts
+++ b/codegen/protocol-test-codegen/build.gradle.kts
@@ -115,7 +115,8 @@ fun generateSmithyBuild(tests: List<ProtocolTest>): String {
                   "homepage": "https://docs.amplify.aws/",
                   "swiftVersion": "5.7.0",
                   "useInterceptors": true,
-                  "mergeModels": false
+                  "mergeModels": false,
+                  "copyrightNotice": "//\n// Copyright Amazon.com Inc. or its affiliates.\n// All Rights Reserved.\n//\n// SPDX-License-Identifier: Apache-2.0\n//\n\n// Code generated by smithy-swift-codegen. DO NOT EDIT!\n\n"
                 }
               }
             }"""
diff --git a/codegen/sdk-codegen/build.gradle.kts b/codegen/sdk-codegen/build.gradle.kts
index ca2acb8656e..66c4963d7d7 100644
--- a/codegen/sdk-codegen/build.gradle.kts
+++ b/codegen/sdk-codegen/build.gradle.kts
@@ -115,7 +115,8 @@ fun generateSmithyBuild(services: List<AwsService>): String {
                           "rootProject": $buildStandaloneSdk
                       },
                       "useInterceptors": ${interceptorsServices.contains(service.packageName)},
-                      "mergeModels": true
+                      "mergeModels": true,
+                      "copyrightNotice": "//\n// Copyright Amazon.com Inc. or its affiliates.\n// All Rights Reserved.\n//\n// SPDX-License-Identifier: Apache-2.0\n//\n\n// Code generated by smithy-swift-codegen. DO NOT EDIT!\n\n"
                     }
                 }
             }
diff --git a/codegen/smithy-aws-swift-codegen/src/test/kotlin/software/amazon/smithy/aws/swift/codegen/TestUtils.kt b/codegen/smithy-aws-swift-codegen/src/test/kotlin/software/amazon/smithy/aws/swift/codegen/TestUtils.kt
index 0ff8d4c2143..0c6f4af8896 100644
--- a/codegen/smithy-aws-swift-codegen/src/test/kotlin/software/amazon/smithy/aws/swift/codegen/TestUtils.kt
+++ b/codegen/smithy-aws-swift-codegen/src/test/kotlin/software/amazon/smithy/aws/swift/codegen/TestUtils.kt
@@ -87,6 +87,7 @@ class TestUtils {
                 .withMember("author", Node.from("Amazon Web Services"))
                 .withMember("gitRepo", Node.from("https://github.com/awslabs/aws-sdk-swift.git"))
                 .withMember("swiftVersion", Node.from("5.7"))
+                .withMember("copyrightNotice", Node.from("//\n// Copyright Amazon.com Inc. or its affiliates.\n// All Rights Reserved.\n//\n// SPDX-License-Identifier: Apache-2.0\n//\n\n// Code generated by smithy-swift-codegen. DO NOT EDIT!\n\n"))
                 .build()
         }
         fun listFilesFromManifest(manifest: MockManifest): String {