cairo-lang-plugins 2.19.0-rc.2

Cairo core plugin implementations.
Documentation 
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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109

use indent::indent_by;
use indoc::formatdoc;
use itertools::Itertools;

use super::PluginTypeInfo;
use crate::plugins::utils::TypeVariant;

/// Adds derive result for the `Serde` trait.
pub fn handle_serde(info: &PluginTypeInfo<'_>, member_access_desnaps: bool) -> String {
    const SERDE_TRAIT: &str = "core::serde::Serde";
    const DESTRUCT_TRAIT: &str = "core::traits::Destruct";
    let ty = &info.name;
    let full_typename = info.full_typename();
    let header = match &info.type_variant {
        TypeVariant::Enum => {
            let impl_additional_generics =
                info.impl_generics(&[SERDE_TRAIT], |trt, ty| format!("{trt}<{ty}>")).join(", ");
            formatdoc! {"
            impl {ty}Serde<
                    {impl_additional_generics}
                > of {SERDE_TRAIT}<{full_typename}>
            "}
        }
        TypeVariant::Struct => info.impl_header(SERDE_TRAIT, &[SERDE_TRAIT, DESTRUCT_TRAIT]),
    };
    let serialize_body = indent_by(
        8,
        match &info.type_variant {
            TypeVariant::Enum => {
                formatdoc! {"
                match self {{
                    {}
                }}",
                info.members_info.iter().enumerate().map(|(idx, variant)| {
                    format!(
                        "{ty}::{variant}(x) => {{ {SERDE_TRAIT}::<felt252>::serialize(@{idx}, ref output); \
                        {imp}::serialize(x, ref output); }},",
                        variant=variant.name,
                        imp=variant.impl_name(SERDE_TRAIT),
                    )
                }).join("\n    ")}
            }
            TypeVariant::Struct => info
                .members_info
                .iter()
                .map(|member| {
                    let imp = member.impl_name(SERDE_TRAIT);
                    let member = &member.name;
                    if member_access_desnaps {
                        format!("{imp}::serialize(@self.{member}, ref output)")
                    } else {
                        format!("{imp}::serialize(self.{member}, ref output)")
                    }
                })
                .join(";\n"),
        },
    );
    let deserialize_body = indent_by(
        8,
        match &info.type_variant {
            TypeVariant::Enum => {
                formatdoc! {"
                    let idx: felt252 = {SERDE_TRAIT}::<felt252>::deserialize(ref serialized)?;
                    core::option::Option::Some(
                        match idx {{
                            {}
                            _ => {{ return core::option::Option::None; }}
                        }}
                    )",
                    info.members_info.iter().enumerate().map(|(idx, variant)| {
                        format!(
                            "{idx} => {ty}::{variant}(\
                                {imp}::deserialize(ref serialized)?),",
                            variant=variant.name,
                            imp=variant.impl_name(SERDE_TRAIT),
                        )
                    }).join("\n        ")
                }
            }
            TypeVariant::Struct => {
                formatdoc! {"
                    {}
                    core::option::Option::Some({ty} {{
                        {}
                    }})",
                    info.members_info.iter().map(|member|format!(
                        "let {member} = {destruct_with} {{ value: {imp}::deserialize(ref serialized)? }};",
                        member=member.name,
                        destruct_with=member.destruct_with(),
                        imp=member.impl_name(SERDE_TRAIT),
                    )).join("\n"),
                    info.members_info.iter().map(|member|format!(
                        "{member}: {member}.value,", member=member.name
                    )).join("\n    "),
                }
            }
        },
    );
    formatdoc! {"
        {header} {{
            fn serialize(self: @{full_typename}, ref output: core::array::Array<felt252>) {{
                {serialize_body}
            }}
            fn deserialize(ref serialized: core::array::Span<felt252>) -> core::option::Option<{full_typename}> {{
                {deserialize_body}
            }}
        }}
    "}
}