BEGIN rust changes for no-optional protos

begin trait updates
penumbra-zone · Aug 16, 2023 · 15afac0 · 15afac0
1 parent 686856d
commit 15afac0
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 9 deletions.
diff --git a/crates/core/component/stake/src/validator/bonding.rs b/crates/core/component/stake/src/validator/bonding.rs
@@ -53,8 +53,9 @@ impl From<State> for pb::BondingState {
                 }
             },
             unbonding_epoch: match v {
-                State::Unbonding { unbonding_epoch } => Some(unbonding_epoch),
-                _ => None,
+                State::Unbonding { unbonding_epoch } => unbonding_epoch,
+                // REVIEW: is setting 0 here equivalent to None when it was an Option?
+                _ => 0,
             },
         }
     }
@@ -70,12 +71,9 @@ impl TryFrom<pb::BondingState> for State {
         match bonding_state {
             pb::bonding_state::BondingStateEnum::Bonded => Ok(State::Bonded),
             pb::bonding_state::BondingStateEnum::Unbonded => Ok(State::Unbonded),
-            pb::bonding_state::BondingStateEnum::Unbonding => {
-                let Some(unbonding_epoch) = v.unbonding_epoch else {
-                    anyhow::bail!("unbonding epoch should be set for unbonding state")
-                };
-                Ok(State::Unbonding { unbonding_epoch })
-            }
+            pb::bonding_state::BondingStateEnum::Unbonding => Ok(State::Unbonding {
+                unbonding_epoch: v.unbonding_epoch,
+            }),
             pb::bonding_state::BondingStateEnum::Unspecified => {
                 Err(anyhow::anyhow!("unspecified bonding state!"))
             }

diff --git a/crates/core/transaction/src/proposal.rs b/crates/core/transaction/src/proposal.rs
@@ -530,6 +530,12 @@ impl<W> Withdrawn<W> {
     }
 }
 
+impl<W> From<std::string::String> for Withdrawn<W> {
+    fn from(reason: String) -> Self {
+        Withdrawn::WithReason { reason }
+    }
+}
+
 impl<W> From<Option<W>> for Withdrawn<W> {
     fn from(reason: Option<W>) -> Self {
         match reason {
@@ -586,7 +592,7 @@ impl From<Outcome<String>> for pb::ProposalOutcome {
             }
             Outcome::Slashed { withdrawn } => {
                 pb::proposal_outcome::Outcome::Slashed(pb::proposal_outcome::Slashed {
-                    withdrawn_with_reason: withdrawn.into(),
+                    withdrawn_with_reason: withdrawn,
                 })
             }
         };