From 804aab288dba9dc553f7ea0e20917bf9b2b9ecd5 Mon Sep 17 00:00:00 2001
From: 0o-de-lally <1364012+0o-de-lally@users.noreply.github.com>
Date: Tue, 19 Dec 2023 17:03:59 +0000
Subject: [PATCH] can get registration script

---
 tools/rescue/src/twin.rs | 15 +++++++++++----
 1 file changed, 11 insertions(+), 4 deletions(-)

diff --git a/tools/rescue/src/twin.rs b/tools/rescue/src/twin.rs
index d322d71ad..f97cb7c82 100644
--- a/tools/rescue/src/twin.rs
+++ b/tools/rescue/src/twin.rs
@@ -1,7 +1,9 @@
 use std::{path::PathBuf, time::Duration};
 
+use anyhow::bail;
 use clap::Parser;
-use libra_cached_packages::libra_framework_sdk_builder::EntryFunctionCall;
+use diem_types::transaction::Script;
+
 use libra_smoke_tests::libra_smoke::LibraSmoke;
 use libra_txs::txs_cli_vals::ValidatorTxs;
 
@@ -40,11 +42,16 @@ impl TwinOpts {
 
     /// create the validator registration entry function payload
     /// needs the file operator.yaml
-    fn register_marlon_tx(file: PathBuf) -> anyhow::Result<EntryFunctionCall> {
+    fn register_marlon_tx(file: PathBuf) -> anyhow::Result<Script> {
         let tx = ValidatorTxs::Register {
             operator_file: Some(file),
-        };
-        tx.make_payload()
+        }
+        .make_payload()?
+        .encode();
+        if let diem_types::transaction::TransactionPayload::Script(s) = tx {
+            return Ok(s);
+        }
+        bail!("function did not return a script")
     }
 
     /// create the rescue blob which has one validator