LibJS: Implement Function.prototype.bind()

Libraries/LibJS/Runtime/BoundFunction.cpp

@@ -0,0 +1,71 @@
+/*
+ * Copyright (c) 2020, Jack Karamanian <karamanian.jack@gmail.com>
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are met:
+ *
+ * 1. Redistributions of source code must retain the above copyright notice, this
+ *    list of conditions and the following disclaimer.
+ *
+ * 2. Redistributions in binary form must reproduce the above copyright notice,
+ *    this list of conditions and the following disclaimer in the documentation
+ *    and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+ * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+ * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+ * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+ * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+ * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+ * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+ * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+ * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+#include <LibJS/Interpreter.h>
+#include <LibJS/Runtime/BoundFunction.h>
+#include <LibJS/Runtime/GlobalObject.h>
+
+namespace JS {
+
+BoundFunction::BoundFunction(Function& target_function, Value bound_this, Vector<Value> arguments, i32 length, Object* constructor_prototype)
+    : Function::Function(*interpreter().global_object().function_prototype(), bound_this, move(arguments))
+    , m_target_function(&target_function)
+    , m_constructor_prototype(constructor_prototype)
+    , m_name(String::format("bound %s", target_function.name().characters()))
+{
+    put("length", Value(length));
+}
+
+BoundFunction::~BoundFunction()
+{
+}
+
+Value BoundFunction::call(Interpreter& interpreter)
+{
+    return m_target_function->call(interpreter);
+}
+
+Value BoundFunction::construct(Interpreter& interpreter)
+{
+    if (auto this_value = interpreter.this_value(); m_constructor_prototype && this_value.is_object()) {
+        this_value.as_object().set_prototype(m_constructor_prototype);
+    }
+    return m_target_function->construct(interpreter);
+}
+
+LexicalEnvironment* BoundFunction::create_environment()
+{
+    return m_target_function->create_environment();
+}
+
+void BoundFunction::visit_children(Visitor& visitor)
+{
+    Function::visit_children(visitor);
+    visitor.visit(m_target_function);
+    visitor.visit(m_constructor_prototype);
+}
+
+}

Libraries/LibJS/Runtime/BoundFunction.h

@@ -0,0 +1,66 @@
+/*
+ * Copyright (c) 2020, Jack Karamanian <karamanian.jack@gmail.com>
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are met:
+ *
+ * 1. Redistributions of source code must retain the above copyright notice, this
+ *    list of conditions and the following disclaimer.
+ *
+ * 2. Redistributions in binary form must reproduce the above copyright notice,
+ *    this list of conditions and the following disclaimer in the documentation
+ *    and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+ * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+ * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+ * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+ * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+ * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+ * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+ * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+ * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+#pragma once
+
+#include <LibJS/Runtime/Function.h>
+
+namespace JS {
+
+class BoundFunction final : public Function {
+public:
+    BoundFunction(Function& target_function, Value bound_this, Vector<Value> arguments, i32 length, Object* constructor_prototype);
+
+    virtual ~BoundFunction();
+
+    virtual Value call(Interpreter& interpreter) override;
+
+    virtual Value construct(Interpreter& interpreter) override;
+
+    virtual LexicalEnvironment* create_environment() override;
+
+    virtual void visit_children(Visitor&) override;
+
+    virtual const FlyString& name() const override
+    {
+        return m_name;
+    }
+
+    Function& target_function() const
+    {
+        return *m_target_function;
+    }
+
+private:
+    virtual bool is_bound_function() const override { return true; }
+    virtual const char* class_name() const override { return "BoundFunction"; }
+
+    Function* m_target_function = nullptr;
+    Object* m_constructor_prototype = nullptr;
+    FlyString m_name;
+};
+
+}

Libraries/LibJS/Runtime/Function.cpp

@@ -25,7 +25,9 @@
  */
 
 #include <LibJS/Interpreter.h>
+#include <LibJS/Runtime/BoundFunction.h>
 #include <LibJS/Runtime/Function.h>
+#include <LibJS/Runtime/GlobalObject.h>
 
 namespace JS {
 
@@ -41,6 +43,50 @@ Function::Function(Object& prototype, Optional<Value> bound_this, Vector<Value>
 {
 }
 
+BoundFunction* Function::bind(Value bound_this_value, Vector<Value> arguments)
+{
+
+    Function& target_function = is_bound_function() ? static_cast<BoundFunction&>(*this).target_function() : *this;
+
+    auto bound_this_object
+        = [bound_this_value, this]() -> Value {
+        if (bound_this().has_value()) {
+            return bound_this().value();
+        }
+        switch (bound_this_value.type()) {
+        case Value::Type::Undefined:
+        case Value::Type::Null:
+            // FIXME: Null or undefined should be passed through in strict mode.
+            return &interpreter().global_object();
+        default:
+            return bound_this_value.to_object(interpreter().heap());
+        }
+    }();
+
+    i32 computed_length = 0;
+    auto length_property = get("length");
+    if (interpreter().exception()) {
+        return nullptr;
+    }
+    if (length_property.has_value() && length_property.value().is_number()) {
+        computed_length = max(0, length_property.value().to_i32() - static_cast<i32>(arguments.size()));
+    }
+
+    Object* constructor_prototype = nullptr;
+    auto prototype_property = target_function.get("prototype");
+    if (interpreter().exception()) {
+        return nullptr;
+    }
+    if (prototype_property.has_value() && prototype_property.value().is_object()) {
+        constructor_prototype = &prototype_property.value().as_object();
+    }
+
+    auto all_bound_arguments = bound_arguments();
+    all_bound_arguments.append(move(arguments));
+
+    return interpreter().heap().allocate<BoundFunction>(target_function, bound_this_object, move(all_bound_arguments), computed_length, constructor_prototype);
+}
+
 void Function::visit_children(Visitor& visitor)
 {
     Object::visit_children(visitor);

Libraries/LibJS/Runtime/Function.h

@@ -42,6 +42,8 @@ public:
 
     virtual void visit_children(Visitor&) override;
 
+    BoundFunction* bind(Value bound_this_value, Vector<Value> arguments);
+
     Optional<Value> bound_this() const
     {
         return m_bound_this;

Libraries/LibJS/Runtime/FunctionPrototype.cpp

@@ -28,6 +28,7 @@
 #include <AK/StringBuilder.h>
 #include <LibJS/AST.h>
 #include <LibJS/Interpreter.h>
+#include <LibJS/Runtime/BoundFunction.h>
 #include <LibJS/Runtime/Error.h>
 #include <LibJS/Runtime/Function.h>
 #include <LibJS/Runtime/FunctionPrototype.h>
@@ -84,8 +85,19 @@ Value FunctionPrototype::bind(Interpreter& interpreter)
     auto* this_object = interpreter.this_value().to_object(interpreter.heap());
     if (!this_object)
         return {};
-    // FIXME: Implement me :^)
-    ASSERT_NOT_REACHED();
+    if (!this_object->is_function())
+        return interpreter.throw_exception<TypeError>("Not a Function object");
+
+    auto& this_function = static_cast<Function&>(*this_object);
+    auto bound_this_arg = interpreter.argument(0);
+
+    Vector<Value> arguments;
+    if (interpreter.argument_count() > 1) {
+        arguments = interpreter.call_frame().arguments;
+        arguments.remove(0);
+    }
+
+    return this_function.bind(bound_this_arg, move(arguments));
 }
 
 Value FunctionPrototype::call(Interpreter& interpreter)
@@ -117,7 +129,7 @@ Value FunctionPrototype::to_string(Interpreter& interpreter)
     String function_parameters = "";
     String function_body;
 
-    if (this_object->is_native_function()) {
+    if (this_object->is_native_function() || this_object->is_bound_function()) {
         function_body = String::format("  [%s]", this_object->class_name());
     } else {
         auto& parameters = static_cast<ScriptFunction*>(this_object)->parameters();

Libraries/LibJS/Runtime/Object.h

@@ -72,6 +72,7 @@ public:
     virtual bool is_error() const { return false; }
     virtual bool is_function() const { return false; }
     virtual bool is_native_function() const { return false; }
+    virtual bool is_bound_function() const { return false; }
     virtual bool is_native_property() const { return false; }
     virtual bool is_string_object() const { return false; }