cs2-dumper/src/output/formatter.rs

use std::fmt::{self, Write};

pub struct Formatter<'a> {
    out: &'a mut String,
    indent_size: usize,
    indent_level: usize,
}

impl<'a> Formatter<'a> {
    pub fn new(out: &'a mut String, indent_size: usize) -> Self {
        Self {
            out,
            indent_size,
            indent_level: 0,
        }
    }

    pub fn block<F>(&mut self, heading: &str, semicolon: bool, f: F) -> fmt::Result
    where
        F: FnOnce(&mut Self) -> fmt::Result,
    {
        writeln!(self, "{} {{", heading)?;

        self.indent(f)?;

        writeln!(self, "{}", if semicolon { "};" } else { "}" })?;

        Ok(())
    }

    pub fn indent<F>(&mut self, f: F) -> fmt::Result
    where
        F: FnOnce(&mut Self) -> fmt::Result,
    {
        self.indent_level += 1;

        f(self)?;

        self.indent_level -= 1;

        Ok(())
    }

    fn push_indentation(&mut self) {
        if self.indent_level > 0 {
            let indentation = " ".repeat(self.indent_level * self.indent_size);

            self.out.push_str(&indentation);
        }
    }
}

impl<'a> Write for Formatter<'a> {
    fn write_str(&mut self, s: &str) -> fmt::Result {
        let mut lines = s.lines().peekable();

        while let Some(line) = lines.next() {
            if self.out.ends_with('\n') && !line.is_empty() {
                self.push_indentation();
            }

            self.out.push_str(line);

            if lines.peek().is_some() || s.ends_with('\n') {
                self.out.push('\n');
            }
        }

        Ok(())
    }
}