register.rs

use crate::{parse_partnerchain_public_keys, PaymentFilePath};
use partner_chains_cardano_offchain::{
	await_tx::FixedDelayRetries,
	register::{run_deregister, run_register},
};
use sidechain_domain::{
	AdaBasedStaking, CandidateRegistration, MainchainSignature, PermissionedCandidateData,
	SidechainSignature, StakePoolPublicKey, UtxoId,
};

#[derive(Clone, Debug, clap::Parser)]
pub struct RegisterCmd {
	#[clap(flatten)]
	common_arguments: crate::CommonArguments,
	/// Genesis UTXO of the partner-chain
	#[arg(long)]
	genesis_utxo: UtxoId,
	/// UTXO that will be spend when executing registration transaction, part of the registration message
	#[arg(long)]
	registration_utxo: UtxoId,
	#[clap(flatten)]
	payment_key_file: PaymentFilePath,
	#[arg(
		long,
		value_name = "PARTNERCHAIN_KEY:AURA_KEY:GRANDPA_KEY",
		alias = "sidechain-public-keys",
		value_parser = parse_partnerchain_public_keys
	)]
	/// Colon separated hex strings representing bytes of the Sidechain, Aura and Grandpa public keys
	partner_chain_public_keys: PermissionedCandidateData,
	/// Hex string of bytes of the registration message signature by partner-chain key, obtained by 'registration-signatures' command
	#[arg(long, alias = "sidechain-signature")]
	partner_chain_signature: SidechainSignature,
	/// Hex string representing bytes of the Stake Pool Verification Key
	#[arg(long)]
	spo_public_key: StakePoolPublicKey,
	/// Hex string of bytes of the registration message signature by main chain key, obtained by 'registration-signatures' command
	#[arg(long)]
	spo_signature: MainchainSignature,
}

impl RegisterCmd {
	pub async fn execute(self) -> crate::CmdResult<()> {
		let payment_key = self.payment_key_file.read_key()?;
		let client = self.common_arguments.get_ogmios_client().await?;
		let candidate_registration = CandidateRegistration {
			stake_ownership: AdaBasedStaking {
				pub_key: self.spo_public_key,
				signature: self.spo_signature,
			},
			partner_chain_pub_key: self.partner_chain_public_keys.sidechain_public_key,
			partner_chain_signature: self.partner_chain_signature,
			own_pkh: payment_key.to_pub_key_hash(),
			registration_utxo: self.registration_utxo,
			aura_pub_key: self.partner_chain_public_keys.aura_public_key,
			grandpa_pub_key: self.partner_chain_public_keys.grandpa_public_key,
		};

		run_register(
			self.genesis_utxo,
			&candidate_registration,
			&payment_key,
			&client,
			FixedDelayRetries::two_minutes(),
		)
		.await?;

		Ok(())
	}
}

#[derive(Clone, Debug, clap::Parser)]
pub struct DeregisterCmd {
	#[clap(flatten)]
	common_arguments: crate::CommonArguments,
	/// Genesis UTXO of the partner-chain
	#[arg(long)]
	genesis_utxo: UtxoId,
	#[clap(flatten)]
	payment_key_file: PaymentFilePath,
	/// Hex string representing bytes of the Stake Pool Verification Key
	#[arg(long)]
	spo_public_key: StakePoolPublicKey,
}

impl DeregisterCmd {
	pub async fn execute(self) -> crate::CmdResult<()> {
		let payment_signing_key = self.payment_key_file.read_key()?;
		let client = self.common_arguments.get_ogmios_client().await?;

		run_deregister(
			self.genesis_utxo,
			&payment_signing_key,
			self.spo_public_key,
			&client,
			FixedDelayRetries::two_minutes(),
		)
		.await?;

		Ok(())
	}
}