katakata_irb 0.1.12 → 0.2.1

data/lib/katakata_irb/scope.rb

@@ -1,42 +1,71 @@
+# frozen_string_literal: true
+
 require 'set'
 require_relative 'types'
 
 module KatakataIrb
   class BaseScope
-    SELF = '%self'
     BREAK_RESULT = '%break'
     NEXT_RESULT = '%next'
     RETURN_RESULT = '%return'
     PATTERNMATCH_BREAK = '%match'
-    RAISE_BREAK = '%raise'
 
-    def initialize(binding, self_object)
-      @binding, @self_object = binding, self_object
-      @cache = { SELF => KatakataIrb::Types.type_from_object(self_object) }
-      @local_variables = binding.local_variables.map(&:to_s).to_set
-      @global_variables = global_variables.map(&:to_s).to_set
+    attr_reader :module_nesting, :self_object
+
+    def initialize(binding, self_object, local_variables)
+      @binding = binding
+      @self_object = self_object
+      @cache = {}
+      modules = [*binding.eval('Module.nesting'), Object]
+      @module_nesting = modules.map { [_1, []] }
+      binding_local_variables = binding.local_variables
+      uninitialized_locals = local_variables - binding_local_variables
+      uninitialized_locals.each { @cache[_1] = KatakataIrb::Types::NIL }
+      @local_variables = (local_variables | binding_local_variables).map(&:to_s).to_set
+      @global_variables = Kernel.global_variables.map(&:to_s).to_set
+      @owned_constants_cache = {}
     end
 
     def level() = 0
 
-    def level_of(name)
-      has?(name) ? 0 : nil
-    end
+    def level_of(_name, _var_type) = 0
 
     def mutable?() = false
 
+    def module_own_constant?(mod, name)
+      set = (@owned_constants_cache[mod] ||= Set.new(mod.constants.map(&:to_s)))
+      set.include? name
+    end
+
+    def get_const(nesting, path, _key = nil)
+      return unless nesting
+
+      result = path.reduce nesting do |mod, name|
+        return nil unless mod.is_a?(Module) && module_own_constant?(mod, name)
+        mod.const_get name
+      end
+      KatakataIrb::Types.type_from_object result
+    end
+
+    def get_cvar(nesting, path, name, _key = nil)
+      return KatakataIrb::Types::NIL unless nesting
+
+      result = path.reduce nesting do |mod, n|
+        return KatakataIrb::Types::NIL unless mod.is_a?(Module) && module_own_constant?(mod, n)
+        mod.const_get n
+      end
+      value = result.class_variable_get name if result.is_a?(Module) && name.size >= 3 && result.class_variable_defined?(name)
+      Types.type_from_object value
+    end
+
     def [](name)
       @cache[name] ||= (
         fallback = KatakataIrb::Types::NIL
         case BaseScope.type_by_name name
-        when :cvar
-          BaseScope.type_of(fallback: fallback) { @self_object.class_variable_get name }
         when :ivar
           BaseScope.type_of(fallback: fallback) { @self_object.instance_variable_get name }
         when :lvar
           BaseScope.type_of(fallback: fallback) { @binding.local_variable_get(name) }
-        when :const
-          BaseScope.type_of(fallback: fallback) { @binding.eval name }
         when :gvar
           BaseScope.type_of(fallback: fallback) { @binding.eval name if @global_variables.include? name }
         end
@@ -44,12 +73,12 @@ module KatakataIrb
     end
 
     def self_type
-      self[SELF]
+      KatakataIrb::Types.type_from_object @self_object
     end
 
-    def local_variables
-      @local_variables.to_a
-    end
+    def local_variables() = @local_variables.to_a
+
+    def global_variables() = @global_variables.to_a
 
     def self.type_of(fallback: KatakataIrb::Types::OBJECT)
       begin
@@ -61,6 +90,7 @@ module KatakataIrb
 
     def self.type_by_name(name)
       if name.start_with? '@@'
+        # "@@cvar" or "@@cvar::[module_id]::[module_path]"
         :cvar
       elsif name.start_with? '@'
         :ivar
@@ -68,49 +98,30 @@ module KatakataIrb
         :gvar
       elsif name.start_with? '%'
         :internal
-      elsif name[0].downcase != name[0]
+      elsif name[0].downcase != name[0] || name[0].match?(/\d/)
+        # "ConstName" or "[module_id]::[const_path]"
         :const
       else
         :lvar
       end
     end
-
-    def has?(name)
-      case BaseScope.type_by_name name
-      when :lvar
-        @local_variables.include? name
-      when :const
-        begin
-          @binding.eval(name)
-          true
-        rescue
-          false
-        end
-      when :gvar, :cvar, :ivar
-        true
-      when :internal
-        true
-      end
-    end
   end
 
   class Scope < BaseScope
-    attr_reader :parent, :jump_branches, :mergeable_changes, :level, :lvars
+    attr_reader :parent, :mergeable_changes, :level, :module_nesting
 
-    def self.from_binding(binding) = new(BaseScope.new(binding, binding.eval('self')))
+    def self.from_binding(binding, locals) = new(BaseScope.new(binding, binding.eval('self'), locals))
 
-    def initialize(parent, table = {}, trace_cvar: true, trace_ivar: true, trace_lvar: true, passthrough: false)
-      @table = table
+    def initialize(parent, table = {}, trace_ivar: true, trace_lvar: true, self_type: nil, nesting: nil)
       @parent = parent
-      @level = parent.level + (passthrough ? 0 : 1)
-      @trace_cvar = trace_cvar
+      @level = parent.level + 1
       @trace_ivar = trace_ivar
       @trace_lvar = trace_lvar
-      @passthrough = passthrough
+      @module_nesting = nesting ? [nesting, *parent.module_nesting] : parent.module_nesting
+      @self_type = self_type
       @terminated = false
       @jump_branches = []
-      @lvars = Set.new
-      @mergeable_changes = @changes = table.transform_values { [level, _1] }
+      @mergeable_changes = @table = table.transform_values { [level, _1] }
     end
 
     def mutable? = true
@@ -138,53 +149,202 @@ module KatakataIrb
     def terminate
       return if terminated?
       @terminated = true
-      @changes = @mergeable_changes.dup
+      @table = @mergeable_changes.dup
     end
 
-    def branch_table_clone() = @tables.last.dup
-
     def trace?(name)
       return false unless @parent
       type = BaseScope.type_by_name(name)
-      type == :cvar ? @trace_cvar : type == :ivar ? @trace_ivar : type == :lvar ? @trace_lvar : true
+      type == :ivar ? @trace_ivar : type == :lvar ? @trace_lvar : true
+    end
+
+    def level_of(name, var_type)
+      case var_type
+      when :ivar
+        return level unless @trace_ivar
+      when :gvar
+        return 0
+      end
+      variable_level, = @table[name]
+      variable_level || parent.level_of(name, var_type)
     end
 
-    def level_of(name)
-      variable_level, = @changes[name]
-      variable_level || parent.level_of(name)
+    def get_const(nesting, path, key = nil)
+      key ||= [nesting.__id__, path].join('::')
+      _l, value = @table[key]
+      value || @parent.get_const(nesting, path, key)
+    end
+
+    def get_cvar(nesting, path, name, key = nil)
+      key ||= [name, nesting.__id__, path].join('::')
+      _l, value = @table[key]
+      value || @parent.get_cvar(nesting, path, name, key)
     end
 
     def [](name)
-      level, value = @changes[name]
+      type = BaseScope.type_by_name(name)
+      if type == :const
+        return get_const(nil, nil, name) || KatakataIrb::Types::NIL if name.include?('::')
+
+        module_nesting.each do |(nesting, path)|
+          value = get_const nesting, [*path, name]
+          return value if value
+        end
+        return KatakataIrb::Types::NIL
+      elsif type == :cvar
+        return get_cvar(nil, nil, nil, name) if name.include?('::')
+
+        nesting, path = module_nesting.first
+        return get_cvar(nesting, path, name)
+      end
+      level, value = @table[name]
       if level
         value
       elsif trace? name
-        @parent[name] if trace? name
+        @parent[name]
+      elsif type == :ivar
+        self_instance_variable_get name
       end
     end
 
+    def set_const(nesting, path, value)
+      key = [nesting.__id__, path].join('::')
+      @table[key] = [0, value]
+    end
+
+    def set_cvar(nesting, path, name, value)
+      key = [name, nesting.__id__, path].join('::')
+      @table[key] = [0, value]
+    end
+
     def []=(name, value)
-      variable_level = level_of(name) || level
-      @lvars << name if level == variable_level && BaseScope.type_by_name(name) == :lvar
-      @changes[name] = [variable_level, value]
+      type = BaseScope.type_by_name(name)
+      if type == :const
+        if name.include?('::')
+          @table[name] = [0, value]
+        else
+          parent_module, parent_path = module_nesting.first
+          set_const parent_module, [*parent_path, name], value
+        end
+        return
+      elsif type == :cvar
+        if name.include?('::')
+          @table[name] = [0, value]
+        else
+          parent_module, parent_path = module_nesting.first
+          set_cvar parent_module, parent_path, name, value
+        end
+        return
+      end
+      variable_level = level_of name, type
+      @table[name] = [variable_level, value] if variable_level
     end
 
     def self_type
-      self[SELF]
+      @self_type || @parent.self_type
+    end
+
+    def global_variables
+      gvar_keys = @table.keys.select do |name|
+        BaseScope.type_by_name(name) == :gvar
+      end
+      gvar_keys | @parent.global_variables
     end
 
     def local_variables
-      lvar_keys = @changes.keys.select do |name|
+      lvar_keys = @table.keys.select do |name|
         BaseScope.type_by_name(name) == :lvar
       end
       lvar_keys |= @parent.local_variables if @trace_lvar
       lvar_keys
     end
 
+    def table_constants
+      constants = module_nesting.flat_map do |mod, path|
+        prefix = [mod.__id__, *path].join('::') + '::'
+        @table.keys.select { _1.start_with? prefix }.map { _1.delete_prefix(prefix).split('::').first }
+      end.uniq
+      constants |= @parent.table_constants if @parent.mutable?
+      constants
+    end
+
+    def table_module_constants(mod)
+      prefix = "#{mod.__id__}::"
+      constants = @table.keys.select { _1.start_with? prefix }.map { _1.delete_prefix(prefix).split('::').first }
+      constants |= @parent.table_constants if @parent.mutable?
+      constants
+    end
+
+    def base_scope
+      @parent.mutable? ? @parent.base_scope : @parent
+    end
+
+    def table_instance_variables
+      ivars = @table.keys.select { BaseScope.type_by_name(_1) == :ivar }
+      ivars |= @parent.table_instance_variables if @parent.mutable? && @trace_ivar
+      ivars
+    end
+
+    def instance_variables
+      self_singleton_types = self_type.types.grep(KatakataIrb::Types::SingletonType)
+      singleton_classes = self_type.types.grep(KatakataIrb::Types::InstanceType).map(&:klass).select(&:singleton_class?)
+      base_self = base_scope.self_object
+      self_instance_variables = singleton_classes.flat_map do |singleton_class|
+        if singleton_class.respond_to? :attached_object
+          singleton_class.attached_object.instance_variables.map(&:to_s)
+        elsif singleton_class == base_self.singleton_class
+          base_self.instance_variables.map(&:to_s)
+        else
+          []
+        end
+      end
+      [
+        self_singleton_types.flat_map { _1.module_or_class.instance_variables.map(&:to_s) },
+        self_instance_variables || [],
+        table_instance_variables
+      ].inject(:|)
+    end
+
+    def self_instance_variable_get(name)
+      self_objects = self_type.types.grep(KatakataIrb::Types::SingletonType).map(&:module_or_class)
+      singleton_classes = self_type.types.grep(KatakataIrb::Types::InstanceType).map(&:klass).select(&:singleton_class?)
+      base_self = base_scope.self_object
+      singleton_classes.each do |singleton_class|
+        if singleton_class.respond_to? :attached_object
+          self_objects << singleton_class.attached_object
+        elsif singleton_class == base_self.singleton_class
+          self_objects << base_self
+        end
+      end
+      types = self_objects.map do |object|
+        BaseScope.type_of(fallback: KatakataIrb::Types::NIL) { object.instance_variable_get name }
+      end
+      KatakataIrb::Types::UnionType[*types]
+    end
+
+    def table_class_variables
+      cvars = @table.keys.filter_map { _1.split('::', 2).first if BaseScope.type_by_name(_1) == :cvar }
+      cvars |= @parent.table_class_variables if @parent.mutable?
+      cvars
+    end
+
+    def class_variables
+      cvars = table_class_variables
+      m, = module_nesting.first
+      cvars |= m.class_variables.map(&:to_s) if m.is_a? Module
+      cvars
+    end
+
+    def constants
+      module_nesting.flat_map do |nest,|
+        nest.constants
+      end.map(&:to_s) | table_constants
+    end
+
     def merge_jumps
       if terminated?
         @terminated = false
-        @changes = @mergeable_changes
+        @table = @mergeable_changes
         merge @jump_branches
         @terminated = true
       else
@@ -197,27 +357,15 @@ module KatakataIrb
     end
 
     def never(&block)
-      block.call Scope.new(self, { BREAK_RESULT => nil, NEXT_RESULT => nil, PATTERNMATCH_BREAK => nil, RETURN_RESULT => nil, RAISE_BREAK => nil }, passthrough: true)
-    end
-
-    def run(*args, **option)
-      scope = Scope.new(self, *args, **option)
-      yield scope
-      merge_jumps
-      update scope
-    end
-
-    def touch_lvars(lvars)
-      lvars.each { self[_1] = self[_1] || KatakataIrb::Types::NIL }
+      block.call Scope.new(self, { BREAK_RESULT => nil, NEXT_RESULT => nil, PATTERNMATCH_BREAK => nil, RETURN_RESULT => nil })
     end
 
     def run_branches(*blocks)
       results = []
       branches = []
       blocks.each do |block|
-        scope = Scope.new self, passthrough: true
+        scope = Scope.new self
         result = block.call scope
-        touch_lvars scope.lvars if scope.level == level
         next if scope.terminated?
         results << result
         branches << scope.mergeable_changes
@@ -227,17 +375,12 @@ module KatakataIrb
       results
     end
 
-    def has?(name)
-      has_own?(name) || (trace?(name) && @parent.has?(name))
-    end
-
     def has_own?(name)
-      @changes.key? name
+      @table.key? name
     end
 
     def update(child_scope)
       current_level = level
-      touch_lvars child_scope.lvars if child_scope.level == current_level
       child_scope.mergeable_changes.each do |name, (level, value)|
         self[name] = value if level <= current_level
       end