[Feature #16254] Allow `Primitive.func` style

2 files changed, 17 insertions, 3 deletions

diff --git a/compile.c b/compile.c
index 13df6a0062..0753792aaf 100644
--- a/compile.c
+++ b/compile.c
@@ -7045,7 +7045,8 @@ iseq_builtin_function_name(const enum node_type type, const NODE *recv, ID mid)
     static const char prefix[] = "__builtin_";
     const size_t prefix_len = sizeof(prefix) - 1;
 
-    if (type == NODE_CALL) {
+    switch (type) {
+      case NODE_CALL:
         if (recv) {
             switch (nd_type(recv)) {
               case NODE_VCALL:
@@ -7053,14 +7054,22 @@ iseq_builtin_function_name(const enum node_type type, const NODE *recv, ID mid)
                     return name;
                 }
                 break;
+              case NODE_CONST:
+                if (recv->nd_vid == rb_intern("Primitive")) {
+                    return name;
+                }
+                break;
               default: break;
             }
         }
-    }
-    else if (type == NODE_VCALL || type == NODE_FCALL) {
+        break;
+      case NODE_VCALL:
+      case NODE_FCALL:
         if (UNLIKELY(strncmp(prefix, name, prefix_len) == 0)) {
             return &name[prefix_len];
         }
+        break;
+      default: break;
     }
     return NULL;
 }
diff --git a/tool/mk_builtin_loader.rb b/tool/mk_builtin_loader.rb
index ae6c3b3812..01551592bf 100644
--- a/tool/mk_builtin_loader.rb
+++ b/tool/mk_builtin_loader.rb
@@ -113,6 +113,11 @@ def collect_builtin base, tree, name, bs, inlines, params = nil
       if recv
         func_name = nil
         case recv.first
+        when :var_ref
+          _, recv = recv
+          if recv.first == :@const and recv[1] == "Primitive"
+            func_name = mid.to_s
+          end
         when :vcall
           _, recv = recv
           if recv.first == :@ident and recv[1] == "__builtin"