remove governance wrapper

Author: popenta

multiversx_sdk_cli/cli_governance.py

@@ -7,7 +7,8 @@
     GovernanceController,
     ProposalInfo,
     ProxyNetworkProvider,
-    TransactionsFactoryConfig,
+    Transaction,
+    VoteType,
 )
 
 from multiversx_sdk_cli import cli_shared, utils
@@ -18,7 +19,10 @@
 )
 from multiversx_sdk_cli.cli_output import CLIOutputBuilder
 from multiversx_sdk_cli.config import get_config_for_network_providers
-from multiversx_sdk_cli.governance import GovernanceWrapper
+from multiversx_sdk_cli.config_env import get_address_hrp
+from multiversx_sdk_cli.guardian_relayer_data import GuardianRelayerData
+from multiversx_sdk_cli.interfaces import IAccount
+from multiversx_sdk_cli.signing_wrapper import SigningWrapper
 
 
 def setup_parser(args: list[str], subparsers: Any) -> Any:
@@ -196,6 +200,31 @@ def _ensure_args(args: Any):
     validate_chain_id_args(args)
 
 
+def _initialize_controller(args: Any) -> GovernanceController:
+    chain = args.chain if hasattr(args, "chain") else None
+    chain_id = cli_shared.get_chain_id(args.proxy, chain)
+    config = get_config_for_network_providers()
+    proxy_url = args.proxy if args.proxy else ""
+    proxy = ProxyNetworkProvider(url=proxy_url, config=config)
+    gas_estimator = cli_shared.initialize_gas_limit_estimator(args)
+
+    return GovernanceController(
+        chain_id=chain_id,
+        network_provider=proxy,
+        address_hrp=get_address_hrp(),
+        gas_limit_estimator=gas_estimator,
+    )
+
+
+def _sign_transaction(transaction: Transaction, sender: IAccount, guardian_and_relayer_data: GuardianRelayerData):
+    signer = SigningWrapper()
+    signer.sign_transaction(
+        transaction=transaction,
+        sender=sender,
+        guardian_and_relayer=guardian_and_relayer_data,
+    )
+
+
 def create_proposal(args: Any):
     _ensure_args(args)
 
@@ -204,24 +233,22 @@ def create_proposal(args: Any):
         sender=sender.address.to_bech32(),
         args=args,
     )
-    chain_id = cli_shared.get_chain_id(args.proxy, args.chain)
-    gas_estimator = cli_shared.initialize_gas_limit_estimator(args)
-    controller = GovernanceWrapper(config=TransactionsFactoryConfig(chain_id), gas_limit_estimator=gas_estimator)
 
+    controller = _initialize_controller(args)
     transaction = controller.create_transaction_for_new_proposal(
         sender=sender,
         nonce=sender.nonce,
         commit_hash=args.commit_hash,
         start_vote_epoch=args.start_vote_epoch,
         end_vote_epoch=args.end_vote_epoch,
         native_token_amount=args.value,
+        guardian=guardian_and_relayer_data.guardian.address if guardian_and_relayer_data.guardian else None,
+        relayer=guardian_and_relayer_data.relayer.address if guardian_and_relayer_data.relayer else None,
         gas_limit=args.gas_limit,
         gas_price=args.gas_price,
-        version=args.version,
-        options=args.options,
-        guardian_and_relayer_data=guardian_and_relayer_data,
     )
 
+    _sign_transaction(transaction, sender, guardian_and_relayer_data)
     cli_shared.send_or_simulate(transaction, args)
 
 
@@ -233,22 +260,22 @@ def vote(args: Any):
         sender=sender.address.to_bech32(),
         args=args,
     )
-    chain_id = cli_shared.get_chain_id(args.proxy, args.chain)
-    gas_estimator = cli_shared.initialize_gas_limit_estimator(args)
-    controller = GovernanceWrapper(config=TransactionsFactoryConfig(chain_id), gas_limit_estimator=gas_estimator)
+
+    [vote_value] = [v for v in VoteType if v.value == args.vote]
+    controller = _initialize_controller(args)
 
     transaction = controller.create_transaction_for_voting(
         sender=sender,
         nonce=sender.nonce,
         proposal_nonce=args.proposal_nonce,
-        vote=args.vote,
+        vote=vote_value,
+        guardian=guardian_and_relayer_data.guardian.address if guardian_and_relayer_data.guardian else None,
+        relayer=guardian_and_relayer_data.relayer.address if guardian_and_relayer_data.relayer else None,
         gas_limit=args.gas_limit,
         gas_price=args.gas_price,
-        version=args.version,
-        options=args.options,
-        guardian_and_relayer_data=guardian_and_relayer_data,
     )
 
+    _sign_transaction(transaction, sender, guardian_and_relayer_data)
     cli_shared.send_or_simulate(transaction, args)
 
 
@@ -260,21 +287,19 @@ def close_proposal(args: Any):
         sender=sender.address.to_bech32(),
         args=args,
     )
-    chain_id = cli_shared.get_chain_id(args.proxy, args.chain)
-    gas_estimator = cli_shared.initialize_gas_limit_estimator(args)
-    controller = GovernanceWrapper(config=TransactionsFactoryConfig(chain_id), gas_limit_estimator=gas_estimator)
 
+    controller = _initialize_controller(args)
     transaction = controller.create_transaction_for_closing_proposal(
         sender=sender,
         nonce=sender.nonce,
         proposal_nonce=args.proposal_nonce,
+        guardian=guardian_and_relayer_data.guardian.address if guardian_and_relayer_data.guardian else None,
+        relayer=guardian_and_relayer_data.relayer.address if guardian_and_relayer_data.relayer else None,
         gas_limit=args.gas_limit,
         gas_price=args.gas_price,
-        version=args.version,
-        options=args.options,
-        guardian_and_relayer_data=guardian_and_relayer_data,
     )
 
+    _sign_transaction(transaction, sender, guardian_and_relayer_data)
     cli_shared.send_or_simulate(transaction, args)
 
 
@@ -286,22 +311,21 @@ def clear_ended_proposals(args: Any):
         sender=sender.address.to_bech32(),
         args=args,
     )
-    chain_id = cli_shared.get_chain_id(args.proxy, args.chain)
-    gas_estimator = cli_shared.initialize_gas_limit_estimator(args)
-    controller = GovernanceWrapper(config=TransactionsFactoryConfig(chain_id), gas_limit_estimator=gas_estimator)
 
     proposers = [Address.new_from_bech32(proposer) for proposer in args.proposers]
+    controller = _initialize_controller(args)
+
     transaction = controller.create_transaction_for_clearing_ended_proposals(
         sender=sender,
         nonce=sender.nonce,
         proposers=proposers,
+        guardian=guardian_and_relayer_data.guardian.address if guardian_and_relayer_data.guardian else None,
+        relayer=guardian_and_relayer_data.relayer.address if guardian_and_relayer_data.relayer else None,
         gas_limit=args.gas_limit,
         gas_price=args.gas_price,
-        version=args.version,
-        options=args.options,
-        guardian_and_relayer_data=guardian_and_relayer_data,
     )
 
+    _sign_transaction(transaction, sender, guardian_and_relayer_data)
     cli_shared.send_or_simulate(transaction, args)
 
 
@@ -313,20 +337,18 @@ def claim_accumulated_fees(args: Any):
         sender=sender.address.to_bech32(),
         args=args,
     )
-    chain_id = cli_shared.get_chain_id(args.proxy, args.chain)
-    gas_estimator = cli_shared.initialize_gas_limit_estimator(args)
-    controller = GovernanceWrapper(config=TransactionsFactoryConfig(chain_id), gas_limit_estimator=gas_estimator)
 
+    controller = _initialize_controller(args)
     transaction = controller.create_transaction_for_claiming_accumulated_fees(
         sender=sender,
         nonce=sender.nonce,
+        guardian=guardian_and_relayer_data.guardian.address if guardian_and_relayer_data.guardian else None,
+        relayer=guardian_and_relayer_data.relayer.address if guardian_and_relayer_data.relayer else None,
         gas_limit=args.gas_limit,
         gas_price=args.gas_price,
-        version=args.version,
-        options=args.options,
-        guardian_and_relayer_data=guardian_and_relayer_data,
     )
 
+    _sign_transaction(transaction, sender, guardian_and_relayer_data)
     cli_shared.send_or_simulate(transaction, args)
 
 
@@ -338,10 +360,8 @@ def change_config(args: Any):
         sender=sender.address.to_bech32(),
         args=args,
     )
-    chain_id = cli_shared.get_chain_id(args.proxy, args.chain)
-    gas_estimator = cli_shared.initialize_gas_limit_estimator(args)
-    controller = GovernanceWrapper(config=TransactionsFactoryConfig(chain_id), gas_limit_estimator=gas_estimator)
 
+    controller = _initialize_controller(args)
     transaction = controller.create_transaction_for_changing_config(
         sender=sender,
         nonce=sender.nonce,
@@ -350,24 +370,20 @@ def change_config(args: Any):
         min_quorum=args.min_quorum,
         min_veto_threshold=args.min_veto_threshold,
         min_pass_threshold=args.min_pass_threshold,
+        guardian=guardian_and_relayer_data.guardian.address if guardian_and_relayer_data.guardian else None,
+        relayer=guardian_and_relayer_data.relayer.address if guardian_and_relayer_data.relayer else None,
         gas_limit=args.gas_limit,
         gas_price=args.gas_price,
-        version=args.version,
-        options=args.options,
-        guardian_and_relayer_data=guardian_and_relayer_data,
     )
 
+    _sign_transaction(transaction, sender, guardian_and_relayer_data)
     cli_shared.send_or_simulate(transaction, args)
 
 
 def get_voting_power(args: Any):
     validate_proxy_argument(args)
 
-    config = get_config_for_network_providers()
-    proxy = ProxyNetworkProvider(url=args.proxy, config=config)
-    chain_id = proxy.get_network_config().chain_id
-    controller = GovernanceController(chain_id, proxy)
-
+    controller = _initialize_controller(args)
     user = Address.new_from_bech32(args.user)
 
     voting_power = controller.get_voting_power(user)
@@ -377,10 +393,7 @@ def get_voting_power(args: Any):
 def get_config(args: Any):
     validate_proxy_argument(args)
 
-    config = get_config_for_network_providers()
-    proxy = ProxyNetworkProvider(url=args.proxy, config=config)
-    chain_id = proxy.get_network_config().chain_id
-    controller = GovernanceController(chain_id, proxy)
+    controller = _initialize_controller(args)
 
     contract_config = controller.get_config()
     utils.dump_out_json(_config_to_dict(contract_config))
@@ -389,10 +402,7 @@ def get_config(args: Any):
 def get_proposal(args: Any):
     validate_proxy_argument(args)
 
-    config = get_config_for_network_providers()
-    proxy = ProxyNetworkProvider(url=args.proxy, config=config)
-    chain_id = proxy.get_network_config().chain_id
-    controller = GovernanceController(chain_id, proxy)
+    controller = _initialize_controller(args)
 
     info = controller.get_proposal(args.proposal_nonce)
     utils.dump_out_json(_proposal_to_dict(info))
@@ -401,10 +411,7 @@ def get_proposal(args: Any):
 def get_delegated_vote_info(args: Any):
     validate_proxy_argument(args)
 
-    config = get_config_for_network_providers()
-    proxy = ProxyNetworkProvider(url=args.proxy, config=config)
-    chain_id = proxy.get_network_config().chain_id
-    controller = GovernanceController(chain_id, proxy)
+    controller = _initialize_controller(args)
 
     contract = Address.new_from_bech32(args.contract)
     user = Address.new_from_bech32(args.user)