rust-codegen 0.1.1

A simple builder API for generating Rust code
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

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

use crate::block::Block;
use crate::formatter::Formatter;

/// Defines the types of content that go in functions and blocks.
#[derive(Debug, Clone)]
pub enum Body {
    /// Used to push lines to blocks.
    String(String),
    /// Used to create blocks.
    Block(Block),
}

impl Body {
    /// Formats the string or block with the given formatter.
    /// 
    /// # Arguments
    /// 
    /// * `fmt` - The formatter to use.
    pub fn fmt(&self, fmt: &mut Formatter<'_>) -> fmt::Result {
        match &self {
            Body::String(s) => write!(fmt, "{}\n", s),
            Body::Block(b) => b.fmt(fmt),
        }
    }
}