opt_aref_with

Author: kddnewton

lib/syntax_tree/compiler.rb

@@ -158,6 +158,21 @@ def visit_tstring_content(node)
         node.value
       end
 
+      def visit_var_ref(node)
+        raise CompilationError unless node.value.is_a?(Kw)
+
+        case node.value.value
+        when "nil"
+          nil
+        when "true"
+          true
+        when "false"
+          false
+        else
+          raise CompilationError
+        end
+      end
+
       def visit_word(node)
         if node.parts.length == 1 && node.parts.first.is_a?(TStringContent)
           node.parts.first.value
@@ -258,6 +273,20 @@ def visit_alias(node)
 
     def visit_aref(node)
       visit(node.collection)
+
+      if !frozen_string_literal && specialized_instruction && (node.index.parts.length == 1)
+        arg = node.index.parts.first
+
+        if arg.is_a?(StringLiteral) && (arg.parts.length == 1)
+          string_part = arg.parts.first
+
+          if string_part.is_a?(TStringContent)
+            iseq.opt_aref_with(string_part.value, :[], 1)
+            return
+          end
+        end
+      end
+
       visit(node.index)
       iseq.send(:[], 1)
     end

lib/syntax_tree/yarv.rb

@@ -527,6 +527,11 @@ def once(postexe_iseq, inline_storage)
         push([:once, postexe_iseq, inline_storage])
       end
 
+      def opt_aref_with(object, method_id, argc, flag = VM_CALL_ARGS_SIMPLE)
+        stack.change_by(-1 + 1)
+        push([:opt_aref_with, object, call_data(method_id, argc, flag)])
+      end
+
       def opt_getconstant_path(names)
         if RUBY_VERSION >= "3.2"
           stack.change_by(+1)

test/compiler_test.rb

@@ -201,6 +201,7 @@ class CompilerTest < Minitest::Test
       "foo[bar] ||= 1",
       "foo[bar] <<= 1",
       "foo[bar] ^= 1",
+      "foo['true']",
       # Constants (single)
       "Foo",
       "Foo = 1",