From 587d36fd6ca8c0021537f417f7533a229dda4377 Mon Sep 17 00:00:00 2001
From: "diamondburned (Forefront)" <datutbrus@gmail.com>
Date: Wed, 19 Feb 2020 19:36:00 -0800
Subject: [PATCH] Gateway: Open now fails at once, only Reconnect retries
 forever.

---
 gateway/gateway.go | 74 ++++++++++++++++++----------------------------
 1 file changed, 28 insertions(+), 46 deletions(-)

diff --git a/gateway/gateway.go b/gateway/gateway.go
index 5e78656..5b556be 100644
--- a/gateway/gateway.go
+++ b/gateway/gateway.go
@@ -177,60 +177,42 @@ func (g *Gateway) Reconnect() error {
 		WSDebug("Gateway is closed asynchronously. Goroutine may not be exited.")
 	}
 
-	// Actually a reconnect at this point.
-	return g.Open()
-}
-
-func (g *Gateway) Open() error {
-	// Reconnect timeout
-	// ctx, cancel := context.WithTimeout(context.Background(), g.WSTimeout)
-	// defer cancel()
-
-	// TODO: this could be of some use.
-	ctx := context.Background()
-
 	for i := 0; ; i++ {
-		/* Context doesn't time out.
+		WSDebug("Trying to dial, attempt", i)
 
-		// Check if context is expired
-		if err := ctx.Err(); err != nil {
-			// Close the connection
-			g.Close()
+		// Condition: err == ErrInvalidSession:
+		// If the connection is rate limited (documented behavior):
+		// https://discordapp.com/developers/docs/topics/gateway#rate-limiting
 
-			// Don't bother if it's expired
-			return err
-		}
-
-		*/
-
-		WSDebug("Trying to dial...", i)
-
-		// Reconnect to the Gateway
-		if err := g.WS.Dial(ctx); err != nil {
-			// Save the error, retry again
-			g.ErrorLog(errors.Wrap(err, "Failed to reconnect"))
-			continue
-		}
-
-		WSDebug("Trying to start...", i)
-
-		// Try to resume the connection
-		if err := g.Start(); err != nil {
-			// If the connection is rate limited (documented behavior):
-			// https://discordapp.com/developers/docs/topics/gateway#rate-limiting
-			if err == ErrInvalidSession {
-				continue
-			}
-
-			// Else, keep retrying
-			g.ErrorLog(errors.Wrap(err, "Failed to start gateway"))
+		if err := g.Open(); err != nil && err != ErrInvalidSession {
+			g.ErrorLog(errors.Wrap(err, "Failed to open gateway"))
 			continue
 		}
 
 		WSDebug("Started after attempt:", i)
-		// Started successfully, return
-		return nil
+		break
 	}
+
+	return nil
+}
+
+func (g *Gateway) Open() error {
+	ctx := context.Background()
+
+	// Reconnect to the Gateway
+	if err := g.WS.Dial(ctx); err != nil {
+		return errors.Wrap(err, "Failed to reconnect")
+	}
+
+	WSDebug("Trying to start...")
+
+	// Try to resume the connection
+	if err := g.Start(); err != nil {
+		return err
+	}
+
+	// Started successfully, return
+	return nil
 }
 
 // Start authenticates with the websocket, or resume from a dead Websocket