codegraph-parser-api 0.2.1

Shared API and types for CodeGraph language parsers
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

use super::function::FunctionEntity;
use serde::{Deserialize, Serialize};

/// Represents a trait/protocol/interface definition
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
pub struct TraitEntity {
    /// Trait name
    pub name: String,

    /// Visibility
    pub visibility: String,

    /// Starting line number
    pub line_start: usize,

    /// Ending line number
    pub line_end: usize,

    /// Required methods
    pub required_methods: Vec<FunctionEntity>,

    /// Parent traits (trait inheritance)
    pub parent_traits: Vec<String>,

    /// Documentation
    pub doc_comment: Option<String>,

    /// Attributes/decorators
    pub attributes: Vec<String>,
}

impl TraitEntity {
    pub fn new(name: impl Into<String>, line_start: usize, line_end: usize) -> Self {
        Self {
            name: name.into(),
            visibility: "public".to_string(),
            line_start,
            line_end,
            required_methods: Vec::new(),
            parent_traits: Vec::new(),
            doc_comment: None,
            attributes: Vec::new(),
        }
    }

    pub fn with_visibility(mut self, vis: impl Into<String>) -> Self {
        self.visibility = vis.into();
        self
    }

    pub fn with_methods(mut self, methods: Vec<FunctionEntity>) -> Self {
        self.required_methods = methods;
        self
    }

    pub fn with_parent_traits(mut self, parents: Vec<String>) -> Self {
        self.parent_traits = parents;
        self
    }

    pub fn with_doc(mut self, doc: impl Into<String>) -> Self {
        self.doc_comment = Some(doc.into());
        self
    }

    pub fn with_attributes(mut self, attrs: Vec<String>) -> Self {
        self.attributes = attrs;
        self
    }
}