oak-jasm 0.0.11

JASM assembly language parser with support for modern assembly syntax and features.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48

//! Formatter implementation for the JASM language.

use oak_core::source::{SourceBuffer, ToSource};
#[cfg(feature = "oak-pretty-print")]
use oak_pretty_print::{Document, PrinterConfig};
#[cfg(feature = "oak-pretty-print")]
use oak_pretty_print::to_doc::AsDocument;

use crate::ast::JasmRoot;

/// Formatter for the JASM language.
pub struct JasmFormatter {
    /// Indentation size.
    indent_size: usize,
}

impl JasmFormatter {
    /// Creates a new `JasmFormatter` with default settings.
    pub fn new() -> Self {
        Self { indent_size: 4 }
    }

    /// Formats a JASM root node into a string.
    pub fn format(&self, root: &JasmRoot) -> String {
        let mut buffer = SourceBuffer::new();
        root.to_source(&mut buffer);
        buffer.to_string()
    }

    /// Formats a JASM root node using pretty-printing.
    #[cfg(feature = "oak-pretty-print")]
    pub fn format_pretty(&self, root: &JasmRoot) -> String {
        let doc = root.as_document(&Default::default());
        doc.render()
    }

    /// Formats a JASM root node using pretty-printing (fallback for no oak-pretty-print).
    #[cfg(not(feature = "oak-pretty-print"))]
    pub fn format_pretty(&self, root: &JasmRoot) -> String {
        self.format(root)
    }
}

impl Default for JasmFormatter {
    fn default() -> Self {
        Self::new()
    }
}