Allow fly pg connect to work with custom images (#3680)

* Allow `fly pg connect` to work with custom images * remove FLY_DEV check * Revert "remove FLY_DEV check" This reverts commit 3b6734c.
superfly · Jul 8, 2024 · 92b9ea9 · 92b9ea9
1 parent 8d1c191
commit 92b9ea9
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/internal/command/postgres/postgres.go b/internal/command/postgres/postgres.go
@@ -57,6 +57,10 @@ func hasRequiredVersionOnMachines(machines []*fly.Machine, cluster, flex, standa
 			return fmt.Errorf("command is not compatible with this image")
 		}
 
+		if machine.ImageVersion() == "custom" {
+			continue
+		}
+
 		imageVersionStr := machine.ImageVersion()[1:]
 
 		imageVersion, err := version.NewVersion(imageVersionStr)