use ruff_formatter::write;
use ruff_python_ast::ModModule;
use ruff_python_trivia::lines_after;
use crate::FormatNodeRule;
use crate::prelude::*;
use crate::statement::suite::SuiteKind;
#[derive(Default)]
pub struct FormatModModule;
impl FormatNodeRule<ModModule> for FormatModModule {
fn fmt_fields(&self, item: &ModModule, f: &mut PyFormatter) -> FormatResult<()> {
let ModModule {
range,
body,
node_index: _,
} = item;
if body.is_empty() {
if !f.context().comments().has_leading(item)
&& lines_after(range.start(), f.context().source()) != 0
{
empty_line().fmt(f)
} else {
Ok(())
}
} else {
write!(
f,
[
body.format().with_options(SuiteKind::TopLevel),
hard_line_break()
]
)
}
}
}