1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78

//! Definition of domain `UpgradeProposal` type for handling upgrade client proposal

use ibc_core_client_types::error::UpgradeClientError;
use ibc_primitives::prelude::*;
use ibc_proto::google::protobuf::Any;
use ibc_proto::ibc::core::client::v1::UpgradeProposal as RawUpgradeProposal;
use ibc_proto::Protobuf;

use super::Plan;

/// Defines a governance proposal of type `Content` that enables the initiation
/// of an IBC breaking upgrade and specifies the new client state that should be
/// utilized following the upgrade.
#[derive(Clone, Debug)]
pub struct UpgradeProposal {
    // Title of the proposal
    pub title: String,
    // Description of the proposal
    pub description: String,
    // The upgrade plan
    pub plan: Plan,
    // The upgraded client state
    pub upgraded_client_state: Any,
}

impl Protobuf<RawUpgradeProposal> for UpgradeProposal {}

impl TryFrom<RawUpgradeProposal> for UpgradeProposal {
    type Error = UpgradeClientError;

    fn try_from(raw: RawUpgradeProposal) -> Result<Self, Self::Error> {
        if raw.title.is_empty() {
            return Err(UpgradeClientError::InvalidUpgradeProposal {
                reason: "title field cannot be empty".to_string(),
            });
        }

        if raw.description.is_empty() {
            return Err(UpgradeClientError::InvalidUpgradeProposal {
                reason: "description field cannot be empty".to_string(),
            });
        }

        let plan = if let Some(plan) = raw.plan {
            plan.try_into()?
        } else {
            return Err(UpgradeClientError::InvalidUpgradeProposal {
                reason: "plan field cannot be empty".to_string(),
            });
        };

        let upgraded_client_state = if let Some(upgraded_client_state) = raw.upgraded_client_state {
            upgraded_client_state
        } else {
            return Err(UpgradeClientError::InvalidUpgradeProposal {
                reason: "upgraded client state cannot be empty".to_string(),
            });
        };

        Ok(Self {
            title: raw.title,
            description: raw.description,
            plan,
            upgraded_client_state,
        })
    }
}

impl From<UpgradeProposal> for RawUpgradeProposal {
    fn from(value: UpgradeProposal) -> Self {
        Self {
            title: value.title,
            description: value.description,
            plan: Some(value.plan.into()),
            upgraded_client_state: Some(value.upgraded_client_state),
        }
    }
}