diff --git a/src/passes/operators/operators.ml b/src/passes/operators/operators.ml index 7316d6650..f91d7d29f 100644 --- a/src/passes/operators/operators.ml +++ b/src/passes/operators/operators.ml @@ -73,6 +73,9 @@ module Simplify = struct ("unit" , "UNIT") ; ("source" , "SOURCE") ; ("sender" , "SENDER") ; + ("address", "ADDRESS") ; + ("self_address", "SELF_ADDRESS") ; + ("implicit_account", "IMPLICIT_ACCOUNT") ; ("failwith" , "FAILWITH") ; ("bitwise_or" , "OR") ; ("bitwise_and" , "AND") ; @@ -136,6 +139,9 @@ module Simplify = struct ("gas", "STEPS_TO_QUOTA") ; ("Current.sender" , "SENDER") ; ("sender", "SENDER") ; + ("Current.address", "ADDRESS") ; + ("Current.self_address", "SELF_ADDRESS") ; + ("Current.implicit_account", "IMPLICIT_ACCOUNT") ; ("Current.source" , "SOURCE") ; ("source", "SOURCE") ; ("Current.failwith", "FAILWITH") ; @@ -466,6 +472,13 @@ module Typer = struct let%bind () = assert_t_contract contract in ok @@ t_address () + let self_address = typer_0 "SELF_ADDRESS" @@ fun _ -> + ok @@ t_address () + + let implicit_account = typer_1 "IMPLICIT_ACCOUNT" @@ fun key_hash -> + let%bind () = assert_t_key_hash key_hash in + ok @@ t_contract (t_unit () ) () + let now = constant "NOW" @@ t_timestamp () let transaction = typer_3 "CALL" @@ fun param amount contract -> @@ -794,6 +807,8 @@ module Typer = struct now ; slice ; address ; + self_address ; + implicit_account ; assertion ; list_cons ; ] @@ -865,6 +880,8 @@ module Compiler = struct ("BALANCE" , simple_constant @@ prim I_BALANCE) ; ("AMOUNT" , simple_constant @@ prim I_AMOUNT) ; ("ADDRESS" , simple_unary @@ prim I_ADDRESS) ; + ("SELF_ADDRESS", simple_constant @@ (seq [prim I_SELF ; prim I_ADDRESS])) ; + ("IMPLICIT_CONTRACT", simple_unary @@ prim I_IMPLICIT_ACCOUNT) ; ("NOW" , simple_constant @@ prim I_NOW) ; ("CALL" , simple_ternary @@ prim I_TRANSFER_TOKENS) ; ("SOURCE" , simple_constant @@ prim I_SOURCE) ; diff --git a/src/test/contracts/address.ligo b/src/test/contracts/address.ligo new file mode 100644 index 000000000..555ab6fe7 --- /dev/null +++ b/src/test/contracts/address.ligo @@ -0,0 +1 @@ +function main (const c: contract(unit)) : address is address(c) diff --git a/src/test/contracts/address.mligo b/src/test/contracts/address.mligo new file mode 100644 index 000000000..efca3d8b8 --- /dev/null +++ b/src/test/contracts/address.mligo @@ -0,0 +1 @@ +let main (c: unit contract) : address = Current.address c diff --git a/src/test/contracts/implicit_account.ligo b/src/test/contracts/implicit_account.ligo new file mode 100644 index 000000000..8883a778d --- /dev/null +++ b/src/test/contracts/implicit_account.ligo @@ -0,0 +1 @@ +function main (const kh: key_hash) : contract(unit) is implicit_account(kh) diff --git a/src/test/contracts/implicit_account.mligo b/src/test/contracts/implicit_account.mligo new file mode 100644 index 000000000..bf3c89dcc --- /dev/null +++ b/src/test/contracts/implicit_account.mligo @@ -0,0 +1 @@ +let main (kh: key_hash) : unit contract = Current.implicit_account kh diff --git a/src/test/contracts/self_address.ligo b/src/test/contracts/self_address.ligo new file mode 100644 index 000000000..72872ce0f --- /dev/null +++ b/src/test/contracts/self_address.ligo @@ -0,0 +1 @@ +function main (const p: unit) : address is self_address diff --git a/src/test/contracts/self_address.mligo b/src/test/contracts/self_address.mligo new file mode 100644 index 000000000..ade0926ee --- /dev/null +++ b/src/test/contracts/self_address.mligo @@ -0,0 +1 @@ +let main (p: unit) : address = Current.self_address diff --git a/src/test/integration_tests.ml b/src/test/integration_tests.ml index ecddfda4f..eb8b36535 100644 --- a/src/test/integration_tests.ml +++ b/src/test/integration_tests.ml @@ -1233,6 +1233,30 @@ let balance_constant_mligo () : unit result = let expected = e_tuple [e_list []; e_mutez 4000000000000] in expect_eq program "main" input expected +let address () : unit result = + let%bind _ = type_file "./contracts/address.ligo" in + ok () + +let address_mligo () : unit result = + let%bind _ = mtype_file "./contracts/address.mligo" in + ok () + +let self_address () : unit result = + let%bind _ = type_file "./contracts/self_address.ligo" in + ok () + +let self_address_mligo () : unit result = + let%bind _ = mtype_file "./contracts/self_address.mligo" in + ok () + +let implicit_account () : unit result = + let%bind _ = type_file "./contracts/implicit_account.ligo" in + ok () + +let implicit_account_mligo () : unit result = + let%bind _ = mtype_file "./contracts/implicit_account.mligo" in + ok () + let is_nat () : unit result = let%bind program = type_file "./contracts/isnat.ligo" in let%bind () = @@ -1398,6 +1422,12 @@ let main = test_suite "Integration (End to End)" [ test "let multiple (mligo)" mligo_let_multiple ; test "balance constant" balance_constant ; test "balance constant (mligo)" balance_constant_mligo ; + test "address" address ; + test "address_mligo" address_mligo ; + test "self address" self_address ; + test "self address (mligo)" self_address_mligo ; + test "implicit account" implicit_account ; + test "implicit account (mligo)" implicit_account_mligo ; test "is_nat" is_nat ; test "is_not (mligo)" is_nat_mligo ; test "simple_access (ligo)" simple_access_ligo;