Merge #276

276: get_operation_id r=Eitu33 a=Eitu33



Co-authored-by: Thomas Plisson <[email protected]>

src/as_execution/abi.rs

@@ -792,6 +792,20 @@ pub(crate) fn assembly_script_send_message(
     Ok(())
 }
 
+/// converts a public key to an address
+#[named]
+pub(crate) fn assembly_script_get_origin_operation_id(
+    mut ctx: FunctionEnvMut<ASEnv>,
+) -> ABIResult<i32> {
+    let env = get_env(&ctx)?;
+    sub_remaining_gas_abi(&env, &mut ctx, function_name!())?;
+    let operation_id = env
+        .get_interface()
+        .get_origin_operation_id()?
+        .unwrap_or_default();
+    Ok(pointer_from_string(&env, &mut ctx, &operation_id)?.offset() as i32)
+}
+
 /// gets the period of the current execution slot
 #[named]
 pub(crate) fn assembly_script_get_current_period(mut ctx: FunctionEnvMut<ASEnv>) -> ABIResult<i64> {

src/as_execution/context.rs

@@ -253,6 +253,7 @@ impl ASContext {
                 "assembly_script_get_call_coins" => Function::new_typed_with_env(store, &fenv, assembly_script_get_call_coins),
                 "assembly_script_get_time" => Function::new_typed_with_env(store, &fenv, assembly_script_get_time),
                 "assembly_script_send_message" => Function::new_typed_with_env(store, &fenv, assembly_script_send_message),
+                "assembly_script_get_origin_operation_id" => Function::new_typed_with_env(store, &fenv, assembly_script_get_origin_operation_id),
                 "assembly_script_get_current_period" => Function::new_typed_with_env(store, &fenv, assembly_script_get_current_period),
                 "assembly_script_get_current_thread" => Function::new_typed_with_env(store, &fenv, assembly_script_get_current_thread),
                 "assembly_script_set_bytecode" => Function::new_typed_with_env(store, &fenv, assembly_script_set_bytecode),

src/types.rs

@@ -105,6 +105,7 @@ impl Default for GasCosts {
         abi_costs.insert(String::from("assembly_script_keccak256_hash"), 83);
         abi_costs.insert(String::from("assembly_script_print"), 35);
         abi_costs.insert(String::from("assembly_script_send_message"), 316);
+        abi_costs.insert(String::from("assembly_script_get_origin_operation_id"), 200);
         abi_costs.insert(String::from("assembly_script_set_bytecode"), 74);
         abi_costs.insert(String::from("assembly_script_set_bytecode_for"), 129);
         abi_costs.insert(String::from("assembly_script_set_data"), 158);
@@ -412,6 +413,11 @@ pub trait Interface: Send + Sync + InterfaceClone {
         unimplemented!("send_message")
     }
 
+    // Returns the operation id that originated the current execution if there is one
+    fn get_origin_operation_id(&self) -> Result<Option<String>> {
+        unimplemented!("get_origin_operation_id")
+    }
+
     // Sha256 hash bytes
     fn hash_sha256(&self, bytes: &[u8]) -> Result<[u8; 32]> {
         unimplemented!("hash_sha256")