bat-cli 0.8.6

Blockchain Auditor Toolkit (BAT)
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

use crate::batbelt::metadata::traits_source_code_metadata::TraitMetadataType;
use crate::batbelt::metadata::{BatMetadata, MetadataError, MetadataId, MetadataResult};
use crate::batbelt::parser::trait_parser::TraitParser;
use error_stack::ResultExt;
use serde::{Deserialize, Serialize};

#[derive(Serialize, Deserialize, Clone, Debug)]
pub struct TraitMetadataFunction {
    pub function_source_code_metadata_id: MetadataId,
    pub trait_signature: String,
}

#[derive(Serialize, Deserialize, Clone, Debug)]
pub struct TraitMetadata {
    pub name: String,
    pub metadata_id: MetadataId,
    pub trait_type: TraitMetadataType,
    pub impl_from: String,
    pub impl_to: String,
    pub external_trait: bool,
    pub trait_source_code_metadata_id: MetadataId,
    pub impl_functions: Vec<TraitMetadataFunction>,
}

impl TraitMetadata {
    pub fn new(
        metadata_id: MetadataId,
        name: String,
        trait_source_code_metadata_id: MetadataId,
        impl_functions: Vec<TraitMetadataFunction>,
        trait_type: TraitMetadataType,
        external_trait: bool,
        impl_from: String,
        impl_to: String,
    ) -> Self {
        Self {
            metadata_id,
            name,
            trait_source_code_metadata_id,
            impl_functions,
            trait_type,
            impl_from,
            impl_to,
            external_trait,
        }
    }

    pub fn to_trait_parser(&self) -> MetadataResult<TraitParser> {
        let trait_sc_metadata = BatMetadata::read_metadata()?
            .source_code
            .get_trait_by_id(self.trait_source_code_metadata_id.clone())?;
        TraitParser::new_from_metadata(trait_sc_metadata).change_context(MetadataError)
    }

    pub fn update_metadata_file(&self) -> MetadataResult<()> {
        let self_clone = self.clone();
        BatMetadata::update_metadata(|bat_metadata| {
            let position = bat_metadata
                .traits
                .iter()
                .position(|ep| {
                    ep.trait_source_code_metadata_id == self_clone.trait_source_code_metadata_id
                });
            match position {
                None => bat_metadata.traits.push(self_clone.clone()),
                Some(pos) => bat_metadata.traits[pos] = self_clone.clone(),
            };
            bat_metadata
                .traits
                .sort_by_key(|trait_meta| trait_meta.name.clone());
        })
    }
}