codeviz_js 0.2.2

Flexible code generator for Rust
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
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

use super::*;
use codeviz_common::VariableFormat;
use codeviz_common::ElementFormat;

/// Variables that are part of statements.
#[derive(Debug, Clone)]
pub enum Variable {
    /// String that will be literally appended.
    Literal(String),
    /// String that will be quoted and appended.
    String(String),
    /// Another statement that will be appended.
    Statement(Statement),
    /// A name that will be appended.
    Name(Name),
}

impl VariableFormat for Variable {
    type Extra = ();

    fn format<E>(&self, out: &mut E, depth: usize, extra: &mut ()) -> Result<()>
    where
        E: ElementFormat,
    {
        match *self {
            Variable::String(ref string) => {
                quote_string(out, string)?;
            }
            Variable::Statement(ref stmt) => {
                stmt.format(out, depth, extra)?;
            }
            Variable::Literal(ref content) => {
                out.write_str(content)?;
            }
            Variable::Name(ref name) => {
                name.format(out)?;
            }
        }

        Ok(())
    }
}

impl<'a, A> From<&'a A> for Variable
where
    A: Into<Variable> + Clone,
{
    fn from(value: &'a A) -> Variable {
        value.clone().into()
    }
}

impl<'a> From<&'a str> for Variable {
    fn from(value: &'a str) -> Variable {
        Variable::Literal(value.to_owned())
    }
}

impl From<String> for Variable {
    fn from(value: String) -> Variable {
        Variable::Literal(value)
    }
}

impl From<Statement> for Variable {
    fn from(value: Statement) -> Variable {
        Variable::Statement(value)
    }
}

impl From<Name> for Variable {
    fn from(value: Name) -> Variable {
        Variable::Name(value)
    }
}

impl From<ImportedName> for Variable {
    fn from(value: ImportedName) -> Variable {
        Variable::Name(value.into())
    }
}

impl From<BuiltInName> for Variable {
    fn from(value: BuiltInName) -> Variable {
        Variable::Name(value.into())
    }
}

impl From<LocalName> for Variable {
    fn from(value: LocalName) -> Variable {
        Variable::Name(value.into())
    }
}

/// Quote a string to make it suitable as a literal JavaScript string.
pub fn quote_string<E>(out: &mut E, input: &str) -> Result<()>
where
    E: ElementFormat,
{
    out.write_char('"')?;

    for c in input.chars() {
        match c {
            '\t' => out.write_str("\\t"),
            '\u{0007}' => out.write_str("\\b"),
            '\n' => out.write_str("\\n"),
            '\r' => out.write_str("\\r"),
            '\u{0014}' => out.write_str("\\f"),
            '\'' => out.write_str("\\'"),
            '"' => out.write_str("\\\""),
            '\\' => out.write_str("\\\\"),
            c => out.write_char(c),
        }?;
    }

    out.write_char('"')?;

    Ok(())
}