nim-libp2p-experimental/libp2p/transport.nim

## Nim-LibP2P
## Copyright (c) 2018 Status Research & Development GmbH
## Licensed under either of
##  * Apache License, version 2.0, ([LICENSE-APACHE](LICENSE-APACHE))
##  * MIT license ([LICENSE-MIT](LICENSE-MIT))
## at your option.
## This file may not be copied, modified, or distributed except according to
## those terms.

import chronos
import peerinfo, connection, multiaddress, multicodec

type
  ConnHandler* = proc (conn: Connection): Future[void] {.gcsafe.}

  ConnHolder* = object
    connection*: Connection
    connFuture*: Future[void]

  Transport* = ref object of RootObj
    ma*: Multiaddress
    connections*: seq[ConnHolder]
    handler*: ConnHandler
    multicodec*: MultiCodec

method connHandler*(t: Transport,
                    server: StreamServer,
                    client: StreamTransport): Future[Connection] {.base, gcsafe, async.} =
  let conn: Connection = newConnection(server, client)
  let handlerFut = if t.handler == nil: nil else: t.handler(conn)
  let connHolder: ConnHolder = ConnHolder(connection: conn,
                                          connFuture: handlerFut)
  t.connections.add(connHolder)
  result = conn

method init*(t: Transport) {.base.} = 
  ## perform protocol initialization
  discard

proc newTransport*(t: typedesc[Transport]): t = 
  new result
  result.init()

method close*(t: Transport) {.base, async.} =
  ## stop and cleanup the transport
  ## including all outstanding connections
  for c in t.connections:
    if c.connection.isOpen:
      await c.connection.close()

method listen*(t: Transport, ma: MultiAddress, handler: ConnHandler) {.base, async.} =
  ## listen for incoming connections
  t.ma = ma
  t.handler = handler

method dial*(t: Transport, address: MultiAddress): Future[Connection] {.base, async.} = 
  ## dial a peer
  discard

method supports(t: Transport, address: MultiAddress): bool {.base.} = 
  ## check if transport supportes the multiaddress
  # TODO: this should implement generic logic that would use the multicodec 
  # declared in the multicodec field and set by each individual transport
  result = true