From 9f38c810978c6ce615e40c942cb400fda3aa1bb0 Mon Sep 17 00:00:00 2001
From: Liu Chuankai <liuchuankai@cryptape.com>
Date: Wed, 10 May 2023 22:55:15 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=F0=9F=8E=B8=20support=20dao=20withdraw?=
 =?UTF-8?q?=20phrase1=20simple=20transaction=20builder?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 examples/dao_withdraw_phrase1_example.rs | 61 ++++++++++++++++++++++++
 src/transaction/handler/dao.rs           | 23 +++++++--
 2 files changed, 80 insertions(+), 4 deletions(-)
 create mode 100644 examples/dao_withdraw_phrase1_example.rs

diff --git a/examples/dao_withdraw_phrase1_example.rs b/examples/dao_withdraw_phrase1_example.rs
new file mode 100644
index 00000000..9de12a70
--- /dev/null
+++ b/examples/dao_withdraw_phrase1_example.rs
@@ -0,0 +1,61 @@
+use ckb_sdk::{
+    transaction::{
+        builder::{CkbTransactionBuilder, SimpleTransactionBuilder},
+        handler::{dao, HandlerContexts},
+        input::InputIterator,
+        signer::{SignContexts, TransactionSigner},
+        TransactionBuilderConfiguration,
+    },
+    Address, CkbRpcClient, NetworkInfo,
+};
+use ckb_types::h256;
+use std::{error::Error as StdErr, str::FromStr};
+
+fn main() -> Result<(), Box<dyn StdErr>> {
+    let network_info = NetworkInfo::testnet();
+    let sender = Address::from_str("ckt1qzda0cr08m85hc8jlnfp3zer7xulejywt49kt2rr0vthywaa50xwsq2qf8keemy2p5uu0g0gn8cd4ju23s5269qk8rg4r")?;
+
+    let configuration = TransactionBuilderConfiguration::new_with_network(network_info.clone())?;
+
+    let iterator = InputIterator::new(vec![(&sender).into()], configuration.network_info());
+    let mut builder = SimpleTransactionBuilder::new(configuration, iterator);
+
+    let mut context = dao::WithdrawPhrase1Context::new(network_info.url.clone());
+    let input_outpoint = serde_json::from_str::<ckb_jsonrpc_types::OutPoint>(
+        r#"
+   {
+      "tx_hash": "0x2aba579894cdf5f6c4afd3ada52792c4405fe6ba64d05226fb63fa5c1ec6f666",
+      "index": "0x0"
+   }
+   "#,
+    )
+    .unwrap();
+    context.add_input_outpoint(input_outpoint.into(), None);
+    let mut contexts = HandlerContexts::default();
+    contexts.add_context(Box::new(context) as Box<_>);
+
+    builder.set_change_lock((&sender).into());
+    let mut tx_with_groups = builder.build(&contexts)?;
+
+    let json_tx = ckb_jsonrpc_types::TransactionView::from(tx_with_groups.get_tx_view().clone());
+    println!("tx: {}", serde_json::to_string_pretty(&json_tx).unwrap());
+
+    let private_keys = vec![h256!(
+        "0x6c9ed03816e3111e49384b8d180174ad08e29feb1393ea1b51cef1c505d4e36a"
+    )];
+    TransactionSigner::new(&network_info).sign_transaction(
+        &mut tx_with_groups,
+        &SignContexts::new_sighash_h256(private_keys)?,
+    )?;
+
+    let json_tx = ckb_jsonrpc_types::TransactionView::from(tx_with_groups.get_tx_view().clone());
+    println!("tx: {}", serde_json::to_string_pretty(&json_tx).unwrap());
+
+    let tx_hash = CkbRpcClient::new(network_info.url.as_str())
+        .send_transaction(json_tx.inner, None)
+        .expect("send transaction");
+    // example tx: b615b9cbb566af18dd2d860836b89e07a86dfcc7af510595dcb404f1b19e6d7e
+    println!(">>> tx {} sent! <<<", tx_hash);
+
+    Ok(())
+}
diff --git a/src/transaction/handler/dao.rs b/src/transaction/handler/dao.rs
index 18c637a5..1eddf418 100644
--- a/src/transaction/handler/dao.rs
+++ b/src/transaction/handler/dao.rs
@@ -62,10 +62,22 @@ pub struct WithdrawPhrase1Context {
 
 impl WithdrawPhrase1Context {
     /// add input.
-    /// If `lock_script` is `None` copy the lock script from input with same
+    /// If `receiver_lock` is `None` copy the lock script from input with same
     /// index, otherwise replace the lock script with the given script.
-    pub fn add_input(&mut self, input: CellInput, lock_script: Option<Script>) {
-        self.items.push(DaoPrepareItem { input, lock_script });
+    pub fn add_input(&mut self, input: CellInput, receiver_lock: Option<Script>) {
+        self.items.push(DaoPrepareItem {
+            input,
+            lock_script: receiver_lock,
+        });
+    }
+
+    pub fn add_input_outpoint(&mut self, input_outpoint: OutPoint, receiver_lock: Option<Script>) {
+        self.items.push(DaoPrepareItem {
+            input: CellInput::new_builder()
+                .previous_output(input_outpoint)
+                .build(),
+            lock_script: receiver_lock,
+        });
     }
 
     pub fn new(rpc_url: String) -> Self {
@@ -136,7 +148,7 @@ impl DaoScriptHandler {
             transaction_inputs.push(transaction_input);
 
             tx_data.dedup_header_dep(deposit_header.hash());
-            tx_data.input(input.clone());
+
             tx_data.output(output);
             tx_data.output_data(output_data.pack());
         }
@@ -208,6 +220,9 @@ impl ScriptHandler for DaoScriptHandler {
                 tx_data.set_witness(*index, witness.as_bytes().pack());
             }
             Ok(true)
+        } else if let Some(_args) = context.as_any().downcast_ref::<WithdrawPhrase1Context>() {
+            tx_data.dedup_cell_deps(self.cell_deps.clone());
+            Ok(true)
         } else {
             Ok(false)
         }