Merge a0fda70 into 641eb1c

Author: aezell

internal/command/launch/plan/postgres_test.go

@@ -111,6 +111,14 @@ func (m *mockUIEXClient) RestoreManagedClusterBackup(ctx context.Context, cluste
 	return uiex.RestoreManagedClusterBackupResponse{}, nil
 }
 
+func (m *mockUIEXClient) CreateAttachment(ctx context.Context, clusterId string, input uiex.CreateAttachmentInput) (uiex.CreateAttachmentResponse, error) {
+	return uiex.CreateAttachmentResponse{}, nil
+}
+
+func (m *mockUIEXClient) DeleteAttachment(ctx context.Context, clusterId string, appName string) (uiex.DeleteAttachmentResponse, error) {
+	return uiex.DeleteAttachmentResponse{}, nil
+}
+
 func (m *mockUIEXClient) CreateBuild(ctx context.Context, in uiex.CreateBuildRequest) (*uiex.BuildResponse, error) {
 	return &uiex.BuildResponse{}, nil
 }

internal/command/mpg/attach.go

@@ -281,6 +281,15 @@ func runAttach(ctx context.Context) error {
 		return err
 	}
 
+	// Create attachment record to track the cluster-app relationship
+	attachInput := uiex.CreateAttachmentInput{
+		AppName: appName,
+	}
+	if _, err := uiexClient.CreateAttachment(ctx, cluster.Id, attachInput); err != nil {
+		// Log warning but don't fail - the secret was set successfully
+		fmt.Fprintf(io.ErrOut, "Warning: failed to create attachment record: %v\n", err)
+	}
+
 	fmt.Fprintf(io.Out, "\nPostgres cluster %s is being attached to %s\n", cluster.Id, appName)
 	fmt.Fprintf(io.Out, "The following secret was added to %s:\n  %s=%s\n", appName, variableName, connectionUri)
 

internal/command/mpg/detach.go

@@ -0,0 +1,91 @@
+package mpg
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/spf13/cobra"
+	"github.com/superfly/flyctl/internal/appconfig"
+	"github.com/superfly/flyctl/internal/command"
+	"github.com/superfly/flyctl/internal/flag"
+	"github.com/superfly/flyctl/internal/flyutil"
+	"github.com/superfly/flyctl/internal/uiexutil"
+	"github.com/superfly/flyctl/iostreams"
+)
+
+func newDetach() *cobra.Command {
+	const (
+		short = "Detach a managed Postgres cluster from an app"
+		long  = short + ". " +
+			`This command will remove the attachment record linking the app to the cluster.
+Note: This does NOT remove any secrets from the app. Use 'fly secrets unset' to remove secrets.`
+		usage = "detach <CLUSTER ID>"
+	)
+
+	cmd := command.New(usage, short, long, runDetach,
+		command.RequireSession,
+		command.RequireAppName,
+	)
+	cmd.Args = cobra.MaximumNArgs(1)
+
+	flag.Add(cmd,
+		flag.App(),
+		flag.AppConfig(),
+		flag.Yes(),
+	)
+
+	return cmd
+}
+
+func runDetach(ctx context.Context) error {
+	// Check token compatibility early
+	if err := validateMPGTokenCompatibility(ctx); err != nil {
+		return err
+	}
+
+	var (
+		clusterId = flag.FirstArg(ctx)
+		appName   = appconfig.NameFromContext(ctx)
+		client    = flyutil.ClientFromContext(ctx)
+		io        = iostreams.FromContext(ctx)
+	)
+
+	// Get app details to determine which org it belongs to
+	app, err := client.GetAppBasic(ctx, appName)
+	if err != nil {
+		return fmt.Errorf("failed retrieving app %s: %w", appName, err)
+	}
+
+	appOrgSlug := app.Organization.RawSlug
+	if appOrgSlug != "" && clusterId == "" {
+		fmt.Fprintf(io.Out, "Listing clusters in organization %s\n", appOrgSlug)
+	}
+
+	// Get cluster details
+	cluster, _, err := ClusterFromArgOrSelect(ctx, clusterId, appOrgSlug)
+	if err != nil {
+		return fmt.Errorf("failed retrieving cluster %s: %w", clusterId, err)
+	}
+
+	clusterOrgSlug := cluster.Organization.Slug
+
+	// Verify that the app and cluster are in the same organization
+	if appOrgSlug != clusterOrgSlug {
+		return fmt.Errorf("app %s is in organization %s, but cluster %s is in organization %s. They must be in the same organization",
+			appName, appOrgSlug, cluster.Id, clusterOrgSlug)
+	}
+
+	uiexClient := uiexutil.ClientFromContext(ctx)
+
+	// Delete the attachment record
+	_, err = uiexClient.DeleteAttachment(ctx, cluster.Id, appName)
+	if err != nil {
+		return fmt.Errorf("failed to detach: %w", err)
+	}
+
+	fmt.Fprintf(io.Out, "\nPostgres cluster %s has been detached from %s\n", cluster.Id, appName)
+	fmt.Fprintf(io.Out, "Note: This only removes the attachment record. Any secrets (like DATABASE_URL) are still set on the app.\n")
+	fmt.Fprintf(io.Out, "Use 'fly secrets unset DATABASE_URL -a %s' to remove the connection string.\n", appName)
+
+	return nil
+}

internal/command/mpg/list.go

@@ -3,10 +3,12 @@ package mpg
 import (
 	"context"
 	"fmt"
+	"strings"
 
 	"github.com/spf13/cobra"
 
 	"github.com/superfly/flyctl/gql"
+	"github.com/superfly/flyctl/internal/uiex"
 	"github.com/superfly/flyctl/iostreams"
 
 	"github.com/superfly/flyctl/internal/command"
@@ -95,8 +97,22 @@ func runList(ctx context.Context) error {
 			cluster.Region,
 			cluster.Status,
 			cluster.Plan,
+			formatAttachedApps(cluster.AttachedApps),
 		})
 	}
 
-	return render.Table(out, "", rows, "ID", "Name", "Org", "Region", "Status", "Plan")
+	return render.Table(out, "", rows, "ID", "Name", "Org", "Region", "Status", "Plan", "Attached Apps")
+}
+
+// formatAttachedApps formats the list of attached apps for display
+func formatAttachedApps(apps []uiex.AttachedApp) string {
+	if len(apps) == 0 {
+		return "<no attached apps>"
+	}
+
+	names := make([]string, len(apps))
+	for i, app := range apps {
+		names[i] = app.Name
+	}
+	return strings.Join(names, ", ")
 }

internal/command/mpg/mpg.go

@@ -76,6 +76,7 @@ func New() *cobra.Command {
 		newProxy(),
 		newConnect(),
 		newAttach(),
+		newDetach(),
 		newStatus(),
 		newList(),
 		newCreate(),