vize_atelier_core 0.167.0

Atelier Core - The core workshop for Vize Vue template parsing and transforms
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
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

//! v-if directive transform.
//!
//! Transforms elements with v-if, v-else-if, and v-else directives into IfNode.

use crate::ast::*;
use crate::transform::TransformContext;

/// Check if an element has a v-if directive
pub fn has_v_if(el: &ElementNode<'_>) -> bool {
    el.props
        .iter()
        .any(|prop| matches!(prop, PropNode::Directive(dir) if dir.name == "if"))
}

/// Check if an element has a v-else-if directive
pub fn has_v_else_if(el: &ElementNode<'_>) -> bool {
    el.props
        .iter()
        .any(|prop| matches!(prop, PropNode::Directive(dir) if dir.name == "else-if"))
}

/// Check if an element has a v-else directive
pub fn has_v_else(el: &ElementNode<'_>) -> bool {
    el.props
        .iter()
        .any(|prop| matches!(prop, PropNode::Directive(dir) if dir.name == "else"))
}

/// Get the v-if/v-else-if expression from an element
pub fn get_if_condition<'a>(el: &'a ElementNode<'a>) -> Option<&'a ExpressionNode<'a>> {
    for prop in el.props.iter() {
        if let PropNode::Directive(dir) = prop
            && (dir.name == "if" || dir.name == "else-if")
        {
            return dir.exp.as_ref();
        }
    }
    None
}

/// Remove v-if/v-else-if/v-else directive from element props
pub fn remove_if_directive(el: &mut ElementNode<'_>) {
    let mut i = 0;
    while i < el.props.len() {
        if let PropNode::Directive(dir) = &el.props[i]
            && (dir.name == "if" || dir.name == "else-if" || dir.name == "else")
        {
            el.props.remove(i);
            return;
        }
        i += 1;
    }
}

/// Process v-if structural directive - adds helpers
pub fn process_v_if(ctx: &mut TransformContext<'_>) {
    ctx.helper(RuntimeHelper::OpenBlock);
    ctx.helper(RuntimeHelper::CreateBlock);
    ctx.helper(RuntimeHelper::CreateElementBlock);
    ctx.helper(RuntimeHelper::Fragment);
    ctx.helper(RuntimeHelper::CreateComment);
}

#[cfg(test)]
mod tests {
    use super::{TemplateChildNode, has_v_else, has_v_else_if, has_v_if};
    use crate::parser::parse;
    use bumpalo::Bump;

    #[test]
    fn test_has_v_if() {
        let allocator = Bump::new();
        let (root, _) = parse(&allocator, r#"<div v-if="show">test</div>"#);

        if let TemplateChildNode::Element(el) = &root.children[0] {
            assert!(has_v_if(el));
            assert!(!has_v_else(el));
            assert!(!has_v_else_if(el));
        }
    }

    #[test]
    fn test_has_v_else() {
        let allocator = Bump::new();
        let (root, _) = parse(&allocator, r#"<div v-else>test</div>"#);

        if let TemplateChildNode::Element(el) = &root.children[0] {
            assert!(has_v_else(el));
            assert!(!has_v_if(el));
        }
    }
}