HTMLCollection: add a matcher by name

This commit is contained in:
Pierre Tachoire
2024-01-04 11:50:47 +01:00
parent a4d44081e5
commit 0b952f5295

View File

@@ -12,6 +12,7 @@ const Element = @import("element.zig").Element;
const Union = @import("element.zig").Union; const Union = @import("element.zig").Union;
const Matcher = union(enum) { const Matcher = union(enum) {
matchByName: MatchByName,
matchByTagName: MatchByTagName, matchByTagName: MatchByTagName,
matchByClassName: MatchByClassName, matchByClassName: MatchByClassName,
matchTrue: struct {}, matchTrue: struct {},
@@ -21,6 +22,7 @@ const Matcher = union(enum) {
inline .matchTrue => return true, inline .matchTrue => return true,
inline .matchByTagName => |case| return case.match(node), inline .matchByTagName => |case| return case.match(node),
inline .matchByClassName => |case| return case.match(node), inline .matchByClassName => |case| return case.match(node),
inline .matchByName => |case| return case.match(node),
} }
} }
@@ -29,6 +31,7 @@ const Matcher = union(enum) {
inline .matchTrue => return, inline .matchTrue => return,
inline .matchByTagName => |case| return case.deinit(alloc), inline .matchByTagName => |case| return case.deinit(alloc),
inline .matchByClassName => |case| return case.deinit(alloc), inline .matchByClassName => |case| return case.deinit(alloc),
inline .matchByName => |case| return case.deinit(alloc),
} }
} }
}; };
@@ -117,6 +120,44 @@ pub fn HTMLCollectionByClassName(
}; };
} }
pub const MatchByName = struct {
name: []const u8,
fn init(alloc: std.mem.Allocator, name: []const u8) !MatchByName {
const names_alloc = try alloc.alloc(u8, name.len);
@memcpy(names_alloc, name);
return MatchByName{
.name = names_alloc,
};
}
pub fn match(self: MatchByName, node: *parser.Node) !bool {
const e = parser.nodeToElement(node);
const nname = try parser.elementGetAttribute(e, "name") orelse return false;
return std.mem.eql(u8, self.name, nname);
}
fn deinit(self: MatchByName, alloc: std.mem.Allocator) void {
alloc.free(self.name);
}
};
pub fn HTMLCollectionByName(
alloc: std.mem.Allocator,
root: ?*parser.Node,
name: []const u8,
include_root: bool,
) !HTMLCollection {
return HTMLCollection{
.root = root,
.walker = Walker{ .walkerDepthFirst = .{} },
.matcher = Matcher{
.matchByName = try MatchByName.init(alloc, name),
},
.include_root = include_root,
};
}
pub fn HTMLCollectionChildren( pub fn HTMLCollectionChildren(
root: ?*parser.Node, root: ?*parser.Node,
include_root: bool, include_root: bool,
@@ -259,7 +300,7 @@ pub const HTMLCollection = struct {
return len; return len;
} }
pub fn _item(self: *HTMLCollection, index: u32) !?Union { pub fn item(self: *HTMLCollection, index: u32) !?*parser.Node {
if (self.root == null) return null; if (self.root == null) return null;
var i: u32 = 0; var i: u32 = 0;
@@ -282,8 +323,7 @@ pub const HTMLCollection = struct {
self.cur_node = node; self.cur_node = node;
self.cur_idx = i; self.cur_idx = i;
const e = @as(*parser.Element, @ptrCast(node)); return node;
return try Element.toInterface(e);
} }
i += 1; i += 1;
@@ -296,6 +336,12 @@ pub const HTMLCollection = struct {
return null; return null;
} }
pub fn _item(self: *HTMLCollection, index: u32) !?Union {
const node = try self.item(index) orelse return null;
const e = @as(*parser.Element, @ptrCast(node));
return try Element.toInterface(e);
}
pub fn _namedItem(self: *HTMLCollection, name: []const u8) !?Union { pub fn _namedItem(self: *HTMLCollection, name: []const u8) !?Union {
if (self.root == null) return null; if (self.root == null) return null;
if (name.len == 0) return null; if (name.len == 0) return null;