reload keys and push own wallet to address book

Signed-off-by: Uncle Stretch <uncle.stretch@ghostchain.io>

Cargo.toml

@@ -2,7 +2,7 @@
 name = "ghost-eye"
 authors = ["str3tch <stretch@ghostchain.io>"]
 description = "Application for interacting with Casper/Ghost nodes that are exposing RPC only to the localhost"
-version = "0.3.41"
+version = "0.3.42"
 edition = "2021"
 homepage = "https://git.ghostchain.io/ghostchain"
 repository = "https://git.ghostchain.io/ghostchain/ghost-eye"

src/action.rs

@@ -105,6 +105,8 @@ pub enum Action {
     GetValidatorPrefs([u8; 32], bool),
     GetSlashingSpans([u8; 32], bool),
     GetValidatorLatestClaim([u8; 32], bool),
+    GetValidatorIsDisabled([u8; 32], bool),
+    GetValidatorInSession([u8; 32], bool),
     GetCurrentValidatorEraRewards,
 
     SetNodeName(Option<String>),
@@ -135,6 +137,7 @@ pub enum Action {
     SetValidatorEraSlash(u32, u128),
     SetValidatorEraUnlocking(u32, u128, [u8; 32]),
     SetValidatorLatestClaim(u32, [u8; 32]),
+    SetValidatorInSession(bool, [u8; 32]),
     SetIsBonded(bool),
     SetStakedAmountRatio(u128, u128, [u8; 32]),
     SetStakedRatio(u128, u128, [u8; 32]),

src/components/validator/reward_details.rs

@@ -24,6 +24,7 @@ pub struct RewardDetails {
     stash: [u8; 32],
     in_staking_validators: bool, 
     in_session_validators: bool,
+    is_disabled: bool,
 }
 
 impl Default for RewardDetails {
@@ -43,6 +44,7 @@ impl RewardDetails {
             stash: [0u8; 32],
             in_staking_validators: false, 
             in_session_validators: false,
+            is_disabled: false,
         }
     }
 
@@ -79,6 +81,13 @@ impl Component for RewardDetails {
     fn update(&mut self, action: Action) -> Result<Option<Action>> {
         match action {
             Action::SetStashAccount(stash) => self.stash = stash,
+            Action::SetCurrentValidatorEraRewards(_, _, individual) => self.is_disabled = individual
+                .iter()
+                .find(|data| data.account_id == self.stash)
+                .map(|data| data.disabled)
+                .unwrap_or_default(),
+            Action::SetValidatorInSession(in_session_validators, account_id) if self.stash == account_id =>
+                self.in_session_validators = in_session_validators,
             Action::SetValidatorPrefs(commission, disabled, account_id) if self.stash == account_id => {
                 self.commission = commission;
                 self.in_staking_validators = commission.is_some();
@@ -95,11 +104,15 @@ impl Component for RewardDetails {
         let [_, _, place] = super::validator_balance_layout(area);
         let (border_style, border_type) = self.palette.create_border_style(false);
 
-        let staking_status = match (self.in_staking_validators, self.in_session_validators) {
-            (false, false) => "Staking status: Nothing".to_string(),
-            (true, false)  => "Staking status: Active".to_string(),
-            (false, true)  => "Staking status: Pending".to_string(),
-            (true, true)   => "Staking status: Active".to_string(),
+        let staking_status = if self.is_disabled {
+            "Staking status: Disabled".to_string()
+        } else {
+            match (self.in_staking_validators, self.in_session_validators) {
+                (false, false) => "Staking status: Nothing".to_string(),
+                (true, false)  => "Staking status: Stopping".to_string(),
+                (false, true)  => "Staking status: Pending".to_string(),
+                (true, true)   => "Staking status: Active".to_string(),
+            }
         };
 
         let table = Table::new(

src/components/validator/stash_info.rs

@@ -159,6 +159,8 @@ impl StashInfo {
             let _ = network_tx.send(Action::GetSessionKeys(account_id, true));
             let _ = network_tx.send(Action::GetValidatorAllRewards(account_id, true));
             let _ = network_tx.send(Action::GetSlashingSpans(account_id, true));
+            let _ = network_tx.send(Action::GetValidatorIsDisabled(account_id, true));
+            let _ = network_tx.send(Action::GetValidatorInSession(account_id, true));
         }
     }
 

src/components/wallet/accounts.rs

@@ -315,6 +315,69 @@ impl Accounts {
         Ok(())
     }
 
+    fn load_initial_keys(&mut self) {
+        self.try_load_and_store_keys("/etc/ghost/wallet-key", "ghostie");
+        self.try_load_and_store_keys("/etc/ghost/stash-key", "stash");
+    }
+
+    fn try_load_and_store_keys(&mut self, path_to_key: &str, name_for_key: &str) {
+        match std::fs::read_to_string(path_to_key) {
+            Ok(content) => {
+                let content = content.replace("\n", "");
+                let content = &content[2..];
+                let seed: [u8; 32] = hex::decode(content)
+                    .expect("stored seed is valid hex string; qed")
+                    .as_slice()
+                    .try_into()
+                    .expect("stored seed is valid length; qed");
+
+                let pair = Pair::from_seed(&seed);
+                let secret_seed = hex::encode(seed);
+                let account_id = pair.public().0;
+                let address = AccountId32::from(account_id)
+                    .to_ss58check_with_version(Ss58AddressFormat::custom(1996));
+
+                if self.wallet_keys.iter().any(|key| key.seed == secret_seed) {
+                    self.log_event(
+                        format!("{} from `{}` already loaded", name_for_key, path_to_key),
+                        ActionLevel::Warn);
+                } else {
+                    self.log_event(
+                        format!("{} from `{}` loaded to ghost-eye", name_for_key, path_to_key),
+                        ActionLevel::Warn);
+
+                    self.send_balance_request(account_id, false);
+                    self.wallet_keys.push(AccountInfo {
+                        name: name_for_key.to_string(), 
+                        address, 
+                        account_id,
+                        seed: secret_seed, 
+                    });
+                    self.save_to_file();
+                }
+
+            }
+            Err(_) => {
+                self.log_event(
+                    format!("nothing found inside `{}`", path_to_key),
+                    ActionLevel::Warn);
+            }
+        };
+    }
+
+    fn push_to_address_book(&mut self) {
+        if let Some(index) = self.table_state.selected() {
+            if let Some(action_tx) = &self.action_tx {
+                let wallet_key = self.wallet_keys
+                    .get(index)
+                    .expect("wallet key index should be in range; qed");
+                let _ = action_tx.send(Action::NewAddressBookRecord(
+                        wallet_key.name.clone(),
+                        wallet_key.address.clone()));
+            }
+        }
+    }
+
     fn first_row(&mut self) {
         if self.wallet_keys.len() > 0 {
             self.table_state.select(Some(0));
@@ -454,6 +517,8 @@ impl Component for Accounts {
                 KeyCode::Char('D') => self.delete_row(),
                 KeyCode::Char('R') => self.update_account_name(),
                 KeyCode::Char('Y') => self.copy_to_clipboard(),
+                KeyCode::Char('L') => self.load_initial_keys(),
+                KeyCode::Char('P') => self.push_to_address_book(),
                 _ => {},
             };
         }

src/components/wallet/address_book.rs

@@ -196,24 +196,40 @@ impl AddressBook {
                         ActionLevel::Error);
                 }
 
-                self.log_event(
-                    format!("account {} with address {} added to address book", &name, &address),
-                    ActionLevel::Info);
+                match self.address_book.iter().position(|record| record.address == address) {
+                    Some(index) => {
+                        let record = self.address_book
+                            .get(index)
+                            .expect("record should be in range of address book; qed");
+                        self.log_event(
+                            format!("record with name `{}` already stores address {}", &record.name, &record.address),
+                            ActionLevel::Warn);
+                        self.table_state.select(Some(index));
+                        self.scroll_state = self.scroll_state.position(index);
+                    },
+                    None => {
+                        let seed_vec = account_id.to_raw_vec();
+                        let mut account_id = [0u8; 32];
+                        account_id.copy_from_slice(&seed_vec);
+                        let seed_str = hex::encode(seed_vec);
 
-                let seed_vec = account_id.to_raw_vec();
-                let mut account_id = [0u8; 32];
-                account_id.copy_from_slice(&seed_vec);
-                let seed_str = hex::encode(seed_vec);
+                        self.log_event(
+                            format!("account {} with address {} added to address book", &name, &address),
+                            ActionLevel::Info);
 
-                self.address_book.push(BookRecord {
-                    name, 
-                    address, 
-                    account_id, 
-                    seed: seed_str,
-                });
-                self.save_to_file();
-                self.last_row();
-                self.send_balance_request(account_id, false);
+                        self.address_book.push(BookRecord {
+                            name, 
+                            address, 
+                            account_id, 
+                            seed: seed_str,
+                        });
+
+                        self.scroll_state = self.scroll_state.content_length(self.address_book.len());
+                        self.save_to_file();
+                        self.send_balance_request(account_id, false);
+                        self.last_row();
+                    },
+                }
             },
             Err(_) => self.log_event(
                 format!("provided account address {} is invalid", &address),

src/network/mod.rs

@@ -118,6 +118,7 @@ impl Network {
                     predefined_calls::get_is_stash_bonded(&self.action_tx, &self.online_client_api, &stash_to_watch).await?;
                     predefined_calls::get_validators_ledger(&self.action_tx, &self.online_client_api, &stash_to_watch).await?;
                     predefined_calls::get_slashing_spans(&self.action_tx, &self.online_client_api, &stash_to_watch).await?;
+                    predefined_calls::get_validator_in_session(&self.action_tx, &self.online_client_api, &stash_to_watch).await?;
 
                     for era_index in self.eras_to_watch.iter() {
                         predefined_calls::get_validator_staking_result(&self.action_tx, &self.online_client_api, &stash_to_watch, *era_index).await?;
@@ -206,6 +207,10 @@ impl Network {
                 Ok(())
             }
 
+            Action::GetValidatorInSession(account_id, is_stash) => {
+                self.store_stash_or_validator_if_possible(account_id, is_stash);
+                predefined_calls::get_validator_in_session(&self.action_tx, &self.online_client_api, &account_id).await
+            }
             Action::GetValidatorLatestClaim(account_id, is_stash) => {
                 self.store_stash_or_validator_if_possible(account_id, is_stash);
                 predefined_calls::get_validator_latest_claim(&self.action_tx, &self.online_client_api, &account_id).await

src/network/predefined_calls.rs

@@ -574,3 +574,18 @@ pub async fn get_validator_latest_claim(
 
     Ok(())
 }
+
+pub async fn get_validator_in_session(
+    action_tx: &UnboundedSender<Action>,
+    api: &OnlineClient<CasperConfig>,
+    account_id: &[u8; 32],
+) -> Result<()> {
+    let in_list = super::raw_calls::session::validators(api, None)
+        .await?
+        .unwrap_or_default()
+        .iter()
+        .position(|validator| validator.0 == *account_id)
+        .is_some();
+    action_tx.send(Action::SetValidatorInSession(in_list, *account_id))?;
+    Ok(())
+}