feat(): support image

3 files changed, 71 insertions, 19 deletions

diff --git a/src/eval/Element.zig b/src/eval/Element.zig
index 5d6fce5..a8b424d 100644
--- a/src/eval/Element.zig
+++ b/src/eval/Element.zig
@@ -4,6 +4,7 @@ pub const HTML = @import("html/Element.zig");
 pub const paragraph = @import("paragraph.zig");
 pub const Title = @import("Title.zig");
 pub const list = @import("list.zig");
+pub const Image = @import("Image.zig");
 
 const Element = @This();
 
diff --git a/src/eval/Image.zig b/src/eval/Image.zig
new file mode 100644
index 0000000..771c8ee
--- /dev/null
+++ b/src/eval/Image.zig
@@ -0,0 +1,68 @@
+const std = @import("std");
+const Allocator = std.mem.Allocator;
+const HTML = @import("html/Element.zig");
+const Element = @import("Element.zig");
+
+const Self = @This();
+
+src: []const u8,
+alt: ?[]const u8 = null,
+source: ?Element = null,
+
+pub fn init(alloc: Allocator, src: []const u8) !*Self {
+    const v = try alloc.create(Self);
+    v.* = .{
+        .src = src,
+    };
+    return v;
+}
+
+pub fn element(self: *Self) Element {
+    return .{ .ptr = self, .vtable = .{ .deinit = destroy, .html = html } };
+}
+
+pub fn deinit(self: *Self, alloc: Allocator) void {
+    destroy(self, alloc);
+}
+
+fn destroy(context: *anyopaque, alloc: Allocator) void {
+    const self: *Self = @ptrCast(@alignCast(context));
+    if (self.source) |it| it.deinit(alloc);
+    alloc.destroy(self);
+}
+
+fn html(context: *anyopaque, alloc: Allocator) HTML.Error!HTML {
+    const self: *Self = @ptrCast(@alignCast(context));
+
+    var img = try HTML.init(alloc, .void, "img");
+    errdefer img.deinit();
+    try img.setAttribute("src", self.src);
+    if (self.alt) |it| try img.setAttribute("alt", it);
+    var el = try HTML.init(alloc, .content, "figure");
+    errdefer el.deinit();
+    try el.appendContent(img);
+
+    const source = self.source orelse return el;
+    var caption = try HTML.init(alloc, .content, "figcaption");
+    errdefer caption.deinit();
+    try caption.content.append(alloc, try source.html(alloc));
+    try el.appendContent(caption);
+    return el;
+}
+
+test "html" {
+    const alloc = std.testing.allocator;
+    const expect = std.testing.expect;
+    const eql = std.mem.eql;
+
+    var img = try init(alloc, "foo");
+    defer img.deinit(alloc);
+    const h = try img.element().renderHTML(alloc);
+    defer alloc.free(h);
+    try expect(eql(u8, h, "<figure><img src=\"foo\"></figure>"));
+
+    img.alt = "bar";
+    const h2 = try img.element().renderHTML(alloc);
+    defer alloc.free(h2);
+    try expect(eql(u8, h2, "<figure><img src=\"foo\" alt=\"bar\"></figure>"));
+}
diff --git a/src/eval/html/Element.zig b/src/eval/html/Element.zig
index 87fd876..666e6f0 100644
--- a/src/eval/html/Element.zig
+++ b/src/eval/html/Element.zig
@@ -178,19 +178,6 @@ pub fn appendContent(self: *Self, content: Self) Error!void {
     return self.content.append(alloc, content);
 }
 
-pub fn initImg(alloc: Allocator, src: []const u8, alt: []const u8) Error!Self {
-    var el = try init(alloc, .void, "img");
-    try el.setAttribute("src", src);
-    try el.setAttribute("alt", alt);
-    return el;
-}
-
-pub fn initContent(alloc: Allocator, tag: []const u8, content: []Self) Error!Self {
-    var el = try init(alloc, .content, tag);
-    for (content) |it| try el.appendContent(it);
-    return el;
-}
-
 fn doTest(alloc: Allocator, el: *Self, exp: []const u8) !void {
     const got = try el.render(alloc);
     defer alloc.free(got);
@@ -214,10 +201,6 @@ test "void element" {
     try img.setAttribute("alt", "bar");
 
     try doTest(alloc, &img, "<img src=\"foo\" alt=\"bar\">");
-
-    var img2 = try initImg(alloc, "foo", "bar");
-    defer img2.deinit();
-    try doTest(alloc, &img2, "<img src=\"foo\" alt=\"bar\">");
 }
 
 test "content element" {
@@ -235,7 +218,7 @@ test "content element" {
     defer div.deinit();
     try div.appendClass("foo-bar");
     try div.appendContent(p);
-    try div.appendContent(try initImg(alloc, "example.org", "example"));
+    try div.appendContent(try init(alloc, .void, "br"));
 
-    try doTest(alloc, &div, "<div class=\"foo-bar\"><p>hello world</p><img src=\"example.org\" alt=\"example\"></div>");
+    try doTest(alloc, &div, "<div class=\"foo-bar\"><p>hello world</p><br></div>");
 }