SeanOMik
/
nimcord
Archived
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Reconnecting was really broken, now fixed, but still needs more testing

This commit is contained in:
SeanOMik 2020-06-22 00:42:59 -05:00
parent 133975b59d
commit c6399b31eb
No known key found for this signature in database
GPG Key ID: FA4D55AC05268A88
1 changed files with 26 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
src/client.nim
View File

@ -55,9 +55,16 @@ proc closeConnection*(client: DiscordClient, code: int = 1000) {.async.} =
await client.ws.close(code) await client.ws.close(code)
proc reconnectClient(client: DiscordClient) {.async.} = proc reconnectClient(client: DiscordClient) {.async.} =
echo "Reconnecting..."
client.reconnecting = true client.reconnecting = true
await client.ws.close(1000) await client.ws.close(1000)
waitFor client.startConnection()
client.ws = await newAsyncWebsocketClient(client.endpoint[6..client.endpoint.high], Port 443,
path = "/v=6&encoding=json", true)
client.reconnecting = false
client.heartbeatAcked = true
#waitFor client.startConnection()
# Handle discord disconnect. If it detects that we can reconnect, it will. # Handle discord disconnect. If it detects that we can reconnect, it will.
proc handleDiscordDisconnect(client: DiscordClient, error: string) {.async.} = proc handleDiscordDisconnect(client: DiscordClient, error: string) {.async.} =
@ -67,7 +74,7 @@ proc handleDiscordDisconnect(client: DiscordClient, error: string) {.async.} =
client.heartbeatAcked = false client.heartbeatAcked = false
# The disconnect code # Get the disconnect code
let c = disconnectData.code let c = disconnectData.code
# 4003, 4004, 4005, 4007, 4010, 4011, 4012, 4013 are not reconnectable. # 4003, 4004, 4005, 4007, 4010, 4011, 4012, 4013 are not reconnectable.
@ -75,7 +82,9 @@ proc handleDiscordDisconnect(client: DiscordClient, error: string) {.async.} =
echo "The Discord gateway sent a disconnect code that we cannot reconnect to." echo "The Discord gateway sent a disconnect code that we cannot reconnect to."
else: else:
if (not client.reconnecting): if (not client.reconnecting):
await client.reconnectClient() waitFor client.reconnectClient()
else:
echo "Gateway is cannot reconnect due to already reconnecting..."
#TODO: Reconnecting may be done, just needs testing. #TODO: Reconnecting may be done, just needs testing.
@ -87,7 +96,7 @@ proc handleWebsocketPacket(client: DiscordClient) {.async.} =
echo "Received gateway payload: ", packet.data echo "Received gateway payload: ", packet.data
if packet.opcode == Opcode.Close: if packet.opcode == Opcode.Close:
asyncCheck client.handleDiscordDisconnect(packet.data) await client.handleDiscordDisconnect(packet.data)
var json: JsonNode var json: JsonNode
@ -108,10 +117,13 @@ proc handleWebsocketPacket(client: DiscordClient) {.async.} =
client.reconnecting = false client.reconnecting = false
let resume = %* { let resume = %* {
"op": opResume, "op": ord(opResume),
"d": {
"token": client.token,
"session_id": client.sessionID, "session_id": client.sessionID,
"seq": client.lastSequence "seq": client.lastSequence
} }
}
await client.sendGatewayRequest(resume) await client.sendGatewayRequest(resume)
else: else:
@ -130,7 +142,7 @@ proc handleWebsocketPacket(client: DiscordClient) {.async.} =
# If the json field `d` is true then the session may be resumable. # If the json field `d` is true then the session may be resumable.
if json["d"].getBool(): if json["d"].getBool():
let resume = %* { let resume = %* {
"op": opResume, "op": ord(opResume),
"session_id": client.sessionID, "session_id": client.sessionID,
"seq": client.lastSequence "seq": client.lastSequence
} }
@ -160,16 +172,17 @@ proc startConnection*(client: DiscordClient) {.async.} =
let urlResult = sendRequest(endpoint("/gateway/bot"), HttpMethod.HttpGet, defaultHeaders()) let urlResult = sendRequest(endpoint("/gateway/bot"), HttpMethod.HttpGet, defaultHeaders())
if (urlResult.contains("url")): if (urlResult.contains("url")):
let url = urlResult["url"].getStr() let url = urlResult["url"].getStr()
client.endpoint = url client.endpoint = url
client.ws = await newAsyncWebsocketClient(url[6..url.high], Port 443, client.ws = await newAsyncWebsocketClient(url[6..url.high], Port 443,
path = "/v=6&encoding=json", true) path = "/v=6&encoding=json", true)
if not client.reconnecting:
asyncCheck client.handleWebsocketPacket() asyncCheck client.handleWebsocketPacket()
runForever() # Now just wait. Dont poll for new events while we're reconnecting
while true:
if not client.reconnecting:
poll()
else: else:
raise newException(IOError, "Failed to get gateway url, token may of been incorrect!") raise newException(IOError, "Failed to get gateway url, token may of been incorrect!")