add get_contract_opt and get_entrypoint_opt to ligo

src/passes/8-compiler/compiler_program.ml

@@ -80,6 +80,12 @@ let get_operator : constant -> type_value -> expression list -> predicate result
             prim ~children:[r_ty] I_CONTRACT ;
             i_assert_some_msg (i_push_string "bad address for get_contract") ;
           ]
+      | C_CONTRACT_OPT -> 
+          let%bind tc = get_t_option ty in
+          let%bind r = get_t_contract tc in
+          let%bind r_ty = Compiler_type.type_ r in
+          ok @@ simple_unary @@ prim ~children:[r_ty] I_CONTRACT ;
+
       | C_CONTRACT_ENTRYPOINT ->
           let%bind r = get_t_contract ty in
           let%bind r_ty = Compiler_type.type_ r in
@@ -94,6 +100,20 @@ let get_operator : constant -> type_value -> expression list -> predicate result
             prim ~annot:[entry] ~children:[r_ty] I_CONTRACT ;
             i_assert_some_msg (i_push_string @@ Format.sprintf "bad address for get_entrypoint (%s)" entry) ;
           ]
+      | C_CONTRACT_ENTRYPOINT_OPT ->
+          let%bind tc = get_t_option ty in
+          let%bind r = get_t_contract tc in
+          let%bind r_ty = Compiler_type.type_ r in
+          let%bind entry = match lst with
+            | [ { content = E_literal (D_string entry); type_value = _ } ; _addr ] -> ok entry
+            | [ _entry ; _addr ] ->
+               fail @@ contract_entrypoint_must_be_literal ~loc:__LOC__
+            | _ ->
+               fail @@ corner_case ~loc:__LOC__ "mini_c . CONTRACT_ENTRYPOINT" in
+          ok @@ simple_binary @@ seq [
+            i_drop ; (* drop the entrypoint... *)
+            prim ~annot:[entry] ~children:[r_ty] I_CONTRACT ;
+          ]
       | x -> simple_fail (Format.asprintf "predicate \"%a\" doesn't exist" Stage_common.PP.constant x)
     )
 

src/passes/operators/operators.ml

@@ -70,7 +70,9 @@ module Simplify = struct
       | "get_chain_id"    -> ok C_CHAIN_ID
       | "transaction"     -> ok C_CALL
       | "get_contract"    -> ok C_CONTRACT
+      | "get_contract_opt"-> ok C_CONTRACT_OPT
       | "get_entrypoint"  -> ok C_CONTRACT_ENTRYPOINT
+      | "get_entrypoint_opt" -> ok C_CONTRACT_ENTRYPOINT_OPT
       | "size"            -> ok C_SIZE
       | "int"             -> ok C_INT
       | "abs"             -> ok C_ABS
@@ -228,7 +230,9 @@ module Simplify = struct
       | "Operation.transaction"    -> ok C_CALL
       | "Operation.set_delegate"   -> ok C_SET_DELEGATE
       | "Operation.get_contract"   -> ok C_CONTRACT
+      | "Operation.get_contract_opt" -> ok C_CONTRACT_OPT
       | "Operation.get_entrypoint" -> ok C_CONTRACT_ENTRYPOINT
+      | "Operation.get_entrypoint_opt" -> ok C_CONTRACT_ENTRYPOINT_OPT
       | "int"                      -> ok C_INT
       | "abs"                      -> ok C_ABS
       | "unit"                     -> ok C_UNIT
@@ -657,6 +661,20 @@ module Typer = struct
       get_t_contract tv in
     ok @@ t_contract tv' ()
 
+  let get_contract_opt = typer_1_opt "CONTRACT OPT" @@ fun addr_tv tv_opt ->
+    if not (type_value_eq (addr_tv, t_address ()))
+    then fail @@ simple_error (Format.asprintf "get_contract_opt expects an address, got %a" PP.type_value addr_tv)
+    else
+    let%bind tv =
+      trace_option (simple_error "get_contract_opt needs a type annotation") tv_opt in
+    let%bind tv =
+      trace_strong (simple_error "get_entrypoint_opt has a not-option annotation") @@
+      get_t_option tv in
+    let%bind tv' =
+      trace_strong (simple_error "get_entrypoint_opt has a not-option(contract) annotation") @@
+      get_t_contract tv in
+    ok @@ t_option (t_contract tv' ()) ()
+
   let get_entrypoint = typer_2_opt "CONTRACT_ENTRYPOINT" @@ fun entry_tv addr_tv tv_opt ->
     if not (type_value_eq (entry_tv, t_string ()))
     then fail @@ simple_error (Format.asprintf "get_entrypoint expects a string entrypoint label for first argument, got %a" PP.type_value entry_tv)
@@ -671,6 +689,23 @@ module Typer = struct
       get_t_contract tv in
     ok @@ t_contract tv' ()
 
+  let get_entrypoint_opt = typer_2_opt "CONTRACT_ENTRYPOINT_OPT" @@ fun entry_tv addr_tv tv_opt ->
+    if not (type_value_eq (entry_tv, t_string ()))
+    then fail @@ simple_error (Format.asprintf "get_entrypoint_opt expects a string entrypoint label for first argument, got %a" PP.type_value entry_tv)
+    else
+    if not (type_value_eq (addr_tv, t_address ()))
+    then fail @@ simple_error (Format.asprintf "get_entrypoint_opt expects an address for second argument, got %a" PP.type_value addr_tv)
+    else
+    let%bind tv =
+      trace_option (simple_error "get_entrypoint_opt needs a type annotation") tv_opt in
+    let%bind tv =
+      trace_strong (simple_error "get_entrypoint_opt has a not-option annotation") @@
+      get_t_option tv in
+    let%bind tv' =
+      trace_strong (simple_error "get_entrypoint_opt has a not-option(contract) annotation") @@
+      get_t_contract tv in
+    ok @@ t_option (t_contract tv' ())()
+
   let set_delegate = typer_1 "SET_DELEGATE" @@ fun delegate_opt ->
     let%bind () = assert_eq_1 delegate_opt (t_option (t_key_hash ()) ()) in
     ok @@ t_operation ()
@@ -1020,7 +1055,9 @@ module Typer = struct
     | C_CHAIN_ID            -> ok @@ chain_id ;
     (*BLOCKCHAIN *)
     | C_CONTRACT            -> ok @@ get_contract ;
+    | C_CONTRACT_OPT        -> ok @@ get_contract_opt ;
     | C_CONTRACT_ENTRYPOINT -> ok @@ get_entrypoint ;
+    | C_CONTRACT_ENTRYPOINT_OPT -> ok @@ get_entrypoint_opt ;
     | C_AMOUNT              -> ok @@ amount ;
     | C_BALANCE             -> ok @@ balance ;
     | C_CALL                -> ok @@ transaction ;

src/stages/common/PP.ml

@@ -108,7 +108,9 @@ let constant ppf : constant -> unit = function
   (* Blockchain *)
   | C_CALL                  -> fprintf ppf "CALL"
   | C_CONTRACT              -> fprintf ppf "CONTRACT"
+  | C_CONTRACT_OPT          -> fprintf ppf "CONTRACT_OPT"
   | C_CONTRACT_ENTRYPOINT   -> fprintf ppf "CONTRACT_ENTRYPOINT"
+  | C_CONTRACT_ENTRYPOINT_OPT -> fprintf ppf "CONTRACT_ENTRYPOINT_OPT"
   | C_AMOUNT                -> fprintf ppf "AMOUNT"
   | C_BALANCE               -> fprintf ppf "BALANCE"
   | C_SOURCE                -> fprintf ppf "SOURCE"

src/stages/common/types.ml

@@ -225,7 +225,9 @@ type constant =
   (* Blockchain *)
   | C_CALL
   | C_CONTRACT
+  | C_CONTRACT_OPT
   | C_CONTRACT_ENTRYPOINT
+  | C_CONTRACT_ENTRYPOINT_OPT
   | C_AMOUNT
   | C_BALANCE
   | C_SOURCE

src/test/contracts/entrypoints.ligo

@@ -3,3 +3,14 @@ function cb(const a : address; const s : unit) : list(operation) * unit is
     const c : contract(unit) = get_entrypoint("%cb", a)
   }
   with (list transaction(unit, 0mutez, c) end, s)
+
+
+function cbo(const a : address; const s : unit) : list(operation) * unit is
+  block {
+    const c : contract(unit) = 
+      case (get_entrypoint_opt("%cbo", a) : option(contract (unit))) of
+        | Some (c) -> c
+        | None -> (failwith ("entrypoint not found") : contract (unit))
+      end
+  }
+  with (list transaction(unit, 0mutez, c) end, s)