From b543f25b4df91f0622614fb369fb9f1ec088443a Mon Sep 17 00:00:00 2001
From: Ashwin Sekar <ashwin@anza.xyz>
Date: Mon, 25 Nov 2024 14:55:08 -0500
Subject: [PATCH] cli: show max vote credits in vote-account based on TVC
 activation epoch (#3776)

---
 cli-output/src/cli_output.rs |  6 ++++--
 cli/src/vote.rs              | 14 +++++++++++++-
 2 files changed, 17 insertions(+), 3 deletions(-)

diff --git a/cli-output/src/cli_output.rs b/cli-output/src/cli_output.rs
index 2f6130e35c2fa3..315d7841ca4582 100644
--- a/cli-output/src/cli_output.rs
+++ b/cli-output/src/cli_output.rs
@@ -1163,8 +1163,9 @@ fn show_votes_and_credits(
         )?;
         writeln!(
             f,
-            "  credits/slots: {}/{}",
-            entry.credits_earned, entry.slots_in_epoch
+            "  credits/max credits: {}/{}",
+            entry.credits_earned,
+            entry.slots_in_epoch * u64::from(entry.max_credits_per_slot)
         )?;
     }
     if let Some(oldest) = epoch_voting_history.iter().next() {
@@ -1740,6 +1741,7 @@ pub struct CliEpochVotingHistory {
     pub credits_earned: u64,
     pub credits: u64,
     pub prev_credits: u64,
+    pub max_credits_per_slot: u8,
 }
 
 #[derive(Serialize, Deserialize)]
diff --git a/cli/src/vote.rs b/cli/src/vote.rs
index 88a44cc6bb0c10..e8ba0acf221908 100644
--- a/cli/src/vote.rs
+++ b/cli/src/vote.rs
@@ -40,7 +40,9 @@ use {
     solana_vote_program::{
         vote_error::VoteError,
         vote_instruction::{self, withdraw, CreateVoteAccountConfig},
-        vote_state::{VoteAuthorize, VoteInit, VoteState, VoteStateVersions},
+        vote_state::{
+            VoteAuthorize, VoteInit, VoteState, VoteStateVersions, VOTE_CREDITS_MAXIMUM_PER_SLOT,
+        },
     },
     std::rc::Rc,
 };
@@ -1293,6 +1295,9 @@ pub fn process_show_vote_account(
         get_vote_account(rpc_client, vote_account_address, config.commitment)?;
 
     let epoch_schedule = rpc_client.get_epoch_schedule()?;
+    let tvc_activation_slot =
+        rpc_client.get_feature_activation_slot(&solana_feature_set::timely_vote_credits::id())?;
+    let tvc_activation_epoch = tvc_activation_slot.map(|s| epoch_schedule.get_epoch(s));
 
     let mut votes: Vec<CliLandedVote> = vec![];
     let mut epoch_voting_history: Vec<CliEpochVotingHistory> = vec![];
@@ -1303,12 +1308,19 @@ pub fn process_show_vote_account(
         for (epoch, credits, prev_credits) in vote_state.epoch_credits().iter().copied() {
             let credits_earned = credits.saturating_sub(prev_credits);
             let slots_in_epoch = epoch_schedule.get_slots_in_epoch(epoch);
+            let is_tvc_active = tvc_activation_epoch.map(|e| epoch >= e).unwrap_or_default();
+            let max_credits_per_slot = if is_tvc_active {
+                VOTE_CREDITS_MAXIMUM_PER_SLOT
+            } else {
+                1
+            };
             epoch_voting_history.push(CliEpochVotingHistory {
                 epoch,
                 slots_in_epoch,
                 credits_earned,
                 credits,
                 prev_credits,
+                max_credits_per_slot,
             });
         }
     }