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

use uuid::Uuid;

use crate::core::parser::markers::PositionMarker;
use crate::core::parser::segments::base::Segment;
use crate::core::parser::segments::fix::SourceFix;

#[derive(Hash, Debug, Clone, PartialEq)]
pub struct RawSegment {
    raw: Option<String>,
    position_marker: Option<PositionMarker>,

    // From BaseSegment
    uuid: Uuid,
}

impl RawSegment {
    pub fn new(
        raw: Option<String>,
        position_marker: Option<PositionMarker>,
        // For legacy and syntactic sugar we allow the simple
        // `type` argument here, but for more precise inheritance
        // we suggest using the `instance_types` option.
        _type: Option<String>,
        _instance_types: Option<Vec<String>>,
        _trim_start: Option<Vec<String>>,
        _trim_cars: Option<Vec<String>>,
        _source_fixes: Option<Vec<SourceFix>>,
        _uuid: Option<String>,
    ) -> Self {
        Self { position_marker, raw, uuid: uuid::Uuid::new_v4() }
    }
}

impl Segment for RawSegment {
    fn get_type(&self) -> &'static str {
        "raw"
    }

    fn is_code(&self) -> bool {
        true
    }

    fn is_comment(&self) -> bool {
        false
    }

    fn is_whitespace(&self) -> bool {
        false
    }

    fn get_raw(&self) -> Option<String> {
        self.raw.clone()
    }

    fn get_position_marker(&self) -> Option<PositionMarker> {
        self.position_marker.clone()
    }

    fn get_raw_segments(&self) -> Vec<Box<dyn Segment>> {
        vec![Box::new(self.clone())]
    }

    fn get_uuid(&self) -> Option<Uuid> {
        Some(self.uuid)
    }

    fn edit(
        &self,
        _raw: Option<String>,
        _source_fixes: Option<Vec<SourceFix>>,
    ) -> Box<dyn Segment> {
        todo!()
    }

    fn set_position_marker(&mut self, _position_marker: Option<PositionMarker>) {
        todo!()
    }
}

#[cfg(test)]
mod test {
    // Test niche case of calling get_raw_segments on a raw segment.
    // TODO Implement
    // #[test]
    // fn test__parser__raw_get_raw_segments() {
    //     let segs = raw_segments();
    //
    //     for seg in segs {
    //         assert_eq!(seg.get_raw_segments(), Some(vec![seg.clone()]));
    //     }
    // }
}