relay-knowledge 1.1.16

Graph-database-based knowledge graph project.
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

//! Static Python string parsing shared by route-framework detectors.

pub(in crate::code::parser) fn extract_quoted_string_python(source: &str) -> Option<String> {
    let source = source.trim_start();
    let string_start = python_static_string_start(source)?;
    let source = &source[string_start..];
    let quote_char = source.chars().next()?;
    if quote_char != '\'' && quote_char != '"' {
        return None;
    }
    let inner = &source[1..];
    let mut result = String::new();
    let mut escaped = false;
    for character in inner.chars() {
        if escaped {
            result.push(character);
            escaped = false;
            continue;
        }
        if character == '\\' {
            escaped = true;
            continue;
        }
        if character == quote_char {
            return Some(result);
        }
        result.push(character);
    }
    Some(result)
}

fn python_static_string_start(source: &str) -> Option<usize> {
    let quote_index = source.find(['\'', '"'])?;
    let prefix = &source[..quote_index];
    if prefix
        .chars()
        .all(|character| matches!(character.to_ascii_lowercase(), 'r' | 'u' | 'b'))
    {
        Some(quote_index)
    } else {
        None
    }
}

#[cfg(test)]
#[path = "python_strings_tests.rs"]
mod tests;