forked from mirror/vulkan-zig
93 lines
2.8 KiB
Zig
93 lines
2.8 KiB
Zig
const std = @import("std");
|
|
const unicode = std.unicode;
|
|
const testing = std.testing;
|
|
|
|
const Entity = struct {
|
|
substitute: []const u8,
|
|
char: u8
|
|
};
|
|
|
|
const entities = [_]Entity{
|
|
Entity{.substitute = "amp;", .char = '&'},
|
|
Entity{.substitute = "lt;", .char = '<'},
|
|
Entity{.substitute = "gt;", .char = '>'},
|
|
Entity{.substitute = "quot;", .char = '"'},
|
|
Entity{.substitute = "apos;", .char = '\''},
|
|
};
|
|
|
|
fn XmlUtf8Parser(comptime ReadError: type) type {
|
|
return struct {
|
|
const Self = @This();
|
|
pub const Stream = std.io.InStream(ReadError);
|
|
|
|
in: *Stream,
|
|
|
|
pub fn init(in: *Stream) Self {
|
|
return Self{
|
|
.in = in
|
|
};
|
|
}
|
|
|
|
pub fn next(self: *Self) !?u32 {
|
|
const cp = (try self.nextCodepoint()) orelse return null;
|
|
return if (cp == '&') try self.nextEntity() else cp;
|
|
}
|
|
|
|
fn nextEntity(self: *Self) !u32 {
|
|
var entity = [_]u8{0} ** 5;
|
|
|
|
for (entity) |*c| {
|
|
const cp = (try self.nextCodepoint()) orelse return error.InvalidEntity;
|
|
c.* = std.math.cast(u8, cp) catch return error.InvalidEntity;
|
|
|
|
if (cp == ';') {
|
|
break;
|
|
}
|
|
} else {
|
|
return error.InvalidEntity;
|
|
}
|
|
|
|
for (entities) |*e| {
|
|
if (std.mem.startsWith(u8, entity, e.substitute)) {
|
|
return e.char;
|
|
}
|
|
}
|
|
|
|
return error.InvalidEntity;
|
|
}
|
|
|
|
fn nextCodepoint(self: *Self) !?u32 {
|
|
var cp: [4]u8 = undefined;
|
|
if ((try self.in.readFull(cp[0 .. 1])) != 1) {
|
|
return null;
|
|
}
|
|
|
|
const cp_len = try unicode.utf8ByteSequenceLength(cp[0]);
|
|
try self.in.readNoEof(cp[1 .. cp_len]);
|
|
return try unicode.utf8Decode(cp[0 .. cp_len]);
|
|
}
|
|
};
|
|
}
|
|
|
|
fn testXmlUtf8Parser(text: []const u8, expected: []const u8) !void {
|
|
var slice_in = std.io.SliceInStream.init(text);
|
|
var p = XmlUtf8Parser(std.io.SliceInStream.Error).init(&slice_in.stream);
|
|
|
|
var i: usize = 0;
|
|
defer testing.expect(i == expected.len);
|
|
|
|
while (try p.next()) |cp| {
|
|
testing.expect(cp == expected[i]);
|
|
i += 1;
|
|
}
|
|
}
|
|
|
|
test "XmlUtf8Parser" {
|
|
try testXmlUtf8Parser("simpleabc", "simpleabc");
|
|
try testXmlUtf8Parser("a<b&c>d"e'f<", "a<b&c>d\"e'f<");
|
|
testing.expectError(error.InvalidEntity, testXmlUtf8Parser("python&", "python"));
|
|
testing.expectError(error.InvalidEntity, testXmlUtf8Parser("python&&", "python"));
|
|
testing.expectError(error.InvalidEntity, testXmlUtf8Parser("python&test;", "python"));
|
|
testing.expectError(error.InvalidEntity, testXmlUtf8Parser("python&boa", "python"));
|
|
}
|