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

use proc_macro2::TokenStream;
use quote::quote;

pub trait LifetimeExt {
    fn is_transparent(&self) -> bool;
    fn needs_lifetime(&self) -> bool;
}

impl LifetimeExt for witx::TypeRef {
    fn is_transparent(&self) -> bool {
        self.type_().is_transparent()
    }
    fn needs_lifetime(&self) -> bool {
        self.type_().needs_lifetime()
    }
}

impl LifetimeExt for witx::Type {
    fn is_transparent(&self) -> bool {
        match self {
            witx::Type::Builtin(b) => b.is_transparent(),
            witx::Type::Record(s) => s.is_transparent(),
            witx::Type::Handle { .. } => true,
            witx::Type::Variant { .. }
            | witx::Type::Pointer { .. }
            | witx::Type::ConstPointer { .. }
            | witx::Type::List { .. } => false,
        }
    }
    fn needs_lifetime(&self) -> bool {
        match self {
            witx::Type::Builtin(b) => b.needs_lifetime(),
            witx::Type::Record(s) => s.needs_lifetime(),
            witx::Type::Variant(u) => u.needs_lifetime(),
            witx::Type::Handle { .. } => false,
            witx::Type::Pointer { .. }
            | witx::Type::ConstPointer { .. }
            | witx::Type::List { .. } => true,
        }
    }
}

impl LifetimeExt for witx::BuiltinType {
    fn is_transparent(&self) -> bool {
        true
    }
    fn needs_lifetime(&self) -> bool {
        false
    }
}

impl LifetimeExt for witx::RecordDatatype {
    fn is_transparent(&self) -> bool {
        self.members.iter().all(|m| m.tref.is_transparent())
    }
    fn needs_lifetime(&self) -> bool {
        self.members.iter().any(|m| m.tref.needs_lifetime())
    }
}

impl LifetimeExt for witx::Variant {
    fn is_transparent(&self) -> bool {
        false
    }
    fn needs_lifetime(&self) -> bool {
        self.cases
            .iter()
            .any(|m| m.tref.as_ref().map(|t| t.needs_lifetime()).unwrap_or(false))
    }
}

pub fn anon_lifetime() -> TokenStream {
    quote!('_)
}