1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
|
const std = @import("std");
const Arena = std.heap.ArenaAllocator;
const Allocator = std.mem.Allocator;
const eql = std.mem.eql;
const html = @import("html.zig");
pub const Void = @import("Void.zig");
pub const Content = @import("Content.zig");
pub const Literal = @import("Literal.zig");
pub const Root = @import("Root.zig");
pub const Error = html.Error || Allocator.Error;
pub fn Wrapper(comptime V: type, comptime r: *const fn (*V, Allocator) Error![]const u8) type {
comptime {
if (!@hasField(V, "node")) @compileError("missing field 'node' for " ++ @typeName(V));
const nd = @FieldType(V, "node");
if (nd != Node) @compileError("invalid node's type: " ++ @typeName(nd) ++ ", want " ++ @typeName(Node));
if (!std.meta.hasMethod(V, "element")) @compileError("missing declaration 'element' for " ++ @typeName(V));
}
return struct {
ptr: *V,
const Self = @This();
fn node(context: *anyopaque) *Node {
const self: *V = @ptrCast(@alignCast(context));
return &self.node;
}
fn render(context: *anyopaque, alloc: Allocator) Error![]const u8 {
const self: *V = @ptrCast(@alignCast(context));
return try r(self, alloc);
}
pub fn init(ptr: *V) Element {
return (Self{ .ptr = ptr }).element();
}
pub fn element(self: Self) Element {
return .{ .ptr = self.ptr, .vtable = .{ .node = Self.node, .render = Self.render } };
}
};
}
pub const Node = struct {
ptr: *anyopaque,
vtable: struct { element: *const fn (*anyopaque) Element },
node: std.DoublyLinkedList.Node = .{},
pub fn from(n: *std.DoublyLinkedList.Node) Element {
const self: *Node = @fieldParentPtr("node", n);
return self.vtable.element(self.ptr);
}
pub fn element(self: Node) Element {
return self.vtable.element(self.ptr);
}
};
const Element = @This();
vtable: struct {
render: *const fn (*anyopaque, Allocator) Error![]const u8,
node: *const fn (*anyopaque) *Node,
},
ptr: *anyopaque,
pub fn render(self: Element, alloc: Allocator) Error![]const u8 {
return self.vtable.render(self.ptr, alloc);
}
pub fn node(self: Element) *Node {
return self.vtable.node(self.ptr);
}
fn doTest(alloc: Allocator, el: Element, exp: []const u8) !void {
const got = try el.render(alloc);
defer alloc.free(got);
std.testing.expect(eql(u8, got, exp)) catch |err| {
std.debug.print("{s}\n", .{got});
return err;
};
}
test "void" {
var arena = Arena.init(std.testing.allocator);
defer arena.deinit();
const alloc = arena.allocator();
var br = try Void.init(alloc, "br");
try doTest(alloc, br.element(), "<br>");
var img = try Void.init(alloc, "img");
try img.setAttribute("src", "foo");
try img.setAttribute("alt", "bar");
try doTest(alloc, img.element(), "<img src=\"foo\" alt=\"bar\">");
}
test "content" {
var arena = Arena.init(std.testing.allocator);
defer arena.deinit();
const alloc = arena.allocator();
var p = try Content.init(alloc, "p");
var root = try Root.init(alloc);
p.content = root.element();
var content = try Literal.init(alloc, "hello world");
try root.append(content.element());
try doTest(alloc, content.element(), "hello world");
try doTest(alloc, p.element(), "<p>hello world</p>");
var div = try Content.init(alloc, "div");
var rootDiv = try Root.init(alloc);
div.content = rootDiv.element();
try div.base.appendClass("foo-bar");
try rootDiv.append(p.element());
try rootDiv.append((try Void.init(alloc, "br")).element());
try doTest(alloc, div.element(), "<div class=\"foo-bar\"><p>hello world</p><br></div>");
}
|
