feat: save artifact size metrics task

build-plugins/build-support/build.gradle.kts

@@ -30,7 +30,7 @@ dependencies {
 gradlePlugin {
     plugins {
         create("artifact-size-metrics") {
-            id = "artifact-size-metrics"
+            id = "aws.sdk.kotlin.gradle.artifactsizemetrics"
             implementationClass = "aws.sdk.kotlin.gradle.plugins.artifactsizemetrics.ArtifactSizeMetricsPlugin"
         }
     }

build-plugins/build-support/src/main/kotlin/aws/sdk/kotlin/gradle/plugins/artifactsizemetrics/ArtifactSizeMetricsPlugin.kt

@@ -4,7 +4,7 @@
  */
 package aws.sdk.kotlin.gradle.plugins.artifactsizemetrics
 
-import org.gradle.api.GradleException
+import aws.sdk.kotlin.gradle.util.verifyRootProject
 import org.gradle.api.Plugin
 import org.gradle.api.Project
 import org.gradle.api.tasks.TaskProvider
@@ -22,9 +22,7 @@ internal const val S3_ARTIFACT_SIZE_METRICS_BUCKET = "artifact-size-metrics" //
  */
 class ArtifactSizeMetricsPlugin : Plugin<Project> {
     override fun apply(target: Project) {
-        if (target != target.rootProject) {
-            throw GradleException("${this::class.java} can only be applied to the root project")
-        }
+        target.verifyRootProject { "${this::class.java} can only be applied to the root project" }
 
         target.extensions.create("artifactSizeMetrics", ArtifactSizeMetricsPluginConfig::class.java)
 
@@ -36,6 +34,7 @@ class ArtifactSizeMetricsPlugin : Plugin<Project> {
         target.tasks.register<CollectDelegatedArtifactSizeMetrics>("collectDelegatedArtifactSizeMetrics") { group = TASK_GROUP }
         target.tasks.register<AnalyzeArtifactSizeMetrics>("analyzeArtifactSizeMetrics") { group = TASK_GROUP }
         target.tasks.register<PutArtifactSizeMetricsInCloudWatch>("putArtifactSizeMetricsInCloudWatch") { group = TASK_GROUP }
+        target.tasks.register<SaveArtifactSizeMetrics>("saveArtifactSizeMetrics") { group = TASK_GROUP }
     }
 }
 

build-plugins/build-support/src/main/kotlin/aws/sdk/kotlin/gradle/plugins/artifactsizemetrics/PutArtifactSizeMetricsInCloudWatch.kt

@@ -4,7 +4,7 @@
  */
 package aws.sdk.kotlin.gradle.plugins.artifactsizemetrics
 
-import aws.sdk.kotlin.gradle.util.AwsSdkGradleException
+import aws.sdk.kotlin.gradle.util.stringPropertyNotNull
 import aws.sdk.kotlin.services.cloudwatch.CloudWatchClient
 import aws.sdk.kotlin.services.cloudwatch.model.Dimension
 import aws.sdk.kotlin.services.cloudwatch.model.MetricDatum
@@ -36,9 +36,7 @@ internal abstract class PutArtifactSizeMetricsInCloudWatch : DefaultTask() {
     fun put() {
         val currentTime = Instant.now()
         val pluginConfig = project.rootProject.extensions.getByType(ArtifactSizeMetricsPluginConfig::class.java)
-        val releaseTag = project.findProperty("release")?.toString()?.also {
-            check(it.isNotEmpty()) { "The release property is set to empty \"-Prelease=\" (no value set). Please specify a value." }
-        } ?: throw AwsSdkGradleException("The release property is not set. Please set a value: \"-Prelease=YOUR_RELEASE_VALUE\"")
+        val releaseTag = project.stringPropertyNotNull("release")
 
         val metrics = metricsFile
             .get()

build-plugins/build-support/src/main/kotlin/aws/sdk/kotlin/gradle/plugins/artifactsizemetrics/SaveArtifactSizeMetrics.kt

@@ -0,0 +1,55 @@
+/*
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+package aws.sdk.kotlin.gradle.plugins.artifactsizemetrics
+
+import aws.sdk.kotlin.gradle.util.stringPropertyNotNull
+import aws.sdk.kotlin.services.s3.S3Client
+import aws.sdk.kotlin.services.s3.putObject
+import aws.smithy.kotlin.runtime.content.ByteStream
+import kotlinx.coroutines.runBlocking
+import org.gradle.api.DefaultTask
+import org.gradle.api.file.RegularFileProperty
+import org.gradle.api.tasks.InputFile
+import org.gradle.api.tasks.TaskAction
+
+/**
+ * Puts artifact size metrics in S3 to save them.
+ * We put them in CloudWatch also (as metrics) but CloudWatch only keeps metrics temporarily.
+ */
+internal abstract class SaveArtifactSizeMetrics : DefaultTask() {
+    /**
+     * File containing the project's artifact size metrics.
+     */
+    @get:InputFile
+    abstract val metricsFile: RegularFileProperty
+
+    init {
+        metricsFile.convention(project.layout.buildDirectory.file(OUTPUT_PATH + "artifact-size-metrics.csv"))
+    }
+
+    private val pluginConfig = project.rootProject.extensions.getByType(ArtifactSizeMetricsPluginConfig::class.java)
+
+    @TaskAction
+    fun save() {
+        val releaseTag = project.stringPropertyNotNull("release")
+        val artifactSizeMetrics = ByteStream.fromString(metricsFile.get().asFile.readText())
+
+        runBlocking {
+            S3Client.fromEnvironment().use { s3 ->
+                s3.putObject {
+                    bucket = S3_ARTIFACT_SIZE_METRICS_BUCKET
+                    key = "${pluginConfig.projectRepositoryName}-latest-release.csv"
+                    body = artifactSizeMetrics
+                }
+
+                s3.putObject {
+                    bucket = S3_ARTIFACT_SIZE_METRICS_BUCKET
+                    key = "${pluginConfig.projectRepositoryName}-$releaseTag-release.csv"
+                    body = artifactSizeMetrics
+                }
+            }
+        }
+    }
+}

build-plugins/build-support/src/main/kotlin/aws/sdk/kotlin/gradle/util/ProjectExt.kt

@@ -65,4 +65,14 @@ public inline fun Project.verifyRootProject(lazyMessage: () -> Any) {
     }
 }
 
+/**
+ * Will access a project property as a String, only if it is not null or not empty.
+ * @throws AwsSdkGradleException If the property is empty or null
+ * @return The property as a String
+ */
+fun Project.stringPropertyNotNull(property: String): String =
+    findProperty(property)?.toString()?.also {
+        check(it.isNotEmpty()) { "The $property property is set to empty \"-P$property=\" (no value set). Please specify a value." }
+    } ?: throw AwsSdkGradleException("The $property property is not set. Please set a value: \"-P$property=YOUR_VALUE\"")
+
 class AwsSdkGradleException(message: String) : GradleException(message)