(**************************************************************************)
(*                                                                        *)
(*    Copyright (c) 2014 - 2018.                                          *)
(*    Dynamic Ledger Solutions, Inc. <contact@tezos.com>                  *)
(*                                                                        *)
(*    All rights reserved. No warranty, explicit or implicit, provided.   *)
(*                                                                        *)
(**************************************************************************)

module S = struct

  let versions =
    RPC_service.post_service
      ~description:"Supported network layer versions."
      ~query: RPC_query.empty
      ~input: Data_encoding.empty
      ~output: (Data_encoding.list P2p_version.encoding)
      RPC_path.(root / "network" / "versions")

  let stat =
    RPC_service.post_service
      ~description:"Global network bandwidth statistics in B/s."
      ~query: RPC_query.empty
      ~input: Data_encoding.empty
      ~output: P2p_stat.encoding
      RPC_path.(root / "network" / "stat")

  let events =
    RPC_service.post_service
      ~description:"Stream of all network events"
      ~query: RPC_query.empty
      ~input: Data_encoding.empty
      ~output: P2p_connection.Pool_event.encoding
      RPC_path.(root / "network" / "log")

  let connect =
    RPC_service.post_service
      ~description:"Connect to a peer"
      ~query: RPC_query.empty
      ~input: Data_encoding.(obj1 (dft "timeout" float 5.))
      ~output: Data_encoding.empty
      RPC_path.(root / "network" / "connect" /: P2p_point.Id.rpc_arg)

end

open RPC_context
let stat ctxt = make_call S.stat ctxt () () ()
let versions ctxt = make_call S.versions ctxt () () ()
let events ctxt = make_streamed_call S.events ctxt () () ()
let connect ctxt ~timeout peer_id =
  make_call1 S.connect ctxt peer_id () timeout

let monitor_encoding = Data_encoding.(obj1 (dft "monitor" bool false))

module Connections = struct

  type connection_info = Connection_metadata.t P2p_connection.Info.t

  let connection_info_encoding =
    P2p_connection.Info.encoding Connection_metadata.encoding

  module S = struct

    let list =
      RPC_service.post_service
        ~description:"List the running P2P connection."
        ~query: RPC_query.empty
        ~input: Data_encoding.empty
        ~output: (Data_encoding.list connection_info_encoding)
        RPC_path.(root / "network" / "connections")

    let info =
      RPC_service.post_service
        ~query: RPC_query.empty
        ~input: Data_encoding.empty
        ~output: connection_info_encoding
        ~description:"Details about the current P2P connection to the given peer."
        RPC_path.(root / "network" / "connections" /: P2p_peer.Id.rpc_arg)

    let kick =
      RPC_service.post_service
        ~query: RPC_query.empty
        ~input: Data_encoding.(obj1 (req "wait" bool))
        ~output: Data_encoding.empty
        ~description:"Forced close of the current P2P connection to the given peer."
        RPC_path.(root / "network" / "connections" /: P2p_peer.Id.rpc_arg / "kick")

  end

  let list ctxt = make_call S.list ctxt () () ()
  let info ctxt peer_id = make_call1 S.info ctxt peer_id () ()
  let kick ctxt ?(wait = false) peer_id = make_call1 S.kick ctxt peer_id () wait

end

module Points = struct

  module S = struct

    let info =
      RPC_service.post_service
        ~query: RPC_query.empty
        ~input: Data_encoding.empty
        ~output: P2p_point.Info.encoding
        ~description: "Details about a given `IP:addr`."
        RPC_path.(root / "network" / "points" /: P2p_point.Id.rpc_arg)

    let events =
      RPC_service.post_service
        ~query: RPC_query.empty
        ~input: monitor_encoding
        ~output: (Data_encoding.list
                    P2p_point.Pool_event.encoding)
        ~description: "Monitor network events related to an `IP:addr`."
        RPC_path.(root / "network" / "points" /: P2p_point.Id.rpc_arg / "log")

    let list =
      let filter =
        let open Data_encoding in
        obj1 (dft "filter" (list P2p_point.State.encoding) []) in
      RPC_service.post_service
        ~query: RPC_query.empty
        ~input: filter
        ~output:
          Data_encoding.(list (tup2
                                 P2p_point.Id.encoding
                                 P2p_point.Info.encoding))
        ~description:"List the pool of known `IP:port` \
                      used for establishing P2P connections."
        RPC_path.(root / "network" / "points")

    let forget =
      RPC_service.post_service
        ~query: RPC_query.empty
        ~input: Data_encoding.empty
        ~output: Data_encoding.empty
        ~description:"Remove the given address from the whitelist/blacklist."
        RPC_path.(root / "network" / "points" /: P2p_point.Id.rpc_arg / "forget" )

    let ban =
      RPC_service.post_service
        ~query: RPC_query.empty
        ~input: Data_encoding.empty
        ~output: Data_encoding.empty
        ~description:"Blacklist the given address."
        RPC_path.(root / "network" / "points" /: P2p_point.Id.rpc_arg / "ban" )

    let trust =
      RPC_service.post_service
        ~query: RPC_query.empty
        ~input: Data_encoding.empty
        ~output: Data_encoding.empty
        ~description:"Trust a given address permanently. \
                      Connections from this address can still be closed \
                      on authentication if the peer is blacklisted or greylisted."
        RPC_path.(root / "network" / "points" /: P2p_point.Id.rpc_arg / "trust" )

    let banned =
      RPC_service.post_service
        ~query: RPC_query.empty
        ~input: Data_encoding.empty
        ~output: Data_encoding.bool
        ~description:"Check is a given address is blacklisted or \
                      greylisted."
        RPC_path.(root / "network" / "points" /: P2p_point.Id.rpc_arg / "banned" )

  end

  open RPC_context
  let info ctxt peer_id = make_call1 S.info ctxt peer_id () ()
  let events ctxt point =
    make_streamed_call S.events ctxt ((), point) () true
  let list ?(filter = []) ctxt = make_call S.list ctxt () () filter
  let forget ctxt peer_id = make_call1 S.forget ctxt peer_id () ()
  let ban ctxt peer_id = make_call1 S.ban ctxt peer_id () ()
  let trust ctxt peer_id = make_call1 S.trust ctxt peer_id () ()
  let banned ctxt peer_id = make_call1 S.banned ctxt peer_id () ()

end

module Peers = struct

  module S = struct

    let info =
      RPC_service.post_service
        ~query: RPC_query.empty
        ~input: Data_encoding.empty
        ~output: (P2p_peer.Info.encoding Connection_metadata.encoding)
        ~description:"Details about a given peer."
        RPC_path.(root / "network" / "peers" /: P2p_peer.Id.rpc_arg)

    let events =
      RPC_service.post_service
        ~query: RPC_query.empty
        ~input: monitor_encoding
        ~output: (Data_encoding.list
                    P2p_peer.Pool_event.encoding)
        ~description:"Monitor network events related to a given peer."
        RPC_path.(root / "network" / "peers" /: P2p_peer.Id.rpc_arg / "log")

    let list =
      let filter =
        let open Data_encoding in
        obj1 (dft "filter" (list P2p_peer.State.encoding) []) in
      RPC_service.post_service
        ~query: RPC_query.empty
        ~input: filter
        ~output:
          Data_encoding.(list (tup2
                                 P2p_peer.Id.encoding
                                 (P2p_peer.Info.encoding Connection_metadata.encoding)))
        ~description:"List the peers the node ever met."
        RPC_path.(root / "network" / "peers")

    let forget =
      RPC_service.post_service
        ~query: RPC_query.empty
        ~input: Data_encoding.empty
        ~output: Data_encoding.empty
        ~description:"Remove the given peer from the whitelist/blacklist."
        RPC_path.(root / "network" / "peers" /: P2p_peer.Id.rpc_arg / "forget" )

    let ban =
      RPC_service.post_service
        ~query: RPC_query.empty
        ~input: Data_encoding.empty
        ~output: Data_encoding.empty
        ~description:"Blacklist the given peer."
        RPC_path.(root / "network" / "peers" /: P2p_peer.Id.rpc_arg / "ban" )

    let trust =
      RPC_service.post_service
        ~query: RPC_query.empty
        ~input: Data_encoding.empty
        ~output: Data_encoding.empty
        ~description:"Trust a given peer permanently: the peer cannot \
                      be blocked (but its host IP still can)."
        RPC_path.(root / "network" / "peers" /: P2p_peer.Id.rpc_arg / "trust" )

    let banned =
      RPC_service.post_service
        ~query: RPC_query.empty
        ~input: Data_encoding.empty
        ~output: Data_encoding.bool
        ~description:"Check if a given peer is blacklisted or \
                      greylisted."
        RPC_path.(root / "network" / "peers" /: P2p_peer.Id.rpc_arg / "banned" )

  end

  let info ctxt peer_id = make_call1 S.info ctxt peer_id () ()
  let events ctxt point =
    make_streamed_call S.events ctxt ((), point) () true
  let list ?(filter = []) ctxt = make_call S.list ctxt () () filter
  let forget ctxt point_id = make_call1 S.forget ctxt point_id () ()
  let ban ctxt point_id = make_call1 S.ban ctxt point_id () ()
  let trust ctxt point_id = make_call1 S.trust ctxt point_id () ()
  let banned ctxt point_id = make_call1 S.banned ctxt point_id () ()

end

module ACL = struct

  module S = struct

    let clear =
      RPC_service.post_service
        ~query: RPC_query.empty
        ~input: Data_encoding.empty
        ~output: Data_encoding.empty
        ~description:"Clear all greylists tables."
        RPC_path.(root / "network" / "greylist" / "clear" )
  end

  let clear ctxt = make_call S.clear ctxt () ()

end