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

use log::trace;

use crate::{env::Env, library::TypeId};

pub fn rustify_pointers(c_type: &str) -> (String, String) {
    let mut input = c_type.trim();
    let leading_const = input.starts_with("const ");
    if leading_const {
        input = &input[6..];
    }
    let end = [
        input.find(" const"),
        input.find("*const"),
        input.find('*'),
        Some(input.len()),
    ]
    .iter()
    .filter_map(|&x| x)
    .min()
    .unwrap();
    let inner = input[..end].trim().into();

    let mut ptrs: Vec<_> = input[end..]
        .rsplit('*')
        .skip(1)
        .map(|s| {
            if s.contains("const") {
                "*const"
            } else {
                "*mut"
            }
        })
        .collect();
    if let (true, Some(p)) = (leading_const, ptrs.last_mut()) {
        *p = "*const";
    }

    let res = (ptrs.join(" "), inner);
    trace!("rustify `{}` -> `{}` `{}`", c_type, res.0, res.1);
    res
}

pub fn is_mut_ptr(c_type: &str) -> bool {
    let (ptr, _inner) = rustify_pointers(c_type);
    ptr.find("*mut") == Some(0)
}

pub fn implements_c_type(env: &Env, tid: TypeId, c_type: &str) -> bool {
    env.class_hierarchy
        .supertypes(tid)
        .iter()
        .any(|&super_tid| env.library.type_(super_tid).get_glib_name() == Some(c_type))
}

#[cfg(test)]
mod tests {
    use super::rustify_pointers as rustify_ptr;

    fn s(x: &str, y: &str) -> (String, String) {
        (x.into(), y.into())
    }

    #[test]
    fn rustify_pointers() {
        assert_eq!(rustify_ptr("char"), s("", "char"));
        assert_eq!(rustify_ptr("char*"), s("*mut", "char"));
        assert_eq!(rustify_ptr("const char*"), s("*const", "char"));
        assert_eq!(rustify_ptr("char const*"), s("*const", "char"));
        assert_eq!(rustify_ptr("char const *"), s("*const", "char"));
        assert_eq!(rustify_ptr(" char * * "), s("*mut *mut", "char"));
        assert_eq!(rustify_ptr("const char**"), s("*mut *const", "char"));
        assert_eq!(rustify_ptr("char const**"), s("*mut *const", "char"));
        assert_eq!(
            rustify_ptr("const char* const*"),
            s("*const *const", "char")
        );
        assert_eq!(
            rustify_ptr("char const * const *"),
            s("*const *const", "char")
        );
        assert_eq!(rustify_ptr("char* const*"), s("*const *mut", "char"));

        assert_eq!(rustify_ptr("GtkWidget*"), s("*mut", "GtkWidget"));
    }
}