如何让Wadler / Leijen漂亮的打印机在单独的行上渲染大括号

Question

我正在使用wl-pprint-extras包。我试图打印精美的LaTeX，改进HaTeX，或者至少根据我的需要进行调整。

我正在寻找的输出示例是：

\foo{bar}{
    \baz{\quux}
}{
    \fizz
}

和

\foo{
    bar
}{
    \baz{\quux}
}{
    \fizz
}

我希望尽可能避免

\foo{
    bar
}{\baz{\quux}}{
    \fizz
}

将我推向墙的输出是

\foo{
    bar
}{\baz{
        \quux
}}{
    \fizz
}

我必须处理的数据是TeXComm String [TeXArg]。

有没有人知道如何在完全避开第四个的同时获得第一个和第二个？

编辑：

迄今为止我所拥有的gist;这两个文件是经过修改的漂亮打印机和一些样本LaTeX我试图打印漂亮。