nim-libp2p/examples/chat.nim

97 lines
2.9 KiB
Nim
Raw Normal View History

2018-11-22 14:12:46 +00:00
import asyncdispatch2, nimcrypto, strutils
import ../libp2p/daemon/daemonapi
const
ConsoleAddress = "/tmp/console-chat.sock"
ServerAddress = "/tmp/remote-chat.sock"
ServerProtocols = @["/test-chat-stream"]
type
CustomData = ref object
api: DaemonAPI
remotes: seq[StreamTransport]
proc threadMain(a: int) {.thread.} =
## This procedure performs reading from `stdin` and sends data over
## unix domain socket to main thread.
var transp = waitFor connect(initTAddress(ConsoleAddress))
while true:
var line = stdin.readLine()
let res = waitFor transp.write(line & "\r\n")
proc serveThread(server: StreamServer,
transp: StreamTransport) {.async.} =
## This procedure perform readin on local unix domain socket and
## sends data to remote clients.
var udata = getUserData[CustomData](server)
proc remoteReader(transp: StreamTransport) {.async.} =
while true:
var line = await transp.readLine()
if len(line) == 0:
break
echo ">> ", line
2018-11-22 14:12:46 +00:00
while true:
try:
var line = await transp.readLine()
if line.startsWith("/connect"):
var parts = line.split(" ")
if len(parts) == 2:
var address = fromHex(parts[1])
echo "= Searching for peer ", toHex(address)
var id = await udata.api.dhtFindPeer(address)
echo "==="
echo repr id
echo "==="
2018-11-22 14:12:46 +00:00
echo "= Connecting to peer ", toHex(address)
await udata.api.connect(id.peer, id.addresses)
echo "= Opening stream to peer chat ", toHex(address)
var stream = await udata.api.openStream(id.peer, ServerProtocols)
udata.remotes.add(stream.transp)
2018-11-22 14:12:46 +00:00
echo "= Connected to peer chat ", toHex(address)
asyncCheck remoteReader(stream.transp)
2018-11-22 14:12:46 +00:00
else:
var msg = line & "\r\n"
echo "<< ", line
2018-11-22 14:12:46 +00:00
var pending = newSeq[Future[int]]()
for item in udata.remotes:
pending.add(item.write(msg))
if len(pending) > 0:
var results = await all(pending)
except:
break
proc main() {.async.} =
var data = new CustomData
data.remotes = newSeq[StreamTransport]()
var lserver = createStreamServer(initTAddress(ConsoleAddress),
serveThread, udata = data)
lserver.start()
var thread: Thread[int]
thread.createThread(threadMain, 0)
echo "= Starting P2P node"
2018-11-22 14:16:34 +00:00
data.api = await newDaemonApi({DHTFull, Bootstrap})
await sleepAsync(3000)
2018-11-22 14:12:46 +00:00
var id = await data.api.identity()
proc streamHandler(api: DaemonAPI, stream: P2PStream) {.async.} =
echo "= Peer ", toHex(stream.peer), " joined chat"
data.remotes.add(stream.transp)
2018-11-22 14:12:46 +00:00
while true:
var line = await stream.transp.readLine()
if len(line) == 0:
break
2018-11-22 14:12:46 +00:00
echo ">> ", line
await data.api.addHandler(ServerProtocols, streamHandler)
echo "= Your PeerID is ", toHex(id.peer)
when isMainModule:
waitFor(main())
while true:
poll()