feat(transaction-builder): Use PTBArgument trait for all builder calls

bindings/kotlin/examples/PrepareMergeCoins.kt

@@ -14,11 +14,11 @@ fun main() = runBlocking {
                 )
 
         val coin0 =
-                ObjectId.fromHex(
+                PtbArgument.objectIdFromHex(
                         "0x0b0270ee9d27da0db09651e5f7338dfa32c7ee6441ccefa1f6e305735bcfc7ab"
                 )
         val coin1 =
-                ObjectId.fromHex(
+                PtbArgument.objectIdFromHex(
                         "0xd04077fe3b6fad13b3d4ed0d535b7ca92afcac8f0f2a0e0925fb9f4f0b30c699"
                 )
 

bindings/python/examples/prepare_merge_coins.py

@@ -14,10 +14,10 @@ async def main():
             "0x611830d3641a68f94a690dcc25d1f4b0dac948325ac18f6dd32564371735f32c"
         )
 
-        coin_0 = ObjectId.from_hex(
+        coin_0 = PtbArgument.object_id_from_hex(
             "0x0b0270ee9d27da0db09651e5f7338dfa32c7ee6441ccefa1f6e305735bcfc7ab"
         )
-        coin_1 = ObjectId.from_hex(
+        coin_1 = PtbArgument.object_id_from_hex(
             "0xd04077fe3b6fad13b3d4ed0d535b7ca92afcac8f0f2a0e0925fb9f4f0b30c699"
         )
 

crates/iota-transaction-builder/src/builder/mod.rs

@@ -4,7 +4,7 @@
 //! Builder for Programmable Transactions.
 
 use std::{
-    collections::{BTreeMap, HashMap},
+    collections::{BTreeMap, HashMap, HashSet},
     marker::PhantomData,
     time::Duration,
 };
@@ -581,7 +581,25 @@ impl<L> TransactionBuilder<Client, L> {
         let mut inputs = Vec::new();
         let mut gas = Vec::new();
         let mut input_map = HashMap::new();
+
         if default_gas && !self.data.inputs.values().any(|i| i.is_gas) {
+            let mut unusable_object_ids = HashSet::new();
+            for cmd in &self.data.commands {
+                for arg in match cmd {
+                    Command::MoveCall(MoveCall { arguments, .. }) => arguments.as_slice(),
+                    Command::TransferObjects(TransferObjects { objects, .. }) => objects.as_slice(),
+                    Command::MergeCoins(MergeCoins { coins_to_merge, .. }) => {
+                        coins_to_merge.as_slice()
+                    }
+                    _ => &[],
+                } {
+                    if let Argument::Input(idx) = arg {
+                        if let Some(obj_id) = self.data.inputs[idx].object_id() {
+                            unusable_object_ids.insert(*obj_id);
+                        }
+                    }
+                }
+            }
             for coin in self
                 .client
                 .objects(
@@ -596,10 +614,12 @@ impl<L> TransactionBuilder<Client, L> {
                 .map_err(Error::Client)?
                 .data
             {
-                self.set_input(
-                    InputKind::Input(iota_types::Input::ImmutableOrOwned(coin.object_ref())),
-                    true,
-                );
+                if !unusable_object_ids.contains(&coin.object_id()) {
+                    self.set_input(
+                        InputKind::Input(iota_types::Input::ImmutableOrOwned(coin.object_ref())),
+                        true,
+                    );
+                }
             }
         }
         for (id, input) in std::mem::take(&mut self.data.inputs) {
@@ -717,6 +737,7 @@ impl<L> TransactionBuilder<Client, L> {
     pub async fn finish(mut self) -> Result<Transaction, Error> {
         let mut txn = self.resolve_ptb(true).await?;
         if self.data.gas_budget.is_none() {
+            println!("{txn:#?}");
             let res = self
                 .client
                 .dry_run_tx_kind(&txn.kind, true, Default::default())