Impl KeyError.initialize with receiver and key kwargs

https://bugs.ruby-lang.org/issues/14313

See #5576

Author: headius

core/src/main/java/org/jruby/RubyKeyError.java

@@ -33,15 +33,18 @@
 
 import org.jruby.anno.JRubyMethod;
 import org.jruby.anno.JRubyClass;
+import org.jruby.ast.util.ArgsUtil;
 import org.jruby.exceptions.KeyError;
 import org.jruby.exceptions.RaiseException;
+import org.jruby.runtime.ThreadContext;
 import org.jruby.runtime.builtin.IRubyObject;
 
 /**
  * @author Miguel Landaeta
  */
 @JRubyClass(name="KeyError", parent="IndexError")
 public class RubyKeyError extends RubyIndexError {
+    private static final String[] VALID_KEYS = {"receiver", "key"};
     private IRubyObject receiver;
     private IRubyObject key;
 
@@ -67,6 +70,49 @@ protected RaiseException constructThrowable(String message) {
         return new KeyError(message, this);
     }
 
+    @JRubyMethod
+    public IRubyObject initialize(ThreadContext context, IRubyObject messageOrKwargs) {
+        IRubyObject[] receiverKey = ArgsUtil.extractKeywordArgs(context, messageOrKwargs, VALID_KEYS);
+
+        if (receiverKey == null) return initialize(context, messageOrKwargs, null);
+
+        return initializeCommon(context, context.nil, receiverKey);
+    }
+
+    @JRubyMethod
+    public IRubyObject initialize(ThreadContext context, IRubyObject message, IRubyObject kwargs) {
+        IRubyObject[] receiverKey = ArgsUtil.extractKeywordArgs(context, kwargs, VALID_KEYS);
+
+        return initializeCommon(context, message, receiverKey);
+    }
+
+    private IRubyObject initializeCommon(ThreadContext context, IRubyObject message, IRubyObject[] receiverKey) {
+        IRubyObject receiver;
+        IRubyObject key;
+        if (receiverKey == null) {
+            receiver = context.nil;
+            key = context.nil;
+        } else {
+            receiver = receiverKey[0];
+            key = receiverKey[1];
+        }
+
+        setMessage(message);
+        this.receiver = receiver;
+        this.key = key;
+
+        return context.nil;
+    }
+
+    @JRubyMethod
+    public IRubyObject initialize(ThreadContext context, IRubyObject message, IRubyObject receiver, IRubyObject key) {
+        setMessage(message);
+        this.receiver = receiver;
+        this.key = key;
+
+        return context.nil;
+    }
+
     @JRubyMethod
     public IRubyObject receiver() {
         return receiver;

core/src/main/java/org/jruby/ast/util/ArgsUtil.java

@@ -145,10 +145,16 @@ public void visit(IRubyObject key, IRubyObject value) {
 
     // not used
     public static IRubyObject[] extractKeywordArgs(ThreadContext context, IRubyObject[] args, String... validKeys) {
-        IRubyObject options = ArgsUtil.getOptionsArg(context.runtime, args);
+        return extractKeywordArgs(context, ArgsUtil.getOptionsArg(context.runtime, args), validKeys);
+    }
+
+    public static IRubyObject[] extractKeywordArgs(ThreadContext context, IRubyObject maybeKwargs, String... validKeys) {
+        IRubyObject options = ArgsUtil.getOptionsArg(context.runtime, maybeKwargs);
+
         if (options instanceof RubyHash) {
             return extractKeywordArgs(context, (RubyHash) options, validKeys);
         }
+
         return null;
     }