feat: add supabase resolve

cmd/resolve.go

@@ -0,0 +1,34 @@
+package cmd
+
+import (
+	"errors"
+
+	"github.com/spf13/cobra"
+	"github.com/supabase/cli/internal/resolve"
+)
+
+var (
+	appliedTimestamp    *string
+	rolledBackTimestamp *string
+
+	resolveCmd = &cobra.Command{
+		Use:   "resolve",
+		Short: "Resolve issues with migration history on the deploy database.",
+		RunE: func(cmd *cobra.Command, args []string) error {
+			if appliedTimestamp != nil {
+				return resolve.ResolveApplied(*appliedTimestamp)
+			} else if rolledBackTimestamp != nil {
+				return resolve.ResolveRolledBack(*rolledBackTimestamp)
+			} else {
+				return errors.New("Must set either --applied or --rolled-back.")
+			}
+		},
+	}
+)
+
+func init() {
+	resolveCmd.Flags().StringVar(appliedTimestamp, "applied", "", "Migration timestamp to be recorded as applied.")
+	resolveCmd.Flags().StringVar(rolledBackTimestamp, "rolled-back", "", "Migration timestamp to be recorded as rolled back.")
+
+	rootCmd.AddCommand(resolveCmd)
+}

internal/deploy/deploy.go

@@ -24,7 +24,7 @@ func Deploy() error {
 	}
 	defer conn.Close(context.Background())
 
-	rows, err := conn.Query(ctx, "SELECT version FROM supabase_migrations.schema_migrations")
+	rows, err := conn.Query(ctx, "SELECT version FROM supabase_migrations.schema_migrations ORDER BY version")
 	if err != nil {
 		return errors.New("❌ supabase_migrations.schema_migrations table does not exist.")
 	}

internal/link/link.go

@@ -123,7 +123,7 @@ func Link(url string) error {
 	fmt.Println("Done pulling images.")
 
 	// sync `migrations`
-	if rows, err := conn.Query(ctx, "SELECT version FROM supabase_migrations.schema_migrations"); err == nil {
+	if rows, err := conn.Query(ctx, "SELECT version FROM supabase_migrations.schema_migrations ORDER BY version"); err == nil {
 		// supabase_migrations.schema_migrations exists.
 		fmt.Println("supabase_migrations.schema_migrations exists on the deploy database.")
 

internal/resolve/resolve.go

@@ -0,0 +1,62 @@
+package resolve
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	"os"
+
+	pgx "github.com/jackc/pgx/v4"
+)
+
+var ctx = context.TODO()
+
+func ResolveApplied(timestamp string) error {
+	url := os.Getenv("SUPABASE_DEPLOY_DB_URL")
+	if url == "" {
+		return errors.New("❌ SUPABASE_DEPLOY_DB_URL is not set.")
+	}
+
+	conn, err := pgx.Connect(ctx, url)
+	if err != nil {
+		return err
+	}
+	defer conn.Close(context.Background())
+
+	if _, err := conn.Query(
+		ctx,
+		"INSERT INTO supabase_migrations.schema_migrations(version) VALUES ($1)",
+		timestamp,
+	); err != nil {
+		return err
+	}
+
+	fmt.Println("Finished supabase resolve.")
+
+	return nil
+}
+
+func ResolveRolledBack(timestamp string) error {
+	url := os.Getenv("SUPABASE_DEPLOY_DB_URL")
+	if url == "" {
+		return errors.New("❌ SUPABASE_DEPLOY_DB_URL is not set.")
+	}
+
+	conn, err := pgx.Connect(ctx, url)
+	if err != nil {
+		return err
+	}
+	defer conn.Close(context.Background())
+
+	if _, err := conn.Query(
+		ctx,
+		"DELETE FROM supabase_migrations.schema_migrations WHERE version = $1",
+		timestamp,
+	); err != nil {
+		return err
+	}
+
+	fmt.Println("Finished supabase resolve.")
+
+	return nil
+}