dan/oxc
1
0
Fork 0
mirror of https://github.com/danbulant/oxc synced 2026-05-24 20:32:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat(printer) Print ArrayExpression (#1279)

Browse source
This commit is contained in:
Cameron 2023-11-14 01:45:26 +00:00 committed by GitHub
parent 41e145b4af
commit 232b93ba41
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 38 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
crates/oxc_prettier/src/doc.rs
View file

@ -19,6 +19,7 @@ pub enum Doc<'a> {
Line,
Softline,
Hardline,
IfBreak(Vec<'a, Doc<'a>>, Vec<'a, Doc<'a>>),
}
/// Doc Builder

36
crates/oxc_prettier/src/format.rs
View file

@ -601,7 +601,11 @@ impl<'a> Format<'a> for Argument<'a> {
impl<'a> Format<'a> for ArrayExpressionElement<'a> {
fn format(&self, p: &mut Prettier<'a>) -> Doc<'a> {
Doc::Line
match self {
Self::SpreadElement(expr) => expr.format(p),
Self::Expression(expr) => expr.format(p),
Self::Elision(elision) => Doc::Str(""),
}
}
}
@ -613,7 +617,35 @@ impl<'a> Format<'a> for SpreadElement<'a> {
impl<'a> Format<'a> for ArrayExpression<'a> {
fn format(&self, p: &mut Prettier<'a>) -> Doc<'a> {
Doc::Line
if self.elements.len() == 0 {
return p.str("[]");
}
let mut parts = p.vec();
parts.push(p.str("["));
let mut parts_inner = p.vec();
parts_inner.push(Doc::Softline);
for (i, element) in self.elements.iter().enumerate() {
if i > 0 {
parts_inner.push(string!(p, ","));
parts_inner.push(Doc::Softline);
}
parts_inner.push(format!(p, element));
}
let mut if_break_comma = p.vec();
if_break_comma.push(Doc::Str(","));
parts_inner.push(Doc::IfBreak(if_break_comma, p.vec()));
parts.push(group!(p, Doc::Indent(parts_inner)));
parts.push(Doc::Softline);
parts.push(p.str("]"));
Doc::Group(parts)
}
}

3
crates/oxc_prettier/src/printer.rs
View file

@ -77,6 +77,9 @@ impl<'a> Printer<'a> {
Doc::Line | Doc::Softline | Doc::Hardline => {
out.push(b'\n');
}
Doc::IfBreak(docs, _) => {
cmds.extend(docs.into_iter().rev().map(|doc| Command::new(indent, mode, doc)));
}
}
}