katakata_irb 0.1.12 → 0.2.0

data/lib/katakata_irb/type_simulator.rb

@@ -1,7 +1,9 @@
-require 'ripper'
+# frozen_string_literal: true
+
 require 'set'
 require_relative 'types'
 require_relative 'scope'
+require 'prism'
 
 class KatakataIrb::TypeSimulator
   class DigTarget
@@ -18,21 +20,6 @@ class KatakataIrb::TypeSimulator
     end
   end
 
-  module LexerElemMatcher
-    refine Ripper::Lexer::Elem do
-      def deconstruct_keys(_keys)
-        {
-          tok: tok,
-          event: event,
-          label: state.allbits?(Ripper::EXPR_LABEL),
-          beg: state.allbits?(Ripper::EXPR_BEG),
-          dot: state.allbits?(Ripper::EXPR_DOT)
-        }
-      end
-    end
-  end
-  using LexerElemMatcher
-
   OBJECT_METHODS = {
     to_s: KatakataIrb::Types::STRING,
     to_str: KatakataIrb::Types::STRING,
@@ -51,22 +38,25 @@ class KatakataIrb::TypeSimulator
     @dig_targets = dig_targets
   end
 
-  def simulate_evaluate(sexp, scope, case_target: nil)
-    result = simulate_evaluate_inner(sexp, scope, case_target: case_target)
-    @dig_targets.resolve result, scope if @dig_targets.target?(sexp)
+  def evaluate(node, scope)
+    result = evaluate_inner(node, scope)
+    @dig_targets.resolve result, scope if @dig_targets.target?(node)
     result
   end
 
-  def simulate_evaluate_inner(sexp, scope, case_target: nil)
-    case sexp
-    in [:program, statements]
-      statements.map { simulate_evaluate _1, scope }.last
-    in [:def | :defs,]
-      sexp in [:def, _method_name_exp, params, body_stmt]
-      sexp in [:defs, receiver_exp, _dot_exp, _method_name_exp, params, body_stmt]
-      if receiver_exp
-        receiver_exp in [:paren, receiver_exp]
-        self_type = simulate_evaluate receiver_exp, scope
+  def evaluate_inner(node, scope)
+    case node
+    when Prism::ProgramNode
+      evaluate node.statements, scope
+    when Prism::StatementsNode
+      if node.body.empty?
+        KatakataIrb::Types::NIL
+      else
+        node.body.map { evaluate _1, scope }.last
+      end
+    when Prism::DefNode
+      if node.receiver
+        self_type = evaluate node.receiver, scope
       else
         current_self_types = scope.self_type.types
         self_types = current_self_types.map do |type|
@@ -78,104 +68,81 @@ class KatakataIrb::TypeSimulator
         end
         self_type = KatakataIrb::Types::UnionType[*self_types]
       end
-      if @dig_targets.dig? sexp
-        params in [:paren, params]
-        params ||= [:params, nil, nil, nil, nil, nil, nil, nil] # params might be nil in ruby 3.0
-        params_table = extract_param_names(params).to_h { [_1, KatakataIrb::Types::NIL] }
+      if @dig_targets.dig?(node.body) || @dig_targets.dig?(node.parameters)
+        params_table = node.locals.to_h { [_1.to_s, KatakataIrb::Types::NIL] }
         method_scope = KatakataIrb::Scope.new(
           scope,
-          { **params_table, KatakataIrb::Scope::SELF => self_type, KatakataIrb::Scope::BREAK_RESULT => nil, KatakataIrb::Scope::NEXT_RESULT => nil, KatakataIrb::Scope::RETURN_RESULT => nil },
-          trace_lvar: false
+          { **params_table, KatakataIrb::Scope::BREAK_RESULT => nil, KatakataIrb::Scope::NEXT_RESULT => nil, KatakataIrb::Scope::RETURN_RESULT => nil },
+          self_type: self_type,
+          trace_lvar: false,
+          trace_ivar: false
         )
-        evaluate_assign_params params, [], method_scope
-        method_scope.conditional { evaluate_param_defaults params, _1 }
-        simulate_evaluate body_stmt, method_scope
+        if node.parameters
+          # node.parameters is Prism::ParametersNode
+          assign_parameters node.parameters, method_scope, [], {}
+        end
+
+        if @dig_targets.dig?(node.body)
+          method_scope.conditional do |s|
+            evaluate node.body, s
+          end
+        end
         method_scope.merge_jumps
         scope.update method_scope
       end
       KatakataIrb::Types::SYMBOL
-    in [:@int,]
+    when Prism::IntegerNode
       KatakataIrb::Types::INTEGER
-    in [:@float,]
+    when Prism::FloatNode
       KatakataIrb::Types::FLOAT
-    in [:@rational,]
+    when Prism::RationalNode
       KatakataIrb::Types::RATIONAL
-    in [:@imaginary,]
+    when Prism::ImaginaryNode
       KatakataIrb::Types::COMPLEX
-    in [:@tstring_content,]
+    when Prism::StringNode
       KatakataIrb::Types::STRING
-    in [:symbol_literal,]
-      KatakataIrb::Types::SYMBOL
-    in [:dyna_symbol, [:string_content, *statements]]
-      statements.each { simulate_evaluate _1, scope }
+    when Prism::XStringNode
+      KatakataIrb::Types::UnionType[KatakataIrb::Types::STRING, KatakataIrb::Types::NIL]
+    when Prism::SymbolNode
       KatakataIrb::Types::SYMBOL
-    in [:@CHAR,]
+    when Prism::RegularExpressionNode
+      KatakataIrb::Types::REGEXP
+    when Prism::StringConcatNode
+      evaluate node.left, scope
+      evaluate node.right, scope
       KatakataIrb::Types::STRING
-    in [:@backref,]
-      KatakataIrb::Types::UnionType[KatakataIrb::Types::STRING, KatakataIrb::Types::NIL]
-    in [:string_literal, [:string_content, *statements]]
-      statements.each { simulate_evaluate _1, scope }
+    when Prism::InterpolatedStringNode
+      node.parts.each { evaluate _1, scope }
       KatakataIrb::Types::STRING
-    in [:xstring_literal, statements]
-      statements.each { simulate_evaluate _1, scope }
+    when Prism::InterpolatedXStringNode
+      node.parts.each { evaluate _1, scope }
       KatakataIrb::Types::STRING
-    in [:string_embexpr, statements]
-      statements.each { simulate_evaluate _1, scope }
+    when Prism::InterpolatedSymbolNode
+      node.parts.each { evaluate _1, scope }
+      KatakataIrb::Types::SYMBOL
+    when Prism::InterpolatedRegularExpressionNode
+      node.parts.each { evaluate _1, scope }
+      KatakataIrb::Types::REGEXP
+    when Prism::EmbeddedStatementsNode
+      node.statements ? evaluate(node.statements, scope) : KatakataIrb::Types::NIL
       KatakataIrb::Types::STRING
-    in [:string_dvar,]
+    when Prism::EmbeddedVariableNode
+      evaluate node.variable, scope
       KatakataIrb::Types::STRING
-    in [:regexp_literal, statements, _regexp_end]
-      statements.each { simulate_evaluate _1, scope }
-      KatakataIrb::Types::REGEXP
-    in [:array, [:args_add_star,] => star]
-      args, kwargs = retrieve_method_args star
-      types = args.flat_map do |elem|
-        if elem.is_a? KatakataIrb::Types::Splat
-          splat = simulate_evaluate elem.item, scope
-          array_elem, non_array = partition_to_array splat.nonnillable, :to_a
-          KatakataIrb::Types::UnionType[*array_elem, *non_array]
-        else
-          simulate_evaluate elem, scope
-        end
-      end
-      types << kwargs_type(kwargs, scope) if kwargs && kwargs.any?
-      KatakataIrb::Types::InstanceType.new Array, Elem: KatakataIrb::Types::UnionType[*types]
-    in [:array, statements]
-      if statements.nil? || statements.empty?
-        KatakataIrb::Types::ARRAY
-      elsif statements.all? { _1 in [Symbol,] }
-        # normal array
-        elem = statements ? KatakataIrb::Types::UnionType[*statements.map { simulate_evaluate _1, scope }] : KatakataIrb::Types::NIL
-        KatakataIrb::Types::InstanceType.new Array, Elem: elem
-      else
-        # %I[] or %W[]
-        statements.each do |sub_statements|
-          sub_statements.each { simulate_evaluate _1, scope }
-        end
-        # TODO: use AST because Ripper.sexp('%I[a]') == Ripper.sexp('%W[a]')
-        elem = KatakataIrb::Types::UnionType[KatakataIrb::Types::STRING, KatakataIrb::Types::SYMBOL]
-        KatakataIrb::Types::InstanceType.new Array, Elem: elem
-      end
-    in [:bare_assoc_hash, args]
-      simulate_evaluate [:hash, [:assoclist_from_args, args]], scope
-    in [:hash, [:assoclist_from_args, args]]
+    when Prism::ArrayNode
+      KatakataIrb::Types.array_of evaluate_list_splat_items(node.elements, scope)
+    when Prism::HashNode, Prism::KeywordHashNode
       keys = []
       values = []
-      args.each do |arg|
-        case arg
-        in [:assoc_new, key, value]
-          if key in [:@label, label, pos]
-            keys << KatakataIrb::Types::SYMBOL
-            name = label.delete ':'
-            value ||= [:__var_ref_or_call, [name =~ /\A[A-Z]/ ? :@const : :@ident, name, pos]]
-          else
-            keys << simulate_evaluate(key, scope)
-          end
-          values << simulate_evaluate(value, scope)
-        in [:assoc_splat, value]
-          hash = simulate_evaluate value, scope
+      node.elements.each do |assoc|
+        case assoc
+        when Prism::AssocNode
+          keys << evaluate(assoc.key, scope)
+          values << evaluate(assoc.value, scope)
+        when Prism::AssocSplatNode
+          hash = evaluate assoc.value, scope
           unless hash.is_a?(KatakataIrb::Types::InstanceType) && hash.klass == Hash
-            hash = simulate_call hash, :to_hash, [], nil, nil
+            hash = simulate_call hash, :to_hash, [], nil, nil, scope
           end
           if hash.is_a?(KatakataIrb::Types::InstanceType) && hash.klass == Hash
             keys << hash.params[:K] if hash.params[:K]
@@ -183,120 +150,89 @@ class KatakataIrb::TypeSimulator
           end
         end
       end
-      KatakataIrb::Types::InstanceType.new Hash, K: KatakataIrb::Types::UnionType[*keys], V: KatakataIrb::Types::UnionType[*values]
-    in [:hash, nil]
-      KatakataIrb::Types::InstanceType.new Hash
-    in [:paren, [Symbol,] | false => statement]
-      # workaround for `p ()` and `p (foo)`
-      simulate_evaluate statement, scope if statement
-    in [:paren | :ensure | :else, statements]
-      statements.map { simulate_evaluate _1, scope }.last
-    in [:const_path_ref, receiver, [:@const, name,]]
-      r = simulate_evaluate receiver, scope
-      r.is_a?(KatakataIrb::Types::SingletonType) ? KatakataIrb::BaseScope.type_of { r.module_or_class.const_get name } : KatakataIrb::Types::NIL
-    in [:__var_ref_or_call, [type, name, pos]]
-      sexp = scope.has?(name) ? [:var_ref, [type, name, pos]] : [:vcall, [:@ident, name, pos]]
-      simulate_evaluate sexp, scope
-    in [:var_ref, [:@kw, name,]]
-      case name
-      in 'self'
-        scope.self_type
-      in 'true'
-        KatakataIrb::Types::TRUE
-      in 'false'
-        KatakataIrb::Types::FALSE
-      in 'nil'
-        KatakataIrb::Types::NIL
-      in '__FILE__'
+      if keys.empty? && values.empty?
+        KatakataIrb::Types::InstanceType.new Hash
+      else
+        KatakataIrb::Types::InstanceType.new Hash, K: KatakataIrb::Types::UnionType[*keys], V: KatakataIrb::Types::UnionType[*values]
+      end
+    when Prism::ParenthesesNode
+      node.body ? evaluate(node.body, scope) : KatakataIrb::Types::NIL
+    when Prism::ConstantPathNode
+      type, = evaluate_constant_node node, scope
+      type
+    when Prism::SelfNode
+      scope.self_type
+    when Prism::TrueNode
+      KatakataIrb::Types::TRUE
+    when Prism::FalseNode
+      KatakataIrb::Types::FALSE
+    when Prism::NilNode
+      KatakataIrb::Types::NIL
+    when Prism::SourceFileNode
         KatakataIrb::Types::STRING
-      in '__LINE__'
+    when Prism::SourceLineNode
         KatakataIrb::Types::INTEGER
-      in '__ENCODING__'
-        KatakataIrb::Types::InstanceType.new Encoding
-      end
-    in [:var_ref, [:@const | :@ivar | :@cvar | :@gvar | :@ident, name,]]
-      scope[name] || KatakataIrb::Types::NIL
-    in [:const_ref, [:@const, name,]]
-      scope[name] || KatakataIrb::Types::NIL
-    in [:aref, receiver, args]
-      receiver_type = simulate_evaluate receiver, scope if receiver
-      args, kwargs, _block = retrieve_method_args args
-      args_type = args.map do |arg|
-        if arg.is_a? KatakataIrb::Types::Splat
-          simulate_evaluate arg.item, scope
-          nil # TODO: splat
-        else
-          simulate_evaluate arg, scope
-        end
-      end
-      simulate_call receiver_type, :[], args_type, kwargs_type(kwargs, scope), nil
-    in [:call | :vcall | :command | :command_call | :method_add_arg | :method_add_block,]
-      if (sexp in [:vcall, [:@ident, name,]]) && scope.has?(name)
-        # workaround for https://bugs.ruby-lang.org/issues/19175
-        return scope[name]
-      end
-      receiver, method, args, kwargs, block, optional_chain = retrieve_method_call sexp
-      if receiver.nil? && method == 'raise'
-        scope.terminate_with KatakataIrb::Scope::RAISE_BREAK, KatakataIrb::Types::TRUE
-        return KatakataIrb::Types::NIL
-      end
-      receiver_type = receiver ? simulate_evaluate(receiver, scope) : scope.self_type
+    when Prism::SourceEncodingNode
+      KatakataIrb::Types::InstanceType.new Encoding
+    when Prism::NumberedReferenceReadNode, Prism::BackReferenceReadNode
+      KatakataIrb::Types::UnionType[KatakataIrb::Types::STRING, KatakataIrb::Types::NIL]
+    when Prism::LocalVariableReadNode
+      scope[node.name.to_s] || KatakataIrb::Types::NIL
+    when Prism::ConstantReadNode, Prism::GlobalVariableReadNode, Prism::InstanceVariableReadNode, Prism::ClassVariableReadNode
+      scope[node.name.to_s] || KatakataIrb::Types::NIL
+    when Prism::CallNode
+      # TODO: return type of []=, field= when operator_loc.nil?
+      receiver_type = node.receiver ? evaluate(node.receiver, scope) : scope.self_type
       evaluate_method = lambda do |scope|
-        args_type = args.map do |arg|
-          if arg.is_a? KatakataIrb::Types::Splat
-            simulate_evaluate arg.item, scope
-            nil # TODO: splat
-          else
-            simulate_evaluate arg, scope
-          end
-        end
+        args_types, kwargs_types, block_sym_node, has_block = evaluate_call_node_arguments node, scope
 
-        if block
-          if block in [:symbol_literal, [:symbol, [:@ident, block_name,]]]
+        if block_sym_node
+          block_sym = block_sym_node.value
+          if @dig_targets.target? block_sym_node
+            # method(args, &:completion_target)
+            call_block_proc = ->(block_args, _self_type) do
+              block_receiver = block_args.first || KatakataIrb::Types::OBJECT
+              @dig_targets.resolve block_receiver, scope
+              KatakataIrb::Types::OBJECT
+            end
+          else
             call_block_proc = ->(block_args, _self_type) do
               block_receiver, *rest = block_args
-              block_receiver ? simulate_call(block_receiver || KatakataIrb::Types::OBJECT, block_name, rest, nil, nil) : KatakataIrb::Types::OBJECT
+              block_receiver ? simulate_call(block_receiver || KatakataIrb::Types::OBJECT, block_sym, rest, nil, nil, scope) : KatakataIrb::Types::OBJECT
             end
-          elsif block in [:do_block | :brace_block => type, block_var, body]
-            block_var in [:block_var, params,]
-            call_block_proc = ->(block_args, block_self_type) do
-              scope.conditional do |s|
-                if params
-                  names = extract_param_names(params)
-                else
-                  names = (1..max_numbered_params(body)).map { "_#{_1}" }
-                  params = [:params, names.map { [:@ident, _1, [0, 0]] }, nil, nil, nil, nil, nil, nil]
-                end
-                params_table = names.zip(block_args).to_h { [_1, _2 || KatakataIrb::Types::NIL] }
-                table = { **params_table, KatakataIrb::Scope::BREAK_RESULT => nil, KatakataIrb::Scope::NEXT_RESULT => nil }
-                table[KatakataIrb::Scope::SELF] = block_self_type if block_self_type
-                block_scope = KatakataIrb::Scope.new s, table
-                evaluate_assign_params params, block_args, block_scope
-                block_scope.conditional { evaluate_param_defaults params, _1 } if params
-                if type == :do_block
-                  result = simulate_evaluate body, block_scope
-                else
-                  result = body.map { simulate_evaluate _1, block_scope }.last
-                end
-                block_scope.merge_jumps
-                s.update block_scope
-                nexts = block_scope[KatakataIrb::Scope::NEXT_RESULT]
-                breaks = block_scope[KatakataIrb::Scope::BREAK_RESULT]
-                if block_scope.terminated?
-                  [KatakataIrb::Types::UnionType[*nexts], breaks]
-                else
-                  [KatakataIrb::Types::UnionType[result, *nexts], breaks]
-                end
+          end
+        elsif node.block.is_a? Prism::BlockNode
+          call_block_proc = ->(block_args, block_self_type) do
+            scope.conditional do |s|
+              numbered_parameters = node.block.locals.grep(/\A_[1-9]/).map(&:to_s)
+              params_table = node.block.locals.to_h { [_1.to_s, KatakataIrb::Types::NIL] }
+              table = { **params_table, KatakataIrb::Scope::BREAK_RESULT => nil, KatakataIrb::Scope::NEXT_RESULT => nil }
+              block_scope = KatakataIrb::Scope.new s, table, self_type: block_self_type, trace_ivar: !block_self_type
+              # TODO kwargs
+              if node.block.parameters&.parameters
+                # node.block.parameters is Prism::BlockParametersNode
+                assign_parameters node.block.parameters.parameters, block_scope, block_args, {}
+              elsif !numbered_parameters.empty?
+                assign_numbered_parameters numbered_parameters, block_scope, block_args, {}
+              end
+              result = node.block.body ? evaluate(node.block.body, block_scope) : KatakataIrb::Types::NIL
+              block_scope.merge_jumps
+              s.update block_scope
+              nexts = block_scope[KatakataIrb::Scope::NEXT_RESULT]
+              breaks = block_scope[KatakataIrb::Scope::BREAK_RESULT]
+              if block_scope.terminated?
+                [KatakataIrb::Types::UnionType[*nexts], breaks]
+              else
+                [KatakataIrb::Types::UnionType[result, *nexts], breaks]
               end
             end
-          else
-            call_block_proc = ->(_block_args, _self_type) { KatakataIrb::Types::OBJECT }
-            simulate_evaluate block, scope
           end
+        elsif has_block
+          call_block_proc = ->(_block_args, _self_type) { KatakataIrb::Types::OBJECT }
         end
-        simulate_call receiver_type, method, args_type, kwargs_type(kwargs, scope), call_block_proc
+        simulate_call receiver_type, node.name, args_types, kwargs_types, call_block_proc, scope
       end
-      if optional_chain
+      if node.call_operator == '&.'
         result = scope.conditional { evaluate_method.call _1 }
         if receiver_type.nillable?
           KatakataIrb::Types::UnionType[result, KatakataIrb::Types::NIL]
@@ -306,381 +242,671 @@ class KatakataIrb::TypeSimulator
       else
         evaluate_method.call scope
       end
-    in [:binary, a, Symbol => op, b]
-      atype = simulate_evaluate a, scope
-      case op
-      when :'&&', :and
-        btype = scope.conditional { simulate_evaluate b, _1 }
-        KatakataIrb::Types::UnionType[btype, KatakataIrb::Types::NIL, KatakataIrb::Types::FALSE]
-      when :'||', :or
-        btype = scope.conditional { simulate_evaluate b, _1 }
-        KatakataIrb::Types::UnionType[atype, btype]
+    when Prism::AndNode, Prism::OrNode
+      left = evaluate node.left, scope
+      right = scope.conditional { evaluate node.right, _1 }
+      if node.operator == '&&'
+        KatakataIrb::Types::UnionType[right, KatakataIrb::Types::NIL, KatakataIrb::Types::FALSE]
+      else
+        KatakataIrb::Types::UnionType[left, right]
+      end
+    when Prism::CallOperatorWriteNode, Prism::CallAndWriteNode, Prism::CallOrWriteNode
+      receiver_type = evaluate node.receiver, scope
+      args_types, kwargs_types, block_sym_node, has_block = evaluate_call_node_arguments node, scope
+      if block_sym_node
+        block_sym = block_sym_node.value
+        call_block_proc = ->(block_args, _self_type) do
+          block_receiver, *rest = block_args
+          block_receiver ? simulate_call(block_receiver || KatakataIrb::Types::OBJECT, block_sym, rest, nil, nil, scope) : KatakataIrb::Types::OBJECT
+        end
+      elsif has_block
+        call_block_proc = ->(_block_args, _self_type) { KatakataIrb::Types::OBJECT }
+      end
+      method = node.write_name.to_s.delete_suffix('=')
+      left = simulate_call receiver_type, method, args_types, kwargs_types, call_block_proc, scope
+      if node.operator == '&&='
+        right = scope.conditional { evaluate node.value, _1 }
+        KatakataIrb::Types::UnionType[right, KatakataIrb::Types::NIL, KatakataIrb::Types::FALSE]
+      elsif node.operator == '||='
+        right = scope.conditional { evaluate node.value, _1 }
+        KatakataIrb::Types::UnionType[left, right]
       else
-        btype = simulate_evaluate b, scope
-        simulate_call atype, op, [btype], nil, nil
-      end
-    in [:unary, op, receiver]
-      simulate_call simulate_evaluate(receiver, scope), op, [], nil, nil
-    in [:lambda, params, statements]
-      params in [:paren, params] # ->{}, -> do end
-      statements in [:bodystmt, statements, _unknown, _unknown, _unknown] # -> do end
-      params in [:paren, params]
-      params_table = extract_param_names(params).to_h { [_1, KatakataIrb::Types::NIL] }
-      block_scope = KatakataIrb::Scope.new scope, { **params_table, KatakataIrb::Scope::BREAK_RESULT => nil, KatakataIrb::Scope::NEXT_RESULT => nil, KatakataIrb::Scope::RETURN_RESULT => nil }
+        right = evaluate node.value, scope
+        simulate_call left, node.operator, [right], nil, nil, scope, name_match: false
+      end
+    when Prism::ClassVariableOperatorWriteNode, Prism::InstanceVariableOperatorWriteNode, Prism::LocalVariableOperatorWriteNode, Prism::GlobalVariableOperatorWriteNode
+      left = scope[node.name.to_s] || KatakataIrb::Types::OBJECT
+      right = evaluate node.value, scope
+      scope[node.name.to_s] = simulate_call left, node.operator, [right], nil, nil, scope, name_match: false
+    when Prism::ClassVariableAndWriteNode, Prism::InstanceVariableAndWriteNode, Prism::LocalVariableAndWriteNode, Prism::GlobalVariableAndWriteNode
+      right = scope.conditional { evaluate node.value, scope }
+      scope[node.name.to_s] = KatakataIrb::Types::UnionType[right, KatakataIrb::Types::NIL, KatakataIrb::Types::FALSE]
+    when Prism::ClassVariableOrWriteNode, Prism::InstanceVariableOrWriteNode, Prism::LocalVariableOrWriteNode, Prism::GlobalVariableOrWriteNode
+      left = scope[node.name.to_s] || KatakataIrb::Types::OBJECT
+      right = scope.conditional { evaluate node.value, scope }
+      scope[node.name.to_s] = KatakataIrb::Types::UnionType[left, right]
+    when Prism::ConstantOperatorWriteNode
+      left = scope[node.name.to_s] || KatakataIrb::Types::OBJECT
+      right = evaluate node.value, scope
+      scope[node.name.to_s] = simulate_call left, node.operator, [right], nil, nil, scope, name_match: false
+    when Prism::ConstantAndWriteNode
+      right = scope.conditional { evaluate node.value, scope }
+      scope[node.name.to_s] = KatakataIrb::Types::UnionType[right, KatakataIrb::Types::NIL, KatakataIrb::Types::FALSE]
+    when Prism::ConstantOrWriteNode
+      left = scope[node.name.to_s] || KatakataIrb::Types::OBJECT
+      right = scope.conditional { evaluate node.value, scope }
+      scope[node.name.to_s] = KatakataIrb::Types::UnionType[left, right]
+    when Prism::ConstantPathOperatorWriteNode
+      left, receiver, _parent_module, name = evaluate_constant_node node.target, scope
+      right = evaluate node.value, scope
+      value = simulate_call left, node.operator, [right], nil, nil, scope, name_match: false
+      const_path_write receiver, name, value, scope
+      value
+    when Prism::ConstantPathAndWriteNode
+      _left, receiver, _parent_module, name = evaluate_constant_node node.target, scope
+      right = scope.conditional { evaluate node.value, scope }
+      value = KatakataIrb::Types::UnionType[right, KatakataIrb::Types::NIL, KatakataIrb::Types::FALSE]
+      const_path_write receiver, name, value, scope
+      value
+    when Prism::ConstantPathOrWriteNode
+      left, receiver, _parent_module, name = evaluate_constant_node node.target, scope
+      right = scope.conditional { evaluate node.value, scope }
+      value = KatakataIrb::Types::UnionType[left, right]
+      const_path_write receiver, name, value, scope
+      value
+    when Prism::ConstantPathWriteNode
+      receiver = evaluate node.target.parent, scope if node.target.parent
+      value = evaluate node.value, scope
+      const_path_write receiver, node.target.child.name.to_s, value, scope
+      value
+    when Prism::LambdaNode
+      local_table = node.locals.to_h { [_1.to_s, KatakataIrb::Types::OBJECT] }
+      block_scope = KatakataIrb::Scope.new scope, { **local_table, KatakataIrb::Scope::BREAK_RESULT => nil, KatakataIrb::Scope::NEXT_RESULT => nil, KatakataIrb::Scope::RETURN_RESULT => nil }
       block_scope.conditional do |s|
-        evaluate_assign_params params, [], s
-        s.conditional { evaluate_param_defaults params, _1 }
-        statements.each { simulate_evaluate _1, s }
+        assign_parameters node.parameters.parameters, s, [], {} if node.parameters&.parameters
+        evaluate node.body, s if node.body
       end
       block_scope.merge_jumps
       scope.update block_scope
       KatakataIrb::Types::ProcType.new
-    in [:assign, [:var_field, [:@gvar | :@ivar | :@cvar | :@ident | :@const, name,]], value]
-      res = simulate_evaluate value, scope
-      scope[name] = res
-      res
-    in [:assign, [:aref_field, receiver, key], value]
-      simulate_evaluate receiver, scope
-      args, kwargs, _block = retrieve_method_args key
-      args.each do |arg|
-        item = arg.is_a?(KatakataIrb::Types::Splat) ? arg.item : arg
-        simulate_evaluate item, scope
-      end
-      kwargs_type kwargs, scope
-      simulate_evaluate value, scope
-    in [:assign, [:field, receiver, period, [:@ident,]], value]
-      simulate_evaluate receiver, scope
-      if period in [:@op, '&.',]
-        scope.conditional { simulate_evaluate value, scope }
-      else
-        simulate_evaluate value, scope
-      end
-    in [:opassign, target, [:@op, op,], value]
-      op = op.to_s.delete('=').to_sym
-      if target in [:var_field, *field]
-        receiver = [:var_ref, *field]
-      elsif target in [:field, *field]
-        receiver = [:call, *field]
-      elsif target in [:aref_field, *field]
-        receiver = [:aref, *field]
-      else
-        receiver = target
-      end
-      simulate_evaluate [:assign, target, [:binary, receiver, op, value]], scope
-    in [:assign, target, value]
-      simulate_evaluate target, scope
-      simulate_evaluate value, scope
-    in [:massign, targets, value]
-      targets in [:mlhs, *targets] # (a,b) = value
-      rhs = simulate_evaluate value, scope
-      evaluate_massign targets, rhs, scope
-      rhs
-    in [:mrhs_new_from_args | :mrhs_add_star,]
-      values, = evaluate_mrhs sexp, scope
-      KatakataIrb::Types::InstanceType.new Array, Elem: KatakataIrb::Types::UnionType[*values]
-    in [:ifop, cond, tval, fval]
-      simulate_evaluate cond, scope
+    when Prism::LocalVariableWriteNode, Prism::GlobalVariableWriteNode, Prism::InstanceVariableWriteNode, Prism::ClassVariableWriteNode, Prism::ConstantWriteNode
+      scope[node.name.to_s] = evaluate node.value, scope
+    when Prism::MultiWriteNode
+      evaluated_receivers = {}
+      evaluate_multi_write_receiver node, scope, evaluated_receivers
+      value = (
+        if node.value.is_a? Prism::ArrayNode
+          if node.value.elements.any?(Prism::SplatNode)
+            evaluate node.value, scope
+          else
+            node.value.elements.map do |n|
+              evaluate n, scope
+            end
+          end
+        elsif node.value
+          evaluate node.value, scope
+        else
+          # For syntax invalid code like `(*a).b`
+          KatakataIrb::Types::NIL
+        end
+      )
+      evaluate_multi_write node, value, scope, evaluated_receivers
+    when Prism::IfNode, Prism::UnlessNode
+      evaluate node.predicate, scope
       KatakataIrb::Types::UnionType[*scope.run_branches(
-        -> { simulate_evaluate tval, _1 },
-        -> { simulate_evaluate fval, _1 }
+        -> { node.statements ? evaluate(node.statements, _1) : KatakataIrb::Types::NIL },
+        -> { node.consequent ? evaluate(node.consequent, _1) : KatakataIrb::Types::NIL }
       )]
-    in [:if_mod | :unless_mod, cond, statement]
-      simulate_evaluate cond, scope
-      KatakataIrb::Types::UnionType[scope.conditional { simulate_evaluate statement, _1 }, KatakataIrb::Types::NIL]
-    in [:if | :unless | :elsif, cond, statements, else_statement]
-      simulate_evaluate cond, scope
-      results = scope.run_branches(
-        ->(s) { statements.map { simulate_evaluate _1, s }.last },
-        ->(s) { else_statement ? simulate_evaluate(else_statement, s) : KatakataIrb::Types::NIL }
-      )
-      results.empty? ? KatakataIrb::Types::NIL : KatakataIrb::Types::UnionType[*results]
-    in [:while | :until, cond, statements]
-      inner_scope = KatakataIrb::Scope.new scope, { KatakataIrb::Scope::BREAK_RESULT => nil }, passthrough: true
-      simulate_evaluate cond, inner_scope
-      inner_scope.conditional { |s| statements.each { simulate_evaluate _1, s } }
-      inner_scope.merge_jumps
-      scope.update inner_scope
-      breaks = inner_scope[KatakataIrb::Scope::BREAK_RESULT]
-      breaks ? KatakataIrb::Types::UnionType[breaks, KatakataIrb::Types::NIL] : KatakataIrb::Types::NIL
-    in [:while_mod | :until_mod, cond, statement]
-      inner_scope = KatakataIrb::Scope.new scope, { KatakataIrb::Scope::BREAK_RESULT => nil }, passthrough: true
-      simulate_evaluate cond, inner_scope
-      inner_scope.conditional { |s| simulate_evaluate statement, s }
+    when Prism::ElseNode
+      node.statements ? evaluate(node.statements, scope) : KatakataIrb::Types::NIL
+    when Prism::WhileNode, Prism::UntilNode
+      inner_scope = KatakataIrb::Scope.new scope, { KatakataIrb::Scope::BREAK_RESULT => nil }
+      evaluate node.predicate, inner_scope
+      if node.statements
+        inner_scope.conditional do |s|
+          evaluate node.statements, s
+        end
+      end
       inner_scope.merge_jumps
       scope.update inner_scope
       breaks = inner_scope[KatakataIrb::Scope::BREAK_RESULT]
       breaks ? KatakataIrb::Types::UnionType[breaks, KatakataIrb::Types::NIL] : KatakataIrb::Types::NIL
-    in [:break | :next | :return => jump_type, value]
-      internal_key = jump_type == :break ? KatakataIrb::Scope::BREAK_RESULT : jump_type == :next ? KatakataIrb::Scope::NEXT_RESULT : KatakataIrb::Scope::RETURN_RESULT
-      if value.empty?
-        jump_value = KatakataIrb::Types::NIL
-      else
-        values, kw = evaluate_mrhs value, scope
-        values << kw if kw
-        jump_value = values.size == 1 ? values.first : KatakataIrb::Types::InstanceType.new(Array, Elem: KatakataIrb::Types::UnionType[*values])
-      end
+    when Prism::BreakNode, Prism::NextNode, Prism::ReturnNode
+      internal_key = (
+        case node
+        when Prism::BreakNode
+          KatakataIrb::Scope::BREAK_RESULT
+        when Prism::NextNode
+          KatakataIrb::Scope::NEXT_RESULT
+        when Prism::ReturnNode
+          KatakataIrb::Scope::RETURN_RESULT
+        end
+      )
+      jump_value = (
+        arguments = node.arguments&.arguments
+        if arguments.nil? || arguments.empty?
+          KatakataIrb::Types::NIL
+        elsif arguments.size == 1 && !arguments.first.is_a?(Prism::SplatNode)
+          evaluate arguments.first, scope
+        else
+          KatakataIrb::Types.array_of evaluate_list_splat_items(arguments, scope)
+        end
+      )
       scope.terminate_with internal_key, jump_value
       KatakataIrb::Types::NIL
-    in [:return0]
-      scope.terminate_with KatakataIrb::Scope::RETURN_RESULT, KatakataIrb::Types::NIL
-      KatakataIrb::Types::NIL
-    in [:yield, args]
-      evaluate_mrhs args, scope
-      KatakataIrb::Types::OBJECT
-    in [:yield0]
+    when Prism::YieldNode
+      evaluate_list_splat_items node.arguments.arguments, scope if node.arguments
       KatakataIrb::Types::OBJECT
-    in [:redo | :retry]
+    when Prism::RedoNode, Prism::RetryNode
       scope.terminate
-    in [:zsuper]
+    when Prism::ForwardingSuperNode
       KatakataIrb::Types::OBJECT
-    in [:super, args]
-      args, kwargs, _block = retrieve_method_args args
-      args.each do |arg|
-        item = arg.is_a?(KatakataIrb::Types::Splat) ? arg.item : arg
-        simulate_evaluate item, scope
-      end
-      kwargs_type kwargs, scope
+    when Prism::SuperNode
+      evaluate_list_splat_items node.arguments.arguments, scope if node.arguments
       KatakataIrb::Types::OBJECT
-    in [:begin, body_stmt]
-      simulate_evaluate body_stmt, scope
-    in [:bodystmt, statements, rescue_stmt, _unknown, ensure_stmt]
-      statements = [statements] if statements in [Symbol,] # oneliner-def body
-      rescue_scope = KatakataIrb::Scope.new scope, { KatakataIrb::Scope::RAISE_BREAK => nil }, passthrough: true if rescue_stmt
-      return_type = statements.map { simulate_evaluate _1, rescue_scope || scope }.last
-      if rescue_stmt
-        rescue_scope.merge_jumps
-        scope.update rescue_scope
-        return_type = KatakataIrb::Types::UnionType[return_type, scope.conditional { simulate_evaluate rescue_stmt, _1 }]
-      end
-      simulate_evaluate ensure_stmt, scope if ensure_stmt
+    when Prism::BeginNode
+      return_type = node.statements ? evaluate(node.statements, scope) : KatakataIrb::Types::NIL
+      if node.rescue_clause
+        if node.else_clause
+          return_types = scope.run_branches(
+            ->{ evaluate node.rescue_clause, _1 },
+            ->{ evaluate node.else_clause, _1 }
+          )
+        else
+          return_types = [
+            return_type,
+            scope.conditional { evaluate node.rescue_clause, _1 }
+          ]
+        end
+        return_type = KatakataIrb::Types::UnionType[*return_types]
+      end
+      if node.ensure_clause&.statements
+        # ensure_clause is Prism::EnsureNode
+        evaluate node.ensure_clause.statements, scope
+      end
       return_type
-    in [:rescue, error_class_stmts, error_var_stmt, statements, rescue_stmt]
-      return_type = scope.conditional do |s|
-        if error_var_stmt in [:var_field, [:@ident, error_var,]]
-          if (error_class_stmts in [:mrhs_new_from_args, Array => stmts, stmt])
-            error_class_stmts = [*stmts, stmt]
+    when Prism::RescueNode
+      run_rescue = lambda do |s|
+        if node.reference
+          error_classes_type = evaluate_list_splat_items node.exceptions, s
+          error_types = error_classes_type.types.filter_map do
+            KatakataIrb::Types::InstanceType.new _1.module_or_class if _1.is_a?(KatakataIrb::Types::SingletonType)
           end
-          error_classes = (error_class_stmts || []).flat_map { simulate_evaluate _1, s }.uniq
-          error_types = error_classes.filter_map { KatakataIrb::Types::InstanceType.new _1.module_or_class if _1.is_a?(KatakataIrb::Types::SingletonType) }
           error_types << KatakataIrb::Types::InstanceType.new(StandardError) if error_types.empty?
-          s[error_var] = KatakataIrb::Types::UnionType[*error_types]
+          error_type = KatakataIrb::Types::UnionType[*error_types]
+          case node.reference
+          when Prism::LocalVariableTargetNode, Prism::InstanceVariableTargetNode, Prism::ClassVariableTargetNode, Prism::GlobalVariableTargetNode, Prism::ConstantTargetNode
+            s[node.reference.name.to_s] = error_type
+          when Prism::CallNode
+            evaluate node.reference, s
+          end
         end
-        statements.map { simulate_evaluate _1, s }.last
+        node.statements ? evaluate(node.statements, s) : KatakataIrb::Types::NIL
       end
-      if rescue_stmt
-        return_type = KatakataIrb::Types::UnionType[return_type, scope.conditional { simulate_evaluate rescue_stmt, _1 }]
+      if node.consequent # begin; rescue A; rescue B; end
+        types = scope.run_branches(
+          run_rescue,
+          -> { evaluate node.consequent, _1 }
+        )
+        KatakataIrb::Types::UnionType[*types]
+      else
+        run_rescue.call scope
       end
-      return_type
-    in [:rescue_mod, statement1, statement2]
-      rescue_scope = KatakataIrb::Scope.new scope, { KatakataIrb::Scope::RAISE_BREAK => nil }, passthrough: true
-      a = simulate_evaluate statement1, rescue_scope
-      rescue_scope.merge_jumps
-      scope.update rescue_scope
-      b = scope.conditional { simulate_evaluate statement2, _1 }
+    when Prism::RescueModifierNode
+      a = evaluate node.expression, scope
+      b = scope.conditional { evaluate node.rescue_expression, _1 }
       KatakataIrb::Types::UnionType[a, b]
-    in [:module, module_stmt, body_stmt]
-      module_types = simulate_evaluate(module_stmt, scope).types.grep(KatakataIrb::Types::SingletonType)
-      module_types << KatakataIrb::Types::MODULE if module_types.empty?
-      module_scope = KatakataIrb::Scope.new(scope, { KatakataIrb::Scope::SELF => KatakataIrb::Types::UnionType[*module_types], KatakataIrb::Scope::BREAK_RESULT => nil, KatakataIrb::Scope::NEXT_RESULT => nil, KatakataIrb::Scope::RETURN_RESULT => nil }, trace_cvar: false, trace_ivar: false, trace_lvar: false)
-      result = simulate_evaluate body_stmt, module_scope
-      scope.update module_scope
-      result
-    in [:sclass, klass_stmt, body_stmt]
-      klass_types = simulate_evaluate(klass_stmt, scope).types.filter_map do |type|
+    when Prism::SingletonClassNode
+      klass_types = evaluate(node.expression, scope).types.filter_map do |type|
         KatakataIrb::Types::SingletonType.new type.klass if type.is_a? KatakataIrb::Types::InstanceType
       end
       klass_types = [KatakataIrb::Types::CLASS] if klass_types.empty?
-      sclass_scope = KatakataIrb::Scope.new(scope, { KatakataIrb::Scope::SELF => KatakataIrb::Types::UnionType[*klass_types], KatakataIrb::Scope::BREAK_RESULT => nil, KatakataIrb::Scope::NEXT_RESULT => nil, KatakataIrb::Scope::RETURN_RESULT => nil }, trace_cvar: false, trace_ivar: false, trace_lvar: false)
-      result = simulate_evaluate body_stmt, sclass_scope
+      table = node.locals.to_h { [_1.to_s, KatakataIrb::Types::NIL] }
+      sclass_scope = KatakataIrb::Scope.new(
+        scope,
+        { **table, KatakataIrb::Scope::BREAK_RESULT => nil, KatakataIrb::Scope::NEXT_RESULT => nil, KatakataIrb::Scope::RETURN_RESULT => nil },
+        trace_ivar: false,
+        trace_lvar: false,
+        self_type: KatakataIrb::Types::UnionType[*klass_types]
+      )
+      result = node.body ? evaluate(node.body, sclass_scope) : KatakataIrb::Types::NIL
       scope.update sclass_scope
       result
-    in [:class, klass_stmt, superclass_stmt, body_stmt]
-      klass_types = simulate_evaluate(klass_stmt, scope).types
-      klass_types += simulate_evaluate(superclass_stmt, scope).types if superclass_stmt
-      klass_types = klass_types.select do |type|
-        type.is_a?(KatakataIrb::Types::SingletonType) && type.module_or_class.is_a?(Class)
-      end
-      klass_types << KatakataIrb::Types::CLASS if klass_types.empty?
-      klass_scope = KatakataIrb::Scope.new(scope, { KatakataIrb::Scope::SELF => KatakataIrb::Types::UnionType[*klass_types], KatakataIrb::Scope::BREAK_RESULT => nil, KatakataIrb::Scope::NEXT_RESULT => nil, KatakataIrb::Scope::RETURN_RESULT => nil }, trace_cvar: false, trace_ivar: false, trace_lvar: false)
-      result = simulate_evaluate body_stmt, klass_scope
-      scope.update klass_scope
+    when Prism::ModuleNode, Prism::ClassNode
+      unless node.constant_path.is_a?(Prism::ConstantReadNode) || node.constant_path.is_a?(Prism::ConstantPathNode)
+        # Syntax error code, example: `module a.b; end`
+        return KatakataIrb::Types::NIL
+      end
+      const_type, _receiver, parent_module, name = evaluate_constant_node node.constant_path, scope
+      if node.is_a? Prism::ModuleNode
+        module_types = const_type.types.select { _1.is_a?(KatakataIrb::Types::SingletonType) && !_1.module_or_class.is_a?(Class) }
+        module_types << KatakataIrb::Types::MODULE if module_types.empty?
+      else
+        select_class_type = -> { _1.is_a?(KatakataIrb::Types::SingletonType) && _1.module_or_class.is_a?(Class) }
+        module_types = const_type.types.select(&select_class_type)
+        module_types += evaluate(node.superclass, scope).types.select(&select_class_type) if node.superclass
+        module_types << KatakataIrb::Types::CLASS if module_types.empty?
+      end
+      return KatakataIrb::Types::NIL unless node.body
+
+      table = node.locals.to_h { [_1.to_s, KatakataIrb::Types::NIL] }
+      if parent_module.is_a?(Module) || parent_module.nil?
+        value = parent_module.const_get name if parent_module&.const_defined?(name)
+        unless value
+          value_type = scope[name]
+          value = value_type.module_or_class if value_type.is_a? KatakataIrb::Types::SingletonType
+        end
+
+        if value.is_a? Module
+          nesting = [value, []]
+        else
+          if parent_module
+            nesting = [parent_module, [name]]
+          else
+            parent_nesting, parent_path = scope.module_nesting.first
+            nesting = [parent_nesting, parent_path + [name]]
+          end
+          nesting_key = [nesting[0].__id__, nesting[1]].join('::')
+          nesting_value = node.is_a?(Prism::ModuleNode) ? KatakataIrb::Types::MODULE : KatakataIrb::Types::CLASS
+        end
+      else
+        # parent_module == :unknown
+        # TODO: dummy module
+      end
+      module_scope = KatakataIrb::Scope.new(
+        scope,
+        { **table, KatakataIrb::Scope::BREAK_RESULT => nil, KatakataIrb::Scope::NEXT_RESULT => nil, KatakataIrb::Scope::RETURN_RESULT => nil },
+        trace_cvar: false,
+        trace_ivar: false,
+        trace_lvar: false,
+        self_type: KatakataIrb::Types::UnionType[*module_types],
+        nesting: nesting
+      )
+      module_scope[nesting_key] = nesting_value if nesting_value
+      result = evaluate(node.body, module_scope)
+      scope.update module_scope
       result
-    in [:for, fields, enum, statements]
-      fields = [fields] if fields in [:var_field | :field | :aref_field,]
-      params = [:params, fields, nil, nil, nil, nil, nil, nil]
-      enum = simulate_evaluate enum, scope
-      extract_param_names(params).each { scope[_1] = KatakataIrb::Types::NIL }
-      response = simulate_call enum, :first, [], nil, nil
-      evaluate_assign_params params, [response], scope
-      inner_scope = KatakataIrb::Scope.new scope, { KatakataIrb::Scope::BREAK_RESULT => nil }, passthrough: true
-      scope.conditional do |s|
-        statements.each { simulate_evaluate _1, s }
+    when Prism::ForNode
+      node.statements
+      collection = evaluate node.collection, scope
+      inner_scope = KatakataIrb::Scope.new scope, { KatakataIrb::Scope::BREAK_RESULT => nil }
+      ary_type = simulate_call collection, :to_ary, [], nil, nil, nil, name_match: false
+      element_types = ary_type.types.filter_map do |ary|
+        ary.params[:Elem] if ary.is_a?(KatakataIrb::Types::InstanceType) && ary.klass == Array
+      end
+      element_type = KatakataIrb::Types::UnionType[*element_types]
+      inner_scope.conditional do |s|
+        evaluate_write node.index, element_type, s, nil
+        evaluate node.statements, s if node.statements
       end
       inner_scope.merge_jumps
       scope.update inner_scope
       breaks = inner_scope[KatakataIrb::Scope::BREAK_RESULT]
-      breaks ? KatakataIrb::Types::UnionType[breaks, enum] : enum
-    in [:when, pattern, if_statements, else_statement]
-      eval_pattern = lambda do |s, pattern, *rest|
-        simulate_evaluate pattern, s
-        scope.conditional { eval_pattern.call(_1, *rest) } if rest.any?
-      end
-      if_branch = lambda do |s|
-        eval_pattern.call(s, *pattern)
-        if_statements.map { simulate_evaluate _1, s }.last
-      end
-      else_branch = lambda do |s|
-        pattern.each { simulate_evaluate _1, s }
-        simulate_evaluate(else_statement, s, case_target: case_target)
-      end
-      if if_statements && else_statement
-        KatakataIrb::Types::UnionType[*scope.run_branches(if_branch, else_branch)]
-      else
-        KatakataIrb::Types::UnionType[scope.conditional { (if_branch || else_branch).call _1 }, KatakataIrb::Types::NIL]
-      end
-    in [:in, [:var_field, [:@ident, name,]], if_statements, else_statement]
-      scope.never { simulate_evaluate else_statement, scope } if else_statement
-      scope[name] = case_target || KatakataIrb::Types::OBJECT
-      if_statements ? if_statements.map { simulate_evaluate _1, scope }.last : KatakataIrb::Types::NIL
-    in [:in, pattern, if_statements, else_statement]
-      pattern_scope = KatakataIrb::Scope.new(scope, { KatakataIrb::Scope::PATTERNMATCH_BREAK => nil }, passthrough: true)
-      results = pattern_scope.run_branches(
-        ->(s) {
-          match_pattern case_target, pattern, s
-          if_statements ? if_statements.map { simulate_evaluate _1, s }.last : KatakataIrb::Types::NIL
-        },
-        ->(s) {
-          else_statement ? simulate_evaluate(else_statement, s, case_target: case_target) : KatakataIrb::Types::NIL
-        }
-      )
-      pattern_scope.merge_jumps
-      scope.update pattern_scope
-      KatakataIrb::Types::UnionType[*results]
-    in [:case, target_exp, match_exp]
-      target = target_exp ? simulate_evaluate(target_exp, scope) : KatakataIrb::Types::NIL
-      simulate_evaluate match_exp, scope, case_target: target
-    in [:void_stmt]
-      KatakataIrb::Types::NIL
-    in [:dot2 | :dot3, range_beg, range_end]
-      beg_type = simulate_evaluate range_beg, scope if range_beg
-      end_type = simulate_evaluate range_end, scope if range_end
+      breaks ? KatakataIrb::Types::UnionType[breaks, collection] : collection
+    when Prism::CaseNode
+      target = evaluate(node.predicate, scope) if node.predicate
+      # TODO
+      branches = node.conditions.map do |condition|
+        ->(s) { evaluate_case_match target, condition, s }
+      end
+      if node.consequent
+        branches << ->(s) { evaluate node.consequent, s }
+      elsif node.conditions.any? { _1.is_a? Prism::WhenNode }
+        branches << ->(s) { KatakataIrb::Types::NIL }
+      end
+      KatakataIrb::Types::UnionType[*scope.run_branches(*branches)]
+    when Prism::MatchRequiredNode
+      value_type = evaluate node.value, scope
+      evaluate_match_pattern value_type, node.pattern, scope
+      KatakataIrb::Types::NIL # void value
+    when Prism::MatchPredicateNode
+      value_type = evaluate node.value, scope
+      scope.conditional { evaluate_match_pattern value_type, node.pattern, _1 }
+      KatakataIrb::Types::BOOLEAN
+    when Prism::RangeNode
+      beg_type = evaluate node.left, scope if node.left
+      end_type = evaluate node.right, scope if node.right
       elem = (KatakataIrb::Types::UnionType[*[beg_type, end_type].compact]).nonnillable
-      KatakataIrb::Types::InstanceType.new Range, { Elem: elem }
-    in [:top_const_ref, [:@const, name,]]
-      KatakataIrb::BaseScope.type_of { Object.const_get name }
-    in [:string_concat, a, b]
-      simulate_evaluate a, scope
-      simulate_evaluate b, scope
-      KatakataIrb::Types::STRING
-    in [:defined, expression]
-      scope.conditional { simulate_evaluate expression, _1 }
+      KatakataIrb::Types::InstanceType.new Range, Elem: elem
+    when Prism::DefinedNode
+      scope.conditional { evaluate node.value, _1 }
       KatakataIrb::Types::UnionType[KatakataIrb::Types::STRING, KatakataIrb::Types::NIL]
+    when Prism::FlipFlopNode
+      scope.conditional { evaluate node.left, _1 } if node.left
+      scope.conditional { evaluate node.right, _1 } if node.right
+      KatakataIrb::Types::BOOLEAN
+    when Prism::MultiTargetNode
+      # Raw MultiTargetNode, incomplete code like `a,b`, `*a`.
+      evaluate_multi_write_receiver node, scope, nil
+      KatakataIrb::Types::NIL
+    when Prism::ImplicitNode
+      evaluate node.value, scope
+    when Prism::MatchWriteNode
+      # /(?<a>)(?<b>)/ =~ string
+      evaluate node.call, scope
+      node.locals.each { scope[_1.to_s] = KatakataIrb::Types::UnionType[KatakataIrb::Types::STRING, KatakataIrb::Types::NIL] }
+      KatakataIrb::Types::BOOLEAN
+    when Prism::MatchLastLineNode
+      KatakataIrb::Types::BOOLEAN
+    when Prism::InterpolatedMatchLastLineNode
+      node.parts.each { evaluate _1, scope }
+      KatakataIrb::Types::BOOLEAN
+    when Prism::PreExecutionNode, Prism::PostExecutionNode
+      node.statements ? evaluate(node.statements, scope) : KatakataIrb::Types::NIL
+    when Prism::AliasMethodNode, Prism::AliasGlobalVariableNode, Prism::UndefNode, Prism::MissingNode
+      # do nothing
+      KatakataIrb::Types::NIL
     else
       KatakataIrb.log_puts
       KatakataIrb.log_puts :NOMATCH
-      KatakataIrb.log_puts sexp.inspect
+      KatakataIrb.log_puts node.inspect
       KatakataIrb::Types::NIL
     end
   end
 
-  def match_pattern(target, pattern, scope)
-    breakable = -> { scope.terminate_with KatakataIrb::Scope::PATTERNMATCH_BREAK, KatakataIrb::Types::NIL }
-    types = target.types
-    case pattern
-    in [:var_field, [:@ident, name,]]
-      scope[name] = target
-    in [:var_ref,] # in Array, in ^a, in nil
-    in [:@int | :@float | :@rational | :@imaginary | :@CHAR | :symbol_literal | :string_literal | :regexp_literal,]
-    in [:begin, statement] # in (statement)
-      simulate_evaluate statement, scope
-      breakable.call
-    in [:binary, lpattern, :|, rpattern]
-      match_pattern target, lpattern, scope
-      scope.conditional { match_pattern target, rpattern, _1 }
-      breakable.call
-    in [:binary, lpattern, :'=>', [:var_field, [:@ident, name,]] => rpattern]
-      if lpattern in [:var_ref, [:@const, _const_name,]]
-        const_value = simulate_evaluate lpattern, scope
-        if const_value.is_a?(KatakataIrb::Types::SingletonType) && const_value.module_or_class.is_a?(Class)
-          scope[name] = KatakataIrb::Types::InstanceType.new const_value.module_or_class
-        else
-          scope[name] = KatakataIrb::Types::OBJECT
-        end
-        breakable.call
+  def evaluate_call_node_arguments(call_node, scope)
+    # call_node.arguments is Prism::ArgumentsNode
+    arguments = call_node.arguments&.arguments&.dup || []
+    block_arg = call_node.block.expression if call_node.respond_to?(:block) && call_node.block.is_a?(Prism::BlockArgumentNode)
+    kwargs = arguments.pop.elements if arguments.last.is_a?(Prism::KeywordHashNode)
+    args_types = arguments.map do |arg|
+      case arg
+      when Prism::ForwardingArgumentsNode
+        # `f(a, ...)` treat like splat
+        nil
+      when Prism::SplatNode
+        evaluate arg.expression, scope
+        nil # TODO: splat
       else
-        match_pattern target, lpattern, scope
-        match_pattern target, rpattern, scope
-      end
-    in [:aryptn, _unknown, items, splat, post_items]
-      # TODO: deconstruct keys
-      array_types = types.select { _1.is_a?(KatakataIrb::Types::InstanceType) && _1.klass == Array }
-      elem = KatakataIrb::Types::UnionType[*array_types.filter_map { _1.params[:Elem] }]
-      items&.each do |item|
-        match_pattern elem, item, scope
-      end
-      if splat in [:var_field, [:@ident, name,]]
-        scope[name] = KatakataIrb::Types::InstanceType.new Array, Elem: elem
-        breakable.call
-      end
-      post_items&.each do |item|
-        match_pattern elem, item, scope
-      end
-    in [:hshptn, _unknown, items, splat]
-      # TODO: deconstruct keys
-      hash_types = types.select { _1.is_a?(KatakataIrb::Types::InstanceType) && _1.klass == Hash }
-      key_type = KatakataIrb::Types::UnionType[*hash_types.filter_map { _1.params[:K] }]
-      value_type = KatakataIrb::Types::UnionType[*hash_types.filter_map { _1.params[:V] }]
-      items&.each do |key_pattern, value_pattern|
-        if (key_pattern in [:@label, label,]) && !value_pattern
-          name = label.delete ':'
-          scope[name] = value_type
-          breakable.call
+        evaluate arg, scope
+      end
+    end
+    if kwargs
+      kwargs_types = kwargs.map do |arg|
+        case arg
+        when Prism::AssocNode
+          if arg.key.is_a?(Prism::SymbolNode)
+            [arg.key.value, evaluate(arg.value, scope)]
+          else
+            evaluate arg.key, scope
+            evaluate arg.value, scope
+            nil
+          end
+        when Prism::AssocSplatNode
+          evaluate arg.value, scope
+          nil
+        end
+      end.compact.to_h
+    end
+    if block_arg.is_a? Prism::SymbolNode
+      block_sym_node = block_arg
+    elsif block_arg
+      evaluate block_arg, scope
+    end
+    [args_types, kwargs_types, block_sym_node, !!block_arg]
+  end
+
+  def const_path_write(receiver, name, value, scope)
+    if receiver # receiver::A = value
+      singleton_type = receiver.types.find { _1.is_a? KatakataIrb::Types::SingletonType }
+      scope.set_const singleton_type.module_or_class, name, value if singleton_type
+    else # ::A = value
+      scope.set_const Object, name, value
+    end
+  end
+
+  def assign_required_parameter(node, value, scope)
+    case node
+    when Prism::RequiredParameterNode
+      scope[node.name.to_s] = value || KatakataIrb::Types::OBJECT
+    when Prism::RequiredDestructuredParameterNode
+      values = value ? sized_splat(value, :to_ary, node.parameters.size) : []
+      node.parameters.zip values do |n, v|
+        assign_required_parameter n, v, scope
+      end
+    when Prism::SplatNode
+      splat_value = value ? KatakataIrb::Types.array_of(value) : KatakataIrb::Types::ARRAY
+      assign_required_parameter node.expression, splat_value, scope
+    end
+  end
+
+  def evaluate_constant_node(node, scope)
+    case node
+    when Prism::ConstantPathNode
+      name = node.child.name.to_s
+      if node.parent
+        receiver = evaluate node.parent, scope
+        if receiver.is_a? KatakataIrb::Types::SingletonType
+          parent_module = receiver.module_or_class
         end
-        match_pattern value_type, value_pattern, scope if value_pattern
-      end
-      if splat in [:var_field, [:@ident, name,]]
-        scope[name] = KatakataIrb::Types::InstanceType.new Hash, K: key_type, V: value_type
-        breakable.call
-      end
-    in [:if_mod, cond, ifpattern]
-      match_pattern target, ifpattern, scope
-      simulate_evaluate cond, scope
-      breakable.call
-    in [:dyna_symbol,]
-    in [:const_path_ref,]
+      else
+        parent_module = Object
+      end
+      if parent_module
+        type = scope.get_const(parent_module, [name]) || KatakataIrb::Types::NIL
+      else
+        parent_module = :unknown
+        type = KatakataIrb::Types::NIL
+      end
+    when Prism::ConstantReadNode
+      name = node.name.to_s
+      type = scope[name]
+    end
+    [type, receiver, parent_module, name]
+  end
+
+
+  def assign_parameters(node, scope, args, kwargs)
+    args = args.dup
+    kwargs = kwargs.dup
+    size = node.requireds.size + node.optionals.size + (node.rest ? 1 : 0) + node.posts.size
+    args = sized_splat(args.first, :to_ary, size) if size >= 2 && args.size == 1
+    reqs = args.shift node.requireds.size
+    if node.rest
+      # node.rest.class is Prism::RestParameterNode
+      posts = []
+      opts = args.shift node.optionals.size
+      rest = args
     else
-      KatakataIrb.log_puts "Unimplemented match pattern: #{pattern}"
+      posts = args.pop node.posts.size
+      opts = args
+      rest = []
+    end
+    node.requireds.zip reqs do |n, v|
+      assign_required_parameter n, v, scope
+    end
+    node.optionals.zip opts do |n, v|
+      # n is Prism::OptionalParameterNode
+      values = [v]
+      values << evaluate(n.value, scope) if n.value
+      scope[n.name.to_s] = KatakataIrb::Types::UnionType[*values.compact]
+    end
+    node.posts.zip posts do |n, v|
+      assign_required_parameter n, v, scope
+    end
+    if node.rest&.name
+      # node.rest is Prism::RestParameterNode
+      scope[node.rest.name.to_s] = KatakataIrb::Types.array_of(*rest)
+    end
+    node.keywords.each do |n|
+      # n is Prism::KeywordParameterNode
+      name = n.name.to_s.delete(':')
+      values = [kwargs.delete(name)]
+      values << evaluate(n.value, scope) if n.value
+      scope[name] = KatakataIrb::Types::UnionType[*values.compact]
+    end
+    # node.keyword_rest is Prism::KeywordRestParameterNode or Prism::ForwardingParameterNode or Prism::NoKeywordsParameterNode
+    if node.keyword_rest.is_a?(Prism::KeywordRestParameterNode) && node.keyword_rest.name
+      scope[node.keyword_rest.name.to_s] = KatakataIrb::Types::InstanceType.new(Hash, K: KatakataIrb::Types::SYMBOL, V: KatakataIrb::Types::UnionType[*kwargs.values])
+    end
+    if node.block&.name
+      # node.block is Prism::BlockParameterNode
+      scope[node.block.name.to_s] = KatakataIrb::Types::PROC
     end
   end
 
-  def evaluate_mrhs(sexp, scope)
-    args, kwargs, = retrieve_method_args sexp
-    values = args.filter_map do |t|
-      if t.is_a? KatakataIrb::Types::Splat
-        simulate_evaluate t.item, scope
-        # TODO
-        nil
+  def assign_numbered_parameters(numbered_parameters, scope, args, _kwargs)
+    return if numbered_parameters.empty?
+    max_num = numbered_parameters.map { _1[1].to_i }.max
+    if max_num == 1
+      if args.size == 0
+        scope['_1'] = KatakataIrb::Types::NIL
+      elsif args.size == 1
+        scope['_1'] = args.first
       else
-        simulate_evaluate t, scope
+        scope['_1'] = KatakataIrb::Types.array_of(*args)
       end
+    else
+      args = sized_splat(args.first, :to_ary, max_num) if args.size == 1
+      numbered_parameters.each do |name|
+        index = name[1].to_i - 1
+        scope[name] = args[index] || KatakataIrb::Types::NIL
+      end
+    end
+  end
+
+  def evaluate_case_match(target, node, scope)
+    case node
+    when Prism::WhenNode
+      node.conditions.each { evaluate _1, scope }
+      node.statements ? evaluate(node.statements, scope) : KatakataIrb::Types::NIL
+    when Prism::InNode
+      pattern = node.pattern
+      if pattern in Prism::IfNode | Prism::UnlessNode
+        cond_node = pattern.predicate
+        pattern = pattern.statements.body.first
+      end
+      evaluate_match_pattern(target, pattern, scope)
+      evaluate cond_node, scope if cond_node # TODO: conditional branch
+      node.statements ? evaluate(node.statements, scope) : KatakataIrb::Types::NIL
     end
-    unless kwargs.empty?
-      kvs = kwargs.map do |t|
-        case t
-        in KatakataIrb::Types::Splat
-          simulate_evaluate t.item, scope
-          # TODO
-          [KatakataIrb::Types::SYMBOL, KatakataIrb::Types::OBJECT]
-        in [key, value]
-          key_type = (key in [:@label,]) ? KatakataIrb::Types::SYMBOL : simulate_evaluate(key, scope)
-          [key_type, simulate_evaluate(value, scope)]
+  end
+
+  def evaluate_match_pattern(value, pattern, scope)
+    # TODO: scope.terminate_with KatakataIrb::Scope::PATTERNMATCH_BREAK, KatakataIrb::Types::NIL
+    case pattern
+    when Prism::FindPatternNode
+      # TODO
+      evaluate_match_pattern KatakataIrb::Types::OBJECT, pattern.left, scope
+      pattern.requireds.each { evaluate_match_pattern KatakataIrb::Types::OBJECT, _1, scope }
+      evaluate_match_pattern KatakataIrb::Types::OBJECT, pattern.right, scope
+    when Prism::ArrayPatternNode
+      # TODO
+      pattern.requireds.each { evaluate_match_pattern KatakataIrb::Types::OBJECT, _1, scope }
+      evaluate_match_pattern KatakataIrb::Types::OBJECT, pattern.rest, scope if pattern.rest
+      pattern.posts.each { evaluate_match_pattern KatakataIrb::Types::OBJECT, _1, scope }
+      KatakataIrb::Types::ARRAY
+    when Prism::HashPatternNode
+      # TODO
+      pattern.assocs.each { evaluate_match_pattern KatakataIrb::Types::OBJECT, _1, scope }
+      KatakataIrb::Types::HASH
+    when Prism::AssocNode
+      evaluate_match_pattern value, pattern.value, scope if pattern.value
+      KatakataIrb::Types::OBJECT
+    when Prism::AssocSplatNode
+      # TODO
+      evaluate_match_pattern KatakataIrb::Types::HASH, pattern.value, scope
+      KatakataIrb::Types::OBJECT
+    when Prism::PinnedVariableNode
+      evaluate pattern.variable, scope
+    when Prism::PinnedExpressionNode
+      evaluate pattern.expression, scope
+    when Prism::LocalVariableTargetNode
+      scope[pattern.name.to_s] = value
+    when Prism::AlternationPatternNode
+      KatakataIrb::Types::UnionType[evaluate_match_pattern(value, pattern.left, scope), evaluate_match_pattern(value, pattern.right, scope)]
+    when Prism::CapturePatternNode
+      capture_type = class_or_value_to_instance evaluate_match_pattern(value, pattern.value, scope)
+      value = capture_type unless capture_type.types.empty? || capture_type.types == [KatakataIrb::Types::OBJECT]
+      evaluate_match_pattern value, pattern.target, scope
+    when Prism::SplatNode
+      value = KatakataIrb::Types.array_of value
+      evaluate_match_pattern value, pattern.expression, scope if pattern.expression
+      value
+    else
+      # literal node
+      type = evaluate(pattern, scope)
+      class_or_value_to_instance(type)
+    end
+  end
+
+  def class_or_value_to_instance(type)
+    instance_types = type.types.map do |t|
+      t.is_a?(KatakataIrb::Types::SingletonType) ? KatakataIrb::Types::InstanceType.new(t.module_or_class) : t
+    end
+    KatakataIrb::Types::UnionType[*instance_types]
+  end
+
+  def evaluate_write(node, value, scope, evaluated_receivers)
+    case node
+    when Prism::MultiTargetNode
+      evaluate_multi_write node, value, scope, evaluated_receivers
+    when Prism::CallNode
+      evaluated_receivers&.[](node.receiver) || evaluate(node.receiver, scope) if node.receiver
+    when Prism::SplatNode
+      evaluate_write node.expression, KatakataIrb::Types.array_of(value), scope, evaluated_receivers
+    when Prism::LocalVariableTargetNode, Prism::GlobalVariableTargetNode, Prism::InstanceVariableTargetNode, Prism::ClassVariableTargetNode, Prism::ConstantTargetNode
+      scope[node.name.to_s] = value
+    when Prism::ConstantPathTargetNode
+      receiver = evaluated_receivers&.[](node.parent) || evaluate(node.parent, scope) if node.parent
+      const_path_write receiver, node.child.name.to_s, value, scope
+      value
+    end
+  end
+
+  def evaluate_multi_write(node, values, scope, evaluated_receivers)
+    values = sized_splat values, :to_ary, node.targets.size unless values.is_a? Array
+    splat_index = node.targets.find_index { _1.is_a? Prism::SplatNode }
+    if splat_index
+      pre_targets = node.targets[0...splat_index]
+      splat_target = node.targets[splat_index]
+      post_targets = node.targets[splat_index + 1..]
+      pre_values = values.shift pre_targets.size
+      post_values = values.pop post_targets.size
+      splat_value = KatakataIrb::Types::UnionType[*values]
+      zips = pre_targets.zip(pre_values) + [[splat_target, splat_value]] + post_targets.zip(post_values)
+    else
+      zips = node.targets.zip(values)
+    end
+    zips.each do |target, value|
+      evaluate_write target, value || KatakataIrb::Types::NIL, scope, evaluated_receivers
+    end
+  end
+
+  def evaluate_multi_write_receiver(node, scope, evaluated_receivers)
+    case node
+    when Prism::MultiWriteNode, Prism::MultiTargetNode
+      node.targets.each { evaluate_multi_write_receiver _1, scope, evaluated_receivers }
+    when Prism::CallNode
+      if node.receiver
+        receiver = evaluate(node.receiver, scope)
+        evaluated_receivers[node.receiver] = receiver if evaluated_receivers
+      end
+      if node.arguments
+        node.arguments.arguments&.each do |arg|
+          if arg.is_a? Prism::SplatNode
+            evaluate arg.expression, scope
+          else
+            evaluate arg, scope
+          end
         end
       end
-      key_type = KatakataIrb::Types::UnionType[*kvs.map(&:first)]
-      value_type = KatakataIrb::Types::UnionType[*kvs.map(&:last)]
-      kw = KatakataIrb::Types::InstanceType.new(Hash, K: key_type, V: value_type)
+    when Prism::SplatNode
+      evaluate_multi_write_receiver node.expression, scope, evaluated_receivers if node.expression
     end
-    [values, kw]
+  end
+
+  def evaluate_list_splat_items(list, scope)
+    items = list.flat_map do |node|
+      if node.is_a? Prism::SplatNode
+        splat = evaluate node.expression, scope
+        array_elem, non_array = partition_to_array splat.nonnillable, :to_a
+        [*array_elem, *non_array]
+      else
+        evaluate node, scope
+      end
+    end.uniq
+    KatakataIrb::Types::UnionType[*items]
   end
 
   def sized_splat(value, method, size)
@@ -693,7 +919,7 @@ class KatakataIrb::TypeSimulator
   def partition_to_array(value, method)
     arrays, non_arrays = value.types.partition { _1.is_a?(KatakataIrb::Types::InstanceType) && _1.klass == Array }
     non_arrays.select! do |type|
-      to_array_result = simulate_call type, method, [], nil, nil, name_match: false
+      to_array_result = simulate_call type, method, [], nil, nil, nil, name_match: false
       if to_array_result.is_a?(KatakataIrb::Types::InstanceType) && to_array_result.klass == Array
         arrays << to_array_result
         false
@@ -706,147 +932,7 @@ class KatakataIrb::TypeSimulator
     [array_elem, non_array]
   end
 
-  def evaluate_massign(sexp, values, scope)
-    values = sized_splat values, :to_ary, sexp.size unless values.is_a? Array
-    rest_index = sexp.find_index { _1 in [:rest_param, ]}
-    if rest_index
-      pre = rest_index ? sexp[0...rest_index] : sexp
-      post = rest_index ? sexp[rest_index + 1..] : []
-      sexp[rest_index] in [:rest_param, rest_field]
-      rest_values = values[pre.size...values.size - post.size] || []
-      rest_type = KatakataIrb::Types::InstanceType.new Array, Elem: KatakataIrb::Types::UnionType[*rest_values]
-      pairs = pre.zip(values.first(pre.size)) + [[rest_field, rest_type]] + post.zip(values.last(post.size))
-    else
-      pairs = sexp.zip values
-    end
-    pairs.each do |field, value|
-      case field
-      in [:@ident, name,]
-        # block arg mlhs
-        scope[name] = value || KatakataIrb::Types::OBJECT
-      in [:var_field, [:@gvar | :@ivar | :@cvar | :@ident | :@const, name,]]
-        # massign
-        scope[name] = value || KatakataIrb::Types::OBJECT
-      in [:mlhs, *mlhs]
-        evaluate_massign mlhs, value || [], scope
-      in [:field, receiver,]
-        # (a=x).b, c = value
-        simulate_evaluate receiver, scope
-      in [:aref_field, *field]
-        # (a=x)[i=y, j=z], b = value
-        simulate_evaluate [:aref, *field], scope
-      in nil
-        # a, *, b = value
-      end
-    end
-  end
-
-  def kwargs_type(kwargs, scope)
-    return if kwargs.empty?
-    keys = []
-    values = []
-    kwargs.each do |kv|
-      if kv.is_a? KatakataIrb::Types::Splat
-        hash = simulate_evaluate kv.item, scope
-        unless hash.is_a?(KatakataIrb::Types::InstanceType) && hash.klass == Hash
-          hash = simulate_call hash, :to_hash, [], nil, nil
-        end
-        if hash.is_a?(KatakataIrb::Types::InstanceType) && hash.klass == Hash
-          keys << hash.params[:K] if hash.params[:K]
-          values << hash.params[:V] if hash.params[:V]
-        end
-      else
-        key, value = kv
-        keys << ((key in [:@label,]) ? KatakataIrb::Types::SYMBOL : simulate_evaluate(key, scope))
-        values << simulate_evaluate(value, scope)
-      end
-    end
-    KatakataIrb::Types::InstanceType.new(Hash, K: KatakataIrb::Types::UnionType[*keys], V: KatakataIrb::Types::UnionType[*values])
-  end
-
-  def retrieve_method_call(sexp)
-    optional = -> { _1 in [:@op, '&.',] }
-    case sexp
-    in [:fcall | :vcall, [:@ident | :@const | :@kw | :@op, method,]] # hoge
-      [nil, method, [], [], nil, false]
-    in [:call, receiver, [:@period,] | [:@op, '&.',] | [:@op, '::',] | :'::' => dot, :call]
-      [receiver, :call, [], [], nil, optional[dot]]
-    in [:call, receiver, [:@period,] | [:@op, '&.',] | [:@op, '::',] | :'::' => dot, method]
-      method => [:@ident | :@const | :@kw | :@op, method,] unless method == :call
-      [receiver, method, [], [], nil, optional[dot]]
-    in [:command, [:@ident | :@const | :@kw | :@op, method,], args] # hoge 1, 2
-      args, kwargs, block = retrieve_method_args args
-      [nil, method, args, kwargs, block, false]
-    in [:command_call, receiver, [:@period,] | [:@op, '&.',] | :'::' => dot, [:@ident | :@const | :@kw | :@op, method,], args] # a.hoge 1; a.hoge 1, 2;
-      args, kwargs, block = retrieve_method_args args
-      [receiver, method, args, kwargs, block, optional[dot]]
-    in [:method_add_arg, call, args]
-      receiver, method, _arg, _kwarg, _block, opt = retrieve_method_call call
-      args, kwargs, block = retrieve_method_args args
-      [receiver, method, args, kwargs, block, opt]
-    in [:method_add_block, call, block]
-      receiver, method, args, kwargs, opt = retrieve_method_call call
-      [receiver, method, args, kwargs, block, opt]
-    end
-  end
-
-  def retrieve_method_args(sexp)
-    case sexp
-    in [:mrhs_add_star, args, star]
-      args, = retrieve_method_args args
-      [[*args, KatakataIrb::Types::Splat.new(star)], [], nil]
-    in [:mrhs_new_from_args, [:args_add_star,] => args]
-      args, = retrieve_method_args args
-      [args, [], nil]
-    in [:mrhs_new_from_args, [:args_add_star,] => args, last_arg]
-      args, = retrieve_method_args args
-      [[*args, last_arg], [], nil]
-    in [:mrhs_new_from_args, args, last_arg]
-      [[*args, last_arg], [], nil]
-    in [:mrhs_new_from_args, args]
-      [args, [], nil]
-    in [:args_add_block, [:args_add_star,] => args, block_arg]
-      args, kwargs, = retrieve_method_args args
-      block_arg = [:void_stmt] if block_arg.nil? # method(*splat, &)
-      [args, kwargs, block_arg]
-    in [:args_add_block, [*args, [:bare_assoc_hash,] => kw], block_arg]
-      block_arg = [:void_stmt] if block_arg.nil? # method(**splat, &)
-      _, kwargs = retrieve_method_args kw
-      [args, kwargs, block_arg]
-    in [:args_add_block, [*args], block_arg]
-      block_arg = [:void_stmt] if block_arg.nil? # method(arg, &)
-      [args, [], block_arg]
-    in [:bare_assoc_hash, kws]
-      kwargs = []
-      kws.each do |kw|
-        if kw in [:assoc_splat, value,]
-          kwargs << KatakataIrb::Types::Splat.new(value) if value
-        elsif kw in [:assoc_new, [:@label, label,] => key, nil]
-          name = label.delete ':'
-          kwargs << [key, [:__var_ref_or_call, [name =~ /\A[A-Z]/ ? :@const : :@ident, name, [0, 0]]]]
-        elsif kw in [:assoc_new, key, value]
-          kwargs << [key, value]
-        end
-      end
-      [[], kwargs, nil]
-    in [:args_add_star, *args, [:bare_assoc_hash,] => kwargs]
-      args, = retrieve_method_args [:args_add_star, *args]
-      _, kwargs = retrieve_method_args kwargs
-      [args, kwargs, nil]
-    in [:args_add_star, pre_args, star_arg, *post_args]
-      pre_args, = retrieve_method_args pre_args if pre_args in [:args_add_star,]
-      args = star_arg ? [*pre_args, KatakataIrb::Types::Splat.new(star_arg), *post_args] : pre_args + post_args
-      [args, [], nil]
-    in [:arg_paren, args]
-      args ? retrieve_method_args(args) : [[], [], nil]
-    in [[:command | :command_call, ] => command_arg] # method(a b, c), method(a.b c, d)
-      [[command_arg], [], nil]
-    else
-      [[], [], nil]
-    end
-  end
-
-  def simulate_call(receiver, method_name, args, kwargs, block, name_match: true)
+  def simulate_call(receiver, method_name, args, kwargs, block, scope, name_match: true)
     methods = KatakataIrb::Types.rbs_methods receiver, method_name.to_sym, args, kwargs, !!block
     block_called = false
     type_breaks = methods.map do |method, given_params, method_params|
@@ -862,10 +948,15 @@ class KatakataIrb::TypeSimulator
         block_called = true
         vars.merge! KatakataIrb::Types.match_free_variables(free_vars - vars.keys.to_set, [method.block.type.return_type], [block_response])
       end
-      [KatakataIrb::Types.from_rbs_type(method.type.return_type, receiver, vars || {}), breaks]
+      if KatakataIrb::Types.method_return_bottom?(method)
+        [nil, breaks]
+      else
+        [KatakataIrb::Types.from_rbs_type(method.type.return_type, receiver, vars || {}), breaks]
+      end
     end
     block&.call [], nil unless block_called
-    types = type_breaks.map(&:first)
+    terminates = !type_breaks.empty? && type_breaks.map(&:first).all?(&:nil?)
+    types = type_breaks.map(&:first).compact
     breaks = type_breaks.map(&:last).compact
     types << OBJECT_METHODS[method_name.to_sym] if name_match && OBJECT_METHODS.has_key?(method_name.to_sym)
 
@@ -876,120 +967,24 @@ class KatakataIrb::TypeSimulator
         end
       end
     end
+    scope&.terminate if terminates && breaks.empty?
     KatakataIrb::Types::UnionType[*types, *breaks]
   end
 
-  def extract_param_names(params)
-    params => [:params, pre_required, optional, rest, post_required, keywords, keyrest, block]
-    names = []
-    extract_mlhs = ->(item) do
-      case item
-      in [:var_field, [:@ident, name,],]
-        names << name
-      in [:@ident, name,]
-        names << name
-      in [:mlhs, *items]
-        items.each(&extract_mlhs)
-      in [:rest_param, item]
-        extract_mlhs.call item if item
-      in [:field | :aref_field,]
-        # a.b, c[i] = value
-      in [:excessed_comma]
-      in [:args_forward]
-      end
-    end
-    [*pre_required, *post_required].each(&extract_mlhs)
-    extract_mlhs.call rest if rest
-    optional&.each do |key, _value|
-      key => [:@ident, name,]
-      names << name
-    end
-    keywords&.each do |key, _value|
-      key => [:@label, label,]
-      names << label.delete(':')
-    end
-    if keyrest in [:kwrest_params, [:@ident, name,]]
-      names << name
-    end
-    if block in [:blockarg, [:@ident, name,]]
-      names << name
-    end
-    names
-  end
-
-  def evaluate_assign_params(params, values, scope)
-    values = values.dup
-    params => [:params, pre_required, optional, rest, post_required, _keywords, keyrest, block]
-    size = (pre_required&.size || 0) + (optional&.size || 0) + (post_required&.size || 0) + (rest ? 1 : 0)
-    values = sized_splat values.first, :to_ary, size if values.size == 1 && size >= 2
-    pre_values = values.shift pre_required.size if pre_required
-    post_values = values.pop post_required.size if post_required
-    opt_values = values.shift optional.size if optional
-    rest_values = values
-    evaluate_massign pre_required, pre_values, scope if pre_required
-    evaluate_massign optional.map(&:first), opt_values, scope if optional
-    if rest in [:rest_param, [:@ident, name,]]
-      scope[name] = KatakataIrb::Types::InstanceType.new Array, Elem: KatakataIrb::Types::UnionType[*rest_values]
-    end
-    evaluate_massign post_required, post_values, scope if post_required
-    # TODO: assign keywords
-    if keyrest in [:kwrest_param, [:@ident, name,]]
-      scope[name] = KatakataIrb::Types::InstanceType.new Hash, K: KatakataIrb::Types::SYMBOL, V: KatakataIrb::Types::OBJECT
-    end
-    if block in [:blockarg, [:@ident, name,]]
-      scope[name] = KatakataIrb::Types::PROC
-    end
-  end
-
-  def evaluate_param_defaults(params, scope)
-    params => [:params, _pre_required, optional, rest, _post_required, keywords, keyrest, block]
-    optional&.each do |item, value|
-      item => [:@ident, name,]
-      scope[name] = simulate_evaluate value, scope
-    end
-    if rest in [:rest_param, [:@ident, name,]]
-      scope[name] = KatakataIrb::Types::ARRAY
-    end
-    keywords&.each do |key, value|
-      key => [:@label, label,]
-      name = label.delete ':'
-      scope[name] = value ? simulate_evaluate(value, scope) : KatakataIrb::Types::OBJECT
-    end
-    if keyrest in [:kwrest_param, [:@ident, name,]]
-        scope[name] = KatakataIrb::Types::HASH
-    end
-    if block in [:blockarg, [:@ident, name,]]
-      scope[name] = KatakataIrb::Types::PROC
-    end
-  end
-
-  def max_numbered_params(sexp)
-    case sexp
-    in [:do_block | :brace_block | :def | :class | :module,]
-      0
-    in [:var_ref, [:@ident, name,]]
-      name.match?(/\A_[1-9]\z/) ? name[1..].to_i : 0
-    else
-      sexp.filter_map do |s|
-        max_numbered_params s if s.is_a? Array
-      end.max || 0
-    end
-  end
-
-  def self.calculate_binding_scope(binding, parents, target)
-    dig_targets = DigTarget.new(parents, target) do |_types, scope|
-      return scope
+  def evaluate_program(program, scope)
+    # statements.body[0] is local variable assign code
+    program.statements.body[1..].each do |statement|
+      evaluate statement, scope
     end
-    scope = KatakataIrb::Scope.from_binding(binding)
-    new(dig_targets).simulate_evaluate parents[0], scope
-    scope
   end
 
-  def self.calculate_receiver(binding, parents, receiver)
-    dig_targets = DigTarget.new(parents, receiver) do |type, _scope|
-      return type
+  def self.calculate_target_type_scope(binding, parents, target)
+    dig_targets = DigTarget.new(parents, target) do |type, scope|
+      return type, scope
     end
-    new(dig_targets).simulate_evaluate parents[0], KatakataIrb::Scope.from_binding(binding)
-    KatakataIrb::Types::NIL
+    program = parents.first
+    scope = KatakataIrb::Scope.from_binding(binding, program.locals)
+    new(dig_targets).evaluate_program program, scope
+    [KatakataIrb::Types::NIL, scope]
   end
 end