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
79
80
81
82
83
84

// Copyright 2019-2024 ChainSafe Systems
// SPDX-License-Identifier: Apache-2.0, MIT

use super::*;
use crate::shim::{address::Address, econ::TokenAmount, state_tree::ActorState};
use ::cid::Cid;

#[derive(Clone, Serialize, Deserialize, JsonSchema)]
#[serde(rename_all = "PascalCase")]
#[schemars(rename = "ActorState")]
pub struct ActorStateLotusJson {
    #[schemars(with = "LotusJson<Cid>")]
    #[serde(with = "crate::lotus_json")]
    code: Cid,
    #[schemars(with = "LotusJson<Cid>")]
    #[serde(with = "crate::lotus_json")]
    head: Cid,
    nonce: u64,
    #[schemars(with = "LotusJson<TokenAmount>")]
    #[serde(with = "crate::lotus_json")]
    balance: TokenAmount,
    #[schemars(with = "LotusJson<Option<Address>>")]
    #[serde(
        with = "crate::lotus_json",
        skip_serializing_if = "Option::is_none",
        default
    )]
    delegated_address: Option<Address>,
}

impl HasLotusJson for ActorState {
    type LotusJson = ActorStateLotusJson;

    #[cfg(test)]
    fn snapshots() -> Vec<(serde_json::Value, Self)> {
        vec![(
            json!({
                "Balance": "0",
                "Code": {
                    "/": "baeaaaaa"
                },
                "Head": {
                    "/": "baeaaaaa"
                },
                "Nonce": 0,
            }),
            Self::new(
                Cid::default(),
                Cid::default(),
                TokenAmount::default(),
                0,
                None,
            ),
        )]
    }

    fn into_lotus_json(self) -> Self::LotusJson {
        let fvm3::state_tree::ActorState {
            code,
            state,
            sequence,
            balance,
            delegated_address,
        } = From::from(self);
        Self::LotusJson {
            code,
            head: state,
            nonce: sequence,
            balance: crate::shim::econ::TokenAmount::from(balance),
            delegated_address: delegated_address.map(crate::shim::address::Address::from),
        }
    }

    fn from_lotus_json(lotus_json: Self::LotusJson) -> Self {
        let ActorStateLotusJson {
            code,
            head,
            nonce,
            balance,
            delegated_address,
        } = lotus_json;
        Self::new(code, head, balance, nonce, delegated_address)
    }
}