## Nim-LibP2P
## Copyright (c) 2019 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 sequtils, tables
import chronos, chronicles
import ../connection,
       ../multiaddress,
       ../multicodec,
       ../errors

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

  TransportFlag* {.pure.} = enum
    ReuseAddr

  TransportFlags* = set[TransportFlag]

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

proc transportFlagsToServerFlags*(flags: TransportFlags): set[ServerFlags] {.gcsafe.} =
  let transportFlagToServerFlagMapping = {
      TransportFlag.ReuseAddr: ServerFlags.ReuseAddr,
    }.toTable()

  for flag in flags:
    result.incl(transportFlagToServerFlagMapping[flag])

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

proc newTransport*(t: typedesc[Transport], flags: TransportFlags = {}): t {.gcsafe.} =
  new result
  result.flags = flags
  result.init()

method close*(t: Transport) {.base, async, gcsafe.} =
  ## stop and cleanup the transport
  ## including all outstanding connections
  let futs = await allFinished(t.connections.mapIt(it.close()))
  checkFutures(futs)

method listen*(t: Transport,
               ma: MultiAddress,
               handler: ConnHandler):
               Future[Future[void]] {.base, async, gcsafe.} =
  ## listen for incoming connections
  t.ma = ma
  t.handler = handler
  trace "starting node", address = $ma

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

method upgrade*(t: Transport) {.base, async, gcsafe.} =
  ## base upgrade method that the transport uses to perform
  ## transport specific upgrades
  discard

method handles*(t: Transport, address: MultiAddress): bool {.base, gcsafe.} =
  ## check if transport supportes the multiaddress

  # by default we skip circuit addresses to avoid
  # having to repeat the check in every transport
  address.protocols.filterIt( it == multiCodec("p2p-circuit") ).len == 0

method localAddress*(t: Transport): MultiAddress {.base, gcsafe.} =
  ## get the local address of the transport in case started with 0.0.0.0:0
  discard