(* Open Source License *)
(* Copyright (c) 2020 Nomadic Labs. <contact@nomadic-labs.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. *)

Require Import String.

Require Import syntax.
Require error.
Require parser_helpers.

Require dexter_parse.

(*! Parameter type !*)

(*+ Extract entrypoint type by name +*)
Definition is_annotation an (oan' : Datatypes.option String.string) :=
  match oan' with
    | None => false
    | Some an' => if String.eqb an an' then true else false
  end.

Fixpoint get_ep_ty_by_name (ty: type) (ep : String.string) : Datatypes.list type :=
  match ty with
    | (or a an1 b an2) =>
      if is_annotation ep an1 then
        cons a nil
      else
        if is_annotation ep an2 then
        cons b nil
        else
          (get_ep_ty_by_name a ep) ++ (get_ep_ty_by_name b ep)
    | _ => nil
  end.

Fixpoint get_ep_ty_by_name1 (ty: type)  (ep : String.string) : Datatypes.option type :=
  match get_ep_ty_by_name ty ep with
    | cons ty nil  => Some ty
    | _ => None
  end.

(*+ Parameter type definitions +*)

Definition parameter_ty := Eval cbv in contract_file_parameter dexter_parse.dex_contract_file.

Definition parameter_ep_addLiquidity_ty := Eval cbv in get_opt (get_ep_ty_by_name1 parameter_ty "%addLiquidity") I.

Definition parameter_ep_removeLiquidity_ty := Eval cbv in get_opt (get_ep_ty_by_name1 parameter_ty "%removeLiquidity") I.

Definition parameter_ep_xtzToToken_ty := Eval cbv in get_opt (get_ep_ty_by_name1 parameter_ty "%xtzToToken") I.

Definition parameter_ep_tokenToXtz_ty := Eval cbv in get_opt (get_ep_ty_by_name1 parameter_ty "%tokenToXtz") I.

Definition parameter_ep_tokenToToken_ty := Eval cbv in get_opt (get_ep_ty_by_name1 parameter_ty "%tokenToToken") I.

Definition parameter_ep_updateTokenPool_ty := Eval cbv in get_opt (get_ep_ty_by_name1 parameter_ty "%updateTokenPool") I.

Definition parameter_ep_updateTokenPoolInternal_ty := Eval cbv in get_opt (get_ep_ty_by_name1 parameter_ty "%updateTokenPoolInternal") I.

Definition parameter_ep_setBaker_ty := Eval cbv in get_opt (get_ep_ty_by_name1 parameter_ty "%setBaker") I.

Definition parameter_ep_setLqtAddress_ty := Eval cbv in get_opt (get_ep_ty_by_name1 parameter_ty "%setLqtAddress") I.

Definition parameter_ep_setManager_ty := Eval cbv in get_opt (get_ep_ty_by_name1 parameter_ty "%setManager") I.

Definition parameter_ep_default_ty := Eval cbv in get_opt (get_ep_ty_by_name1 parameter_ty "%default") I.

(*! Storage type !*)

Definition storage_ty : type :=
  Eval cbv in contract_file_storage dexter_parse.dex_contract_file.

Require contract_semantics.
Module Storage.
  Import contract_semantics.
  
  (* Generated by split.py *)
  Definition sto_selfIsUpdatingTokenPool (sto : data storage_ty) : data bool :=
    match sto with
    | (tokenPool, (xtzPool, (lqtTotal, (selfIsUpdatingTokenPool, (freezeBaker, (manager, (tokenAddress, lqtAddress))))))) =>
      selfIsUpdatingTokenPool
    end.

  Definition sto_freezeBaker (sto : data storage_ty) : data bool :=
    match sto with
    | (tokenPool, (xtzPool, (lqtTotal, (selfIsUpdatingTokenPool, (freezeBaker, (manager, (tokenAddress, lqtAddress))))))) =>
      freezeBaker
    end.

  Definition sto_lqtTotal (sto : data storage_ty) : data nat :=
    match sto with
    | (tokenPool, (xtzPool, (lqtTotal, (selfIsUpdatingTokenPool, (freezeBaker, (manager, (tokenAddress, lqtAddress))))))) =>
      lqtTotal
    end.

  Definition sto_manager (sto : data storage_ty) : data address :=
    match sto with
    | (tokenPool, (xtzPool, (lqtTotal, (selfIsUpdatingTokenPool, (freezeBaker, (manager, (tokenAddress, lqtAddress))))))) =>
      manager
    end.

  Definition sto_tokenAddress (sto : data storage_ty) : data address :=
    match sto with
    | (tokenPool, (xtzPool, (lqtTotal, (selfIsUpdatingTokenPool, (freezeBaker, (manager, (tokenAddress, lqtAddress))))))) =>
      tokenAddress
    end.

  Definition sto_lqtAddress (sto : data storage_ty) : data address :=
    match sto with
    | (tokenPool, (xtzPool, (lqtTotal, (selfIsUpdatingTokenPool, (freezeBaker, (manager, (tokenAddress, lqtAddress))))))) =>
      lqtAddress
    end.

  Definition sto_tokenPool (sto : data storage_ty) : data nat :=
    match sto with
    | (tokenPool, (xtzPool, (lqtTotal, (selfIsUpdatingTokenPool, (freezeBaker, (manager, (tokenAddress, lqtAddress))))))) =>
      tokenPool
    end.

  Definition sto_xtzPool (sto : data storage_ty) : data mutez :=
    match sto with
    | (tokenPool, (xtzPool, (lqtTotal, (selfIsUpdatingTokenPool, (freezeBaker, (manager, (tokenAddress, lqtAddress))))))) =>
      xtzPool
    end.




  Definition sto_set_selfIsUpdatingTokenPool (sto : data storage_ty) (selfIsUpdatingTokenPool' : data bool) : data storage_ty :=
    match sto with
    | (tokenPool, (xtzPool, (lqtTotal, (selfIsUpdatingTokenPool, (freezeBaker, (manager, (tokenAddress, lqtAddress))))))) =>
      (tokenPool, (xtzPool, (lqtTotal, (selfIsUpdatingTokenPool', (freezeBaker, (manager, (tokenAddress, lqtAddress)))))))
    end.

  Definition sto_set_freezeBaker (sto : data storage_ty) (freezeBaker' : data bool) : data storage_ty :=
    match sto with
    | (tokenPool, (xtzPool, (lqtTotal, (selfIsUpdatingTokenPool, (freezeBaker, (manager, (tokenAddress, lqtAddress))))))) =>
      (tokenPool, (xtzPool, (lqtTotal, (selfIsUpdatingTokenPool, (freezeBaker', (manager, (tokenAddress, lqtAddress)))))))
    end.

  Definition sto_set_lqtTotal (sto : data storage_ty) (lqtTotal' : data nat) : data storage_ty :=
    match sto with
    | (tokenPool, (xtzPool, (lqtTotal, (selfIsUpdatingTokenPool, (freezeBaker, (manager, (tokenAddress, lqtAddress))))))) =>
      (tokenPool, (xtzPool, (lqtTotal', (selfIsUpdatingTokenPool, (freezeBaker, (manager, (tokenAddress, lqtAddress)))))))
    end.

  Definition sto_set_manager (sto : data storage_ty) (manager' : data address) : data storage_ty :=
    match sto with
    | (tokenPool, (xtzPool, (lqtTotal, (selfIsUpdatingTokenPool, (freezeBaker, (manager, (tokenAddress, lqtAddress))))))) =>
      (tokenPool, (xtzPool, (lqtTotal, (selfIsUpdatingTokenPool, (freezeBaker, (manager', (tokenAddress, lqtAddress)))))))
    end.

  Definition sto_set_tokenAddress (sto : data storage_ty) (tokenAddress' : data address) : data storage_ty :=
    match sto with
    | (tokenPool, (xtzPool, (lqtTotal, (selfIsUpdatingTokenPool, (freezeBaker, (manager, (tokenAddress, lqtAddress))))))) =>
      (tokenPool, (xtzPool, (lqtTotal, (selfIsUpdatingTokenPool, (freezeBaker, (manager, (tokenAddress', lqtAddress)))))))
    end.

  Definition sto_set_lqtAddress (sto : data storage_ty) (lqtAddress' : data address) : data storage_ty :=
    match sto with
    | (tokenPool, (xtzPool, (lqtTotal, (selfIsUpdatingTokenPool, (freezeBaker, (manager, (tokenAddress, lqtAddress))))))) =>
      (tokenPool, (xtzPool, (lqtTotal, (selfIsUpdatingTokenPool, (freezeBaker, (manager, (tokenAddress, lqtAddress')))))))
    end.

  Definition sto_set_tokenPool (sto : data storage_ty) (tokenPool' : data nat) : data storage_ty :=
    match sto with
    | (tokenPool, (xtzPool, (lqtTotal, (selfIsUpdatingTokenPool, (freezeBaker, (manager, (tokenAddress, lqtAddress))))))) =>
      (tokenPool', (xtzPool, (lqtTotal, (selfIsUpdatingTokenPool, (freezeBaker, (manager, (tokenAddress, lqtAddress)))))))
    end.

  Definition sto_set_xtzPool (sto : data storage_ty) (xtzPool' : data mutez) : data storage_ty :=
    match sto with
    | (tokenPool, (xtzPool, (lqtTotal, (selfIsUpdatingTokenPool, (freezeBaker, (manager, (tokenAddress, lqtAddress))))))) =>
      (tokenPool, (xtzPool', (lqtTotal, (selfIsUpdatingTokenPool, (freezeBaker, (manager, (tokenAddress, lqtAddress)))))))
    end.

  Ltac destruct_sto sto tokenPool xtzPool lqtTotal selfIsUpdatingTokenPool freezeBaker manager tokenAddress lqtAddress :=
    destruct sto as
      (tokenPool, (xtzPool, (lqtTotal, (selfIsUpdatingTokenPool, (freezeBaker, (manager, (tokenAddress, lqtAddress))))))).
End Storage.

(*! Contract definition !*)

Definition self_type : self_info := Eval cbv in (Some (parameter_ty, None)).

Import error.Notations.

Definition instr_seq_from_typer_result s self_type A B :
  error.M (sigT (fun b => typer.instruction_seq self_type b A B)) :=
  parser_helpers.michelson_type_checked_seq_M
    s 100 self_type A B.

Definition instr_seq_from_sig_to_error self_type A B
           (v : (sigT (fun b => typer.instruction_seq self_type b A B))) :
  error.M (typer.instruction_seq self_type false A B) :=
  match v  with
  | existT _ true code => error.Failed _ (error.Assertion_Failure _ tt)
  | existT _ false code => error.Return code
  end.

Definition m_instr_seq_from_string s self_type A B:
  error.M (typer.instruction_seq self_type false A B) :=
  let! v := instr_seq_from_typer_result s self_type A B in
  let! v' := instr_seq_from_sig_to_error self_type A B v in
  error.Return v'.

Definition ep_from_string s (ep_ty : type) :
  error.M (typer.instruction_seq self_type false
                                 (ep_ty ::: storage_ty ::: nil)
                                 (pair (list operation) storage_ty ::: nil)) :=
  m_instr_seq_from_string s self_type
                          (ep_ty ::: storage_ty ::: nil)
                          (pair (list operation) storage_ty ::: nil).

Definition ep_addLiquidity := Eval cbv in (error.extract (ep_from_string dexter_string.ep_addLiquidity parameter_ep_addLiquidity_ty) I).

Definition ep_removeLiquidity := Eval cbv in (error.extract (ep_from_string dexter_string.ep_removeLiquidity parameter_ep_removeLiquidity_ty) I).

Definition ep_xtzToToken := Eval cbv in (error.extract (ep_from_string dexter_string.ep_xtzToToken parameter_ep_xtzToToken_ty) I).

Definition ep_tokenToXtz := Eval cbv in (error.extract (ep_from_string dexter_string.ep_tokenToXtz parameter_ep_tokenToXtz_ty) I).

Definition ep_tokenToToken := Eval cbv in (error.extract (ep_from_string dexter_string.ep_tokenToToken parameter_ep_tokenToToken_ty) I).

Definition ep_updateTokenPool := Eval cbv in (error.extract (ep_from_string dexter_string.ep_updateTokenPool parameter_ep_updateTokenPool_ty) I).

Definition ep_updateTokenPoolInternal := Eval cbv in (error.extract (ep_from_string dexter_string.ep_updateTokenPoolInternal parameter_ep_updateTokenPoolInternal_ty) I).

Definition ep_setBaker := Eval cbv in (error.extract (ep_from_string dexter_string.ep_setBaker parameter_ep_setBaker_ty) I).

Definition ep_setManager := Eval cbv in (error.extract (ep_from_string dexter_string.ep_setManager parameter_ep_setManager_ty) I).

Definition ep_setLqtAddress := Eval cbv in (error.extract (ep_from_string dexter_string.ep_setLqtAddress parameter_ep_setLqtAddress_ty) I).

Definition ep_default := Eval cbv in (error.extract (ep_from_string dexter_string.ep_default parameter_ep_default_ty) I).


(*! Main !*)

Definition main :=
  Eval cbv in contract_file_code dexter_parse.dex_contract_file.


(*! Alternate main !*)

(* This should be the same as [main], however, the contract above
   renders something that is too slow to work on. *)

Require Import macros.

Definition main2 : instruction_seq
                     self_type _
                     (pair parameter_ty storage_ty ::: nil)
                     (pair (list operation) storage_ty ::: nil)
  :=  { DUP ;
        CDR ;
        SWAP ;
        CAR ;
        IF_LEFT
          { IF_LEFT
              { IF_LEFT
                  { IF_LEFT
                      ( ep_addLiquidity )
                      ( ep_default ) }
                  { IF_LEFT
                      ( ep_removeLiquidity )
                      ( ep_setBaker ) } }
              { IF_LEFT
                  { IF_LEFT
                      ( ep_setLqtAddress )
                      ( ep_setManager ) }
                  { IF_LEFT
                      ( ep_tokenToToken )
                      ( ep_tokenToXtz ) } } }
          { IF_LEFT
              { IF_LEFT
                  ( ep_updateTokenPool )
                  ( ep_updateTokenPoolInternal ) }
              ( ep_xtzToToken ) } }.

Lemma main_eq_main2 : main = main2.
  reflexivity.
Qed.