chore: unban all ip on failure to connect

Author: sweatybridge

internal/utils/flags/db_url.go

@@ -171,13 +171,6 @@ func initLoginRole(ctx context.Context, projectRef string, config *pgconn.Config
 }
 
 func initPoolerLogin(ctx context.Context, projectRef string, poolerConfig *pgconn.Config) error {
-	// Prefer password prompt because temp login is unstable using pooler
-	resetUrl := fmt.Sprintf("%s/project/%s/settings/database", utils.GetSupabaseDashboardURL(), projectRef)
-	fmt.Fprintln(os.Stderr, "Forgot your password? Reset it from the Dashboard:", utils.Bold(resetUrl))
-	fmt.Fprint(os.Stderr, "Enter your database password (or leave blank to use temporary login): ")
-	if poolerConfig.Password = credentials.PromptMasked(os.Stdin); len(poolerConfig.Password) > 0 {
-		return nil
-	}
 	poolerUser := poolerConfig.User
 	if err := initLoginRole(ctx, projectRef, poolerConfig); err != nil {
 		return err
@@ -195,10 +188,18 @@ func initPoolerLogin(ctx context.Context, projectRef string, poolerConfig *pgcon
 		return conn.Close(ctx)
 	}
 	notify := utils.NewErrorCallback(func(attempt uint) error {
-		if attempt%3 > 0 {
+		if attempt > 2 {
+			return nil
+		}
+		resp, err := utils.GetSupabase().V1ListAllNetworkBansWithResponse(ctx, projectRef)
+		if err != nil {
+			return errors.Errorf("failed to list network bans: %w", err)
+		} else if resp.JSON201 == nil {
+			return errors.Errorf("unexpected list bans status %d: %s", resp.StatusCode(), string(resp.Body))
+		} else if len(resp.JSON201.BannedIpv4Addresses) == 0 {
 			return nil
 		}
-		return UnbanIP(ctx, projectRef)
+		return UnbanIP(ctx, projectRef, resp.JSON201.BannedIpv4Addresses...)
 	})
 	return backoff.RetryNotify(login, utils.NewBackoffPolicy(ctx), notify)
 }