| ... | @@ -9,6 +9,7 @@ | ... | @@ -9,6 +9,7 @@ |
| 9 | const std = @import("std"); | 9 | const std = @import("std"); |
| 10 | const range = @import("range").range; | 10 | const range = @import("range").range; |
| 11 | const htmlentities = @import("htmlentities"); | 11 | const htmlentities = @import("htmlentities"); |
| | 12 | const root = @import("root"); |
| 12 | | 13 | |
| 13 | const tokenize = @import("./tokenize.zig"); | 14 | const tokenize = @import("./tokenize.zig"); |
| 14 | const astgen = @import("./astgen.zig"); | 15 | const astgen = @import("./astgen.zig"); |
| ... | @@ -134,6 +135,27 @@ fn do(alloc: *std.mem.Allocator, writer: anytype, comptime value: astgen.Value, | ... | @@ -134,6 +135,27 @@ fn do(alloc: *std.mem.Allocator, writer: anytype, comptime value: astgen.Value, |
| 134 | try do(alloc, writer, val, data, ctx, indent, flag1); | 135 | try do(alloc, writer, val, data, ctx, indent, flag1); |
| 135 | } | 136 | } |
| 136 | }, | 137 | }, |
| | 138 | .function => |v| { |
| | 139 | if (@hasDecl(root, "pek_" ++ v.name)) { |
| | 140 | const func = @field(root, "pek_" ++ v.name); |
| | 141 | var args: FnArgsTuple(func) = undefined; |
| | 142 | args.@"0" = alloc; |
| | 143 | inline for (v.args) |arg, i| { |
| | 144 | const field_name = comptime std.fmt.comptimePrint("{d}", .{i + 1}); |
| | 145 | @field(args, field_name) = if (comptime std.mem.eql(u8, arg[0], "this")) search(arg[1..], data) else search(arg, ctx); |
| | 146 | } |
| | 147 | const s: []const u8 = try @call(.{}, func, args); |
| | 148 | for (s) |c| { |
| | 149 | if (entityLookupBefore(&[_]u8{c})) |ent| { |
| | 150 | try writer.writeAll(ent.entity); |
| | 151 | } else { |
| | 152 | try writer.writeAll(&[_]u8{c}); |
| | 153 | } |
| | 154 | } |
| | 155 | return; |
| | 156 | } |
| | 157 | @compileError("pek: unknown custom function: " ++ v.name); |
| | 158 | }, |
| 137 | else => unreachable, | 159 | else => unreachable, |
| 138 | } | 160 | } |
| 139 | } | 161 | } |
Meghan Denny <hello@nektro.net> 2021-08-14 13:58:57 -07:00