summaryrefslogtreecommitdiff
path: root/systems/sheet/src/mapping/parse.rs
blob: e203c9653bbec17c49ce858af8a899956d66d721 (plain)
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
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
use just_fmt::fmt_path::fmt_path_str;

use crate::{
    index_source::IndexSource,
    mapping::{LocalMapping, LocalMappingForward, error::ParseMappingError},
};

impl TryFrom<&str> for LocalMapping {
    type Error = ParseMappingError;

    fn try_from(s: &str) -> Result<Self, Self::Error> {
        // Remove surrounding quotes if present
        let s = s.trim_matches('"');

        // Helper function to remove quotes from a string
        fn remove_quotes(s: &str) -> String {
            // Simply remove all quotes from the string
            s.replace('"', "").trim().to_string()
        }

        // Helper function to split by operator, handling both spaced and non-spaced versions
        fn split_by_operator<'a>(s: &'a str, operator: &'a str) -> Vec<&'a str> {
            let mut result = Vec::new();
            let mut start = 0;

            // Find all occurrences of the operator
            let mut search_from = 0;
            while let Some(pos) = s[search_from..].find(operator) {
                let actual_pos = search_from + pos;
                result.push(&s[start..actual_pos]);
                start = actual_pos + operator.len();
                search_from = start;
            }

            if start < s.len() {
                result.push(&s[start..]);
            }

            result
        }

        // Helper function to find operator position
        fn find_operator<'a>(s: &'a str, operator: &'a str) -> Option<usize> {
            s.find(operator)
        }

        // Try to parse "path" => "source" == "version" pattern
        if let Some(arrow_pos) = find_operator(s, "=>") {
            let after_arrow = &s[arrow_pos + 2..];
            if let Some(equal_pos) = find_operator(after_arrow, "==") {
                // Format: "path" => "source" == "version"
                let path = remove_quotes(s[..arrow_pos].trim());
                let middle_part = after_arrow[..equal_pos].trim();
                let version_part = after_arrow[equal_pos + 2..].trim();

                let middle = remove_quotes(middle_part);
                let version_part_str = remove_quotes(version_part);

                let val = fmt_path_str(path)
                    .map_err(|_| ParseMappingError::InvalidMapping)?
                    .split('/')
                    .map(|s| s.to_string())
                    .collect();

                let source = IndexSource::try_from(middle.as_str())
                    .map_err(|_| ParseMappingError::InvalidMapping)?;

                let version = version_part_str
                    .parse::<u16>()
                    .map_err(|_| ParseMappingError::InvalidMapping)?;

                return Ok(LocalMapping {
                    val,
                    source,
                    forward: LocalMappingForward::Version { version },
                });
            }
        }

        // Split by "=>" to parse the format
        let parts = split_by_operator(s, "=>");

        match parts.len() {
            1 => {
                // Check for "==" operator
                if let Some(equal_pos) = find_operator(s, "==") {
                    // Format: "path" == "source" (when mapped_version equals version)
                    let path_raw = s[..equal_pos].trim();
                    let source_part_raw = s[equal_pos + 2..].trim();
                    let path = remove_quotes(path_raw);
                    let source_str = remove_quotes(source_part_raw);

                    let val = fmt_path_str(path)
                        .map_err(|_| ParseMappingError::InvalidMapping)?
                        .split('/')
                        .map(|s| s.to_string())
                        .collect();

                    let source = IndexSource::try_from(source_str.as_str())
                        .map_err(|_| ParseMappingError::InvalidMapping)?;

                    let version = source.version();

                    Ok(LocalMapping {
                        val,
                        source,
                        forward: LocalMappingForward::Version { version },
                    })
                } else {
                    Err(ParseMappingError::InvalidMapping)
                }
            }
            2 => {
                // Check if the second part contains "=="
                if let Some(equal_pos) = find_operator(parts[1], "==") {
                    // Format: "path" => "source" == "version"
                    let path = remove_quotes(parts[0].trim());
                    let middle_part = parts[1][..equal_pos].trim();
                    let version_part = parts[1][equal_pos + 2..].trim();

                    let middle = remove_quotes(middle_part);
                    let version_part_str = remove_quotes(version_part);

                    let val = fmt_path_str(path)
                        .map_err(|_| ParseMappingError::InvalidMapping)?
                        .split('/')
                        .map(|s| s.to_string())
                        .collect();

                    let source = IndexSource::try_from(middle.as_str())
                        .map_err(|_| ParseMappingError::InvalidMapping)?;

                    let version = version_part_str
                        .parse::<u16>()
                        .map_err(|_| ParseMappingError::InvalidMapping)?;

                    return Ok(LocalMapping {
                        val,
                        source,
                        forward: LocalMappingForward::Version { version },
                    });
                }

                // Format: "path" => "source"
                let path = remove_quotes(parts[0].trim());
                let source_str = remove_quotes(parts[1].trim());

                let val = fmt_path_str(path)
                    .map_err(|_| ParseMappingError::InvalidMapping)?
                    .split('/')
                    .map(|s| s.to_string())
                    .collect();

                let source = IndexSource::try_from(source_str.as_str())
                    .map_err(|_| ParseMappingError::InvalidMapping)?;

                Ok(LocalMapping {
                    val,
                    source,
                    forward: LocalMappingForward::Latest,
                })
            }
            3 => {
                // Format: "path" => "source" => "sheet_name"
                let path = remove_quotes(parts[0].trim());
                let source_str = remove_quotes(parts[1].trim());
                let sheet_name = remove_quotes(parts[2].trim());

                let val = fmt_path_str(path)
                    .map_err(|_| ParseMappingError::InvalidMapping)?
                    .split('/')
                    .map(|s| s.to_string())
                    .collect();

                let source = IndexSource::try_from(source_str.as_str())
                    .map_err(|_| ParseMappingError::InvalidMapping)?;

                Ok(LocalMapping {
                    val,
                    source,
                    forward: LocalMappingForward::Ref {
                        sheet_name: sheet_name,
                    },
                })
            }
            _ => Err(ParseMappingError::InvalidMapping),
        }
    }
}