From c73979fbc3b585c4641a89c1f297d52172d733f3 Mon Sep 17 00:00:00 2001
From: Ish Shah <ish.shah@wandb.com>
Date: Mon, 2 Dec 2024 15:07:41 -0800
Subject: [PATCH] generate status messages for resource migration

---
 migrate-state.sh | 41 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 41 insertions(+)
 create mode 100644 migrate-state.sh

diff --git a/migrate-state.sh b/migrate-state.sh
new file mode 100644
index 0000000..23fa8ee
--- /dev/null
+++ b/migrate-state.sh
@@ -0,0 +1,41 @@
+#!/bin/bash
+set -e
+
+echo "Starting Terraform state migration..."
+
+# Function to check if a resource exists in terraform state
+check_resource() {
+    terraform state list | grep -q "$1" && return 0 || return 1
+}
+
+# Function to migrate a single resource
+migrate_resource() {
+    local OLD_RESOURCE="$1"
+    local NEW_RESOURCE="$2"
+    
+    if check_resource "$OLD_RESOURCE"; then
+        echo "Migrating ${OLD_RESOURCE} to ${NEW_RESOURCE}"
+        terraform state mv -lock=false "$OLD_RESOURCE" "$NEW_RESOURCE" || {
+            echo "Failed to migrate ${OLD_RESOURCE}"
+            return 1
+        }
+    else
+        echo "Resource ${OLD_RESOURCE} not found in state, skipping..."
+    fi
+}
+
+# Backup the current state
+echo "Creating state backup..."
+terraform state pull > terraform.tfstate.backup.$(date +%Y%m%d-%H%M%S)
+
+# Migrate each resource
+echo "Migrating resources..."
+
+# Migrate the operator release
+migrate_resource "helm_release.operator" "helm_release.operator[0]"
+
+# Migrate the wandb release
+migrate_resource "helm_release.wandb" "helm_release.wandb[0]"
+
+echo "Migration completed!"
+echo "Please review the state and run 'terraform plan' to verify the migration" 
\ No newline at end of file