web3-cli/src/web3cli/controllers/replay_controller.py at master · Dreamworks-dev/web3-cli · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
from cement import ex

from web3cli.exceptions import Web3CliError
from web3cli.framework.controller import Controller
from web3cli.helpers import args
from web3cli.helpers.client_factory import make_client, make_wallet
from web3cli.helpers.render import render, render_web3py
from web3core.helpers.misc import yes_or_exit


class ReplayController(Controller):
    """Handler of the `w3 replay` commands"""

    class Meta:
        label = "replay"
        help = "replay transactions on the blockchain"
        stacked_type = "embedded"
        stacked_on = "base"

    @ex(
        help="Replay the given transaction on the blockchain",
        arguments=[
            (["hash"], {"help": "hash of the transaction"}),
            *args.chain_and_rpc(),
            args.signer(),
            (
                ["--fee-multiplier"],
                {
                    "help": "multiply gas price by this factor with respect to the original tx; set to zero to estimate it based on current chain conditions",
                    "default": 0,
                    "type": float,
                },
            ),
            (
                ["--gas-multiplier"],
                {
                    "help": "multiply the gas limit by this factor with respect to the gas spent by the original tx",
                    "default": 1.2,
                    "type": float,
                },
            ),
            (
                ["--type"],
                {"help": "Override the type of the original tx", "type": int},
            ),
            (
                ["--data", "--input"],
                {"help": "Override the input data of the original tx"},
            ),
            args.force(),
        ],
    )
    def replay(self) -> None:
        original_tx = make_client(self.app).get_tx(self.app.pargs.hash)
        # Build replay transaction
        signer = make_wallet(self.app)
        tx = signer.build_base_tx()
        # Make sure you call the same contract
        tx["to"] = original_tx["to"]
        # Make sure you call the same method with same args
        if original_tx.get("input"):
            tx["data"] = original_tx["input"]
        # Make sure you pay the same amount
        tx["value"] = original_tx["value"]
        # Make sure you send the same type of transaction
        if original_tx.get("type"):
            tx["type"] = original_tx["type"]
        # Make sure you use the right amount of gas
        tx["gas"] = int(int(original_tx["gas"]) * self.app.pargs.gas_multiplier)
        # For post EIP-1559 transactions, use the same tip and max gas
        if self.app.pargs.fee_multiplier != 0 and original_tx.get("maxFeePerGas"):
            tx["maxFeePerGas"] = int(
                original_tx["maxFeePerGas"] * self.app.pargs.fee_multiplier
            )
        if self.app.pargs.fee_multiplier != 0 and original_tx.get(
            "maxPriorityFeePerGas"
        ):
            tx["maxPriorityFeePerGas"] = int(
                original_tx["maxPriorityFeePerGas"] * self.app.pargs.fee_multiplier
            )
        # For pre EIP-1559 transactions, use the same gas price
        if (
            self.app.pargs.fee_multiplier != 0
            and original_tx.get("gasPrice")
            and not original_tx.get("maxFeePerGas")
            and not original_tx.get("maxPriorityFeePerGas")
        ):
            tx["gasPrice"] = int(
                original_tx["gasPrice"] * self.app.pargs.fee_multiplier
            )
        # Optionally override type field
        if self.app.pargs.type is not None:
            tx["type"] = self.app.pargs.type
        # Optionally override data field
        if self.app.pargs.data is not None:
            tx["data"] = self.app.pargs.data
        # Ask for confirmation
        if not self.app.pargs.force:
            self.app.log.info(f"Replaying the following transaction:")
            render_web3py(self.app, original_tx)
            print("\n")
            self.app.log.info(f"Transaction that will be sent:")
            render_web3py(self.app, tx)
            print("\n")
            yes_or_exit("Continue? ")
        # Send transaction
        try:
            tx_hash = signer.sign_and_send_tx(tx)
        except TypeError as e:
            raise Web3CliError(
                f"Failed to send transaction: {e}. "
                f"Try using the flag --type {self.app.chain.tx_type} to prevent type-related errors."
            )
        render(self.app, tx_hash)