ISA: mana remainder

cli/src/cli.rs

@@ -64,7 +64,7 @@ pub struct InitParameters {
     #[arg(short, long, value_name = "URL", env = "NODE_URL", default_value = DEFAULT_NODE_URL)]
     pub node_url: String,
     /// Set the BIP path, `4219/0/0/0` if not provided.
-    #[arg(short, long, value_parser = parse_bip_path)]
+    #[arg(short, long, value_parser = parse_bip_path, default_value = "4219/0/0/0")]
     pub bip_path: Option<Bip44>,
     /// Set the Bech32-encoded wallet address.
     #[arg(short, long)]

sdk/src/client/api/block_builder/input_selection/mod.rs

@@ -155,6 +155,7 @@ impl InputSelection {
         available_inputs: impl Into<Vec<InputSigningData>>,
         outputs: impl Into<Vec<Output>>,
         addresses: impl IntoIterator<Item = Address>,
+        slot_index: impl Into<SlotIndex>,
         protocol_parameters: ProtocolParameters,
     ) -> Self {
         let available_inputs = available_inputs.into();
@@ -189,9 +190,8 @@ impl InputSelection {
             burn: None,
             remainder_address: None,
             protocol_parameters,
-            // TODO may want to make this mandatory at some point
             // Should be set from a commitment context input
-            slot_index: SlotIndex::from(0),
+            slot_index: slot_index.into(),
             requirements: Vec::new(),
             automatically_transitioned: HashSet::new(),
             mana_allotments: 0,
@@ -222,12 +222,6 @@ impl InputSelection {
         self
     }
 
-    /// Sets the slot index of an [`InputSelection`].
-    pub fn with_slot_index(mut self, slot_index: impl Into<SlotIndex>) -> Self {
-        self.slot_index = slot_index.into();
-        self
-    }
-
     /// Sets the mana allotments sum of an [`InputSelection`].
     pub fn with_mana_allotments<'a>(mut self, mana_allotments: impl Iterator<Item = &'a ManaAllotment>) -> Self {
         self.mana_allotments = mana_allotments.map(ManaAllotment::mana).sum();

sdk/src/client/api/block_builder/input_selection/remainder.rs

@@ -14,7 +14,10 @@ use crate::{
     client::api::RemainderData,
     types::block::{
         address::{Address, Ed25519Address},
-        output::{unlock_condition::AddressUnlockCondition, BasicOutputBuilder, NativeTokensBuilder, Output},
+        output::{
+            unlock_condition::AddressUnlockCondition, AccountOutputBuilder, BasicOutputBuilder, NativeTokensBuilder,
+            NftOutputBuilder, Output,
+        },
     },
 };
 
@@ -73,17 +76,18 @@ impl InputSelection {
                 ))));
 
         // TODO https://github.com/iotaledger/iota-sdk/issues/1631
-        // if let Some(native_tokens) = native_tokens_diff {
-        //     remainder_builder = remainder_builder.with_native_tokens(native_tokens);
-        // }
+        // TODO Only putting one in remainder atm so we can at least create foundries
+        if let Some(native_tokens) = native_tokens_diff {
+            remainder_builder = remainder_builder.with_native_token(*native_tokens.first().unwrap());
+        }
 
         Ok((remainder_builder.finish_output()?.amount(), native_tokens_remainder))
     }
 
     pub(crate) fn remainder_and_storage_deposit_return_outputs(
-        &self,
+        &mut self,
     ) -> Result<(Option<RemainderData>, Vec<Output>), Error> {
-        let (inputs_sum, outputs_sum, inputs_sdr, outputs_sdr) =
+        let (input_amount, output_amount, inputs_sdr, outputs_sdr) =
             amount_sums(&self.selected_inputs, &self.outputs, self.slot_index);
         let mut storage_deposit_returns = Vec::new();
 
@@ -118,30 +122,77 @@ impl InputSelection {
 
         let native_tokens_diff = get_native_tokens_diff(&input_native_tokens, &output_native_tokens)?;
 
-        if inputs_sum == outputs_sum && native_tokens_diff.is_none() {
+        let mut input_mana = 0;
+
+        for input in &self.selected_inputs {
+            input_mana += input.output.all_mana(
+                &self.protocol_parameters,
+                input.output_id().transaction_id().slot_index(),
+                self.slot_index,
+            )?;
+            // TODO rewards https://github.com/iotaledger/iota-sdk/issues/1310
+        }
+
+        let output_mana = self.outputs.iter().map(|o| o.mana()).sum::<u64>() + self.mana_allotments;
+
+        if input_amount == output_amount && input_mana == output_mana && native_tokens_diff.is_none() {
             log::debug!("No remainder required");
             return Ok((None, storage_deposit_returns));
         }
 
+        let amount_diff = input_amount - output_amount;
+        let mana_diff = input_mana - output_mana;
+
+        // If there is only a mana remainder, try to fit it in an automatically transitioned output.
+        if input_amount == output_amount && input_mana != output_mana && native_tokens_diff.is_none() {
+            let output = self
+                .outputs
+                .iter_mut()
+                .filter(|output| {
+                    output
+                        .chain_id()
+                        .as_ref()
+                        .map(|chain_id| self.automatically_transitioned.contains(chain_id))
+                        .unwrap_or(false)
+                        // Foundries can't hold mana so they are not considered here.
+                        && !output.is_foundry()
+                })
+                .next();
+
+            if let Some(output) = output {
+                *output = match output {
+                    Output::Account(output) => AccountOutputBuilder::from(&*output)
+                        .with_mana(output.mana() + mana_diff)
+                        .finish_output()?,
+                    Output::Nft(output) => NftOutputBuilder::from(&*output)
+                        .with_mana(output.mana() + mana_diff)
+                        .finish_output()?,
+                    _ => panic!("only account, nft can be automatically created and can hold mana"),
+                };
+
+                return Ok((None, storage_deposit_returns));
+            }
+        }
+
         let Some((remainder_address, chain)) = self.get_remainder_address()? else {
             return Err(Error::MissingInputWithEd25519Address);
         };
 
-        let diff = inputs_sum - outputs_sum;
-        let mut remainder_builder = BasicOutputBuilder::new_with_amount(diff);
+        let mut remainder_builder = BasicOutputBuilder::new_with_amount(amount_diff).with_mana(mana_diff);
 
         remainder_builder =
             remainder_builder.add_unlock_condition(AddressUnlockCondition::new(remainder_address.clone()));
 
         // TODO https://github.com/iotaledger/iota-sdk/issues/1631
-        // if let Some(native_tokens) = native_tokens_diff {
-        //     log::debug!("Adding {native_tokens:?} to remainder output for {remainder_address:?}");
-        //     remainder_builder = remainder_builder.with_native_tokens(native_tokens);
-        // }
+        // TODO Only putting one in remainder atm so we can at least create foundries
+        if let Some(native_tokens) = native_tokens_diff {
+            log::debug!("Adding {native_tokens:?} to remainder output for {remainder_address:?}");
+            remainder_builder = remainder_builder.with_native_token(*native_tokens.first().unwrap());
+        }
 
         let remainder = remainder_builder.finish_output()?;
 
-        log::debug!("Created remainder output of {diff} for {remainder_address:?}");
+        log::debug!("Created remainder output of amount {amount_diff} and mana {mana_diff} for {remainder_address:?}");
 
         remainder.verify_storage_deposit(self.protocol_parameters.storage_score_parameters())?;
 

sdk/src/client/api/high_level.rs

@@ -111,23 +111,23 @@ impl Client {
 
     // Returns the slot index corresponding to the current timestamp.
     pub async fn get_slot_index(&self) -> Result<SlotIndex> {
-        let current_time = unix_timestamp_now().as_nanos() as u64;
+        let unix_timestamp = unix_timestamp_now();
+        let current_time_nanos = unix_timestamp.as_nanos() as u64;
 
         let network_info = self.get_network_info().await?;
 
         if let Some(tangle_time) = network_info.tangle_time {
             // Check the local time is in the range of +-5 minutes of the node to prevent locking funds by accident
             if !(tangle_time - FIVE_MINUTES_IN_NANOSECONDS..tangle_time + FIVE_MINUTES_IN_NANOSECONDS)
-                .contains(&current_time)
+                .contains(&current_time_nanos)
             {
                 return Err(Error::TimeNotSynced {
-                    current_time,
+                    current_time: current_time_nanos,
                     tangle_time,
                 });
             }
         }
 
-        // TODO double check with TIP if this should be seconds or nanoseconds
-        Ok(network_info.protocol_parameters.slot_index(current_time))
+        Ok(network_info.protocol_parameters.slot_index(unix_timestamp.as_secs()))
     }
 }

sdk/src/types/block/output/mod.rs

@@ -217,6 +217,33 @@ impl Output {
         }
     }
 
+    /// Returns all the mana held by the output, which is potential + stored, all decayed.
+    pub fn all_mana(
+        &self,
+        protocol_parameters: &ProtocolParameters,
+        creation_index: SlotIndex,
+        target_index: SlotIndex,
+    ) -> Result<u64, Error> {
+        let (amount, mana) = match self {
+            Self::Basic(output) => (output.amount(), output.mana()),
+            Self::Account(output) => (output.amount(), output.mana()),
+            Self::Anchor(output) => (output.amount(), output.mana()),
+            Self::Foundry(output) => (output.amount(), 0),
+            Self::Nft(output) => (output.amount(), output.mana()),
+            Self::Delegation(output) => (output.amount(), 0),
+        };
+
+        let min_deposit = self.minimum_amount(protocol_parameters.storage_score_parameters());
+        let generation_amount = amount.saturating_sub(min_deposit);
+        let potential_mana =
+            protocol_parameters.generate_mana_with_decay(generation_amount, creation_index, target_index)?;
+        let stored_mana = protocol_parameters.mana_with_decay(mana, creation_index, target_index)?;
+
+        Ok(potential_mana
+            .checked_add(stored_mana)
+            .ok_or(Error::ConsumedManaOverflow)?)
+    }
+
     /// Returns the unlock conditions of an [`Output`], if any.
     pub fn unlock_conditions(&self) -> Option<&UnlockConditions> {
         match self {

sdk/src/types/block/payload/signed_transaction/transaction.rs

@@ -61,8 +61,8 @@ impl TransactionBuilder {
     }
 
     /// Sets the context inputs of a [`TransactionBuilder`].
-    pub fn with_context_inputs(mut self, context_inputs: impl Into<Vec<ContextInput>>) -> Self {
-        self.context_inputs = context_inputs.into();
+    pub fn with_context_inputs(mut self, context_inputs: impl IntoIterator<Item = ContextInput>) -> Self {
+        self.context_inputs = context_inputs.into_iter().collect();
         self
     }
 
@@ -164,6 +164,8 @@ impl TransactionBuilder {
             .try_into()
             .map_err(Error::InvalidContextInputCount)?;
 
+        verify_context_inputs(&context_inputs)?;
+
         let inputs: BoxedSlicePrefix<Input, InputCount> = self
             .inputs
             .into_boxed_slice()

sdk/src/types/block/rand/output/mod.rs

@@ -35,13 +35,22 @@ use crate::types::block::{
                 rand_state_controller_address_unlock_condition_different_from,
             },
         },
-        transaction::rand_transaction_id,
+        transaction::rand_transaction_id_with_slot_index,
     },
+    slot::SlotIndex,
 };
 
+/// Generates a random output id with a given slot index.
+pub fn rand_output_id_with_slot_index(slot_index: impl Into<SlotIndex>) -> OutputId {
+    OutputId::new(
+        rand_transaction_id_with_slot_index(slot_index),
+        rand_number_range(OUTPUT_INDEX_RANGE),
+    )
+}
+
 /// Generates a random [`OutputId`].
 pub fn rand_output_id() -> OutputId {
-    OutputId::new(rand_transaction_id(), rand_number_range(OUTPUT_INDEX_RANGE))
+    rand_output_id_with_slot_index(rand_number::<u32>())
 }
 
 /// Generates a random [`BasicOutput`].

sdk/src/types/block/semantic/mod.rs

@@ -16,9 +16,7 @@ pub use self::{
 };
 use crate::types::block::{
     address::Address,
-    output::{
-        AccountId, AnchorOutput, ChainId, FoundryId, MinimumOutputAmount, NativeTokens, Output, OutputId, TokenId,
-    },
+    output::{AccountId, AnchorOutput, ChainId, FoundryId, NativeTokens, Output, OutputId, TokenId},
     payload::signed_transaction::{Transaction, TransactionCapabilityFlag, TransactionSigningHash},
     protocol::ProtocolParameters,
     unlock::Unlock,
@@ -162,34 +160,13 @@ impl<'a> SemanticValidationContext<'a> {
                 .checked_add(amount)
                 .ok_or(Error::ConsumedAmountOverflow)?;
 
-            let potential_mana = {
-                // Deposit amount doesn't generate mana
-                let min_deposit = consumed_output.minimum_amount(self.protocol_parameters.storage_score_parameters());
-                let generation_amount = consumed_output.amount().saturating_sub(min_deposit);
-
-                self.protocol_parameters.generate_mana_with_decay(
-                    generation_amount,
-                    output_id.transaction_id().slot_index(),
-                    self.transaction.creation_slot(),
-                )
-            }?;
-
-            // Add potential mana
-            self.input_mana = self
-                .input_mana
-                .checked_add(potential_mana)
-                .ok_or(Error::ConsumedManaOverflow)?;
-
-            let stored_mana = self.protocol_parameters.mana_with_decay(
-                mana,
-                output_id.transaction_id().slot_index(),
-                self.transaction.creation_slot(),
-            )?;
-
-            // Add stored mana
             self.input_mana = self
                 .input_mana
-                .checked_add(stored_mana)
+                .checked_add(consumed_output.all_mana(
+                    &self.protocol_parameters,
+                    output_id.transaction_id().slot_index(),
+                    self.transaction.creation_slot(),
+                )?)
                 .ok_or(Error::ConsumedManaOverflow)?;
 
             // TODO: Add reward mana https://github.com/iotaledger/iota-sdk/issues/1310

sdk/src/wallet/operations/transaction/account.rs

@@ -51,14 +51,15 @@ where
     where
         crate::wallet::Error: From<S::Error>,
     {
-        let implicit_account_data = self.data().await.unspent_outputs.get(output_id).cloned();
-
-        let implicit_account = if let Some(implicit_account_data) = &implicit_account_data {
-            if implicit_account_data.output.is_implicit_account() {
-                implicit_account_data.output.as_basic()
-            } else {
-                return Err(Error::ImplicitAccountNotFound);
-            }
+        let implicit_account_data = self
+            .data()
+            .await
+            .unspent_outputs
+            .get(output_id)
+            .cloned()
+            .ok_or(Error::ImplicitAccountNotFound)?;
+        let implicit_account = if implicit_account_data.output.is_implicit_account() {
+            implicit_account_data.output.as_basic()
         } else {
             return Err(Error::ImplicitAccountNotFound);
         };
@@ -86,7 +87,11 @@ where
 
         let account_id = AccountId::from(output_id);
         let account = AccountOutput::build_with_amount(implicit_account.amount(), account_id)
-            .with_mana(implicit_account.mana())
+            .with_mana(implicit_account_data.output.all_mana(
+                &self.client().get_protocol_parameters().await?,
+                implicit_account_data.output_id.transaction_id().slot_index(),
+                self.client().get_slot_index().await?,
+            )?)
             .with_unlock_conditions([AddressUnlockCondition::from(Address::from(
                 *implicit_account
                     .address()