From a270b04ad68f942c5b5217f737720bcb77af3b72 Mon Sep 17 00:00:00 2001
From: Xun Zhang <xunzh@amazon.com>
Date: Fri, 24 Feb 2023 14:23:50 -0800
Subject: [PATCH] Publish snapshots to maven via GHA

Signed-off-by: Xun Zhang <xunzh@amazon.com>
---
 .github/workflows/maven-publish.yml | 39 +++++++++++++++++++++++++++++
 plugin/build.gradle                 | 11 ++++++++
 2 files changed, 50 insertions(+)
 create mode 100644 .github/workflows/maven-publish.yml

diff --git a/.github/workflows/maven-publish.yml b/.github/workflows/maven-publish.yml
new file mode 100644
index 0000000000..647fdd1420
--- /dev/null
+++ b/.github/workflows/maven-publish.yml
@@ -0,0 +1,39 @@
+name: Publish snapshots to maven
+
+on:
+  workflow_dispatch:
+  push:
+    branches: [
+        main
+          1.*
+          2.*
+    ]
+
+jobs:
+  build-and-publish-snapshots:
+    strategy:
+      fail-fast: false
+    if: github.repository == 'opensearch-project/ml-commons'
+    runs-on: ubuntu-latest
+
+    permissions:
+      id-token: write
+      contents: write
+
+    steps:
+      - uses: actions/setup-java@v3
+        with:
+          distribution: temurin # Temurin is a distribution of adoptium
+          java-version: 17
+      - uses: actions/checkout@v3
+      - uses: aws-actions/configure-aws-credentials@v1
+        with:
+          role-to-assume: ${{ secrets.PUBLISH_SNAPSHOTS_ROLE }}
+          aws-region: us-east-1
+      - name: publish snapshots to maven
+        run: |
+          export SONATYPE_USERNAME=$(aws secretsmanager get-secret-value --secret-id maven-snapshots-username --query SecretString --output text)
+          export SONATYPE_PASSWORD=$(aws secretsmanager get-secret-value --secret-id maven-snapshots-password --query SecretString --output text)
+          echo "::add-mask::$SONATYPE_USERNAME"
+          echo "::add-mask::$SONATYPE_PASSWORD"
+          ./gradlew publishPluginZipPublicationToSnapshotsRepository
\ No newline at end of file
diff --git a/plugin/build.gradle b/plugin/build.gradle
index d3fbb04e58..685877e728 100644
--- a/plugin/build.gradle
+++ b/plugin/build.gradle
@@ -78,6 +78,17 @@ publishing {
             }
         }
     }
+
+    repositories {
+        maven {
+            name = "Snapshots"
+            url = "https://aws.oss.sonatype.org/content/repositories/snapshots"
+            credentials {
+                username "$System.env.SONATYPE_USERNAME"
+                password "$System.env.SONATYPE_PASSWORD"
+            }
+        }
+    }
 }
 
 compileJava {