(*****************************************************************************)
(*                                                                           *)
(* Open Source License                                                       *)
(* Copyright (c) 2018 Dynamic Ledger Solutions, Inc. <contact@tezos.com>     *)
(*                                                                           *)
(* Permission is hereby granted, free of charge, to any person obtaining a   *)
(* copy of this software and associated documentation files (the "Software"),*)
(* to deal in the Software without restriction, including without limitation *)
(* the rights to use, copy, modify, merge, publish, distribute, sublicense,  *)
(* and/or sell copies of the Software, and to permit persons to whom the     *)
(* Software is furnished to do so, subject to the following conditions:      *)
(*                                                                           *)
(* The above copyright notice and this permission notice shall be included   *)
(* in all copies or substantial portions of the Software.                    *)
(*                                                                           *)
(* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR*)
(* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,  *)
(* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL   *)
(* THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER*)
(* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING   *)
(* FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER       *)
(* DEALINGS IN THE SOFTWARE.                                                 *)
(*                                                                           *)
(*****************************************************************************)

open Client_config

let disable_disclaimer =
  match Sys.getenv_opt "TEZOS_CLIENT_UNSAFE_DISABLE_DISCLAIMER" with
  | Some ("yes" | "y" | "YES" | "Y") -> true
  | _ -> false

let default () =
  if not disable_disclaimer then
    Format.eprintf
      "@[<v 2>@{<warning>@{<title>Warning@}@}@,@,\
      \                       This is @{<warning>NOT@} the Tezos Mainnet.@,\
      \                    The Tezos Mainnet is not yet released.@,\
       @,\
      \                 Use your fundraiser keys @{<warning>AT YOUR OWN RISK@}.@,\
       All transactions happening on the Betanet @{<warning>are expected to be valid in the Mainnet@}.@,\
      \        In doubt, we recommend that you wait for the lunch of the Mainnet.@]@\n@."

let zeronet () =
  if not disable_disclaimer then
    Format.eprintf
      "@[<v 2>@{<warning>@{<title>Warning@}@}@,@,\
      \               This is @{<warning>NOT@} the Tezos Mainnet.@,\
      \            The Tezos Mainnet is not yet released.@,\
       @,\
      \    The node you are connecting to claims to be running on the@,\
      \               @{<warning>Tezos Zeronet DEVELOPMENT NETWORK@}.@,\
      \         Do @{<warning>NOT@} use your fundraiser keys on this network.@,\
       Zeronet is a testing network, with free tokens and frequent resets.@]@\n@."

let alphanet () =
  if not disable_disclaimer then
    Format.eprintf
      "@[<v 2>@{<warning>@{<title>Warning@}@}@,@,\
      \               This is @{<warning>NOT@} the Tezos Mainnet.@,\
      \            The Tezos Mainnet is not yet released.@,\
       @,\
      \   The node you are connecting to claims to be running on the@,\
      \             @{<warning>Tezos Alphanet DEVELOPMENT NETWORK.@}@,\
      \        Do @{<warning>NOT@} use your fundraiser keys on this network.@,\
      \        Alphanet is a testing network, with free tokens.@]@\n@."

let betanet () =
  if not disable_disclaimer then
    Format.eprintf
      "@[<v 2>@{<warning>@{<title>Warning@}@}@,@,\
      \                         This is @{<warning>NOT@} the Tezos Mainnet.@,\
      \                      The Tezos Mainnet is not yet released.@,\
       @,\
      \            The node you are connecting to claims to be running on the@,\
      \                      @{<warning>Tezos Betanet EXPERIMENTAL NETWORK@}.@,\
      \    Betanet is a pre-release experimental network and comes with no warranty.@,\
      \            Use your fundraiser keys on this network @{<warning>AT YOUR OWN RISK@}.@,\
      \  All transactions happening on the Betanet @{<warning>are expected to be valid in the Mainnet@}.@,\
      \          If in doubt, we recommend that you wait for the Mainnet lunch.@]@\n@."

let sandbox () =
  if not disable_disclaimer then
    Format.eprintf
      "@[<v 2>@{<warning>@{<title>Warning@}@}@,@,\
      \             This is @{<warning>NOT@} the Tezos Mainnet.@,\
      \          The Tezos Mainnet is not yet released.@,\
       @,\
      \ The node you are connecting to claims to be running in a@,\
      \                  @{<warning>Tezos TEST SANDBOX@}.@,\
      \    Do @{<warning>NOT@} use your fundraiser keys on this network.@,\
       You should not see this message if you are not a developer.@]@\n@."

let check_network ctxt =
  Shell_services.P2p.versions ctxt >>= function
  | Error _ ->
      default () ;
      Lwt.return_none
  | Ok versions ->
      match String.split_on_char '_' (List.hd versions).name with
      | "TEZOS" :: "ZERONET" :: _date :: [] ->
          zeronet () ;
          Lwt.return_some `Zeronet
      | "TEZOS" :: "ALPHANET" :: _date :: [] ->
          alphanet () ;
          Lwt.return_some `Alphanet
      | "TEZOS" :: "BETANET" :: _date :: [] ->
          betanet () ;
          Lwt.return_some `Betanet
      | "TEZOS" :: _date :: [] ->
          sandbox () ;
          Lwt.return_some `Sandbox
      | _ ->
          default () ;
          Lwt.return_none

let get_commands_for_version ctxt network block protocol =
  Shell_services.Blocks.protocols ctxt ~block () >>= function
  | Ok { next_protocol = version } -> begin
      match protocol with
      | None ->
          return (Some version, Client_commands.commands_for_version version network)
      | Some given_version -> begin
          if not (Protocol_hash.equal version given_version) then
            Format.eprintf
              "@[<v 2>@{<warning>@{<title>Warning@}@}@,\
               The protocol provided via `--protocol` (%a)@,\
               is not the one retrieved from the node (%a).@]@\n@."
              Protocol_hash.pp_short given_version
              Protocol_hash.pp_short version ;
          return (Some version, Client_commands.commands_for_version given_version network)
        end
    end
  | Error errs -> begin
      match protocol with
      | None -> begin
          Format.eprintf
            "@[<v 2>@{<warning>@{<title>Warning@}@}@,\
             Failed to acquire the protocol version from the node@,%a@]@\n@."
            (Format.pp_print_list pp) errs ;
          return (None, [])
        end
      | Some version ->
          return (Some version, Client_commands.commands_for_version version network)
    end

let select_commands ctxt { block ; protocol } =
  check_network ctxt >>= fun network ->
  get_commands_for_version ctxt network block protocol >>|? fun (_, commands_for_version)  ->
  Client_rpc_commands.commands @
  List.map
    (Clic.map_command
       (fun (o : Client_context.full) -> (o :> Client_context.io_wallet)))
    (Tezos_signer_backends.Ledger.commands () @
     Client_keys_commands.commands network) @
  Client_helpers_commands.commands () @
  commands_for_version

let () = Client_main_run.run select_commands