aboutsummaryrefslogtreecommitdiff
path: root/src/string_immutable.zig
diff options
context:
space:
mode:
Diffstat (limited to 'src/string_immutable.zig')
-rw-r--r--src/string_immutable.zig54
1 files changed, 54 insertions, 0 deletions
diff --git a/src/string_immutable.zig b/src/string_immutable.zig
index 15e0fc032..2527b6898 100644
--- a/src/string_immutable.zig
+++ b/src/string_immutable.zig
@@ -686,6 +686,25 @@ pub fn endsWithAny(self: string, str: string) bool {
return false;
}
+pub fn quotedWriter(writer: anytype, self: string) !void {
+ var remain = self;
+ if (strings.containsNewlineOrNonASCIIOrQuote(remain)) {
+ try bun.js_printer.writeJSONString(self, @TypeOf(writer), writer, strings.Encoding.utf8);
+ } else {
+ try writer.writeAll("\"");
+ try writer.writeAll(self);
+ try writer.writeAll("\"");
+ }
+}
+
+pub const QuotedFormatter = struct {
+ text: []const u8,
+
+ pub fn format(this: QuotedFormatter, comptime _: []const u8, _: std.fmt.FormatOptions, writer: anytype) !void {
+ try strings.quotedWriter(writer, this.text);
+ }
+};
+
pub fn quotedAlloc(allocator: std.mem.Allocator, self: string) !string {
var count: usize = 0;
for (self) |char| {
@@ -3214,6 +3233,41 @@ pub fn indexOfNewlineOrNonASCIICheckStart(slice_: []const u8, offset: u32, compt
return null;
}
+pub fn containsNewlineOrNonASCIIOrQuote(slice_: []const u8) bool {
+ const slice = slice_;
+ var remaining = slice;
+
+ if (remaining.len == 0)
+ return false;
+
+ if (comptime Environment.enableSIMD) {
+ while (remaining.len >= ascii_vector_size) {
+ const vec: AsciiVector = remaining[0..ascii_vector_size].*;
+ const cmp = @bitCast(AsciiVectorU1, (vec > max_16_ascii)) | @bitCast(AsciiVectorU1, (vec < min_16_ascii)) |
+ @bitCast(AsciiVectorU1, vec == @splat(ascii_vector_size, @as(u8, '\r'))) |
+ @bitCast(AsciiVectorU1, vec == @splat(ascii_vector_size, @as(u8, '\n'))) |
+ @bitCast(AsciiVectorU1, vec == @splat(ascii_vector_size, @as(u8, '"')));
+
+ if (@reduce(.Max, cmp) > 0) {
+ return true;
+ }
+
+ remaining = remaining[ascii_vector_size..];
+ }
+
+ if (comptime Environment.allow_assert) std.debug.assert(remaining.len < ascii_vector_size);
+ }
+
+ for (remaining) |*char_| {
+ const char = char_.*;
+ if (char > 127 or char < 0x20 or char == '\n' or char == '\r' or char == '"') {
+ return true;
+ }
+ }
+
+ return false;
+}
+
pub fn indexOfNeedsEscape(slice: []const u8) ?u32 {
var remaining = slice;
if (remaining.len == 0)