Rendering setup

generator/render/util.zig

@@ -0,0 +1,232 @@
+const std = @import("std");
+const reg = @import("../registry.zig");
+const mem = std.mem;
+const Allocator = mem.Allocator;
+
+// Lifted from src-self-hosted/translate_c.zig
+pub fn isValidZigIdentifier(name: []const u8) bool {
+    for (name) |c, i| {
+        switch (c) {
+            '_', 'a'...'z', 'A'...'Z' => {},
+            '0' ... '9' => if (i == 0) return false,
+            else => return false
+        }
+    }
+
+    return true;
+}
+
+// Lifted from src-self-hosted/translate_c.zig
+pub fn isZigReservedIdentifier(name: []const u8) bool {
+    if (name.len > 1 and (name[0] == 'u' or name[0] == 'i')) {
+        for (name[1..]) |c| {
+            switch (c) {
+                '0'...'9' => {},
+                else => return false,
+            }
+        }
+        return true;
+    }
+
+    const reserved_names = [_][]const u8 {
+        "void", "comptime_float", "comptime_int", "bool", "isize",
+        "usize", "f16", "f32", "f64", "f128", "c_longdouble",
+        "noreturn", "type", "anyerror", "c_short", "c_ushort",
+        "c_int", "c_uint", "c_long", "c_ulong", "c_longlong", "c_ulonglong"
+    };
+
+    for (reserved_names) |reserved| {
+        if (mem.eql(u8, reserved, name)) {
+            return true;
+        }
+    }
+
+    return false;
+}
+
+pub fn needZigEscape(name: []const u8) bool {
+    return !isValidZigIdentifier(name)
+        or isZigReservedIdentifier(name)
+        or std.zig.Token.getKeyword(name) != null;
+}
+
+pub fn writeIdentifier(out: var, id: []const u8) !void {
+    if (needZigEscape(id)) {
+        try out.print("@\"{}\"", .{id});
+    } else {
+        try out.writeAll(id);
+    }
+}
+
+pub const CaseStyle = enum {
+    snake,
+    screaming_snake,
+    title,
+    camel,
+};
+
+pub fn trimVkNamespace(id: []const u8) []const u8 {
+    const prefixes = [_][]const u8{"VK_", "vk", "Vk", "PFN_vk"};
+    for (prefixes) |prefix| {
+        if (mem.startsWith(u8, id, prefix)) {
+            return id[prefix.len..];
+        }
+    }
+
+    return id;
+}
+
+pub fn getAuthorTag(id: []const u8, tags: []const reg.Tag) ?[]const u8 {
+    for (tags) |tag| {
+        if (mem.endsWith(u8, id, tag.name)) {
+            return tag.name;
+        }
+    }
+
+    return null;
+}
+
+pub const SegmentIterator = struct {
+    text: []const u8,
+    offset: usize,
+
+    pub fn init(text: []const u8) SegmentIterator {
+        return .{
+            .text = text,
+            .offset = 0,
+        };
+    }
+
+    fn nextBoundary(self: SegmentIterator) usize {
+        var i = self.offset + 1;
+
+        while (true) {
+            if (i == self.text.len or self.text[i] == '_') {
+                return i;
+            }
+
+            const prev_lower = std.ascii.isLower(self.text[i - 1]);
+            const next_lower = std.ascii.isLower(self.text[i]);
+
+            if (prev_lower and !next_lower) {
+                return i;
+            } else if (i != self.offset + 1 and !prev_lower and next_lower) {
+                return i - 1;
+            }
+
+            i += 1;
+        }
+    }
+
+    pub fn next(self: *SegmentIterator) ?[]const u8 {
+        while (self.offset < self.text.len and self.text[self.offset] == '_') {
+            self.offset += 1;
+        }
+
+        if (self.offset == self.text.len) {
+            return null;
+        }
+
+        const end = self.nextBoundary();
+        const word = self.text[self.offset .. end];
+        self.offset = end;
+        return word;
+    }
+
+    pub fn rest(self: SegmentIterator) []const u8 {
+        if (self.offset >= self.text.len) {
+            return &[_]u8{};
+        } else {
+            return self.text[self.offset..];
+        }
+    }
+};
+
+pub const IdRenderer = struct {
+    tags: []const reg.Tag,
+    text_cache: std.ArrayList(u8),
+
+    pub fn init(allocator: *Allocator, tags: []const reg.Tag) IdRenderer {
+        return .{
+            .tags = tags,
+            .text_cache = std.ArrayList(u8).init(allocator),
+        };
+    }
+
+    pub fn deinit(self: IdRenderer) void {
+        self.text_cache.deinit();
+    }
+
+    fn renderSnake(self: *IdRenderer, screaming: bool, id: []const u8, tag: ?[]const u8) !void {
+        var it = SegmentIterator.init(id);
+        var first = true;
+        const transform = if (screaming) std.ascii.toUpper else std.ascii.toLower;
+
+        while (it.next()) |segment| {
+            if (first) {
+                first = false;
+            } else {
+                try self.text_cache.append('_');
+            }
+
+            for (segment) |c| {
+                try self.text_cache.append(transform(c));
+            }
+        }
+
+        if (tag) |name| {
+            try self.text_cache.append('_');
+
+            for (name) |c| {
+                try self.text_cache.append(transform(c));
+            }
+        }
+    }
+
+    fn renderCamel(self: *IdRenderer, title: bool, id: []const u8, tag: ?[]const u8) !void {
+        var it = SegmentIterator.init(id);
+        var lower_first = !title;
+
+        while (it.next()) |segment| {
+            var i: usize = 0;
+            while (i < segment.len and std.ascii.isDigit(segment[i])) {
+                try self.text_cache.append(segment[i]);
+                i += 1;
+            }
+
+            if (i == segment.len) {
+                continue;
+            }
+
+            if (i == 0 and lower_first) {
+                try self.text_cache.append(std.ascii.toLower(segment[i]));
+            } else {
+                try self.text_cache.append(std.ascii.toUpper(segment[i]));
+            }
+            lower_first = false;
+
+            for (segment[i + 1..]) |c| {
+                try self.text_cache.append(std.ascii.toLower(c));
+            }
+        }
+
+        if (tag) |name| {
+            try self.text_cache.appendSlice(name);
+        }
+    }
+
+    pub fn render(self: *IdRenderer, out: var, case_style: CaseStyle, id: []const u8) !void {
+        const tag = getAuthorTag(id, self.tags);
+        const adjusted_id = if (tag) |name| id[0 .. id.len - name.len] else id;
+        self.text_cache.items.len = 0;
+
+        switch (case_style) {
+            .snake => try self.renderSnake(false, adjusted_id, tag),
+            .screaming_snake => try self.renderSnake(true, adjusted_id, tag),
+            .title => try self.renderCamel(true, adjusted_id, tag),
+            .camel => try self.renderCamel(false, adjusted_id, tag),
+        }
+
+        try writeIdentifier(out, self.text_cache.items);
+    }
+};