miden_client/transaction/request/

builder.rs

//! Contains structures and functions related to transaction creation.
use alloc::collections::{BTreeMap, BTreeSet};
use alloc::string::ToString;
use alloc::vec::Vec;
use core::borrow::Borrow;

use miden_protocol::account::{AccountCode, AccountCodeUpgrade, AccountId};
use miden_protocol::asset::{Asset, AssetAmount, FungibleAsset};
use miden_protocol::block::BlockNumber;
use miden_protocol::crypto::SequentialCommit;
use miden_protocol::crypto::merkle::InnerNodeInfo;
use miden_protocol::crypto::merkle::store::MerkleStore;
use miden_protocol::crypto::rand::FeltRng;
use miden_protocol::errors::NoteError;
use miden_protocol::note::{
    Note,
    NoteAssets,
    NoteAttachment,
    NoteDetails,
    NoteDetailsCommitment,
    NoteId,
    NoteRecipient,
    NoteScript,
    NoteStorage,
    NoteTag,
    NoteType,
    PartialNote,
    PartialNoteMetadata,
};
use miden_protocol::transaction::{InputNote, TransactionScript};
use miden_protocol::vm::AdviceMap;
use miden_protocol::{Felt, Word};
use miden_standards::note::{P2idNote, P2ideNote, PswapNote, PswapNoteStorage, SwapNote};

use super::code_upgrade::account_code_upgrade_script;
use super::{
    ForeignAccount,
    NoteArgs,
    TransactionRequest,
    TransactionRequestError,
    TransactionScriptTemplate,
};
use crate::ClientRng;

// TRANSACTION REQUEST BUILDER
// ================================================================================================

/// A builder for a [`TransactionRequest`].
///
/// Use this builder to construct a [`TransactionRequest`] by adding input notes, specifying
/// scripts, and setting other transaction parameters.
#[derive(Clone, Debug)]
pub struct TransactionRequestBuilder {
    /// Blocks that the transaction must be able to authenticate against its reference block.
    block_numbers: BTreeSet<BlockNumber>,
    /// Notes to be consumed by the transaction, in consumption order.
    ///
    /// A note with an entry in `explicit_input_notes` is consumed in the mode that entry pins. The
    /// executing client infers the mode of every other note from its store.
    input_notes: Vec<Note>,
    /// Optional arguments of the Notes to be consumed by the transaction. This includes both
    /// authenticated and unauthenticated notes.
    input_notes_args: Vec<(NoteId, Option<NoteArgs>)>,
    /// Pinned consumption mode of selected input notes.
    explicit_input_notes: BTreeMap<NoteId, InputNote>,
    /// Notes to be created by the transaction. The full note data is needed internally to build the
    /// transaction script template.
    own_output_notes: Vec<Note>,
    /// A map of recipients of the output notes expected to be generated by the transaction.
    expected_output_recipients: BTreeMap<Word, NoteRecipient>,
    /// A map of details and tags of notes we expect to be created as part of future transactions
    /// with their respective tags.
    ///
    /// For example, after a swap note is consumed, a payback note is expected to be created.
    expected_future_notes: BTreeMap<NoteDetailsCommitment, (NoteDetails, NoteTag)>,
    /// Custom transaction script to be used.
    custom_script: Option<TransactionScript>,
    /// Initial state of the `AdviceMap` that provides data during runtime.
    advice_map: AdviceMap,
    /// Initial state of the `MerkleStore` that provides data during runtime.
    merkle_store: MerkleStore,
    /// Foreign account data requirements. At execution time, account data will be retrieved from
    /// the network, and injected as advice inputs. Additionally, the account's code will be added
    /// to the executor and prover.
    foreign_accounts: BTreeMap<AccountId, ForeignAccount>,
    /// The number of blocks in relation to the transaction's reference block after which the
    /// transaction will expire. If `None`, the transaction will not expire.
    expiration_delta: Option<u16>,
    /// Indicates whether to **silently** ignore invalid input notes when executing the transaction.
    /// This will allow the transaction to be executed even if some input notes are invalid.
    ignore_invalid_input_notes: bool,
    /// Optional [`Word`] that will be pushed to the operand stack before the transaction script
    /// execution. If the advice map is extended with some user defined entries, this script
    /// argument could be used as a key to access the corresponding value.
    script_arg: Option<Word>,
    /// Optional [`Word`] that will be pushed to the stack for the authentication procedure during
    /// transaction execution.
    auth_arg: Option<Word>,
    /// Salt the native fee conversion info is committed under when the transaction is prepared, set
    /// through [`TransactionRequestBuilder::fee_conversion_salt`]. `None` leaves the client to use
    /// its fixed default salt.
    fee_conversion_salt: Option<Word>,
    /// Note scripts that the node's NTX builder will need in its script registry.
    ///
    /// See [`TransactionRequestBuilder::expected_ntx_scripts`] for details.
    expected_ntx_scripts: Vec<NoteScript>,
    /// New code of the executing account.
    ///
    /// See [`TransactionRequestBuilder::account_code_upgrade`] for details.
    account_code_upgrade: Option<AccountCodeUpgrade>,
}

impl TransactionRequestBuilder {
    // CONSTRUCTORS
    // --------------------------------------------------------------------------------------------

    /// Creates a new, empty [`TransactionRequestBuilder`].
    pub fn new() -> Self {
        Self {
            block_numbers: BTreeSet::new(),
            input_notes: vec![],
            input_notes_args: vec![],
            explicit_input_notes: BTreeMap::new(),
            own_output_notes: Vec::new(),
            expected_output_recipients: BTreeMap::new(),
            expected_future_notes: BTreeMap::new(),
            custom_script: None,
            advice_map: AdviceMap::default(),
            merkle_store: MerkleStore::default(),
            expiration_delta: None,
            foreign_accounts: BTreeMap::default(),
            ignore_invalid_input_notes: false,
            script_arg: None,
            auth_arg: None,
            fee_conversion_salt: None,
            expected_ntx_scripts: vec![],
            account_code_upgrade: None,
        }
    }

    /// Adds blocks that the transaction must be able to authenticate against its reference block.
    ///
    /// The client adds each block header and its authentication path to the transaction's partial
    /// blockchain. The client fetches a header from the node when it is not in the local store.
    /// Each block must be at or before the transaction's reference block.
    #[must_use]
    pub fn block_numbers<I, B>(mut self, block_numbers: I) -> Self
    where
        I: IntoIterator<Item = B>,
        B: Borrow<BlockNumber>,
    {
        self.block_numbers
            .extend(block_numbers.into_iter().map(|block_num| *block_num.borrow()));
        self
    }

    /// Adds the specified notes as input notes to the transaction request.
    ///
    /// The executing client consumes a note as authenticated when its store holds the note's
    /// inclusion proof and as unauthenticated otherwise. Use [`Self::explicit_input_notes`] when
    /// the mode must not depend on the executing client.
    #[must_use]
    pub fn input_notes(
        mut self,
        notes: impl IntoIterator<Item = (Note, Option<NoteArgs>)>,
    ) -> Self {
        for (note, argument) in notes {
            self.input_notes_args.push((note.id(), argument));
            self.input_notes.push(note);
        }
        self
    }

    /// Adds the specified [`InputNote`]s as input notes to the transaction request. Each note is
    /// consumed in the mode it carries: an [`InputNote::Authenticated`] note with its proof, an
    /// [`InputNote::Unauthenticated`] note as unauthenticated even if the executing client's store
    /// holds a proof for it. The executing client does not classify these notes from its store, so
    /// every client that executes the request commits to the same input notes and produces the same
    /// transaction summary. Use this for a request that is shared across clients.
    ///
    /// To consume an authenticated note, the executing client must be able to serve the header of
    /// the note's creation block, from its store or from the
    /// [`ChainAnchor`](crate::transaction::ChainAnchor) the request executes against.
    #[must_use]
    pub fn explicit_input_notes(
        mut self,
        notes: impl IntoIterator<Item = (InputNote, Option<NoteArgs>)>,
    ) -> Self {
        for (input_note, argument) in notes {
            let note_id = input_note.id();

            self.input_notes_args.push((note_id, argument));
            self.input_notes.push(input_note.note().clone());
            self.explicit_input_notes.insert(note_id, input_note);
        }
        self
    }

    /// Specifies the output notes that should be created in the transaction script and will be used
    /// as a transaction script template. These notes will also be added to the expected output
    /// recipients of the transaction.
    ///
    /// If a transaction script template is already set (e.g. by calling `with_custom_script`), the
    /// [`TransactionRequestBuilder::build`] method will return an error.
    #[must_use]
    pub fn own_output_notes(mut self, notes: impl IntoIterator<Item = Note>) -> Self {
        for note in notes {
            self.expected_output_recipients
                .insert(note.recipient().digest(), note.recipient().clone());
            self.own_output_notes.push(note);
        }

        self
    }

    /// Specifies a custom transaction script to be used.
    ///
    /// If a script template is already set (e.g. by calling `with_own_output_notes`), the
    /// [`TransactionRequestBuilder::build`] method will return an error.
    #[must_use]
    pub fn custom_script(mut self, script: TransactionScript) -> Self {
        self.custom_script = Some(script);
        self
    }

    /// Specifies one or more foreign accounts (public or private) that contain data utilized by the
    /// transaction.
    ///
    /// At execution, the client queries the node and retrieves the appropriate data, depending on
    /// whether each foreign account is public or private:
    ///
    /// - **Public accounts**: the node retrieves the state and code for the account and injects
    ///   them as advice inputs. Public accounts can be omitted here, as they will be lazily loaded
    ///   through RPC calls. Undeclared accounts may trigger additional RPC calls for storage map
    ///   accesses during execution.
    /// - **Private accounts**: the node retrieves a proof of the account's existence and injects
    ///   that as advice inputs. Private accounts must always be declared here with their
    ///   [`PartialAccount`](miden_protocol::account::PartialAccount) state.
    /// - **Prefetched accounts**: the caller supplies the state and inclusion witness as
    ///   [`ForeignAccount::Prefetched`] and nothing is fetched for them. The witness must open
    ///   against the transaction's reference block.
    ///   [`Client::get_foreign_account_inputs`](crate::Client::get_foreign_account_inputs) fetches
    ///   inputs for a given block.
    ///
    /// Declaring an account ID more than once keeps the last declaration.
    #[must_use]
    pub fn foreign_accounts(
        mut self,
        foreign_accounts: impl IntoIterator<Item = impl Into<ForeignAccount>>,
    ) -> Self {
        for account in foreign_accounts {
            let foreign_account: ForeignAccount = account.into();
            self.foreign_accounts.insert(foreign_account.account_id(), foreign_account);
        }

        self
    }

    /// Specifies a transaction's expected output note recipients.
    ///
    /// The set of specified recipients is treated as a subset of the recipients for notes that may
    /// be created by a transaction. That is, the transaction must create notes for all the
    /// specified expected recipients, but it may also create notes for other recipients not
    /// included in this set.
    #[must_use]
    pub fn expected_output_recipients(
        mut self,
        recipients: impl IntoIterator<Item = impl Into<NoteRecipient>>,
    ) -> Self {
        self.expected_output_recipients = recipients
            .into_iter()
            .map(|recipient| {
                let recipient: NoteRecipient = recipient.into();
                (recipient.digest(), recipient)
            })
            .collect::<BTreeMap<_, _>>();
        self
    }

    /// Specifies a set of notes which may be created when a transaction's output notes are
    /// consumed.
    ///
    /// For example, after a SWAP note is consumed, a payback note is expected to be created. This
    /// allows the client to track this note accordingly.
    #[must_use]
    pub fn expected_future_notes(mut self, notes: Vec<(NoteDetails, NoteTag)>) -> Self {
        self.expected_future_notes = notes
            .into_iter()
            .map(|note| (note.0.commitment(), note))
            .collect::<BTreeMap<_, _>>();
        self
    }

    /// Extends the advice map with the specified `([Word], Vec<[Felt]>)` pairs.
    #[must_use]
    pub fn extend_advice_map<I, V>(mut self, iter: I) -> Self
    where
        I: IntoIterator<Item = (Word, V)>,
        V: AsRef<[Felt]>,
    {
        self.advice_map.extend(iter.into_iter().map(|(w, v)| (w, v.as_ref().to_vec())));
        self
    }

    /// Extends the merkle store with the specified [`InnerNodeInfo`] elements.
    #[must_use]
    pub fn extend_merkle_store<T: IntoIterator<Item = InnerNodeInfo>>(mut self, iter: T) -> Self {
        self.merkle_store.extend(iter);
        self
    }

    /// The number of blocks in relation to the transaction's reference block after which the
    /// transaction will expire. By default, the transaction will not expire.
    ///
    /// Setting transaction expiration delta defines an upper bound for transaction expiration, but
    /// other code executed during the transaction may impose an even smaller transaction expiration
    /// delta.
    #[must_use]
    pub fn expiration_delta(mut self, expiration_delta: u16) -> Self {
        self.expiration_delta = Some(expiration_delta);
        self
    }

    /// The resulting transaction will **silently** ignore invalid input notes when being executed.
    /// By default, this will not happen.
    #[must_use]
    pub fn ignore_invalid_input_notes(mut self) -> Self {
        self.ignore_invalid_input_notes = true;
        self
    }

    /// Sets an optional [`Word`] that will be pushed to the operand stack before the transaction
    /// script execution. If the advice map is extended with some user defined entries, this script
    /// argument could be used as a key to access the corresponding value.
    #[must_use]
    pub fn script_arg(mut self, script_arg: Word) -> Self {
        self.script_arg = Some(script_arg);
        self
    }

    /// Sets the raw word passed to the authentication procedure during transaction execution.
    ///
    /// This method does not add an advice-map preimage. Use [`Self::auth_args`] for auth components
    /// that resolve committed auth args from the advice map.
    #[deprecated(
        since = "0.17.4",
        note = "use `auth_args` to set the auth args commitment and its advice-map preimage together"
    )]
    #[must_use]
    pub fn auth_arg(mut self, auth_arg: Word) -> Self {
        self.auth_arg = Some(auth_arg);
        self.fee_conversion_salt = None;
        self
    }

    /// Sets the auth arg to the commitment of `auth_args` and adds their preimage to the advice
    /// map.
    ///
    /// Use this method for auth components that load their auth args from the advice map, for
    /// example [`MultisigAuthArgs`](miden_standards::account::auth::MultisigAuthArgs). Use
    /// [`Self::auth_arg`] for auth components that read the auth arg directly.
    ///
    /// The client does not change auth args that the caller sets. The client tracks the bound block
    /// of multisig auth args.
    #[must_use]
    pub fn auth_args<A: SequentialCommit<Commitment = Word>>(mut self, auth_args: &A) -> Self {
        let auth_arg = auth_args.to_commitment();
        self.advice_map.insert(auth_arg, auth_args.to_elements());
        self.auth_arg = Some(auth_arg);
        self.fee_conversion_salt = None;
        self
    }

    /// Declares the salt the fee conversion info is committed under.
    ///
    /// Fees are settled in the chain's native fee asset at rate 1/1. The client commits that info
    /// through the transaction's auth args when preparing the transaction.
    ///
    /// Multisig accounts require a caller-declared salt. The client also binds their auth args to
    /// the transaction's reference block. Use the same reference block when collecting signatures
    /// across repeated executions. See [`crate::Client::execute_transaction_at`].
    #[must_use]
    pub fn fee_conversion_salt(mut self, salt: Word) -> Self {
        self.fee_conversion_salt = Some(salt);
        self.auth_arg = None;
        self
    }

    /// Specifies note scripts that the node's network transaction (NTX) builder will need in its
    /// script registry.
    ///
    /// When a transaction creates notes destined for a network account, the node's NTX builder must
    /// have the scripts of any public output notes in its registry. If a required script is
    /// missing, the NTX will silently fail on the node side.
    ///
    /// When this field is set, the client will check each script against the node before executing
    /// the main transaction. For any script not yet registered, the client automatically creates
    /// and submits a separate registration transaction (a public note carrying that script) so the
    /// node's registry is populated before the NTX executes.
    ///
    /// Standard note scripts are ignored here — the NTX builder resolves them directly.
    #[must_use]
    pub fn expected_ntx_scripts(mut self, scripts: Vec<NoteScript>) -> Self {
        self.expected_ntx_scripts = scripts;
        self
    }

    /// Gives `code` to the transaction as the new code of the executing account.
    ///
    /// The built request adds the serialized code to the transaction advice map. The account
    /// upgrade procedure reads the code after a transaction script initializes an upgrade with the
    /// matching code commitment. This method does not initialize the upgrade or change the
    /// transaction script.
    ///
    /// Use [`Self::build_account_code_upgrade`] when the transaction only upgrades the code.
    #[must_use]
    pub fn account_code_upgrade(mut self, code: AccountCode) -> Self {
        self.account_code_upgrade = Some(AccountCodeUpgrade::new(code));
        self
    }

    // STANDARDIZED REQUESTS
    // --------------------------------------------------------------------------------------------

    /// Consumes the builder and returns a [`TransactionRequest`] for a transaction to consume the
    /// specified notes.
    ///
    /// - `notes` is a list of notes to be consumed.
    pub fn build_consume_notes(
        self,
        notes: Vec<Note>,
    ) -> Result<TransactionRequest, TransactionRequestError> {
        let input_notes = notes.into_iter().map(|id| (id, None));
        self.input_notes(input_notes).build()
    }

    /// Consumes the builder and returns a [`TransactionRequest`] for a transaction to mint fungible
    /// assets. This request must be executed against a fungible faucet account.
    ///
    /// - `asset` is the fungible asset to be minted. The amount must be non-zero: minting nothing
    ///   would emit a P2ID note the target cannot draw anything from.
    /// - `target_id` is the account ID of the account to receive the minted asset.
    /// - `note_type` determines the visibility of the note to be created.
    /// - `rng` is the random number generator used to generate the serial number for the created
    ///   note.
    ///
    /// This function cannot be used with a previously set custom script.
    pub fn build_mint_fungible_asset(
        self,
        asset: FungibleAsset,
        target_id: AccountId,
        note_type: NoteType,
        rng: &mut ClientRng,
    ) -> Result<TransactionRequest, TransactionRequestError> {
        // Minting emits a P2ID note, and a P2ID note carrying nothing is rejected on the transfer
        // path for the same reason: it costs a transaction and leaves the target a note with
        // nothing to consume.
        if asset.amount() == AssetAmount::ZERO {
            return Err(TransactionRequestError::P2IDNoteWithoutAsset);
        }

        let created_note = P2idNote::builder()
            .sender(asset.faucet_id())
            .target(target_id)
            .asset(asset)
            .note_type(note_type)
            .generate_serial_number(rng)
            .build()?
            .into();

        self.own_output_notes(vec![created_note]).build()
    }

    /// Consumes the builder and returns a [`TransactionRequest`] for a transaction to send a P2ID
    /// or P2IDE note. This request must be executed against the wallet sender account.
    ///
    /// - `payment_data` is the data for the payment transaction that contains the asset to be
    ///   transferred, the sender account ID, and the target account ID. If the recall or timelock
    ///   heights are set, a P2IDE note will be created; otherwise, a P2ID note will be created.
    /// - `note_type` determines the visibility of the note to be created.
    /// - `rng` is the random number generator used to generate the serial number for the created
    ///   note.
    ///
    /// This function cannot be used with a previously set custom script.
    pub fn build_pay_to_id(
        self,
        payment_data: PaymentNoteDescription,
        note_type: NoteType,
        rng: &mut ClientRng,
    ) -> Result<TransactionRequest, TransactionRequestError> {
        if payment_data
            .assets()
            .iter()
            .all(|asset| asset.is_fungible() && asset.unwrap_fungible().amount().as_u64() == 0)
        {
            return Err(TransactionRequestError::P2IDNoteWithoutAsset);
        }

        let created_note = payment_data.into_note(note_type, rng)?;

        self.own_output_notes(vec![created_note]).build()
    }

    /// Consumes the builder and returns a [`TransactionRequest`] for a transaction to send a SWAP
    /// note. This request must be executed against the wallet sender account.
    ///
    /// - `swap_data` is the data for the swap transaction that contains the sender account ID, the
    ///   offered asset, and the requested asset. Neither asset may be a zero-amount fungible asset:
    ///   the SWAP note holds the offered one and its payback note holds the requested one.
    /// - `note_type` determines the visibility of the note to be created.
    /// - `payback_note_type` determines the visibility of the payback note.
    /// - `rng` is the random number generator used to generate the serial number for the created
    ///   note.
    ///
    /// This function cannot be used with a previously set custom script.
    pub fn build_swap(
        self,
        swap_data: &SwapTransactionData,
        note_type: NoteType,
        payback_note_type: NoteType,
        rng: &mut ClientRng,
    ) -> Result<TransactionRequest, TransactionRequestError> {
        // Both sides carry value: the SWAP note holds the offered asset, and filling it emits a
        // P2ID payback carrying the requested one. A zero amount on either side leaves one of those
        // two notes empty.
        if is_zero_fungible(&swap_data.offered_asset()) {
            return Err(TransactionRequestError::SwapNoteWithZeroAsset("offered"));
        }
        if is_zero_fungible(&swap_data.requested_asset()) {
            return Err(TransactionRequestError::SwapNoteWithZeroAsset("requested"));
        }

        // The created note is the one that we need as the output of the tx, the other one is the
        // one that we expect to receive and consume eventually.
        let swap_note = SwapNote::builder()
            .sender(swap_data.account_id())
            .offered_asset(swap_data.offered_asset())
            .requested_asset(swap_data.requested_asset())
            .note_type(note_type)
            .payback_note_type(payback_note_type)
            .generate_serial_number(rng)
            .build()?;

        let payback_note_details = swap_note.payback_note_details();
        let created_note = Note::from(swap_note);

        let payback_tag = NoteTag::with_account_target(swap_data.account_id());

        self.expected_future_notes(vec![(payback_note_details, payback_tag)])
            .own_output_notes(vec![created_note])
            .build()
    }

    /// Consumes the builder and returns a [`TransactionRequest`] for a transaction that registers
    /// note scripts in the node's script registry.
    ///
    /// This creates one public output note per script, each with empty assets and storage. The node
    /// indexes the script of every public note it processes, so submitting this transaction makes
    /// the scripts available for future network transactions (NTX).
    ///
    /// - `sender_account_id` is the account executing the transaction.
    /// - `scripts` is the list of note scripts to register.
    /// - `rng` is used to generate serial numbers for the registration notes.
    ///
    /// This function cannot be used with a previously set custom script.
    pub fn build_register_note_scripts(
        self,
        sender_account_id: AccountId,
        scripts: Vec<NoteScript>,
        rng: &mut ClientRng,
    ) -> Result<TransactionRequest, TransactionRequestError> {
        let registration_notes: Vec<Note> = scripts
            .into_iter()
            .map(|script| {
                let serial_num = rng.draw_word();
                let note_storage = NoteStorage::new(vec![])?;
                let recipient = NoteRecipient::new(serial_num, script, note_storage);
                let note_assets = NoteAssets::new(vec![])?;
                let metadata = PartialNoteMetadata::new(sender_account_id, NoteType::Public);
                Ok(Note::new(note_assets, metadata, recipient))
            })
            .collect::<Result<_, NoteError>>()?;

        self.own_output_notes(registration_notes).build()
    }

    /// Consumes the builder and returns a [`TransactionRequest`] for a transaction that creates a
    /// partial swap (PSWAP) note. This request must be executed against the creator account.
    ///
    /// - `pswap_data` is the data for the partial swap that contains the creator account ID, the
    ///   offered fungible asset, and the requested fungible asset.
    /// - `note_type` determines the visibility of the PSWAP note itself.
    /// - `payback_note_type` determines the visibility of the payback note that fillers emit back
    ///   to the creator. Typically [`NoteType::Private`] (cheaper; the fill amount is already
    ///   visible in the executing transaction).
    /// - `note_attachment` is the optional attachment for the PSWAP note. Pass `None` when there is
    ///   nothing to attach.
    /// - `rng` is the random number generator used to generate the serial number for the created
    ///   note.
    ///
    /// This function cannot be used with a previously set custom script.
    pub fn build_pswap_create(
        self,
        pswap_data: &PswapTransactionData,
        note_type: NoteType,
        payback_note_type: NoteType,
        note_attachment: Option<NoteAttachment>,
        rng: &mut ClientRng,
    ) -> Result<TransactionRequest, TransactionRequestError> {
        // Same exchange invariant as build_swap; PSWAP is fungible on both sides.
        if is_zero_fungible(&pswap_data.offered_asset().into()) {
            return Err(TransactionRequestError::SwapNoteWithZeroAsset("offered"));
        }
        if is_zero_fungible(&pswap_data.requested_asset().into()) {
            return Err(TransactionRequestError::SwapNoteWithZeroAsset("requested"));
        }

        let storage = PswapNoteStorage::builder()
            .min_requested_asset(pswap_data.requested_asset())
            .creator_account_id(pswap_data.creator_account_id())
            .payback_note_type(payback_note_type)
            .build();

        let pswap_note = PswapNote::builder()
            .sender(pswap_data.creator_account_id())
            .storage(storage)
            .serial_number(rng.draw_word())
            .note_type(note_type)
            .offered_asset(pswap_data.offered_asset())
            .maybe_attachment(note_attachment)
            .build()
            .map_err(TransactionRequestError::NoteCreationError)?;

        let note: Note = pswap_note.into();
        self.own_output_notes(vec![note]).build()
    }

    /// Consumes the builder and returns a [`TransactionRequest`] for a transaction that consumes
    /// (fills) a partial swap (PSWAP) note. This request must be executed against the consumer
    /// account.
    ///
    /// - `pswap_note` is the PSWAP note being consumed.
    /// - `consumer_account_id` is the account consuming the swap.
    /// - `account_fill_amount` is the amount of the requested asset being provided by the consumer
    ///   account.
    /// - `note_fill_amount` is any additional amount being provided by other (in-flight) notes.
    ///
    /// This function cannot be used with a previously set custom script.
    pub fn build_pswap_consume(
        self,
        pswap_note: &Note,
        consumer_account_id: AccountId,
        account_fill_amount: AssetAmount,
        note_fill_amount: AssetAmount,
    ) -> Result<TransactionRequest, TransactionRequestError> {
        let pswap = PswapNote::try_from(pswap_note)
            .map_err(TransactionRequestError::NoteValidationError)?;

        let requested_faucet_id = pswap.storage().min_requested_asset().faucet_id();

        let account_fill_asset =
            FungibleAsset::new(requested_faucet_id, account_fill_amount.as_u64())?;
        let note_fill_asset = FungibleAsset::new(requested_faucet_id, note_fill_amount.as_u64())?;

        let (payback_note, remainder_pswap) = pswap
            .execute(consumer_account_id, Some(account_fill_asset), Some(note_fill_asset))
            .map_err(TransactionRequestError::NoteExecutionError)?;

        let note_args =
            PswapNote::create_args(account_fill_amount.as_u64(), note_fill_amount.as_u64())
                .map_err(TransactionRequestError::NoteArgError)?;

        // Payback and remainder both settle to the creator, not the consumer. Declare them as
        // expected recipients so the transaction is validated against them, but don't register them
        // as expected future notes — that's the creator's concern, and doing so here would leave
        // stale, un-consumable notes in the consumer's store.
        let mut expected_recipients = vec![payback_note.recipient().clone()];

        if let Some(remainder) = remainder_pswap {
            let remainder_note: Note = remainder.into();
            expected_recipients.push(remainder_note.recipient().clone());
        }

        self.input_notes(vec![(pswap_note.clone(), Some(note_args))])
            .expected_output_recipients(expected_recipients)
            .build()
    }

    /// Consumes the builder and returns a [`TransactionRequest`] for a transaction that cancels a
    /// partial swap (PSWAP) note. This request must be executed against the creator account.
    ///
    /// - `pswap_note` is the PSWAP note to cancel.
    /// - `creator_account_id` is the account that created the note. The note's stored creator must
    ///   match this ID; this is the account the resulting transaction must be executed against.
    ///
    /// This function cannot be used with a previously set custom script.
    pub fn build_pswap_cancel(
        self,
        pswap_note: Note,
        creator_account_id: AccountId,
    ) -> Result<TransactionRequest, TransactionRequestError> {
        let pswap = PswapNote::try_from(&pswap_note)
            .map_err(TransactionRequestError::NoteValidationError)?;

        let note_creator = pswap.storage().creator_account_id();
        if note_creator != creator_account_id {
            return Err(TransactionRequestError::PswapCancelCreatorMismatch {
                expected: note_creator,
                actual: creator_account_id,
            });
        }

        self.input_notes(vec![(pswap_note, None)]).build()
    }

    /// Consumes the builder and returns a [`TransactionRequest`] for a transaction that upgrades
    /// the code of the executing account to `code`. This request must be executed against an
    /// account with the [`UpgradeManager`](crate::account::component::UpgradeManager) component and
    /// the [`Authority::AuthControlled`](crate::account::component::Authority::AuthControlled)
    /// authority.
    ///
    /// - `code` is the new code of the account.
    ///
    /// An upgrade does not change the account storage. The new code must use the same storage
    /// layout as the current code, otherwise the account can become unusable.
    ///
    /// To upgrade a network account, build an [`UpgradeNote`](crate::note::UpgradeNote) and add it
    /// with [`Self::own_output_notes`].
    ///
    /// The request uses a custom script and gives it the new code commitment as the script
    /// argument. This function replaces a previously set custom script and script argument.
    ///
    /// # Errors
    /// - If own output notes are set.
    /// - If an expiration delta is set.
    pub fn build_account_code_upgrade(
        self,
        code: AccountCode,
    ) -> Result<TransactionRequest, TransactionRequestError> {
        let new_code_commitment = code.commitment();

        self.custom_script(account_code_upgrade_script())
            .script_arg(new_code_commitment)
            .account_code_upgrade(code)
            .build()
    }

    // FINALIZE BUILDER
    // --------------------------------------------------------------------------------------------

    /// Consumes the builder and returns a [`TransactionRequest`].
    ///
    /// # Errors
    /// - If both a custom script and own output notes are set.
    /// - If an expiration delta is set when a custom script is set.
    /// - If an invalid note variant is encountered in the own output notes.
    pub fn build(self) -> Result<TransactionRequest, TransactionRequestError> {
        if self.expiration_delta == Some(0) {
            return Err(TransactionRequestError::ZeroExpirationDelta);
        }

        let script_template = match (self.custom_script, self.own_output_notes.is_empty()) {
            (Some(_), false) => {
                return Err(TransactionRequestError::ScriptTemplateError(
                    "Cannot set both a custom script and own output notes".to_string(),
                ));
            },
            (Some(script), true) => {
                if self.expiration_delta.is_some() {
                    return Err(TransactionRequestError::ScriptTemplateError(
                        "Cannot set expiration delta when a custom script is set".to_string(),
                    ));
                }

                Some(TransactionScriptTemplate::CustomScript(script))
            },
            (None, false) => {
                let partial_notes: Vec<PartialNote> =
                    self.own_output_notes.into_iter().map(Into::into).collect();

                Some(TransactionScriptTemplate::SendNotes(partial_notes))
            },
            (None, true) => None,
        };

        let request = TransactionRequest {
            block_numbers: self.block_numbers,
            input_notes: self.input_notes,
            input_notes_args: self.input_notes_args,
            explicit_input_notes: self.explicit_input_notes,
            script_template,
            expected_output_recipients: self.expected_output_recipients,
            expected_future_notes: self.expected_future_notes,
            advice_map: self.advice_map,
            merkle_store: self.merkle_store,
            foreign_accounts: self.foreign_accounts,
            expiration_delta: self.expiration_delta,
            ignore_invalid_input_notes: self.ignore_invalid_input_notes,
            script_arg: self.script_arg,
            auth_arg: self.auth_arg,
            fee_conversion_salt: self.fee_conversion_salt,
            expected_ntx_scripts: self.expected_ntx_scripts,
            account_code_upgrade: self.account_code_upgrade,
        };
        request.validate()?;

        Ok(request)
    }
}

/// Returns `true` when `asset` is a fungible asset carrying no value. Non-fungible assets always
/// carry value, so they are never zero.
fn is_zero_fungible(asset: &Asset) -> bool {
    asset.is_fungible() && asset.unwrap_fungible().amount() == AssetAmount::ZERO
}

// PAYMENT NOTE DESCRIPTION
// ================================================================================================

/// Contains information needed to create a payment note.
#[derive(Clone, Debug)]
pub struct PaymentNoteDescription {
    /// Assets that are meant to be sent to the target account.
    assets: Vec<Asset>,
    /// Account ID of the sender account.
    sender_account_id: AccountId,
    /// Account ID of the receiver account.
    target_account_id: AccountId,
    /// Optional reclaim height for the P2IDE note. It allows the possibility for the sender to
    /// reclaim the assets if the note has not been consumed by the target before this height.
    reclaim_height: Option<BlockNumber>,
    /// Optional timelock height for the P2IDE note. It allows the possibility to add a timelock to
    /// the asset transfer, meaning that the note can only be consumed after this height.
    timelock_height: Option<BlockNumber>,
}

impl PaymentNoteDescription {
    // CONSTRUCTORS
    // --------------------------------------------------------------------------------------------

    /// Creates a new [`PaymentNoteDescription`].
    pub fn new(
        assets: Vec<Asset>,
        sender_account_id: AccountId,
        target_account_id: AccountId,
    ) -> PaymentNoteDescription {
        PaymentNoteDescription {
            assets,
            sender_account_id,
            target_account_id,
            reclaim_height: None,
            timelock_height: None,
        }
    }

    /// Modifies the [`PaymentNoteDescription`] to set a reclaim height for payment note.
    #[must_use]
    pub fn with_reclaim_height(mut self, reclaim_height: BlockNumber) -> PaymentNoteDescription {
        self.reclaim_height = Some(reclaim_height);
        self
    }

    /// Modifies the [`PaymentNoteDescription`] to set a timelock height for payment note.
    #[must_use]
    pub fn with_timelock_height(mut self, timelock_height: BlockNumber) -> PaymentNoteDescription {
        self.timelock_height = Some(timelock_height);
        self
    }

    /// Returns the executor [`AccountId`].
    pub fn account_id(&self) -> AccountId {
        self.sender_account_id
    }

    /// Returns the target [`AccountId`].
    pub fn target_account_id(&self) -> AccountId {
        self.target_account_id
    }

    /// Returns the transaction's list of [`Asset`].
    pub fn assets(&self) -> &Vec<Asset> {
        &self.assets
    }

    /// Returns the reclaim height for the P2IDE note, if set.
    pub fn reclaim_height(&self) -> Option<BlockNumber> {
        self.reclaim_height
    }

    /// Returns the timelock height for the P2IDE note, if set.
    pub fn timelock_height(&self) -> Option<BlockNumber> {
        self.timelock_height
    }

    // CONVERSION
    // --------------------------------------------------------------------------------------------

    /// Converts the payment transaction data into a [`Note`] based on the specified fields. If the
    /// reclaim and timelock heights are not set, a P2ID note is created; otherwise, a P2IDE note is
    /// created.
    pub(crate) fn into_note(
        self,
        note_type: NoteType,
        rng: &mut ClientRng,
    ) -> Result<Note, NoteError> {
        if self.reclaim_height.is_none() && self.timelock_height.is_none() {
            // Create a P2ID note
            Ok(P2idNote::builder()
                .sender(self.sender_account_id)
                .target(self.target_account_id)
                .assets(self.assets)
                .note_type(note_type)
                .generate_serial_number(rng)
                .build()?
                .into())
        } else {
            // Create a P2IDE note
            Ok(P2ideNote::builder()
                .sender(self.sender_account_id)
                .target(self.target_account_id)
                .assets(self.assets)
                .note_type(note_type)
                .maybe_reclaim_height(self.reclaim_height)
                .maybe_timelock_height(self.timelock_height)
                .generate_serial_number(rng)
                .build()?
                .into())
        }
    }
}

// SWAP TRANSACTION DATA
// ================================================================================================

/// Contains information related to a swap transaction.
///
/// A swap transaction involves creating a SWAP note, which will carry the offered asset and which,
/// when consumed, will create a payback note that carries the requested asset taken from the
/// consumer account's vault.
#[derive(Clone, Debug)]
pub struct SwapTransactionData {
    /// Account ID of the sender account.
    sender_account_id: AccountId,
    /// Asset that is offered in the swap.
    offered_asset: Asset,
    /// Asset that is expected in the payback note generated as a result of the swap.
    requested_asset: Asset,
}

impl SwapTransactionData {
    // CONSTRUCTORS
    // --------------------------------------------------------------------------------------------

    /// Creates a new [`SwapTransactionData`].
    pub fn new(
        sender_account_id: AccountId,
        offered_asset: Asset,
        requested_asset: Asset,
    ) -> SwapTransactionData {
        SwapTransactionData {
            sender_account_id,
            offered_asset,
            requested_asset,
        }
    }

    /// Returns the executor [`AccountId`].
    pub fn account_id(&self) -> AccountId {
        self.sender_account_id
    }

    /// Returns the transaction offered [`Asset`].
    pub fn offered_asset(&self) -> Asset {
        self.offered_asset
    }

    /// Returns the transaction requested [`Asset`].
    pub fn requested_asset(&self) -> Asset {
        self.requested_asset
    }
}

// PSWAP TRANSACTION DATA
// ================================================================================================

/// Contains information related to a partial swap (PSWAP) transaction.
///
/// A PSWAP transaction involves creating a PSWAP note that carries the offered fungible asset and,
/// when consumed (filled), produces a payback note carrying the requested fungible asset taken from
/// the filler's vault. Both legs are restricted to fungible assets so that fills can be denominated
/// in arbitrary amounts.
#[derive(Clone, Debug)]
pub struct PswapTransactionData {
    /// Account ID of the creator account.
    creator_account_id: AccountId,
    /// Fungible asset offered in the swap.
    offered_asset: FungibleAsset,
    /// Fungible asset expected in the payback note generated when the PSWAP is filled.
    requested_asset: FungibleAsset,
}

impl PswapTransactionData {
    // CONSTRUCTORS
    // --------------------------------------------------------------------------------------------

    /// Creates a new [`PswapTransactionData`].
    pub fn new(
        creator_account_id: AccountId,
        offered_asset: FungibleAsset,
        requested_asset: FungibleAsset,
    ) -> PswapTransactionData {
        PswapTransactionData {
            creator_account_id,
            offered_asset,
            requested_asset,
        }
    }

    /// Returns the creator [`AccountId`].
    pub fn creator_account_id(&self) -> AccountId {
        self.creator_account_id
    }

    /// Returns the offered [`FungibleAsset`].
    pub fn offered_asset(&self) -> FungibleAsset {
        self.offered_asset
    }

    /// Returns the requested [`FungibleAsset`].
    pub fn requested_asset(&self) -> FungibleAsset {
        self.requested_asset
    }
}