42 lines
1.2 KiB
Rust
42 lines
1.2 KiB
Rust
use pretty::{DocAllocator, DocBuilder, Pretty};
|
|
|
|
use crate::ast::{TableConstr, TableConstrElem};
|
|
|
|
use super::NEST_DEPTH;
|
|
|
|
impl<'a, D: DocAllocator<'a>> Pretty<'a, D> for TableConstrElem {
|
|
fn pretty(self, allocator: &'a D) -> DocBuilder<'a, D> {
|
|
match self {
|
|
Self::Lit(lit) => lit.pretty(allocator),
|
|
Self::Indexed {
|
|
s0,
|
|
index,
|
|
s1,
|
|
s2,
|
|
s3,
|
|
value,
|
|
span: _,
|
|
} => index
|
|
.pretty(allocator)
|
|
.brackets()
|
|
.append(allocator.text(": "))
|
|
.append(value.pretty(allocator)),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl<'a, D: DocAllocator<'a>> Pretty<'a, D> for TableConstr {
|
|
fn pretty(self, allocator: &'a D) -> DocBuilder<'a, D> {
|
|
self.elems
|
|
.pretty(
|
|
allocator,
|
|
|e| allocator.line().append(e.pretty(allocator)),
|
|
|(s0, s1)| allocator.text(","),
|
|
|s| allocator.text(","),
|
|
)
|
|
.nest(NEST_DEPTH)
|
|
.append(allocator.line())
|
|
.braces()
|
|
.group()
|
|
}
|
|
}
|