Garmelon/tada
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Simplify pretty printing separated elements

Browse source
This commit is contained in:
Joscha 2022-11-21 00:20:33 +01:00
parent 9d6cd580d4
commit 6eee1ba930
5 changed files with 24 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

20
src/pretty/basic.rs
View file

@ -18,9 +18,9 @@ impl<E, S1, S2> Separated<E, S1, S2> {
) -> DocBuilder<'a, D>
where
D: DocAllocator<'a>,
FE: Fn(&'a D, E) -> DocBuilder<'a, D>,
FS1: Fn(&'a D, S1) -> DocBuilder<'a, D>,
FS2: Fn(&'a D, S2) -> DocBuilder<'a, D>,
FE: Fn(E) -> DocBuilder<'a, D>,
FS1: Fn(S1) -> DocBuilder<'a, D>,
FS2: Fn(S2) -> DocBuilder<'a, D>,
{
match self {
Self::Empty(_) => allocator.nil(),
@ -29,13 +29,17 @@ impl<E, S1, S2> Separated<E, S1, S2> {
last_elems,
trailing,
span: _span,
} => elem_to_doc(allocator, first_elem)
.append(allocator.concat(last_elems.into_iter().map(|(s, e)| {
separator_to_doc(allocator, s).append(elem_to_doc(allocator, e))
})))
} => elem_to_doc(first_elem)
.append(
allocator.concat(
last_elems
.into_iter()
.map(|(s, e)| separator_to_doc(s).append(elem_to_doc(e))),
),
)
.append(
trailing
.map(|s| trailing_separator_to_doc(allocator, s))
.map(trailing_separator_to_doc)
.unwrap_or_else(|| allocator.nil()),
),
}

6
src/pretty/lit.rs
View file

@ -55,9 +55,9 @@ impl<'a, D: DocAllocator<'a>> Pretty<'a, D> for TableLit {
self.elems
.pretty(
allocator,
|a, e| a.line().append(e.pretty(a)),
|a, (s0, s1)| a.text(","),
|a, s| a.text(","),
|e| allocator.line().append(e.pretty(allocator)),
|(s0, s1)| allocator.text(","),
|s| allocator.text(","),
)
.nest(NEST_DEPTH)
.append(allocator.line())

6
src/pretty/program.rs
View file

@ -23,9 +23,9 @@ impl<'a, D: DocAllocator<'a>> Pretty<'a, D> for Program {
.append(allocator.line())
.append(elems.pretty(
allocator,
|a, e| e.pretty(a),
|a, (s0, s1)| a.text(",").append(a.line()),
|a, s| a.text(","),
|e| e.pretty(allocator),
|(s0, s1)| allocator.text(",").append(allocator.line()),
|s| allocator.text(","),
)),
}
}

6
src/pretty/table_constr.rs
View file

@ -30,9 +30,9 @@ impl<'a, D: DocAllocator<'a>> Pretty<'a, D> for TableConstr {
self.elems
.pretty(
allocator,
|a, e| a.line().append(e.pretty(a)),
|a, (s0, s1)| a.text(","),
|a, s| a.text(","),
|e| allocator.line().append(e.pretty(allocator)),
|(s0, s1)| allocator.text(","),
|s| allocator.text(","),
)
.nest(NEST_DEPTH)
.append(allocator.line())

6
src/pretty/table_destr.rs
View file

@ -27,9 +27,9 @@ impl<'a, D: DocAllocator<'a>> Pretty<'a, D> for TablePattern {
self.elems
.pretty(
allocator,
|a, e| a.line().append(e.pretty(a)),
|a, (s0, s1)| a.text(","),
|a, s| a.text(","),
|e| allocator.line().append(e.pretty(allocator)),
|(s0, s1)| allocator.text(","),
|s| allocator.text(","),
)
.nest(NEST_DEPTH)
.append(allocator.line())