97 lines
2.9 KiB
Nim
97 lines
2.9 KiB
Nim
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
|
|
|
|
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 "==="
|
|
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)
|
|
echo "= Connected to peer chat ", toHex(address)
|
|
asyncCheck remoteReader(stream.transp)
|
|
else:
|
|
var msg = line & "\r\n"
|
|
echo "<< ", line
|
|
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"
|
|
data.api = await newDaemonApi({DHTFull, Bootstrap})
|
|
await sleepAsync(3000)
|
|
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)
|
|
while true:
|
|
var line = await stream.transp.readLine()
|
|
if len(line) == 0:
|
|
break
|
|
echo ">> ", line
|
|
|
|
await data.api.addHandler(ServerProtocols, streamHandler)
|
|
echo "= Your PeerID is ", toHex(id.peer)
|
|
|
|
when isMainModule:
|
|
waitFor(main())
|
|
while true:
|
|
poll()
|