mass payout for all unclaimed eras added

Signed-off-by: Uncle Stretch <uncle.stretch@ghostchain.io>
This commit is contained in:
Uncle Stretch 2025-05-17 16:49:14 +03:00
parent 5de724fc19
commit 0e4cb343dc
Signed by: str3tch
GPG Key ID: 84F3190747EE79AA
4 changed files with 176 additions and 1 deletions

View File

@ -36,6 +36,7 @@ pub enum Action {
ClosePopup,
RotateSessionKeys,
PayoutValidatorPopup(u32),
PayoutAllValidatorPopup(Vec<u32>),
WithdrawValidatorPopup,
BalanceRequest([u8; 32], bool),

View File

@ -65,6 +65,25 @@ impl History {
}
}
fn payout_all_available(&mut self) {
let unclaimed_keys = self.rewards
.iter()
.filter_map(|(k, v)| (!v.is_claimed).then(|| *k))
.collect::<Vec<_>>();
if let Some(action_tx) = &self.action_tx {
let _ = if unclaimed_keys.len() == 0 {
action_tx.send(Action::EventLog(
String::from("no available payouts found for current validator"),
ActionLevel::Warn,
ActionTarget::ValidatorLog))
} else {
self.pending_payout.extend(&unclaimed_keys);
action_tx.send(Action::PayoutAllValidatorPopup(unclaimed_keys))
};
}
}
fn payout_by_era_index(&mut self) {
if let Some(index) = self.table_state.selected() {
let rev_index = self.rewards.len()
@ -248,6 +267,7 @@ impl Component for History {
KeyCode::Down | KeyCode::Char('j') => self.next_row(),
KeyCode::Char('g') => self.first_row(),
KeyCode::Char('G') => self.last_row(),
KeyCode::Char('H') => self.payout_all_available(),
KeyCode::Enter => self.payout_by_era_index(),
_ => {},
};

View File

@ -23,6 +23,7 @@ mod staking_details;
mod reward_details;
mod bond_popup;
mod payout_popup;
mod payout_all_popup;
mod rotate_popup;
mod validate_popup;
mod chill_popup;
@ -43,6 +44,7 @@ use history::History;
use withdrawals::Withdrawals;
use bond_popup::BondPopup;
use payout_popup::PayoutPopup;
use payout_all_popup::PayoutAllPopup;
use rotate_popup::RotatePopup;
use validate_popup::ValidatePopup;
use chill_popup::ChillPopup;
@ -62,6 +64,7 @@ pub enum CurrentTab {
EventLogs,
BondPopup,
PayoutPopup,
PayoutAllPopup,
RotatePopup,
ValidatePopup,
ChillPopup,
@ -101,6 +104,7 @@ impl Default for Validator {
Box::new(EventLogs::default()),
Box::new(BondPopup::default()),
Box::new(PayoutPopup::default()),
Box::new(PayoutAllPopup::default()),
Box::new(RotatePopup::default()),
Box::new(ValidatePopup::default()),
Box::new(ChillPopup::default()),
@ -171,7 +175,8 @@ impl Component for Validator {
CurrentTab::UnbondPopup |
CurrentTab::RebondPopup |
CurrentTab::WithdrawPopup |
CurrentTab::PayoutPopup => {
CurrentTab::PayoutPopup |
CurrentTab::PayoutAllPopup => {
for component in self.components.iter_mut() {
component.handle_key_event(key)?;
}
@ -244,6 +249,10 @@ impl Component for Validator {
self.previous_tab = self.current_tab;
self.current_tab = CurrentTab::PayoutPopup;
}
Action::PayoutAllValidatorPopup(_) => {
self.previous_tab = self.current_tab;
self.current_tab = CurrentTab::PayoutAllPopup;
}
Action::WithdrawValidatorPopup => {
self.previous_tab = self.current_tab;
self.current_tab = CurrentTab::WithdrawPopup;

View File

@ -0,0 +1,145 @@
use crossterm::event::{KeyCode, KeyEvent, KeyEventKind};
use color_eyre::Result;
use ratatui::{
layout::{Alignment, Constraint, Flex, Layout, Rect},
widgets::{Block, Clear, Paragraph},
Frame
};
use tokio::sync::mpsc::UnboundedSender;
use std::sync::mpsc::Sender;
use super::{Component, PartialComponent, CurrentTab};
use crate::{
action::Action,
config::Config,
palette::StylePalette,
};
#[derive(Debug)]
pub struct PayoutAllPopup {
is_active: bool,
action_tx: Option<UnboundedSender<Action>>,
network_tx: Option<Sender<Action>>,
secret_seed: [u8; 32],
stash_account_id: [u8; 32],
era_indexes: Vec<u32>,
palette: StylePalette
}
impl Default for PayoutAllPopup {
fn default() -> Self {
Self::new()
}
}
impl PayoutAllPopup {
pub fn new() -> Self {
Self {
is_active: false,
secret_seed: [0u8; 32],
stash_account_id: [0u8; 32],
era_indexes: Default::default(),
action_tx: None,
network_tx: None,
palette: StylePalette::default(),
}
}
fn close_popup(&mut self) {
self.is_active = false;
self.era_indexes = Default::default();
if let Some(action_tx) = &self.action_tx {
let _ = action_tx.send(Action::ClosePopup);
}
}
fn start_payout(&mut self) {
if let Some(network_tx) = &self.network_tx {
for era_index in &self.era_indexes {
let _ = network_tx.send(Action::PayoutStakers(
self.secret_seed,
self.stash_account_id,
*era_index));
}
}
self.close_popup();
}
}
impl PartialComponent for PayoutAllPopup {
fn set_active(&mut self, current_tab: CurrentTab) {
match current_tab {
CurrentTab::PayoutAllPopup => self.is_active = true,
_ => self.is_active = false,
};
}
}
impl Component for PayoutAllPopup {
fn register_network_handler(&mut self, tx: Sender<Action>) -> Result<()> {
self.network_tx = Some(tx);
Ok(())
}
fn register_action_handler(&mut self, tx: UnboundedSender<Action>) -> Result<()> {
self.action_tx = Some(tx);
Ok(())
}
fn register_config_handler(&mut self, config: Config) -> Result<()> {
if let Some(style) = config.styles.get(&crate::app::Mode::Wallet) {
self.palette.with_normal_style(style.get("normal_style").copied());
self.palette.with_normal_border_style(style.get("normal_border_style").copied());
self.palette.with_normal_title_style(style.get("normal_title_style").copied());
self.palette.with_popup_style(style.get("popup_style").copied());
self.palette.with_popup_title_style(style.get("popup_title_style").copied());
}
Ok(())
}
fn handle_key_event(&mut self, key: KeyEvent) -> Result<Option<Action>> {
if self.is_active && key.kind == KeyEventKind::Press {
match key.code {
KeyCode::Enter => self.start_payout(),
KeyCode::Esc => self.close_popup(),
_ => {},
};
}
Ok(None)
}
fn update(&mut self, action: Action) -> Result<Option<Action>> {
match action {
Action::PayoutAllValidatorPopup(era_indexes) => self.era_indexes = era_indexes,
Action::SetStashSecret(secret_seed) => self.secret_seed = secret_seed,
Action::SetStashAccount(account_id) => self.stash_account_id = account_id,
_ => {}
};
Ok(None)
}
fn draw(&mut self, frame: &mut Frame, area: Rect) -> Result<()> {
if self.is_active {
let (border_style, border_type) = self.palette.create_popup_style();
let popup = Paragraph::new(
format!(" Do payout for {} eras, from {} to {}",
self.era_indexes.len(),
self.era_indexes.first().expect("Length of unclaimed indexes always more then one; qed"),
self.era_indexes.last().expect("Length of unclaimed indexes always more then one; qed")))
.block(Block::bordered()
.border_style(border_style)
.border_type(border_type)
.title_style(self.palette.create_popup_title_style())
.title_alignment(Alignment::Right)
.title("Enter to proceed / Esc to close"));
let v = Layout::vertical([Constraint::Max(3)]).flex(Flex::Center);
let h = Layout::horizontal([Constraint::Max(50)]).flex(Flex::Center);
let [area] = v.areas(area);
let [area] = h.areas(area);
frame.render_widget(Clear, area);
frame.render_widget(popup, area);
}
Ok(())
}
}