tiny.pretty.FlatFitWriter

1pub const FlatFitWriter = struct {2    writer: std.Io.Writer,3    limit: usize,4    count: usize = 0,56    pub fn init(limit: usize) FlatFitWriter {7        return .{8            .writer = .{9                .vtable = &.{ .drain = drain },10                .buffer = &.{},11            },12            .limit = limit,13        };14    }1516    fn drain(17        writer: *std.Io.Writer,18        data: []const []const u8,19        splat: usize,20    ) std.Io.Writer.Error!usize {21        const self: *FlatFitWriter = @alignCast(@fieldParentPtr("writer", writer));22        try self.accept(writer.buffered());23        writer.end = 0;2425        var consumed: usize = 0;26        for (data[0 .. data.len - 1]) |bytes| {27            try self.accept(bytes);28            consumed += bytes.len;29        }3031        const pattern = data[data.len - 1];32        for (0..splat) |_| {33            try self.accept(pattern);34            consumed += pattern.len;35        }36        return consumed;37    }3839    fn accept(self: *FlatFitWriter, bytes: []const u8) std.Io.Writer.Error!void {40        var budget = self.limit -| self.count;41        if (!position.consumeFlatBytes(bytes, &budget)) return error.WriteFailed;42        self.count = self.limit - budget;43    }44};

1pub const FlatFitWriter = render.FlatFitWriter;