Add program checksum to deploy process

Author: iamalwaysuncomfortable

wasm/Cargo.toml

@@ -88,6 +88,7 @@ version = "1.8"
 
 [dependencies.reqwest]
 version = "0.11.18"
+features = [ "json" ]
 
 [dependencies.serde]
 version = "1.0.183"

wasm/src/lib.rs

@@ -176,9 +176,9 @@ pub use types::{Field, Group, Scalar};
 mod thread_pool;
 
 mod utilities;
-pub use utilities::EncryptionToolkit;
 #[cfg(test)]
 pub use utilities::test;
+pub use utilities::{EncryptionToolkit, get, get_network, latest_block_height};
 
 #[cfg(test)]
 mod thread_pool {

wasm/src/programs/manager/deploy.rs

@@ -22,10 +22,13 @@ use crate::{
     RecordPlaintext,
     Transaction,
     execute_fee,
+    latest_block_height,
     log,
     types::native::{
+        AddressNative,
         CurrentAleo,
         CurrentNetwork,
+        PrivateKeyNative,
         ProcessNative,
         ProgramIDNative,
         ProgramNative,
@@ -34,6 +37,7 @@ use crate::{
         TransactionNative,
     },
 };
+use snarkvm_console::prelude::{ConsensusVersion, Network};
 
 use js_sys::Object;
 use rand::{SeedableRng, rngs::StdRng};
@@ -84,11 +88,24 @@ impl ProgramManager {
 
         log("Creating deployment");
         let node_url = url.as_deref().unwrap_or(DEFAULT_URL);
-        let deployment = process.deploy::<CurrentAleo, _>(&program, rng).map_err(|err| err.to_string())?;
+        let mut deployment = process.deploy::<CurrentAleo, _>(&program, rng).map_err(|err| err.to_string())?;
         if deployment.program().functions().is_empty() {
             return Err("Attempted to create an empty transaction deployment".to_string());
         }
 
+        log("Setting program checksum and owner");
+        let latest_height = latest_block_height(node_url).await?;
+        let consensus_version = CurrentNetwork::CONSENSUS_VERSION(latest_height).map_err(|err| err.to_string())?;
+        let private_key_native = PrivateKeyNative::from(private_key);
+        if consensus_version < ConsensusVersion::V9 {
+            deployment.set_program_checksum_raw(None);
+            deployment.set_program_owner_raw(None)
+        } else {
+            deployment.set_program_checksum_raw(Some(deployment.program().to_checksum()));
+            deployment
+                .set_program_owner_raw(Some(AddressNative::try_from(private_key_native).map_err(|e| e.to_string())?));
+        }
+
         log("Ensuring the fee is sufficient to pay for the deployment");
         let (minimum_deployment_cost, (_, _, _, _)) =
             deployment_cost::<CurrentNetwork>(&process, &deployment).map_err(|err| err.to_string())?;

wasm/src/utilities/mod.rs

@@ -21,3 +21,6 @@ pub mod test;
 
 pub mod encrypt;
 pub use encrypt::EncryptionToolkit;
+
+pub mod network;
+pub use network::{get, get_network, latest_block_height};

wasm/src/utilities/network.rs

@@ -0,0 +1,46 @@
+// Copyright (C) 2019-2025 Provable Inc.
+// This file is part of the Provable SDK library.
+
+// The Provable SDK library is free software: you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+
+// The Provable SDK library is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+// GNU General Public License for more details.
+
+// You should have received a copy of the GNU General Public License
+// along with the Provable SDK library. If not, see <https://www.gnu.org/licenses/>.
+
+use crate::types::native::CurrentNetwork;
+
+use snarkvm_console::network::Network;
+
+/// Get the current network name.
+pub fn get_network() -> &'static str {
+    match CurrentNetwork::ID {
+        snarkvm_console::network::MainnetV0::ID => "mainnet",
+        snarkvm_console::network::TestnetV0::ID => "testnet",
+        snarkvm_console::network::CanaryV0::ID => "canary",
+        _ => panic!("Invalid network"),
+    }
+}
+
+/// Get the latest block height.
+pub async fn latest_block_height(base_url: &str) -> Result<u32, String> {
+    let url = format!("{base_url}/{}/block/height/latest", get_network());
+    let res = get(&url).await?;
+    Ok(res)
+}
+
+/// Make a GET request to the service.
+pub async fn get<T>(url: &str) -> Result<T, String>
+where
+    T: serde::de::DeserializeOwned,
+{
+    let client = reqwest::Client::new();
+    let res = client.get(url).send().await.map_err(|e| e.to_string())?.json().await.map_err(|e| e.to_string())?;
+    Ok(res)
+}