LibJS: Generate bytecode for UpdateExpression with MemberExpression arg

1 files changed, 13 insertions, 19 deletions

diff --git a/Userland/Libraries/LibJS/Bytecode/ASTCodegen.cpp b/Userland/Libraries/LibJS/Bytecode/ASTCodegen.cpp
index b55a79dddd..c9c89557d1 100644
--- a/Userland/Libraries/LibJS/Bytecode/ASTCodegen.cpp
+++ b/Userland/Libraries/LibJS/Bytecode/ASTCodegen.cpp
@@ -1134,29 +1134,23 @@ void TaggedTemplateLiteral::generate_bytecode(Bytecode::Generator& generator) co
 
 void UpdateExpression::generate_bytecode(Bytecode::Generator& generator) const
 {
-    if (is<Identifier>(*m_argument)) {
-        auto& identifier = static_cast<Identifier const&>(*m_argument);
-        generator.emit<Bytecode::Op::GetVariable>(generator.intern_identifier(identifier.string()));
-
-        Optional<Bytecode::Register> previous_value_for_postfix_reg;
-        if (!m_prefixed) {
-            previous_value_for_postfix_reg = generator.allocate_register();
-            generator.emit<Bytecode::Op::Store>(*previous_value_for_postfix_reg);
-        }
+    generator.emit_load_from_reference(*m_argument);
 
-        if (m_op == UpdateOp::Increment)
-            generator.emit<Bytecode::Op::Increment>();
-        else
-            generator.emit<Bytecode::Op::Decrement>();
+    Optional<Bytecode::Register> previous_value_for_postfix_reg;
+    if (!m_prefixed) {
+        previous_value_for_postfix_reg = generator.allocate_register();
+        generator.emit<Bytecode::Op::Store>(*previous_value_for_postfix_reg);
+    }
 
-        generator.emit<Bytecode::Op::SetVariable>(generator.intern_identifier(identifier.string()));
+    if (m_op == UpdateOp::Increment)
+        generator.emit<Bytecode::Op::Increment>();
+    else
+        generator.emit<Bytecode::Op::Decrement>();
 
-        if (!m_prefixed)
-            generator.emit<Bytecode::Op::Load>(*previous_value_for_postfix_reg);
-        return;
-    }
+    generator.emit_store_to_reference(*m_argument);
 
-    TODO();
+    if (!m_prefixed)
+        generator.emit<Bytecode::Op::Load>(*previous_value_for_postfix_reg);
 }
 
 void ThrowStatement::generate_bytecode(Bytecode::Generator& generator) const