feat: unlink

cmd/unlink.go

@@ -0,0 +1,33 @@
+package cmd
+
+import (
+	"os"
+	"os/signal"
+
+	"github.com/spf13/afero"
+	"github.com/spf13/cobra"
+	"github.com/supabase/cli/internal/unlink"
+)
+
+var (
+	unlinkCmd = &cobra.Command{
+		GroupID: groupLocalDev,
+		Use:     "unlink",
+		Short:   "Unlink to a Supabase project",
+		PreRunE: func(cmd *cobra.Command, args []string) error {
+			return unlink.PreRun("", afero.NewOsFs())
+		},
+		RunE: func(cmd *cobra.Command, args []string) error {
+			ctx, _ := signal.NotifyContext(cmd.Context(), os.Interrupt)
+			fsys := afero.NewOsFs()
+			return unlink.Run(ctx, dbPassword, fsys)
+		},
+		PostRunE: func(cmd *cobra.Command, args []string) error {
+			return unlink.PostRun("", os.Stdout, afero.NewOsFs())
+		},
+	}
+)
+
+func init() {
+	rootCmd.AddCommand(unlinkCmd)
+}

internal/unlink/unlink.go

@@ -0,0 +1,216 @@
+package unlink
+
+import (
+	"context"
+	"fmt"
+	"io"
+	"os"
+	"strconv"
+	"strings"
+	"sync"
+
+	"github.com/BurntSushi/toml"
+	"github.com/go-errors/errors"
+	"github.com/jackc/pgconn"
+	"github.com/jackc/pgx/v4"
+	"github.com/spf13/afero"
+	"github.com/spf13/viper"
+	"github.com/supabase/cli/internal/migration/repair"
+	"github.com/supabase/cli/internal/utils"
+	"github.com/supabase/cli/internal/utils/credentials"
+	"github.com/supabase/cli/internal/utils/tenant"
+	"github.com/supabase/cli/pkg/api"
+)
+
+var updatedConfig ConfigCopy
+
+type ConfigCopy struct {
+	Api    interface{} `toml:"api"`
+	Db     interface{} `toml:"db"`
+	Pooler interface{} `toml:"db.pooler"`
+}
+
+func (c ConfigCopy) IsEmpty() bool {
+	return c.Api == nil && c.Db == nil && c.Pooler == nil
+}
+
+func PreRun(projectRef string, fsys afero.Fs) error {
+	return utils.LoadConfigFS(fsys)
+}
+
+func Run(ctx context.Context, password string, fsys afero.Fs, options ...func(*pgx.ConnConfig)) error {
+	unlinkServices(ctx, fsys)
+
+	// 3. Save project ref
+	return utils.WriteFile(utils.ProjectRefPath, []byte(""), fsys)
+}
+
+func PostRun(projectRef string, stdout io.Writer, fsys afero.Fs) error {
+	fmt.Fprintln(stdout, "Finished "+utils.Aqua("supabase unlink")+".")
+	if updatedConfig.IsEmpty() {
+		return nil
+	}
+	fmt.Fprintln(os.Stderr, "Local config differs from linked project. Try updating", utils.Bold(utils.ConfigPath))
+	enc := toml.NewEncoder(stdout)
+	enc.Indent = ""
+	if err := enc.Encode(updatedConfig); err != nil {
+		return errors.Errorf("failed to marshal toml config: %w", err)
+	}
+	return nil
+}
+
+func unlinkServices(ctx context.Context, fsys afero.Fs) {
+	// Ignore non-fatal errors linking services
+	var wg sync.WaitGroup
+	wg.Add(6)
+	go func() {
+		defer wg.Done()
+		if err := unlinkDatabaseVersion(ctx, fsys); err != nil && viper.GetBool("DEBUG") {
+			fmt.Fprintln(os.Stderr, err)
+		}
+	}()
+	go func() {
+		defer wg.Done()
+		if err := linkPostgrest(ctx, ""); err != nil && viper.GetBool("DEBUG") {
+			fmt.Fprintln(os.Stderr, err)
+		}
+	}()
+	go func() {
+		defer wg.Done()
+		if err := unlinkPostgrestVersion(ctx, fsys); err != nil && viper.GetBool("DEBUG") {
+			fmt.Fprintln(os.Stderr, err)
+		}
+	}()
+	go func() {
+		defer wg.Done()
+		if err := unlinkGotrueVersion(ctx, fsys); err != nil && viper.GetBool("DEBUG") {
+			fmt.Fprintln(os.Stderr, err)
+		}
+	}()
+	go func() {
+		defer wg.Done()
+		if err := unlinkStorageVersion(ctx, fsys); err != nil && viper.GetBool("DEBUG") {
+			fmt.Fprintln(os.Stderr, err)
+		}
+	}()
+	go func() {
+		defer wg.Done()
+		if err := unlinkPooler(ctx, fsys); err != nil && viper.GetBool("DEBUG") {
+			fmt.Fprintln(os.Stderr, err)
+		}
+	}()
+	wg.Wait()
+}
+
+func linkPostgrest(ctx context.Context, projectRef string) error {
+	resp, err := utils.GetSupabase().GetPostgRESTConfigWithResponse(ctx, projectRef)
+	if err != nil {
+		return errors.Errorf("failed to get postgrest config: %w", err)
+	}
+	if resp.JSON200 == nil {
+		return errors.Errorf("%w: %s", tenant.ErrAuthToken, string(resp.Body))
+	}
+	updateApiConfig(*resp.JSON200)
+	return nil
+}
+
+func unlinkPostgrestVersion(ctx context.Context, fsys afero.Fs) error {
+	return utils.WriteFile(utils.RestVersionPath, []byte(""), fsys)
+}
+
+func updateApiConfig(config api.PostgrestConfigWithJWTSecretResponse) {
+	copy := utils.Config.Api
+	copy.MaxRows = uint(config.MaxRows)
+	copy.ExtraSearchPath = readCsv(config.DbExtraSearchPath)
+	copy.Schemas = readCsv(config.DbSchema)
+	changed := utils.Config.Api.MaxRows != copy.MaxRows ||
+		!utils.SliceEqual(utils.Config.Api.ExtraSearchPath, copy.ExtraSearchPath) ||
+		!utils.SliceEqual(utils.Config.Api.Schemas, copy.Schemas)
+	if changed {
+		updatedConfig.Api = copy
+	}
+}
+
+func readCsv(line string) []string {
+	var result []string
+	tokens := strings.Split(line, ",")
+	for _, t := range tokens {
+		trimmed := strings.TrimSpace(t)
+		if len(trimmed) > 0 {
+			result = append(result, trimmed)
+		}
+	}
+	return result
+}
+
+func unlinkGotrueVersion(ctx context.Context, fsys afero.Fs) error {
+	return utils.WriteFile(utils.GotrueVersionPath, []byte(""), fsys)
+}
+
+func unlinkStorageVersion(ctx context.Context, fsys afero.Fs) error {
+	return utils.WriteFile(utils.StorageVersionPath, []byte(""), fsys)
+}
+
+func unlinkDatabase(ctx context.Context, config pgconn.Config, options ...func(*pgx.ConnConfig)) error {
+	conn, err := utils.ConnectRemotePostgres(ctx, config, options...)
+	if err != nil {
+		return err
+	}
+	defer conn.Close(context.Background())
+	updatePostgresConfig(conn)
+	// If `schema_migrations` doesn't exist on the remote database, create it.
+	return repair.CreateMigrationTable(ctx, conn)
+}
+
+func unlinkDatabaseVersion(ctx context.Context, fsys afero.Fs) error {
+	return utils.WriteFile(utils.PostgresVersionPath, []byte(""), fsys)
+}
+
+func updatePostgresConfig(conn *pgx.Conn) {
+	serverVersion := conn.PgConn().ParameterStatus("server_version")
+	// Safe to assume that supported Postgres version is 10.0 <= n < 100.0
+	majorDigits := len(serverVersion)
+	if majorDigits > 2 {
+		majorDigits = 2
+	}
+	dbMajorVersion, err := strconv.ParseUint(serverVersion[:majorDigits], 10, 7)
+	// Treat error as unchanged
+	if err == nil && uint64(utils.Config.Db.MajorVersion) != dbMajorVersion {
+		copy := utils.Config.Db
+		copy.MajorVersion = uint(dbMajorVersion)
+		updatedConfig.Db = copy
+	}
+}
+
+func unlinkPooler(ctx context.Context, fsys afero.Fs) error {
+	return utils.WriteFile(utils.PoolerUrlPath, []byte(""), fsys)
+}
+
+func updatePoolerConfig(config api.V1PgbouncerConfigResponse) {
+	copy := utils.Config.Db.Pooler
+	if config.PoolMode != nil {
+		copy.PoolMode = utils.PoolMode(*config.PoolMode)
+	}
+	if config.DefaultPoolSize != nil {
+		copy.DefaultPoolSize = uint(*config.DefaultPoolSize)
+	}
+	if config.MaxClientConn != nil {
+		copy.MaxClientConn = uint(*config.MaxClientConn)
+	}
+	changed := utils.Config.Db.Pooler.PoolMode != copy.PoolMode ||
+		utils.Config.Db.Pooler.DefaultPoolSize != copy.DefaultPoolSize ||
+		utils.Config.Db.Pooler.MaxClientConn != copy.MaxClientConn
+	if changed {
+		updatedConfig.Pooler = copy
+	}
+}
+
+func PromptPassword(stdin *os.File) string {
+	fmt.Fprint(os.Stderr, "Enter your database password: ")
+	return credentials.PromptMasked(stdin)
+}
+
+func PromptPasswordAllowBlank(stdin *os.File) string {
+	fmt.Fprint(os.Stderr, "Enter your database password (or leave blank to skip): ")
+	return credentials.PromptMasked(stdin)
+}