add unsafe exec resource

pkg/provider/provider.go

@@ -476,6 +476,7 @@ func getResources() map[string]*schema.Resource {
 		"snowflake_tag_association":                         resources.TagAssociation(),
 		"snowflake_tag_masking_policy_association":          resources.TagMaskingPolicyAssociation(),
 		"snowflake_task":                                    resources.Task(),
+		"snowflake_unsafe_migration":                        resources.UnsafeMigration(),
 		"snowflake_user":                                    resources.User(),
 		"snowflake_user_ownership_grant":                    resources.UserOwnershipGrant(),
 		"snowflake_user_public_keys":                        resources.UserPublicKeys(),

pkg/resources/testdata/TestAcc_UnsafeMigration_basic/test.tf

@@ -0,0 +1,4 @@
+resource "snowflake_unsafe_migration" "migration" {
+  up   = var.up
+  down = var.down
+}

pkg/resources/testdata/TestAcc_UnsafeMigration_basic/variables.tf

@@ -0,0 +1,7 @@
+variable "up" {
+  type = string
+}
+
+variable "down" {
+  type = string
+}

pkg/resources/testdata/TestAcc_UnsafeMigration_downChanged/test.tf

@@ -0,0 +1,4 @@
+resource "snowflake_unsafe_migration" "migration" {
+  up   = var.up
+  down = var.down
+}

pkg/resources/testdata/TestAcc_UnsafeMigration_downChanged/variables.tf

@@ -0,0 +1,7 @@
+variable "up" {
+  type = string
+}
+
+variable "down" {
+  type = string
+}

pkg/resources/testdata/TestAcc_UnsafeMigration_grants/test.tf

@@ -0,0 +1,4 @@
+resource "snowflake_unsafe_migration" "migration" {
+  up   = var.up
+  down = var.down
+}

pkg/resources/testdata/TestAcc_UnsafeMigration_grants/variables.tf

@@ -0,0 +1,7 @@
+variable "up" {
+  type = string
+}
+
+variable "down" {
+  type = string
+}

pkg/resources/testdata/TestAcc_UnsafeMigration_upChanged/test.tf

@@ -0,0 +1,4 @@
+resource "snowflake_unsafe_migration" "migration" {
+  up   = var.up
+  down = var.down
+}

pkg/resources/testdata/TestAcc_UnsafeMigration_upChanged/variables.tf

@@ -0,0 +1,7 @@
+variable "up" {
+  type = string
+}
+
+variable "down" {
+  type = string
+}

pkg/resources/unsafe_migration.go

@@ -0,0 +1,74 @@
+package resources
+
+import (
+	"context"
+	"database/sql"
+	"github.com/Snowflake-Labs/terraform-provider-snowflake/pkg/sdk"
+	"github.com/hashicorp/go-uuid"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
+	"log"
+)
+
+var unsafeMigrationSchema = map[string]*schema.Schema{
+	"up": {
+		Type:        schema.TypeString,
+		Required:    true,
+		ForceNew:    true,
+		Description: "TODO",
+	},
+	"down": {
+		Type:        schema.TypeString,
+		Required:    true,
+		Description: "TODO",
+	},
+}
+
+func UnsafeMigration() *schema.Resource {
+	return &schema.Resource{
+		Create: ApplyUnsafeMigration,
+		Read:   schema.Noop,
+		Delete: RevertUnsafeMigration,
+		Update: schema.Noop,
+
+		Schema: unsafeMigrationSchema,
+	}
+}
+
+func ApplyUnsafeMigration(d *schema.ResourceData, meta interface{}) error {
+	db := meta.(*sql.DB)
+	ctx := context.Background()
+	client := sdk.NewClientFromDB(db)
+
+	id, err := uuid.GenerateUUID()
+	if err != nil {
+		return err
+	}
+
+	upStatement := d.Get("up").(string)
+	_, err = client.ExecUnsafe(ctx, upStatement)
+	if err != nil {
+		return err
+	}
+
+	d.SetId(id)
+	log.Printf(`[DEBUG] SQL "%s" applied successfully\n`, upStatement)
+
+	return nil
+}
+
+func RevertUnsafeMigration(d *schema.ResourceData, meta interface{}) error {
+	db := meta.(*sql.DB)
+	ctx := context.Background()
+	client := sdk.NewClientFromDB(db)
+
+	downStatement := d.Get("down").(string)
+	_, err := client.ExecUnsafe(ctx, downStatement)
+	if err != nil {
+		return err
+	}
+
+	d.SetId("")
+	log.Printf(`[DEBUG] SQL "%s" applied successfully\n`, downStatement)
+
+	return nil
+}