open Proto_alpha_utils
open Trace
open Compiler.Program
open Memory_proto_alpha.Protocol.Script_ir_translator
open Memory_proto_alpha.X

type options = Memory_proto_alpha.options

let run ?options (* ?(is_input_value = false) *) (program:compiled_program) (input_michelson:Michelson.t) : ex_typed_value result =
  let Compiler.Program.{input;output;body} : compiled_program = program in
  let (Ex_ty input_ty) = input in
  let (Ex_ty output_ty) = output in
  (* let%bind input_ty_mich =
   *   Trace.trace_tzresult_lwt (simple_error "error unparsing input ty") @@
   *   Memory_proto_alpha.unparse_michelson_ty input_ty in
   * let%bind output_ty_mich =
   *   Trace.trace_tzresult_lwt (simple_error "error unparsing output ty") @@
   *   Memory_proto_alpha.unparse_michelson_ty output_ty in
   * Format.printf "code: %a\n" Michelson.pp program.body ;
   * Format.printf "input_ty: %a\n" Michelson.pp input_ty_mich ;
   * Format.printf "output_ty: %a\n" Michelson.pp output_ty_mich ;
   * Format.printf "input: %a\n" Michelson.pp input_michelson ; *)
  let%bind input =
    Trace.trace_tzresult_lwt (simple_error "error parsing input") @@
    Memory_proto_alpha.parse_michelson_data input_michelson input_ty
  in
  let body = Michelson.(strip_nops @@ strip_annots body) in
  let%bind descr =
    Trace.trace_tzresult_lwt (simple_error "error parsing program code") @@
    Memory_proto_alpha.parse_michelson body
      (Item_t (input_ty, Empty_t, None)) (Item_t (output_ty, Empty_t, None)) in
  let open! Memory_proto_alpha.Protocol.Script_interpreter in
  let%bind (Item(output, Empty)) =
    Trace.trace_tzresult_lwt (simple_error "error of execution") @@
    Memory_proto_alpha.interpret ?options descr (Item(input, Empty)) in
  ok (Ex_typed_value (output_ty, output))

let evaluate ?options program = run ?options program Michelson.d_unit

let ex_value_ty_to_michelson (v : ex_typed_value) : Michelson.t result =
  let (Ex_typed_value (value , ty)) = v in
  Trace.trace_tzresult_lwt (simple_error "error unparsing michelson result") @@
  Memory_proto_alpha.unparse_michelson_data value ty

let evaluate_michelson ?options program =
  let%bind etv = evaluate ?options program in
  ex_value_ty_to_michelson etv