implement two-arg version of {public,private,protected}_method_defined?

Author: Nicolas Laurent

CHANGELOG.md

@@ -39,6 +39,7 @@ Compatibility:
 * Switched to the io-console C extension from C ruby for better performance and compatibility in `irb`.
 * Coerce the message to a `String` for `BasicSocket#send` (#2209, @HoneyryderChuck).
 * Support buffer argument for `UDPSocket#recvfrom_nonblock` (#2209, @HoneyryderChuck).
+* Support the `inherit` parameter for `Module#{private, protected, public}_method_defined?`.
 
 Performance:
 

spec/tags/core/module/private_method_defined_tags.txt

@@ -1687,6 +1687,7 @@ public PrivateInstanceMethodsNode() {
 
     @NodeChild(value = "module", type = RubyNode.class)
     @NodeChild(value = "name", type = RubyNode.class)
+    @NodeChild(value = "inherit", type = RubyNode.class)
     protected abstract static class AbstractMethodDefinedNode extends CoreMethodNode {
 
         final Visibility visibility;
@@ -1700,39 +1701,47 @@ protected RubyNode coerceToString(RubyNode name) {
             return NameToJavaStringNode.create(name);
         }
 
-        @Specialization
-        protected boolean isMethodDefined(RubyModule module, String name) {
-            // TODO (pitr-ch 30-Mar-2016): cache lookup
-            return ModuleOperations.lookupMethod(module, name, visibility) != null;
+        @CreateCast("inherit")
+        protected RubyNode coerceToBoolean(RubyNode inherit) {
+            return BooleanCastWithDefaultNodeGen.create(true, inherit);
+        }
+
+        // TODO (pitr-ch 30-Mar-2016): cache lookup
+
+        @Specialization(guards = "inherit")
+        protected boolean isMethodDefinedInherit(RubyModule module, String name, boolean inherit) {
+            final InternalMethod method = ModuleOperations.lookupMethodUncached(module, name, null);
+            return method != null && !method.isUndefined() && !method.isUnimplemented() &&
+                    method.getVisibility() == visibility;
         }
 
+        @Specialization(guards = "!inherit")
+        protected boolean isMethodDefinedDontInherit(RubyModule module, String name, boolean inherit) {
+            final InternalMethod method = module.fields.getMethod(name);
+            return method != null && !method.isUndefined() && !method.isUnimplemented() &&
+                    method.getVisibility() == visibility;
+        }
     }
 
-    @CoreMethod(names = "public_method_defined?", required = 1)
+    @CoreMethod(names = "public_method_defined?", required = 1, optional = 1)
     public abstract static class PublicMethodDefinedNode extends AbstractMethodDefinedNode {
-
         public PublicMethodDefinedNode() {
             super(Visibility.PUBLIC);
         }
-
     }
 
-    @CoreMethod(names = "protected_method_defined?", required = 1)
+    @CoreMethod(names = "protected_method_defined?", required = 1, optional = 1)
     public abstract static class ProtectedMethodDefinedNode extends AbstractMethodDefinedNode {
-
         public ProtectedMethodDefinedNode() {
             super(Visibility.PROTECTED);
         }
-
     }
 
-    @CoreMethod(names = "private_method_defined?", required = 1)
+    @CoreMethod(names = "private_method_defined?", required = 1, optional = 1)
     public abstract static class PrivateMethodDefinedNode extends AbstractMethodDefinedNode {
-
         public PrivateMethodDefinedNode() {
             super(Visibility.PRIVATE);
         }
-
     }
 
     @CoreMethod(names = "instance_methods", optional = 1)

spec/tags/core/module/protected_method_defined_tags.txt

@@ -24,7 +24,6 @@
 import org.truffleruby.core.string.StringUtils;
 import org.truffleruby.language.LexicalScope;
 import org.truffleruby.language.RubyConstant;
-import org.truffleruby.language.Visibility;
 import org.truffleruby.language.control.RaiseException;
 import org.truffleruby.language.methods.DeclarationContext;
 import org.truffleruby.language.methods.InternalMethod;
@@ -483,14 +482,6 @@ private static InternalMethod lookupMethodUncached(RubyModule module, RubyModule
         return null;
     }
 
-    public static InternalMethod lookupMethod(RubyModule module, String name, Visibility visibility) {
-        final InternalMethod method = lookupMethodUncached(module, name, null);
-        if (method == null || method.isUndefined()) {
-            return null;
-        }
-        return method.getVisibility() == visibility ? method : null;
-    }
-
     public static MethodLookupResult lookupSuperMethod(InternalMethod currentMethod, RubyModule objectMetaClass) {
         final String name = currentMethod.getSharedMethodInfo().getMethodNameForNotBlock(); // use the original name