logos-storage/nim-json-rpc
1
0
Fork 0
mirror of https://github.com/logos-storage/nim-json-rpc.git synced 2026-01-07 08:03:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Allow custom headers in ws backend. Set Origin to http://localhost by default

Browse Source
This commit is contained in:
Yuriy Glukhov 2019-07-02 16:15:41 +03:00
parent af33cb9fbc
commit 881c62ed54
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
json_rpc/clients/websocketclient.nim
View File

@ -45,8 +45,15 @@ proc processData(client: RpcWebSocketClient) {.async.} =
# async loop reconnection and waiting
client.transport = await newWebSocket(client.uri)
proc connect*(client: RpcWebSocketClient, uri: string) {.async.} =
client.transport = await newWebSocket(uri)
proc connect*(client: RpcWebSocketClient, uri: string, headers: StringTableRef = nil) {.async.} =
var headers = headers
if headers.isNil:
headers = newStringTable({"Origin": "http://localhost"})
elif "Origin" notin headers:
# TODO: This is a hack, because the table might be case sensitive. Ideally strtabs module has
# to be extended with case insensitive accessors.
headers["Origin"] = "http://localhost"
client.transport = await newWebSocket(uri, headers)
client.uri = uri
client.loop = processData(client)