dprint-plugin-pug 0.1.0

A super minimal Pug formatter plugin for dprint.
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

use crate::ast::{Document, Node, RawTextNode, StatementNode};
use crate::config::Configuration;

pub fn format(document: &Document, config: &Configuration) -> String {
    let mut output = String::new();

    for (index, node) in document.children.iter().enumerate() {
        if index > 0 {
            output.push('\n');
        }
        write_node(
            &mut output,
            node,
            0,
            config.indent_width(),
            config.use_tabs(),
        );
    }

    if !output.ends_with('\n') {
        output.push('\n');
    }

    output
}

fn write_node(output: &mut String, node: &Node, depth: usize, indent_width: usize, use_tabs: bool) {
    match node {
        Node::Statement(statement) => {
            write_statement(output, statement, depth, indent_width, use_tabs)
        }
        Node::Comment(text) => {
            write_indent(output, depth, indent_width, use_tabs);
            output.push_str("// ");
            output.push_str(text.trim());
        }
        Node::Text(text) => {
            write_indent(output, depth, indent_width, use_tabs);
            output.push('|');
            output.push_str(text);
        }
        Node::RawText(text) => write_raw_text(output, text, depth, indent_width, use_tabs),
    }
}

fn write_statement(
    output: &mut String,
    element: &StatementNode,
    depth: usize,
    indent_width: usize,
    use_tabs: bool,
) {
    write_indent(output, depth, indent_width, use_tabs);
    output.push_str(&element.head.to_source());
    if element.is_text_block {
        output.push('.');
    }

    for child in &element.children {
        output.push('\n');
        write_node(output, child, depth + 1, indent_width, use_tabs);
    }
}

fn write_raw_text(
    output: &mut String,
    text: &RawTextNode,
    depth: usize,
    indent_width: usize,
    use_tabs: bool,
) {
    write_indent(output, depth, indent_width, use_tabs);
    for _ in 0..text.extra_indent {
        output.push(' ');
    }
    output.push_str(&text.content);
}

fn write_indent(output: &mut String, depth: usize, indent_width: usize, use_tabs: bool) {
    if depth == 0 {
        return;
    }

    if use_tabs {
        for _ in 0..depth {
            output.push('\t');
        }
        return;
    }

    if indent_width == 0 {
        return;
    }

    for _ in 0..depth {
        for _ in 0..indent_width {
            output.push(' ');
        }
    }
}