From 135e9b81217e2e8e6cdd4053e6df097a02dd9a12 Mon Sep 17 00:00:00 2001
From: Stephen Gutekanst <stephen@hexops.com>
Date: Fri, 15 Mar 2024 18:17:58 -0700
Subject: [PATCH] module: add GlobalEvent enum

Signed-off-by: Stephen Gutekanst <stephen@hexops.com>
---
 src/module.zig | 39 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)

diff --git a/src/module.zig b/src/module.zig
index 54bacb10..d05dac08 100644
--- a/src/module.zig
+++ b/src/module.zig
@@ -295,6 +295,38 @@ fn UninjectedArgsTuple(comptime Function: type, comptime Injectable: type) type
     return std.meta.Tuple(std_args);
 }
 
+/// enum describing every possible comptime-known global event name
+fn GlobalEvent(comptime mods: anytype) type {
+    var enum_fields: []const std.builtin.Type.EnumField = &[0]std.builtin.Type.EnumField{};
+    var i: u32 = 0;
+    for (mods) |M| {
+        // Global event handlers
+        for (@typeInfo(M).Struct.decls) |decl| {
+            switch (@typeInfo(@TypeOf(@field(M, decl.name)))) {
+                .Fn => {
+                    const exists_already = blk2: {
+                        for (enum_fields) |existing| if (std.mem.eql(u8, existing.name, decl.name)) break :blk2 true;
+                        break :blk2 false;
+                    };
+                    if (!exists_already) {
+                        enum_fields = enum_fields ++ [_]std.builtin.Type.EnumField{.{ .name = decl.name, .value = i }};
+                        i += 1;
+                    }
+                },
+                else => {},
+            }
+        }
+    }
+    return @Type(.{
+        .Enum = .{
+            .tag_type = std.math.IntFittingRange(0, enum_fields.len - 1),
+            .fields = enum_fields,
+            .decls = &[_]std.builtin.Type.Declaration{},
+            .is_exhaustive = true,
+        },
+    });
+}
+
 /// enum describing every possible comptime-known event name
 fn EventName(comptime mods: anytype) type {
     var enum_fields: []const std.builtin.Type.EnumField = &[0]std.builtin.Type.EnumField{};
@@ -571,6 +603,13 @@ test EventName {
     try testing.expect([]const u8, "bam").eql(info.fields[3].name);
     try testing.expect([]const u8, "tick").eql(info.fields[4].name);
     try testing.expect([]const u8, "foobar").eql(info.fields[5].name);
+
+    const global_info = @typeInfo(GlobalEvent(Mods.modules)).Enum;
+    try testing.expect(type, u2).eql(global_info.tag_type);
+    try testing.expect(usize, 3).eql(global_info.fields.len);
+    try testing.expect([]const u8, "foo").eql(global_info.fields[0].name);
+    try testing.expect([]const u8, "bar").eql(global_info.fields[1].name);
+    try testing.expect([]const u8, "tick").eql(global_info.fields[2].name);
 }
 
 test ModuleName {