typeprof 0.21.11 → 0.30.0

data/lib/typeprof/core/env/static_read.rb

@@ -0,0 +1,155 @@
+module TypeProf::Core
+  class StaticRead
+    def initialize(name)
+      @name = name
+      @followers = Set[]
+      @source_modules = Set[]
+    end
+
+    attr_reader :name, :followers
+
+    def propagate(genv)
+      @followers.each do |follower|
+        case follower
+        when ModuleEntity
+          genv.add_static_eval_queue(:parent_modules_changed, follower)
+        when ScopedStaticRead
+          follower.on_cbase_updated(genv)
+        when Box, IsAFilter
+          genv.add_run(follower)
+        else
+          raise follower.inspect
+        end
+      end
+    end
+
+    def destroy(genv)
+      @source_modules.each do |mod|
+        mod.static_reads[@name].delete(self) || raise
+      end
+      @source_modules.clear
+    end
+
+    def resolve(genv, cref, break_object)
+      destroy(genv)
+
+      first = true
+      while cref
+        scope = cref.cpath
+        mod = genv.resolve_cpath(scope)
+        genv.each_superclass(mod, false) do |mod, _singleton|
+          break if mod == genv.mod_object && break_object
+
+          unless @source_modules.include?(mod)
+            @source_modules << mod
+            (mod.static_reads[@name] ||= Set[]) << self
+          end
+
+          return if check_module(genv, mod)
+
+          break unless first
+        end
+        first = false
+        cref = cref.outer
+      end
+      resolution_failed(genv)
+    end
+  end
+
+  class BaseStaticRead < StaticRead
+    def initialize(genv, name, cref)
+      super(name)
+      @cref = cref
+      genv.add_static_eval_queue(:static_read_changed, self)
+    end
+
+    attr_reader :cref
+
+    def on_scope_updated(genv)
+      resolve(genv, @cref, false)
+    end
+  end
+
+  class ScopedStaticRead < StaticRead
+    def initialize(name, cbase)
+      super(name)
+      @cbase = cbase
+      @cbase.followers << self if @cbase
+    end
+
+    def on_cbase_updated(genv)
+      if @cbase && @cbase.cpath
+        resolve(genv, CRef.new(@cbase.cpath, :class, nil, nil), true)
+      else
+        resolution_failed(genv)
+      end
+    end
+  end
+
+  module ConstRead
+    def check_module(genv, mod)
+      cdef = mod.consts[@name]
+      if cdef && cdef.exist?
+        inner_mod = genv.resolve_cpath(mod.cpath + [@name]) # TODO
+        cpath = inner_mod.exist? ? inner_mod.cpath : nil
+        update_module(genv, cpath, cdef)
+        return true
+      end
+      return false
+    end
+
+    def resolution_failed(genv)
+      update_module(genv, nil, nil)
+    end
+
+    def update_module(genv, cpath, cdef)
+      if cpath != @cpath || cdef != @cdef
+        @cpath = cpath
+        @cdef = cdef
+        propagate(genv)
+      end
+    end
+
+    attr_reader :cpath, :cdef
+  end
+
+  class BaseConstRead < BaseStaticRead
+    include ConstRead
+  end
+
+  class ScopedConstRead < ScopedStaticRead
+    include ConstRead
+  end
+
+  module TypeAliasRead
+    def check_module(genv, mod)
+      tae = mod.type_aliases[@name]
+      if tae && tae.exist?
+        update_type_alias(genv, tae)
+        return true
+      end
+      return false
+    end
+
+    def resolution_failed(genv)
+      update_type_alias(genv, nil)
+    end
+
+    def update_type_alias(genv, tae)
+      if tae != @type_alias_entity
+        @type_alias_entity = tae
+        propagate(genv)
+      end
+    end
+
+    attr_reader :type_alias_entity
+  end
+
+  class BaseTypeAliasRead < BaseStaticRead
+    include TypeAliasRead
+  end
+
+  class ScopedTypeAliasRead < ScopedStaticRead
+    include TypeAliasRead
+  end
+end

data/lib/typeprof/core/env/type_alias_entity.rb

@@ -0,0 +1,27 @@
+module TypeProf::Core
+  class TypeAliasEntity
+    def initialize
+      @decls = Set[]
+      @type = nil
+    end
+
+    attr_reader :decls, :type
+
+    def exist?
+      !@decls.empty?
+    end
+
+    def add_decl(decl)
+      @decls << decl
+      @type = decl.type unless @type
+      # TODO: report an error if there are duplicated declarations
+    end
+
+    def remove_decl(decl)
+      @decls.delete(decl) || raise
+      if @type == decl.type
+        @type = @decls.empty? ? nil : @decls.to_a.first.type
+      end
+    end
+  end
+end

data/lib/typeprof/core/env/value_entity.rb

@@ -0,0 +1,32 @@
+module TypeProf::Core
+  class ValueEntity
+    def initialize
+      @decls = Set[]
+      @defs = Set[]
+      @read_boxes = Set[]
+      @vtx = Vertex.new(self)
+    end
+
+    attr_reader :decls, :defs, :read_boxes, :vtx
+
+    def add_decl(decl)
+      @decls << decl
+    end
+
+    def remove_decl(decl)
+      @decls.delete(decl) || raise
+    end
+
+    def add_def(def_)
+      @defs << def_
+    end
+
+    def remove_def(def_)
+      @defs.delete(def_) || raise
+    end
+
+    def exist?
+      !@decls.empty? || !@defs.empty?
+    end
+  end
+end

data/lib/typeprof/core/env.rb

@@ -0,0 +1,360 @@
+module TypeProf::Core
+  class GlobalEnv
+    def initialize
+      @type_table = {}
+
+      @static_eval_queue = []
+
+      @run_queue = []
+      @run_queue_set = Set[]
+
+      @mod_object = ModuleEntity.new([])
+      @mod_object.inner_modules[:Object] = @mod_object
+
+      @mod_basic_object = resolve_cpath([:BasicObject])
+      @mod_class = resolve_cpath([:Class])
+      @mod_module = resolve_cpath([:Module])
+
+      @gvars = {}
+      @mod_ary = resolve_cpath([:Array])
+      @mod_hash = resolve_cpath([:Hash])
+      @mod_range = resolve_cpath([:Range])
+      @mod_str = resolve_cpath([:String])
+
+      @cls_type = Type::Instance.new(self, @mod_class, [])
+      @mod_type = Type::Instance.new(self, @mod_module, [])
+      @obj_type = Type::Instance.new(self, resolve_cpath([:Object]), [])
+      @nil_type = Type::Instance.new(self, resolve_cpath([:NilClass]), [])
+      @true_type = Type::Instance.new(self, resolve_cpath([:TrueClass]), [])
+      @false_type = Type::Instance.new(self, resolve_cpath([:FalseClass]), [])
+      @str_type = Type::Instance.new(self, resolve_cpath([:String]), [])
+      @int_type = Type::Instance.new(self, resolve_cpath([:Integer]), [])
+      @float_type = Type::Instance.new(self, resolve_cpath([:Float]), [])
+      @rational_type = Type::Instance.new(self, resolve_cpath([:Rational]), [])
+      @complex_type = Type::Instance.new(self, resolve_cpath([:Complex]), [])
+      @proc_type = Type::Instance.new(self, resolve_cpath([:Proc]), [])
+      @symbol_type = Type::Instance.new(self, resolve_cpath([:Symbol]), [])
+      @set_type = Type::Instance.new(self, resolve_cpath([:Set]), [])
+      @regexp_type = Type::Instance.new(self, resolve_cpath([:Regexp]), [])
+
+      @run_count = 0
+    end
+
+    attr_reader :type_table
+
+    attr_reader :mod_class, :mod_object, :mod_ary, :mod_hash, :mod_range, :mod_str
+    attr_reader :cls_type, :mod_type
+    attr_reader :obj_type, :nil_type, :true_type, :false_type, :str_type
+    attr_reader :int_type, :float_type, :rational_type, :complex_type
+    attr_reader :proc_type, :symbol_type, :set_type, :regexp_type
+
+    def gen_ary_type(elem_vtx)
+      Type::Instance.new(self, @mod_ary, [elem_vtx])
+    end
+
+    def gen_hash_type(key_vtx, val_vtx)
+      Type::Instance.new(self, @mod_hash, [key_vtx, val_vtx])
+    end
+
+    def gen_range_type(elem_vtx)
+      Type::Instance.new(self, @mod_range, [elem_vtx])
+    end
+
+    attr_accessor :run_count
+
+    def each_direct_superclass(mod, singleton)
+      while mod
+        yield mod, singleton
+        singleton, mod = get_superclass(singleton, mod)
+      end
+    end
+
+    def each_superclass(mod, singleton, &blk)
+      while mod
+        # TODO: prepended modules
+        yield mod, singleton
+        if singleton
+          # TODO: extended modules
+        else
+          each_included_module(mod, &blk)
+        end
+        singleton, mod = get_superclass(singleton, mod)
+      end
+    end
+
+    def each_included_module(mod, &blk)
+      mod.included_modules.each do |_inc_decl, inc_mod|
+        yield inc_mod, false
+        each_included_module(inc_mod, &blk)
+      end
+    end
+
+    def get_superclass(singleton, mod)
+      super_mod = mod.superclass
+      if super_mod
+        return [singleton, super_mod]
+      else
+        if mod == @mod_basic_object
+          if singleton
+            return [false, @mod_class]
+          else
+            return nil
+          end
+        elsif mod == @mod_module && !singleton
+          return nil
+        else
+          return [false, @mod_module]
+        end
+      end
+    end
+
+    def get_instance_type(mod, type_args, changes, base_ty_env, base_ty)
+      ty_env = base_ty_env.dup
+      if base_ty.is_a?(Type::Instance)
+        base_ty.mod.type_params.zip(base_ty.args) do |param, arg|
+          ty_env[param] = arg
+        end
+      end
+      args = mod.type_params.zip(type_args).map do |param, arg|
+        arg && changes ? arg.covariant_vertex(self, changes, ty_env) : Source.new
+      end
+      Type::Instance.new(self, mod, args)
+    end
+
+    def get_superclass_type(ty, changes, base_ty_env)
+      singleton, super_mod = get_superclass(ty.is_a?(Type::Singleton), ty.mod)
+      return unless super_mod
+
+      if singleton
+        Type::Singleton.new(self, super_mod)
+      else
+        get_instance_type(super_mod, ty.mod.superclass_type_args || [], changes, base_ty_env, ty)
+      end
+    end
+
+    def add_static_eval_queue(change_type, arg)
+      @static_eval_queue << [change_type, arg]
+    end
+
+    def define_all
+      until @static_eval_queue.empty?
+        change_type, arg = @static_eval_queue.shift
+        case change_type
+        when :inner_modules_changed
+          arg[0].on_inner_modules_changed(self, arg[1])
+        when :static_read_changed
+          case arg
+          when BaseStaticRead
+            arg.on_scope_updated(self)
+          when ScopedStaticRead
+            arg.on_cbase_updated(self)
+          else
+            raise
+          end
+        when :parent_modules_changed
+          arg.on_parent_modules_changed(self)
+        else
+          raise change_type.to_s
+        end
+      end
+    end
+
+    def add_run(obj)
+      unless @run_queue_set.include?(obj)
+        @run_queue << obj
+        @run_queue_set << obj
+      end
+    end
+
+    def run_all
+      run_count = 0
+      until @run_queue.empty?
+        obj = @run_queue.shift
+        @run_queue_set.delete(obj)
+        unless obj.destroyed
+          run_count += 1
+          obj.run(self)
+        end
+      end
+      @run_count += run_count
+    end
+
+    # just for validation
+    def get_vertexes(vtxs)
+      @mod_object.get_vertexes(vtxs)
+      @gvars.each_value do |gvar_entity|
+        vtxs << gvar_entity.vtx
+      end
+    end
+
+    # classes and modules
+
+    def resolve_cpath(cpath)
+      mod = @mod_object
+      raise unless cpath # annotation
+      cpath.each do |cname|
+        mod = mod.inner_modules[cname] ||= ModuleEntity.new(mod.cpath + [cname], mod)
+      end
+      mod
+    end
+
+    # constants
+
+    def resolve_const(cpath)
+      mod = resolve_cpath(cpath[0..-2])
+      mod.get_const(cpath[-1])
+    end
+
+    def resolve_method(cpath, singleton, mid)
+      mod = resolve_cpath(cpath)
+      mod.get_method(singleton, mid)
+    end
+
+    def resolve_gvar(name)
+      @gvars[name] ||= ValueEntity.new
+    end
+
+    def resolve_ivar(cpath, singleton, name)
+      # TODO: include はあとで考える
+      mod = resolve_cpath(cpath)
+      mod.get_ivar(singleton, name)
+    end
+
+    def resolve_cvar(cpath, name)
+      # TODO: include はあとで考える
+      mod = resolve_cpath(cpath)
+      mod.get_cvar(name)
+    end
+
+    def resolve_type_alias(cpath, name)
+      # TODO: include はあとで考える
+      mod = resolve_cpath(cpath)
+      mod.get_type_alias(name)
+    end
+
+    def load_core_rbs(raw_decls)
+      lenv = LocalEnv.new(nil, CRef::Toplevel, {}, [])
+      decls = raw_decls.map do |raw_decl|
+        AST.create_rbs_decl(raw_decl, lenv)
+      end.compact
+
+      decls += AST.parse_rbs("typeprof-rbs-shim.rbs", <<-RBS)
+        class Exception
+          include _Exception
+        end
+        class String
+          include _ToS
+          include _ToStr
+        end
+        class Array[Elem]
+          include _ToAry[Elem]
+          include _Each[Elem]
+        end
+        class Hash[K, V]
+          include _Each[[K, V]]
+        end
+      RBS
+
+      # Loading frequently used modules first will reduces constant resolution
+      # which makes loading faster :-)
+      critical_modules = [
+        decls.find {|decl| decl.cpath == [:Object] },
+        decls.find {|decl| decl.cpath == [:Module] },
+        decls.find {|decl| decl.cpath == [:Numeric] },
+        decls.find {|decl| decl.cpath == [:Integer] },
+        decls.find {|decl| decl.cpath == [:String] },
+        decls.find {|decl| decl.cpath == [:Array] },
+        decls.find {|decl| decl.cpath == [:Hash] },
+        decls.find {|decl| decl.cpath == [:Enumerator] },
+      ]
+      decls = critical_modules + (decls - critical_modules)
+
+      decls.each {|decl| decl.define(self) }
+      define_all
+      decls.each {|decl| decl.install(self) }
+      run_all
+    end
+  end
+
+  class LocalEnv
+    def initialize(path, cref, locals, return_boxes)
+      @path = path
+      @cref = cref
+      @locals = locals
+      @return_boxes = return_boxes
+      @next_boxes = []
+      @filters = {}
+    end
+
+    attr_reader :path, :cref, :locals, :return_boxes, :next_boxes
+
+    def new_var(name, node)
+      @locals[name] = Vertex.new(node)
+    end
+
+    def set_var(name, vtx)
+      @locals[name] = vtx
+    end
+
+    def get_var(name)
+      @locals[name] || raise("#{ name }")
+    end
+
+    def exist_var?(name)
+      !!@locals[name]
+    end
+
+    def add_return_box(box)
+      @return_boxes << box
+    end
+
+    def add_next_box(box)
+      @next_boxes << box
+    end
+
+    def push_read_filter(name, type)
+      (@filters[name] ||= []) << type
+    end
+
+    def pop_read_filter(name)
+      (@filters[name] ||= []).pop
+    end
+
+    def apply_read_filter(genv, node, name, vtx)
+      if @filters[name] && !@filters[name].empty?
+        case @filters[name].last
+        when :non_nil
+          return NilFilter.new(genv, node, vtx, false).next_vtx
+        end
+      end
+      vtx
+    end
+  end
+
+  class CRef
+    def initialize(cpath, scope_level, mid, outer)
+      @cpath = cpath
+      @scope_level = scope_level
+      @mid = mid
+      @outer = outer
+    end
+
+    attr_reader :cpath, :scope_level, :mid, :outer
+
+    def get_self(genv)
+      case @scope_level
+      when :instance
+        mod = genv.resolve_cpath(@cpath || [])
+        type_params = mod.type_params.map {|ty_param| Source.new() } # TODO: better support
+        ty = Type::Instance.new(genv, mod, type_params)
+        Source.new(ty)
+      when :class
+        Source.new(Type::Singleton.new(genv, genv.resolve_cpath(@cpath || [])))
+      else
+        Source.new()
+      end
+    end
+
+    Toplevel = self.new([], :instance, nil, nil)
+  end
+end