feat: assemble ExpirationTransactionScript on build-time

Author: TomasArrachea

CHANGELOG.md

@@ -56,6 +56,7 @@
 - Added the canonical `ExpirationTransactionScript` to `miden-standards`, with a delta-independent script root that network accounts can allowlist ([#3051](https://github.com/0xMiden/protocol/pull/3051)).
 - [BREAKING] Migrated the `miden-standards` library to a `miden-project.toml` project ([#3107](https://github.com/0xMiden/protocol/pull/3107)).
 - [BREAKING] Migrated `miden-protocol` MASM assembly to a `miden-project.toml` project ([#3094](https://github.com/0xMiden/protocol/pull/3094)).
+- Added `ExpirationTransactionScript` assembly in build-time into a package ([#3111](https://github.com/0xMiden/protocol/pull/3111)).
 - [BREAKING] Replaced `AccountInterface::build_send_notes_script` with a standalone `SendNotesTransactionScript` built against `AccountCodeInterface` ([#3055](https://github.com/0xMiden/protocol/pull/3055)).
 - Added an `AccountCode::interface` helper that returns the public `AccountCodeInterface` ([#3080](https://github.com/0xMiden/protocol/pull/3080)).
 - [BREAKING] Tightened `AccountStorage::get_map_item` to take a `StorageMapKey` instead of a raw `Word` ([#3080](https://github.com/0xMiden/protocol/pull/3080)).

crates/miden-standards/asm/tx_scripts/expiration/expiration.masm

@@ -0,0 +1,15 @@
+use miden::protocol::tx
+
+#! Set the transaction's expiration delta.
+#!
+#! Inputs:  [[delta, 0, 0, 0], pad(12)]
+#! Outputs: [pad(16)]
+#!
+#! Panics if:
+#! - delta is 0 or not a u32 in the range 1..=0xFFFF (ERR_TX_INVALID_EXPIRATION_DELTA).
+#!
+#! Invocation: call
+begin
+    exec.tx::update_expiration_block_delta
+    # => [pad(16)]
+end

crates/miden-standards/asm/tx_scripts/expiration/miden-project.toml

@@ -0,0 +1,11 @@
+[package]
+name              = "expiration-tx-script"
+version.workspace = true
+
+[[bin]]
+name = "expiration"
+path = "expiration.masm"
+
+[dependencies]
+miden-core.workspace     = true
+miden-protocol.workspace = true

crates/miden-standards/asm/tx_scripts/miden-project.toml

@@ -0,0 +1,9 @@
+[workspace]
+members = ["expiration"]
+
+[workspace.package]
+version = "0.16.0"
+
+[workspace.dependencies]
+miden-core     = { linkage = "dynamic", version = "0.24" }
+miden-protocol = { linkage = "dynamic", version = "0.16.0" }

crates/miden-standards/build.rs

@@ -19,6 +19,7 @@ const ASSETS_DIR: &str = "assets";
 const ASM_DIR: &str = "asm";
 const ASM_STANDARDS_DIR: &str = "standards";
 const ASM_COMPONENTS_DIR: &str = "components";
+const ASM_TX_SCRIPTS_DIR: &str = "tx_scripts";
 
 /// Name of the manifest file defining a Miden project.
 const PROJECT_MANIFEST: &str = "miden-project.toml";
@@ -35,6 +36,7 @@ const STANDARDS_ERRORS_ARRAY_NAME: &str = "STANDARDS_ERRORS";
 /// Read and parse the contents from `./asm`.
 /// - Compiles the contents of asm/standards directory into a package. Note scripts are included in
 ///   this library.
+/// - Compiles the contents of asm/tx_scripts directory into individual executable packages.
 /// - Compiles the contents of asm/components directory into individual packages.
 fn main() -> Result<()> {
     // re-build when the MASM code changes
@@ -59,6 +61,15 @@ fn main() -> Result<()> {
     // compile standards library (includes note scripts) and seed it into the registry
     compile_standards_lib(&source_dir, &target_dir, assembler.clone(), &mut registry)?;
 
+    // compile transaction scripts
+    compile_tx_scripts(
+        &source_dir.join(ASM_TX_SCRIPTS_DIR),
+        &target_dir.join(ASM_TX_SCRIPTS_DIR),
+        &assembler,
+        &mut registry,
+        source_manager.clone(),
+    )?;
+
     // compile account components
     compile_account_components(
         &source_dir.join(ASM_COMPONENTS_DIR),
@@ -148,6 +159,45 @@ fn compile_account_components(
     Ok(())
 }
 
+// COMPILE TRANSACTION SCRIPTS
+// ================================================================================================
+
+/// Assembles each member of the tx-scripts workspace in `source_dir` into an executable MAST
+/// package and saves it to `target_dir`. Each file is named after its `[[bin]]` target (e.g.
+/// `expiration.masp`), so the include path used by the Rust side is the script name.
+///
+/// Each script's dependencies are declared in its `miden-project.toml` and resolved against the
+/// `registry`, which is seeded with the core, kernel, protocol, and standards packages.
+fn compile_tx_scripts(
+    source_dir: &Path,
+    target_dir: &Path,
+    assembler: &Assembler,
+    registry: &mut InMemoryPackageRegistry,
+    source_manager: Arc<dyn SourceManager>,
+) -> Result<()> {
+    let manifest =
+        source_manager.load_file(&source_dir.join(PROJECT_MANIFEST)).into_diagnostic()?;
+    let workspace = Workspace::load(manifest, source_manager.as_ref())?;
+
+    std::fs::create_dir_all(target_dir).into_diagnostic()?;
+
+    for script in workspace.members() {
+        for target in script.executable_targets() {
+            let target_name = target.name.inner().as_ref();
+
+            let package = assembler
+                .clone()
+                .for_project(script.clone(), registry)?
+                .assemble(ProjectTargetSelector::Executable(target_name), BUILD_PROFILE)?;
+
+            let output_file = target_dir.join(target_name).with_extension(Package::EXTENSION);
+            package.write_to_file(output_file).into_diagnostic()?;
+        }
+    }
+
+    Ok(())
+}
+
 // ERROR CONSTANTS FILE GENERATION
 // ================================================================================================
 

crates/miden-standards/src/tx_script/expiration_script.rs

@@ -1,38 +1,27 @@
 use core::num::NonZeroU16;
 
 use miden_protocol::transaction::{TransactionScript, TransactionScriptRoot};
+use miden_protocol::utils::serde::Deserializable;
 use miden_protocol::utils::sync::LazyLock;
+use miden_protocol::vm::Package;
 use miden_protocol::{Felt, Word};
 
-use crate::code_builder::CodeBuilder;
-
-// EXPIRATION TRANSACTION SCRIPT
+// CONSTANTS
 // ================================================================================================
 
-/// Transaction script that sets the expiration delta.
-const EXPIRATION_TX_SCRIPT_SOURCE: &str = "\
-use miden::protocol::tx
+/// The canonical expiration transaction script, assembled at build time from
+/// `asm/tx_scripts/expiration/expiration.masm` into an executable package.
+const EXPIRATION_TX_SCRIPT_BYTES: &[u8] =
+    include_bytes!(concat!(env!("OUT_DIR"), "/assets/tx_scripts/expiration.masp"));
 
-#! Set the transaction's expiration delta.
-#!
-#! Inputs:  [[delta, 0, 0, 0], pad(12)]
-#! Outputs: [pad(16)]
-#!
-#! Panics if:
-#! - delta is 0 or not a u32 in the range 1..=0xFFFF (ERR_TX_INVALID_EXPIRATION_DELTA).
-#!
-#! Invocation: call
-@transaction_script
-pub proc main
-    exec.tx::update_expiration_block_delta
-    # => [pad(16)]
-end
-";
+// EXPIRATION TRANSACTION SCRIPT
+// ================================================================================================
 
 static EXPIRATION_TX_SCRIPT: LazyLock<TransactionScript> = LazyLock::new(|| {
-    CodeBuilder::default()
-        .compile_tx_script(EXPIRATION_TX_SCRIPT_SOURCE)
-        .expect("canonical expiration tx script should compile")
+    let package = Package::read_from_bytes(EXPIRATION_TX_SCRIPT_BYTES)
+        .expect("expiration tx script masp should be well-formed");
+    TransactionScript::from_package(&package)
+        .expect("expiration tx script package should contain an executable program")
 });
 
 /// The canonical transaction script that sets the transaction's expiration delta to the value