tiny.pretty.Builder

1pub const Builder = struct {2    allocator: std.mem.Allocator,34    pub fn init(allocator: std.mem.Allocator) Builder {5        return .{ .allocator = allocator };6    }78    pub fn empty(_: Builder) Doc {9        return .empty;10    }1112    pub fn text(_: Builder, bytes: []const u8) Doc {13        if (bytes.len == 0) return .empty;14        return .{ .text = bytes };15    }1617    pub fn textFmt(18        self: Builder,19        comptime fmt: []const u8,20        args: anytype,21    ) std.mem.Allocator.Error!Doc {22        return .{ .text = try std.fmt.allocPrint(self.allocator, fmt, args) };23    }2425    pub fn spaces(26        self: Builder,27        amount: usize,28    ) std.mem.Allocator.Error!Doc {29        if (amount == 0) return self.empty();30        const bytes = try self.allocator.alloc(u8, amount);31        @memset(bytes, ' ');32        return self.text(bytes);33    }3435    pub fn concat(36        self: Builder,37        parts: []const Doc,38    ) std.mem.Allocator.Error!Doc {39        if (parts.len == 0) return .empty;40        if (parts.len == 1) return parts[0];41        return .{ .concat = try self.allocator.dupe(Doc, parts) };42    }4344    pub fn nest(45        self: Builder,46        amount: usize,47        doc: Doc,48    ) std.mem.Allocator.Error!Doc {49        if (amount == 0) return doc;50        const owned = try self.allocator.create(Doc);51        owned.* = doc;52        return .{ .nest = .{ .amount = amount, .doc = owned } };53    }5455    pub fn group(56        self: Builder,57        doc: Doc,58    ) std.mem.Allocator.Error!Doc {59        const owned = try self.allocator.create(Doc);60        owned.* = doc;61        return .{ .group = owned };62    }6364    pub fn style(65        self: Builder,66        style_kind: Style,67        doc: Doc,68    ) std.mem.Allocator.Error!Doc {69        if (style_kind == .plain) return doc;70        const owned = try self.allocator.create(Doc);71        owned.* = doc;72        return .{ .styled = .{ .style = style_kind, .doc = owned } };73    }7475    pub fn styledText(76        self: Builder,77        style_kind: Style,78        bytes: []const u8,79    ) std.mem.Allocator.Error!Doc {80        return try self.style(style_kind, self.text(bytes));81    }8283    pub fn styledFmt(84        self: Builder,85        style_kind: Style,86        comptime fmt: []const u8,87        args: anytype,88    ) std.mem.Allocator.Error!Doc {89        return try self.style(style_kind, try self.textFmt(fmt, args));90    }9192    pub fn words(93        self: Builder,94        bytes: []const u8,95    ) std.mem.Allocator.Error!Doc {96        var parts: std.ArrayListUnmanaged(Doc) = .empty;97        defer parts.deinit(self.allocator);9899        var line_start = true;100        var index: usize = 0;101        while (index < bytes.len) {102            while (index < bytes.len) : (index += 1) {103                switch (bytes[index]) {104                    ' ', '\t', '\r' => {},105                    '\n' => {106                        try parts.append(self.allocator, hardline);107                        line_start = true;108                    },109                    else => break,110                }111            }112113            if (index >= bytes.len) break;114115            const word_start = index;116            while (index < bytes.len and bytes[index] != ' ' and bytes[index] != '\t' and bytes[index] != '\r' and bytes[index] != '\n') : (index += 1) {}117            const word = self.text(bytes[word_start..index]);118119            if (line_start) {120                try parts.append(self.allocator, word);121                line_start = false;122            } else {123                try parts.append(self.allocator, try self.group(try self.concat(&.{124                    softline,125                    word,126                })));127            }128        }129130        return try self.concat(parts.items);131    }132133    pub fn styledWords(134        self: Builder,135        style_kind: Style,136        bytes: []const u8,137    ) std.mem.Allocator.Error!Doc {138        return try self.style(style_kind, try self.words(bytes));139    }140141    pub fn punct(142        self: Builder,143        bytes: []const u8,144    ) std.mem.Allocator.Error!Doc {145        return try self.styledText(.punctuation, bytes);146    }147148    pub fn join(149        self: Builder,150        items: []const Doc,151        separator: Doc,152    ) std.mem.Allocator.Error!Doc {153        if (items.len == 0) return self.empty();154        if (items.len == 1) return items[0];155156        const parts = try self.allocator.alloc(Doc, items.len * 2 - 1);157        var index: usize = 0;158        for (items, 0..) |item, item_index| {159            if (item_index != 0) {160                parts[index] = separator;161                index += 1;162            }163            parts[index] = item;164            index += 1;165        }166        return .{ .concat = parts };167    }168};

1pub const Builder = builder_product.Builder;