aboutsummaryrefslogtreecommitdiff
path: root/src/string_builder.zig
diff options
context:
space:
mode:
Diffstat (limited to 'src/string_builder.zig')
-rw-r--r--src/string_builder.zig57
1 files changed, 57 insertions, 0 deletions
diff --git a/src/string_builder.zig b/src/string_builder.zig
index abed901dd..7aba5cd89 100644
--- a/src/string_builder.zig
+++ b/src/string_builder.zig
@@ -53,6 +53,36 @@ pub fn append(this: *StringBuilder, slice: string) string {
return result;
}
+pub fn add(this: *StringBuilder, len: usize) bun.StringPointer {
+ if (comptime Environment.allow_assert) {
+ assert(this.len <= this.cap); // didn't count everything
+ assert(this.ptr != null); // must call allocate first
+ }
+
+ const start = this.len;
+ this.len += len;
+
+ if (comptime Environment.allow_assert) assert(this.len <= this.cap);
+
+ return bun.StringPointer{ .offset = @truncate(u32, start), .length = @truncate(u32, len) };
+}
+pub fn appendCount(this: *StringBuilder, slice: string) bun.StringPointer {
+ if (comptime Environment.allow_assert) {
+ assert(this.len <= this.cap); // didn't count everything
+ assert(this.ptr != null); // must call allocate first
+ }
+
+ const start = this.len;
+ bun.copy(u8, this.ptr.?[this.len..this.cap], slice);
+ const result = this.ptr.?[this.len..this.cap][0..slice.len];
+ _ = result;
+ this.len += slice.len;
+
+ if (comptime Environment.allow_assert) assert(this.len <= this.cap);
+
+ return bun.StringPointer{ .offset = @truncate(u32, start), .length = @truncate(u32, slice.len) };
+}
+
pub fn fmt(this: *StringBuilder, comptime str: string, args: anytype) string {
if (comptime Environment.allow_assert) {
assert(this.len <= this.cap); // didn't count everything
@@ -68,6 +98,25 @@ pub fn fmt(this: *StringBuilder, comptime str: string, args: anytype) string {
return out;
}
+pub fn fmtAppendCount(this: *StringBuilder, comptime str: string, args: anytype) bun.StringPointer {
+ if (comptime Environment.allow_assert) {
+ assert(this.len <= this.cap); // didn't count everything
+ assert(this.ptr != null); // must call allocate first
+ }
+
+ var buf = this.ptr.?[this.len..this.cap];
+ const out = std.fmt.bufPrint(buf, str, args) catch unreachable;
+ const off = this.len;
+ this.len += out.len;
+
+ if (comptime Environment.allow_assert) assert(this.len <= this.cap);
+
+ return bun.StringPointer{
+ .offset = @truncate(u32, off),
+ .length = @truncate(u32, out.len),
+ };
+}
+
pub fn fmtCount(this: *StringBuilder, comptime str: string, args: anytype) void {
this.cap += std.fmt.count(str, args);
}
@@ -79,3 +128,11 @@ pub fn allocatedSlice(this: *StringBuilder) []u8 {
}
return ptr[0..this.cap];
}
+
+pub fn writable(this: *StringBuilder) []u8 {
+ var ptr = this.ptr orelse return &[_]u8{};
+ if (comptime Environment.allow_assert) {
+ assert(this.cap > 0);
+ }
+ return ptr[this.len..this.cap];
+}