test recovery

.github/workflows/recover_s3_repository.yml

@@ -1,21 +1,18 @@
-name: . ⚠️⚠️⚠️ Recover S3 Repository back in time ⚠️⚠️⚠️
+name: . ⚠️⚠️⚠️ Recover S3 Repository back in time  ⚠️⚠️⚠️
 
 on:
-  workflow_dispatch:
-    inputs:
-      date_time:
-        description: 'UTC DateTime to recover the S3 repository back in time (MM-DD-YYYY HH:MM:SS +0)'
-        type: string
-        required: true
-      path:
-        description: 'Path under infrastructure_agent folder to recover (w/o leading slash, with trailing slash)'
-        type: string
-        required: true
+  push:
+    branches:
+      - 'NR-229418_disaster_recovery_action_TEST'
 
 env:
-  BUCKET: 'nr-downloads-main'
-  MANDATORY_PREFIX: 'infrastructure_agent/'
+  MANDATORY_PREFIX: '20240108-test-disaster-recovery/'
   IMAGE: 'ghcr.io/newrelic-forks/s3-pit-restore:latest'
+  DATETIME: '01-09-2025 15:06:00 +0000'
+  S3_PATH: 'test1'
+  ENVIRONMENT: 'staging'
+  AWS_REGION: "us-east-1"
+  TEMP_AWS_PROFILE: temp_staging
 
 jobs:
   recover-s3-repository:
@@ -24,41 +21,66 @@ jobs:
     steps:
       - name: Validate datetime
         run: |
-          datetime="${{ github.event.inputs.date_time }}"
+          datetime="${{ env.DATETIME }}"
           # Use Python's strptome (same as s3-pit-restore) to check if it's a valid datetime
           python3 -c "from datetime import datetime; datetime.strptime('$datetime', '%m-%d-%Y %H:%M:%S %z')" 2> /dev/null
-          exit_code=$?
-          
-          if [ $exit_code -ne 0 ]]; then
+          exit_code=$?          
+          if [ $exit_code -ne 0 ]; then
             exit 1
           fi
 
       - name: Validate path input does not have leading slash and has trailing slash
         run: |
-          path="${{ github.event.inputs.path }}"
+          s3_path="${{ env.S3_PATH }}"
           # Check if the path has a leading slash
-          if [[ "$path" == /* ]]; then
+          if [[ "$s3_path" == /* ]]; then
             echo "Invalid path: should not have a leading slash."
             return 1
           fi
       
           # Check if the path has a trailing slash
-          if [[ "$path" != */ ]]; then
-            echo "Invalid path: should have a trailing slash."
+          if [[ "$s3_path" == */ ]]; then
+            echo "Invalid path: should not have a trailing slash."
             return 1
           fi
 
       - name: Checkout repository
         uses: actions/checkout@v4
         with:
           repository: newrelic-forks/s3-pit-restore
-          ref: master
+          ref: NR-229418_aws_credentials
 
-      - name: Run S3 PIT restore
+#      - name: Run S3 PIT restore Production S3
+#        if: ${{ github.event.inputs.environment == 'production' }}
+#        run: |
+#          BUCKET="nr-downloads-main" \
+#          PREFIX="${{ env.MANDATORY_PREFIX }}${{ github.event.inputs.path }}" \
+#          TIME="${{ github.event.inputs.date_time }}" \
+#          IMAGE="${{ env.IMAGE }}" \
+#          make validate-input
+#        env:
+#          AWS_ACCESS_KEY_ID: ${{ secrets.OHAI_AWS_ACCESS_KEY_ID_PRODUCTION }}
+#          AWS_SECRET_ACCESS_KEY: ${{ secrets.OHAI_AWS_SECRET_ACCESS_KEY_PRODUCTION }}
+#          AWS_ROLE_ARN: ${{ secrets.OHAI_AWS_ROLE_ARN_PRODUCTION }}
+#          AWS_ROLE_SESSION_NAME: ${{ secrets.OHAI_AWS_ROLE_SESSION_NAME_PRODUCTION }}
+
+      - name: Setup AWS credentials for Staging
+        if: ${{ env.ENVIRONMENT == 'staging' }}
+        run: |
+            ./setup_aws_credentials.sh
+        env:
+          AWS_ACCESS_KEY_ID: ${{ secrets.OHAI_AWS_ACCESS_KEY_ID_STAGING }}
+          AWS_SECRET_ACCESS_KEY: ${{ secrets.OHAI_AWS_SECRET_ACCESS_KEY_STAGING }}
+          AWS_ROLE_ARN: ${{ secrets.OHAI_AWS_ROLE_ARN_STAGING }}
+          AWS_ROLE_SESSION_NAME: ${{ secrets.OHAI_AWS_ROLE_SESSION_NAME_STAGING }}
+
+      - name: Run S3 PIT restore in Staging S3
+        if: ${{ env.ENVIRONMENT == 'staging' }}
         run: |
-          AWS_PROFILE=primary/okta_onhostintegrations \
-          BUCKET="nr-downloads-main" \
-          PREFIX="${{ env.MANDATORY_PREFIX }}${{ github.event.inputs.path }}" \
-          TIME="${{ github.event.inputs.date_time }}" \
+          BUCKET="nr-downloads-ohai-staging" \
+          PREFIX="${{ env.MANDATORY_PREFIX }}${{ env.S3_PATH }}" \
+          TIME="${{ env.DATETIME }}" \
           IMAGE="${{ env.IMAGE }}" \
+          AWS_PROFILE="${{ env.TEMP_AWS_PROFILE }}" \
           make restore
+