module TypeProf::Builtin

Public Class Methods

file_load(path, ep, env, scratch, &ctn) click to toggle source
# File typeprof-0.21.2/lib/typeprof/builtin.rb, line 626
def self.file_load(path, ep, env, scratch, &ctn)
  iseq, = ISeq.compile(path)
  callee_ep, callee_env = TypeProf.starting_state(iseq)
  scratch.merge_env(callee_ep, callee_env)

  scratch.add_callsite!(callee_ep.ctx, ep, env) do |_ret_ty, ep|
    ret_ty = Type::Instance.new(Type::Builtin[:true])
    ctn[ret_ty, ep, env]
  end
end
file_require(feature, scratch) click to toggle source
# File typeprof-0.21.2/lib/typeprof/builtin.rb, line 637
def self.file_require(feature, scratch)
  # XXX: dynamic RBS load is really needed??  Another idea:
  #
  # * RBS should be loaded in advance of analysis
  # * require "some_gem/foo" should be ignored
  # * require "app/foo" should always load .rb file (in this case, app/foo.rb)
  return :done, :true if Import.import_library(scratch, feature)

  # Try to analyze the source code of the gem
  begin
    gem feature
  rescue Gem::MissingSpecError, Gem::LoadError
  end

  begin
    filetype, path = $LOAD_PATH.resolve_feature_path(feature)

    if filetype == :rb
      return :done, :false if scratch.loaded_files[path]
      scratch.loaded_files[path] = true

      return :do, path if File.readable?(path)

      return :error, "failed to load: #{ path }"
    else
      return :error, "cannot load a .so file: #{ path }"
    end
  rescue LoadError
    return :error, "failed to require: #{ feature }"
  end
end
setup_initial_global_env(scratch) click to toggle source
# File typeprof-0.21.2/lib/typeprof/builtin.rb, line 785
def self.setup_initial_global_env(scratch)
  klass_basic_obj = scratch.new_class(nil, :BasicObject, [], :__root__, nil) # cbase, name, superclass
  klass_obj = scratch.new_class(nil, :Object, [], klass_basic_obj, nil)
  scratch.add_constant(klass_obj, :Object, klass_obj, nil)
  scratch.add_constant(klass_obj, :BasicObject, klass_basic_obj, nil)

  Type::Builtin[:basic_obj] = klass_basic_obj
  Type::Builtin[:obj]   = klass_obj

  Import.import_builtin(scratch)

  Type::Builtin[:vmcore]     , = scratch.new_class(klass_obj, :VMCore, [], klass_obj, nil)
  Type::Builtin[:int]        , = scratch.get_constant(klass_obj, :Integer)
  Type::Builtin[:float]      , = scratch.get_constant(klass_obj, :Float)
  Type::Builtin[:rational]   , = scratch.get_constant(klass_obj, :Rational)
  Type::Builtin[:complex]    , = scratch.get_constant(klass_obj, :Complex)
  Type::Builtin[:sym]        , = scratch.get_constant(klass_obj, :Symbol)
  Type::Builtin[:str]        , = scratch.get_constant(klass_obj, :String)
  Type::Builtin[:struct]     , = scratch.get_constant(klass_obj, :Struct)
  Type::Builtin[:ary]        , = scratch.get_constant(klass_obj, :Array)
  Type::Builtin[:hash]       , = scratch.get_constant(klass_obj, :Hash)
  Type::Builtin[:io]         , = scratch.get_constant(klass_obj, :IO)
  Type::Builtin[:proc]       , = scratch.get_constant(klass_obj, :Proc)
  Type::Builtin[:range]      , = scratch.get_constant(klass_obj, :Range)
  Type::Builtin[:regexp]     , = scratch.get_constant(klass_obj, :Regexp)
  Type::Builtin[:matchdata]  , = scratch.get_constant(klass_obj, :MatchData)
  Type::Builtin[:class]      , = scratch.get_constant(klass_obj, :Class)
  Type::Builtin[:module]     , = scratch.get_constant(klass_obj, :Module)
  Type::Builtin[:exc]        , = scratch.get_constant(klass_obj, :Exception)
  Type::Builtin[:encoding]   , = scratch.get_constant(klass_obj, :Encoding)
  Type::Builtin[:enumerator] , = scratch.get_constant(klass_obj, :Enumerator)
  Type::Builtin[:kernel]     , = scratch.get_constant(klass_obj, :Kernel)

  klass_vmcore = Type::Builtin[:vmcore]
  klass_ary    = Type::Builtin[:ary]
  klass_hash   = Type::Builtin[:hash]
  klass_struct = Type::Builtin[:struct]
  klass_proc   = Type::Builtin[:proc]
  klass_module = Type::Builtin[:module]

  scratch.set_custom_method(klass_vmcore, :"core#set_method_alias", Builtin.method(:vmcore_set_method_alias))
  scratch.set_custom_method(klass_vmcore, :"core#undef_method", Builtin.method(:vmcore_undef_method))
  scratch.set_custom_method(klass_vmcore, :"core#hash_merge_kwd", Builtin.method(:vmcore_hash_merge_kwd))
  scratch.set_custom_method(klass_vmcore, :"core#raise", Builtin.method(:vmcore_raise))

  scratch.set_custom_method(klass_vmcore, :lambda, Builtin.method(:lambda))
  scratch.set_singleton_custom_method(klass_obj, :"new", Builtin.method(:object_s_new))
  scratch.set_custom_method(klass_obj, :p, Builtin.method(:kernel_p), false)
  scratch.set_custom_method(klass_obj, :is_a?, Builtin.method(:object_is_a?))
  scratch.set_custom_method(klass_obj, :respond_to?, Builtin.method(:object_respond_to?))
  scratch.set_custom_method(klass_obj, :class, Builtin.method(:object_class))
  scratch.set_custom_method(klass_obj, :send, Builtin.method(:object_send))
  scratch.set_custom_method(klass_obj, :instance_eval, Builtin.method(:object_instance_eval))
  scratch.set_custom_method(klass_obj, :proc, Builtin.method(:lambda), false)
  scratch.set_custom_method(klass_obj, :__method__, Builtin.method(:object_privitive_method), false)
  scratch.set_custom_method(klass_obj, :block_given?, Builtin.method(:object_block_given?), false)

  scratch.set_custom_method(klass_obj, :enum_for, Builtin.method(:object_enum_for))
  scratch.set_custom_method(klass_obj, :to_enum, Builtin.method(:object_enum_for))

  scratch.set_custom_method(klass_module, :include, Builtin.method(:module_include))
  scratch.set_custom_method(klass_module, :extend, Builtin.method(:module_extend))
  scratch.set_custom_method(klass_module, :prepend, Builtin.method(:module_prepend))
  scratch.set_custom_method(klass_module, :module_function, Builtin.method(:module_module_function), false)
  scratch.set_custom_method(klass_module, :public, Builtin.method(:module_public), false)
  scratch.set_custom_method(klass_module, :private, Builtin.method(:module_private), false)
  scratch.set_custom_method(klass_module, :define_method, Builtin.method(:module_define_method))
  scratch.set_custom_method(klass_module, :attr_accessor, Builtin.method(:module_attr_accessor))
  scratch.set_custom_method(klass_module, :attr_reader, Builtin.method(:module_attr_reader))
  scratch.set_custom_method(klass_module, :attr_writer, Builtin.method(:module_attr_writer))
  scratch.set_custom_method(klass_module, :class_eval, Builtin.method(:object_module_eval))
  scratch.set_custom_method(klass_module, :module_eval, Builtin.method(:object_module_eval))
  scratch.set_custom_method(klass_module, :===, Builtin.method(:module_eqq))

  scratch.set_custom_method(klass_proc, :[], Builtin.method(:proc_call))
  scratch.set_custom_method(klass_proc, :call, Builtin.method(:proc_call))

  scratch.set_custom_method(klass_ary, :[], Builtin.method(:array_aref))
  scratch.set_custom_method(klass_ary, :[]=, Builtin.method(:array_aset))
  scratch.set_custom_method(klass_ary, :pop, Builtin.method(:array_pop))

  scratch.set_custom_method(klass_hash, :[], Builtin.method(:hash_aref))
  scratch.set_custom_method(klass_hash, :[]=, Builtin.method(:hash_aset))

  scratch.set_custom_method(klass_struct, :initialize, Builtin.method(:struct_initialize))
  scratch.set_singleton_custom_method(klass_struct, :new, Builtin.method(:struct_s_new))

  scratch.set_custom_method(klass_obj, :require, Builtin.method(:kernel_require), false)
  scratch.set_custom_method(klass_obj, :require_relative, Builtin.method(:kernel_require_relative), false)
  scratch.set_custom_method(klass_obj, :Array, Builtin.method(:kernel_Array), false)
  scratch.set_custom_method(klass_obj, :autoload, Builtin.method(:kernel_autoload), false)
  scratch.set_custom_method(klass_module, :autoload, Builtin.method(:module_autoload))

  # remove BasicObject#method_missing
  scratch.set_method(klass_basic_obj, :method_missing, false, nil)

  # ENV: Hash[String, String]
  str_ty = Type::Instance.new(Type::Builtin[:str])
  env_ty = Type.gen_hash {|h| h[str_ty] = Type.optional(str_ty) }
  scratch.add_constant(klass_obj, :ENV, env_ty, nil)

  scratch.search_method(Type::Builtin[:kernel], false, :sprintf) do |mdefs,|
    mdefs.each do |mdef|
      scratch.add_method(klass_vmcore, :"core#sprintf", false, mdef)
    end
  end
end

Public Instance Methods

array_aref(recv, mid, aargs, ep, env, scratch, &ctn) click to toggle source
# File typeprof-0.21.2/lib/typeprof/builtin.rb, line 451
def array_aref(recv, mid, aargs, ep, env, scratch, &ctn)
  return ctn[Type.any, ep, env] unless recv.is_a?(Type::Local) && recv.kind == Type::Array

  case aargs.lead_tys.size
  when 1
    idx = aargs.lead_tys.first
    if idx.is_a?(Type::Literal)
      idx = idx.lit
      idx = nil if !idx.is_a?(Integer) && !idx.is_a?(Range)
    elsif idx == Type::Instance.new(Type::Builtin[:range])
      idx = (nil..nil)
    else
      idx = nil
    end
    ty = scratch.get_array_elem_type(env, ep, recv.id, idx)
    ctn[ty, ep, env]
  when 2
    ty = scratch.get_array_elem_type(env, ep, recv.id)
    base_ty = Type::Instance.new(Type::Builtin[:ary])
    ret_ty = Type::Array.new(Type::Array::Elements.new([], ty), base_ty)
    ctn[ret_ty, ep, env]
  else
    ctn[Type.any, ep, env]
  end
end
array_aset(recv, mid, aargs, ep, env, scratch, &ctn) click to toggle source
# File typeprof-0.21.2/lib/typeprof/builtin.rb, line 477
def array_aset(recv, mid, aargs, ep, env, scratch, &ctn)
  return ctn[Type.any, ep, env] unless recv.is_a?(Type::Local) && recv.kind == Type::Array

  if aargs.lead_tys.size != 2
    # XXX: Support `ary[idx, len] = val`
    #raise NotImplementedError # XXX
    return ctn[Type.any, ep, env]
  end

  idx = aargs.lead_tys.first
  if idx.is_a?(Type::Literal)
    idx = idx.lit
    if !idx.is_a?(Integer)
      # XXX: Support `ary[idx..end] = val`
      #raise NotImplementedError # XXX
      return ctn[Type.any, ep, env]
    end
  else
    idx = nil
  end

  ty = aargs.lead_tys.last

  env = scratch.update_container_elem_types(env, ep, recv.id, recv.base_type) do |elems|
    elems.update(idx, ty)
  end

  ctn[ty, ep, env]
end
array_pop(recv, mid, aargs, ep, env, scratch, &ctn) click to toggle source
# File typeprof-0.21.2/lib/typeprof/builtin.rb, line 507
def array_pop(recv, mid, aargs, ep, env, scratch, &ctn)
  return ctn[Type.any, ep, env] unless recv.is_a?(Type::Local) && recv.kind == Type::Array

  if aargs.lead_tys.size != 0
    ctn[Type.any, ep, env]
    return
  end

  ty = scratch.get_array_elem_type(env, ep, recv.id)
  ctn[ty, ep, env]
end
get_sym(target, ty, ep, scratch) click to toggle source
# File typeprof-0.21.2/lib/typeprof/builtin.rb, line 5
def get_sym(target, ty, ep, scratch)
  unless ty.is_a?(Type::Symbol)
    scratch.warn(ep, "symbol expected")
    return
  end
  sym = ty.sym
  unless sym
    scratch.warn(ep, "dynamic symbol is given to #{ target }; ignored")
    return
  end
  sym
end
hash_aref(recv, mid, aargs, ep, env, scratch, &ctn) click to toggle source
# File typeprof-0.21.2/lib/typeprof/builtin.rb, line 519
def hash_aref(recv, mid, aargs, ep, env, scratch, &ctn)
  return ctn[Type.any, ep, env] unless recv.is_a?(Type::Local) && recv.kind == Type::Hash

  if aargs.lead_tys.size != 1
    ctn[Type.any, ep, env]
    return
  end
  idx = aargs.lead_tys.first
  recv.each_child do |recv|
    if recv.is_a?(Type::Local) && recv.kind == Type::Hash
      ty = scratch.get_hash_elem_type(env, ep, recv.id, idx)
      ty = Type.nil if ty == Type.bot
    else
      ty = Type.any
    end
    ctn[ty, ep, env]
  end
end
hash_aset(recv, mid, aargs, ep, env, scratch, &ctn) click to toggle source
# File typeprof-0.21.2/lib/typeprof/builtin.rb, line 538
def hash_aset(recv, mid, aargs, ep, env, scratch, &ctn)
  return ctn[Type.any, ep, env] unless recv.is_a?(Type::Local) && recv.kind == Type::Hash

  if aargs.lead_tys.size != 2
    # XXX: error?
    ctn[Type.any, ep, env]
    return
  end

  idx = aargs.lead_tys.first
  idx = scratch.globalize_type(idx, env, ep)
  ty = aargs.lead_tys.last

  unless recv.is_a?(Type::Local) && recv.kind == Type::Hash
    # to ignore: class OptionMap < Hash
    return ctn[ty, ep, env]
  end

  env = scratch.update_container_elem_types(env, ep, recv.id, recv.base_type) do |elems|
    elems.update(idx, ty)
  end

  ctn[ty, ep, env]
end
kernel_Array(recv, mid, aargs, ep, env, scratch, &ctn) click to toggle source
# File typeprof-0.21.2/lib/typeprof/builtin.rb, line 768
def kernel_Array(recv, mid, aargs, ep, env, scratch, &ctn)
  raise NotImplementedError if aargs.lead_tys.size != 1
  ty = aargs.lead_tys.first
  ty = scratch.globalize_type(ty, env, ep)
  all_ty = Type.bot
  ty.each_child_global do |ty|
    if ty.is_a?(Type::Array)
      all_ty = all_ty.union(ty)
    else
      base_ty = Type::Instance.new(Type::Builtin[:ary])
      ret_ty = Type::Array.new(Type::Array::Elements.new([ty]), base_ty)
      all_ty = all_ty.union(ret_ty)
    end
  end
  ctn[all_ty, ep, env]
end
kernel_autoload(recv, mid, aargs, ep, env, scratch, &ctn) click to toggle source
# File typeprof-0.21.2/lib/typeprof/builtin.rb, line 738
def kernel_autoload(recv, mid, aargs, ep, env, scratch, &ctn)
  if aargs.lead_tys.size != 2
    # XXX: handle correctly
    ctn[Type.any, ep, env]
    return
  end

  feature = aargs.lead_tys[1]
  if feature.is_a?(Type::Literal)
    feature = feature.lit

    action, arg = Builtin.file_require(feature, scratch)
    case action
    when :do
      Builtin.file_load(arg, ep, env, scratch, &ctn)
    when :done
    when :error
      scratch.warn(ep, arg)
    end
    ctn[Type.nil, ep, env]
  else
    scratch.warn(ep, "autoload target cannot be identified statically")
    ctn[Type.nil, ep, env]
  end
end
kernel_p(recv, mid, aargs, ep, env, scratch, &ctn) click to toggle source
# File typeprof-0.21.2/lib/typeprof/builtin.rb, line 444
def kernel_p(recv, mid, aargs, ep, env, scratch, &ctn)
  aargs.lead_tys.each do |aarg|
    scratch.reveal_type(ep, scratch.globalize_type(aarg, env, ep))
  end
  ctn[aargs.lead_tys.size == 1 ? aargs.lead_tys.first : Type.any, ep, env]
end
kernel_require(recv, mid, aargs, ep, env, scratch, &ctn) click to toggle source
# File typeprof-0.21.2/lib/typeprof/builtin.rb, line 669
def kernel_require(recv, mid, aargs, ep, env, scratch, &ctn)
  if aargs.lead_tys.size != 1
    # XXX: handle correctly
    ctn[Type.any, ep, env]
    return
  end

  feature = aargs.lead_tys.first
  if feature.is_a?(Type::Literal)
    feature = feature.lit

    unless feature.is_a?(String)
      return ctn[Type.any, ep, env]
    end

    action, arg = Builtin.file_require(feature, scratch)
    case action
    when :do
      Builtin.file_load(arg, ep, env, scratch, &ctn)
    when :done
      result = Type::Instance.new(Type::Builtin[arg])
      ctn[result, ep, env]
    when :error
      scratch.warn(ep, arg)
      result = Type.bool
      ctn[result, ep, env]
    end
  else
    scratch.warn(ep, "require target cannot be identified statically")
    result = Type.bool
    ctn[result, ep, env]
  end
end
kernel_require_relative(recv, mid, aargs, ep, env, scratch, &ctn) click to toggle source
# File typeprof-0.21.2/lib/typeprof/builtin.rb, line 703
def kernel_require_relative(recv, mid, aargs, ep, env, scratch, &ctn)
  if aargs.lead_tys.size != 1
    # XXX: handle correctly
    ctn[Type.any, ep, env]
    return
  end

  feature = aargs.lead_tys.first
  if feature.is_a?(Type::Literal)
    feature = feature.lit

    unless feature.is_a?(String)
      return ctn[Type.any, ep, env]
    end

    path = File.join(File.dirname(ep.ctx.iseq.absolute_path), feature) + ".rb" # XXX

    if scratch.loaded_files[path]
      result = Type::Instance.new(Type::Builtin[:false])
      return ctn[result, ep, env]
    end
    scratch.loaded_files[path] = true

    return Builtin.file_load(path, ep, env, scratch, &ctn) if File.readable?(path)

    scratch.warn(ep, "failed to load: #{ path }")
  else
    scratch.warn(ep, "require target cannot be identified statically")
    feature = nil
  end

  result = Type::Instance.new(Type::Builtin[:true])
  ctn[result, ep, env]
end
lambda(recv, mid, aargs, ep, env, scratch, &ctn) click to toggle source
# File typeprof-0.21.2/lib/typeprof/builtin.rb, line 61
def lambda(recv, mid, aargs, ep, env, scratch, &ctn)
  ctn[aargs.blk_ty, ep, env]
end
module_attr_accessor(recv, mid, aargs, ep, env, scratch, &ctn) click to toggle source
# File typeprof-0.21.2/lib/typeprof/builtin.rb, line 417
def module_attr_accessor(recv, mid, aargs, ep, env, scratch, &ctn)
  aargs.lead_tys.each do |aarg|
    sym = get_sym("attr_accessor", aarg, ep, scratch) or next
    cref = ep.ctx.cref
    scratch.add_attr_method(cref.klass, sym, :"@#{ sym }", :accessor, env.static_env.pub_meth, ep)
  end
  ctn[Type.nil, ep, env]
end
module_attr_reader(recv, mid, aargs, ep, env, scratch, &ctn) click to toggle source
# File typeprof-0.21.2/lib/typeprof/builtin.rb, line 426
def module_attr_reader(recv, mid, aargs, ep, env, scratch, &ctn)
  aargs.lead_tys.each do |aarg|
    sym = get_sym("attr_reader", aarg, ep, scratch) or next
    cref = ep.ctx.cref
    scratch.add_attr_method(cref.klass, sym, :"@#{ sym }", :reader, env.static_env.pub_meth, ep)
  end
  ctn[Type.nil, ep, env]
end
module_attr_writer(recv, mid, aargs, ep, env, scratch, &ctn) click to toggle source
# File typeprof-0.21.2/lib/typeprof/builtin.rb, line 435
def module_attr_writer(recv, mid, aargs, ep, env, scratch, &ctn)
  aargs.lead_tys.each do |aarg|
    sym = get_sym("attr_writer", aarg, ep, scratch) or next
    cref = ep.ctx.cref
    scratch.add_attr_method(cref.klass, sym, :"@#{ sym }", :writer, env.static_env.pub_meth, ep)
  end
  ctn[Type.nil, ep, env]
end
module_autoload(recv, mid, aargs, ep, env, scratch, &ctn) click to toggle source
# File typeprof-0.21.2/lib/typeprof/builtin.rb, line 764
def module_autoload(recv, mid, aargs, ep, env, scratch, &ctn)
  kernel_autoload(recv, mid, aargs, ep, env, scratch, &ctn)
end
module_define_method(recv, mid, aargs, ep, env, scratch, &ctn) click to toggle source
# File typeprof-0.21.2/lib/typeprof/builtin.rb, line 386
def module_define_method(recv, mid, aargs, ep, env, scratch, &ctn)
  if aargs.lead_tys.size != 1
    scratch.warn(ep, "Module#define with #{ aargs.lead_tys.size } argument is ignored")
    ctn[Type.any, ep, env]
    return
  end

  mid, = aargs.lead_tys
  mid.each_child do |mid|
    if mid.is_a?(Type::Symbol)
      mid = mid.sym
      aargs.blk_ty.each_child do |blk_ty|
        if blk_ty.is_a?(Type::Proc)
          blk = blk_ty.block_body
          case blk
          when ISeqBlock
            scratch.do_define_iseq_method(ep, env, mid, blk.iseq, blk.outer_ep)
          else
            # XXX: what to do?
          end
        else
          # XXX: what to do?
        end
      end
    else
      # XXX: what to do?
    end
  end
  ctn[Type.any, ep, env]
end
module_eqq(recv, mid, aargs, ep, env, scratch, &ctn) click to toggle source
# File typeprof-0.21.2/lib/typeprof/builtin.rb, line 166
def module_eqq(recv, mid, aargs, ep, env, scratch, &ctn)
  if aargs.lead_tys.size == 1
    aargs.lead_tys[0].each_child do |aarg|
      aarg = aarg.base_type if aarg.is_a?(Type::Symbol) # XXX
      if aarg.is_a?(Type::Instance)
        if aarg.klass == recv # XXX: inheritance
          true_val = Type::Instance.new(Type::Builtin[:true])
          ctn[true_val, ep, env]
        else
          false_val = Type::Instance.new(Type::Builtin[:false])
          ctn[false_val, ep, env]
        end
      else
        ctn[Type.bool, ep, env]
      end
    end
  else
    ctn[Type.bool, ep, env]
  end
end
module_extend(recv, mid, aargs, ep, env, scratch, &ctn) click to toggle source
# File typeprof-0.21.2/lib/typeprof/builtin.rb, line 286
def module_extend(recv, mid, aargs, ep, env, scratch, &ctn)
  if aargs.lead_tys.size != 1
    scratch.warn(ep, "Module#extend without an argument is ignored")
    ctn[Type.any, ep, env]
    return
  end

  unless recv.is_a?(Type::Class)
    # XXX: warn?
    return ctn[Type.any, ep, env]
  end

  arg = aargs.lead_tys[0]
  arg.each_child do |arg|
    if arg.is_a?(Type::Class)
      aargs = ActualArguments.new([recv], nil, {}, Type.nil)
      scratch.do_send(arg, :extended, aargs, ep, env) {|_ret_ty, _ep| }
      # if ep.ctx.cref.singleton is true, the meta-meta level is ignored. Should we warn?
      scratch.mix_module(:after, recv, arg, nil, true, ep)
    end
  end
  ctn[recv, ep, env]
end
module_include(recv, mid, aargs, ep, env, scratch, &ctn) click to toggle source
# File typeprof-0.21.2/lib/typeprof/builtin.rb, line 262
def module_include(recv, mid, aargs, ep, env, scratch, &ctn)
  if aargs.lead_tys.size != 1
    scratch.warn(ep, "Module#include without an argument is ignored")
    ctn[Type.any, ep, env]
    return
  end

  unless recv.is_a?(Type::Class)
    # XXX: warn?
    return ctn[Type.any, ep, env]
  end

  # support multiple arguments: include M1, M2
  arg = aargs.lead_tys[0]
  arg.each_child do |arg|
    if arg.is_a?(Type::Class)
      aargs = ActualArguments.new([recv], nil, {}, Type.nil)
      scratch.do_send(arg, :included, aargs, ep, env) {|_ret_ty, _ep| }
      scratch.mix_module(:after, recv, arg, nil, ep.ctx.cref.singleton, ep)
    end
  end
  ctn[recv, ep, env]
end
module_module_function(recv, mid, aargs, ep, env, scratch, &ctn) click to toggle source
# File typeprof-0.21.2/lib/typeprof/builtin.rb, line 331
def module_module_function(recv, mid, aargs, ep, env, scratch, &ctn)
  if aargs.lead_tys.empty?
    ctn[recv, ep, env.enable_module_function]
  else
    aargs.lead_tys.each do |aarg|
      sym = get_sym("module_function", aarg, ep, scratch) or next
      meths = scratch.get_method(recv, false, false, sym)
      meths.each do |mdef|
        scratch.add_method(recv, sym, true, mdef)
      end
    end
    ctn[recv, ep, env]
  end
end
module_prepend(recv, mid, aargs, ep, env, scratch, &ctn) click to toggle source
# File typeprof-0.21.2/lib/typeprof/builtin.rb, line 310
def module_prepend(recv, mid, aargs, ep, env, scratch, &ctn)
  if aargs.lead_tys.size != 1
    scratch.warn(ep, "Module#prepend without an argument is ignored")
    ctn[Type.any, ep, env]
    return
  end

  unless recv.is_a?(Type::Class)
    # XXX: warn?
    return ctn[Type.any, ep, env]
  end

  arg = aargs.lead_tys[0]
  arg.each_child do |arg|
    if arg.is_a?(Type::Class)
      scratch.mix_module(:before, recv, arg, nil, ep.ctx.cref.singleton, ep)
    end
  end
  ctn[recv, ep, env]
end
module_private(recv, mid, aargs, ep, env, scratch, &ctn) click to toggle source
# File typeprof-0.21.2/lib/typeprof/builtin.rb, line 366
def module_private(recv, mid, aargs, ep, env, scratch, &ctn)
  if aargs.lead_tys.empty?
    ctn[recv, ep, env.method_public_set(false)]
  else
    if recv.is_a?(Type::Class)
      aargs.lead_tys.each do |aarg|
        sym = get_sym("private", aarg, ep, scratch) or next
        meths = scratch.get_method(recv, false, false, sym)
        next unless meths
        meths.each do |mdef|
          mdef.pub_meth = false if mdef.respond_to?(:pub_meth=)
        end
      end
    else
      # XXX: warn?
    end
    ctn[recv, ep, env]
  end
end
module_public(recv, mid, aargs, ep, env, scratch, &ctn) click to toggle source
# File typeprof-0.21.2/lib/typeprof/builtin.rb, line 346
def module_public(recv, mid, aargs, ep, env, scratch, &ctn)
  if aargs.lead_tys.empty?
    ctn[recv, ep, env.method_public_set(true)]
  else
    if recv.is_a?(Type::Class)
      aargs.lead_tys.each do |aarg|
        sym = get_sym("public", aarg, ep, scratch) or next
        meths = scratch.get_method(recv, false, false, sym)
        next unless meths
        meths.each do |mdef|
          mdef.pub_meth = true if mdef.respond_to?(:pub_meth=)
        end
      end
    else
      # XXX: warn?
    end
    ctn[recv, ep, env]
  end
end
object_block_given?(recv, mid, aargs, ep, env, scratch, &ctn) click to toggle source
# File typeprof-0.21.2/lib/typeprof/builtin.rb, line 241
def object_block_given?(recv, mid, aargs, ep, env, scratch, &ctn)
  procs = Type.bot
  no_proc = false
  env.static_env.blk_ty.each_child do |blk_ty|
    case blk_ty
    when Type::Proc
      procs = procs.union(blk_ty)
    when Type.nil
      no_proc = true
    else
      ctn[Type.bool, ep, env]
    end
  end
  if procs != Type.bot
    ctn[Type::Instance.new(Type::Builtin[:true]), ep, env.replace_blk_ty(procs)]
  end
  if no_proc
    ctn[Type::Instance.new(Type::Builtin[:false]), ep, env.replace_blk_ty(Type.nil)]
  end
end
object_class(recv, mid, aargs, ep, env, scratch, &ctn) click to toggle source
# File typeprof-0.21.2/lib/typeprof/builtin.rb, line 122
def object_class(recv, mid, aargs, ep, env, scratch, &ctn)
  if recv.is_a?(Type::Instance)
    ctn[recv.klass, ep, env]
  else
    ctn[Type.any, ep, env]
  end
end
object_enum_for(recv, mid, aargs, ep, env, scratch, &ctn) click to toggle source
# File typeprof-0.21.2/lib/typeprof/builtin.rb, line 202
def object_enum_for(recv, mid, aargs, ep, env, scratch, &ctn)
  if aargs.lead_tys.size >= 1
    mid_ty, = aargs.lead_tys
    naargs = ActualArguments.new(aargs.lead_tys[1..], aargs.rest_ty, aargs.kw_tys, aargs.blk_ty)
  elsif aargs.rest_ty
    mid_ty = aargs.rest_ty
    naargs = aargs
  else
    mid_ty = Type::Symbol.new(:each, Type::Instance.new(Type::Builtin[:sym]))
    naargs = aargs
  end

  elem_ty = Type.bot
  enum_for_blk = CustomBlock.new(ep, mid) do |aargs, caller_ep, caller_env, scratch, replace_recv_ty:, replace_cref:, &blk_ctn|
    if aargs.lead_tys.size >= 1
      elem_ty = elem_ty.union(aargs.lead_tys[0])
    else
      elem_ty = elem_ty.union(Type.any)
    end
    ctn[Type::Cell.new(Type::Cell::Elements.new([elem_ty, Type.any]), Type::Instance.new(Type::Builtin[:enumerator])), ep, env]
    blk_ctn[Type.any, caller_ep, caller_env]
  end
  enum_for_blk_ty = Type::Proc.new(enum_for_blk, Type::Instance.new(Type::Builtin[:proc]))

  naargs = ActualArguments.new(naargs.lead_tys, naargs.rest_ty, naargs.kw_tys, enum_for_blk_ty)
  mid_ty.each_child do |mid|
    if mid.is_a?(Type::Symbol)
      mid = mid.sym
      scratch.do_send(recv, mid, naargs, ep, env) do |_ret_ty, _ep|
        ctn[Type::Cell.new(Type::Cell::Elements.new([elem_ty, Type.any]), Type::Instance.new(Type::Builtin[:enumerator])), ep, env]
      end
    end
  end
end
object_instance_eval(recv, mid, aargs, ep, env, scratch, &ctn) click to toggle source
# File typeprof-0.21.2/lib/typeprof/builtin.rb, line 152
def object_instance_eval(recv, mid, aargs, ep, env, scratch, &ctn)
  if aargs.lead_tys.size >= 1
    scratch.warn(ep, "instance_eval with arguments is ignored")
    ctn[Type.any, ep, env]
    return
  end
  naargs = ActualArguments.new([recv], nil, {}, Type.nil)
  nrecv = recv
  nrecv = nrecv.base_type if nrecv.is_a?(Type::ContainerType)
  scratch.do_invoke_block(aargs.blk_ty, naargs, ep, env, replace_recv_ty: nrecv) do |ret_ty, ep|
    ctn[ret_ty, ep, env]
  end
end
object_is_a?(recv, mid, aargs, ep, env, scratch, &ctn) click to toggle source
# File typeprof-0.21.2/lib/typeprof/builtin.rb, line 84
def object_is_a?(recv, mid, aargs, ep, env, scratch, &ctn)
  if aargs.lead_tys.size == 1
    if recv.is_a?(Type::Instance)
      if recv.klass == aargs.lead_tys[0] # XXX: inheritance
        true_val = Type::Instance.new(Type::Builtin[:true])
        ctn[true_val, ep, env]
      else
        false_val = Type::Instance.new(Type::Builtin[:false])
        ctn[false_val, ep, env]
      end
    else
      ctn[Type.bool, ep, env]
    end
  else
    ctn[Type.bool, ep, env]
  end
end
object_module_eval(recv, mid, aargs, ep, env, scratch, &ctn) click to toggle source
# File typeprof-0.21.2/lib/typeprof/builtin.rb, line 187
def object_module_eval(recv, mid, aargs, ep, env, scratch, &ctn)
  if aargs.lead_tys.size >= 1
    scratch.warn(ep, "class_eval with arguments is ignored")
    ctn[Type.any, ep, env]
    return
  end
  naargs = ActualArguments.new([recv], nil, {}, Type.nil)
  nrecv = recv
  nrecv = nrecv.base_type if nrecv.is_a?(Type::ContainerType)
  ncref = ep.ctx.cref.extend(nrecv, true)
  scratch.do_invoke_block(aargs.blk_ty, naargs, ep, env, replace_recv_ty: nrecv, replace_cref: ncref) do |_ret_ty, ep|
    ctn[recv, ep, env]
  end
end
object_privitive_method(recv, mid, aargs, ep, env, scratch, &ctn) click to toggle source
# File typeprof-0.21.2/lib/typeprof/builtin.rb, line 237
def object_privitive_method(recv, mid, aargs, ep, env, scratch, &ctn)
  ctn[Type::Symbol.new(ep.ctx.mid, Type::Instance.new(Type::Builtin[:sym])), ep, env]
end
object_respond_to?(recv, mid, aargs, ep, env, scratch, &ctn) click to toggle source
# File typeprof-0.21.2/lib/typeprof/builtin.rb, line 102
def object_respond_to?(recv, mid, aargs, ep, env, scratch, &ctn)
  if aargs.lead_tys.size == 1
    sym = get_sym("respond_to?", aargs.lead_tys[0], ep, scratch)
    if sym
      klass, singleton = recv.method_dispatch_info
      if scratch.get_method(klass, singleton, false, sym)
        true_val = Type::Instance.new(Type::Builtin[:true])
        ctn[true_val, ep, env]
      else
        false_val = Type::Instance.new(Type::Builtin[:false])
        ctn[false_val, ep, env]
      end
    else
      ctn[Type.bool, ep, env]
    end
  else
    ctn[Type.bool, ep, env]
  end
end
object_s_new(recv, mid, aargs, ep, env, scratch, &ctn) click to toggle source
# File typeprof-0.21.2/lib/typeprof/builtin.rb, line 69
def object_s_new(recv, mid, aargs, ep, env, scratch, &ctn)
  if recv.type_params.size >= 1
    ty = Type::ContainerType.create_empty_instance(recv)
    env, ty = scratch.localize_type(ty, env, ep, AllocationSite.new(ep).add_id(:object_s_new))
  else
    ty = Type::Instance.new(recv)
  end
  meths = scratch.get_method(recv, false, false, :initialize)
  meths.flat_map do |meth|
    meth.do_send(ty, :initialize, aargs, ep, env, scratch) do |_ret_ty, ep, env|
      ctn[ty, ep, env]
    end
  end
end
object_send(recv, mid, aargs, ep, env, scratch, &ctn) click to toggle source
# File typeprof-0.21.2/lib/typeprof/builtin.rb, line 130
def object_send(recv, mid, aargs, ep, env, scratch, &ctn)
  if aargs.lead_tys.size >= 1
    mid_ty, = aargs.lead_tys
  elsif aargs.rest_ty
    mid_ty = aargs.rest_ty
  else
    return ctn[Type.any, ep, env]
  end
  aargs = ActualArguments.new(aargs.lead_tys[1..] || [], aargs.rest_ty, aargs.kw_tys, aargs.blk_ty)
  found = false
  mid_ty.each_child do |mid|
    if mid.is_a?(Type::Symbol)
      found = true
      mid = mid.sym
      scratch.do_send(recv, mid, aargs, ep, env, &ctn)
    end
  end
  unless found
    ctn[Type.any, ep, env]
  end
end
proc_call(recv, mid, aargs, ep, env, scratch, &ctn) click to toggle source
# File typeprof-0.21.2/lib/typeprof/builtin.rb, line 65
def proc_call(recv, mid, aargs, ep, env, scratch, &ctn)
  scratch.do_invoke_block(recv, aargs, ep, env, &ctn)
end
struct_initialize(recv, mid, aargs, ep, env, scratch, &ctn) click to toggle source
# File typeprof-0.21.2/lib/typeprof/builtin.rb, line 563
def struct_initialize(recv, mid, aargs, ep, env, scratch, &ctn)
  struct_klass = recv.klass
  while struct_klass.superclass != Type::Builtin[:struct]
    struct_klass = struct_klass.superclass
  end
  if struct_klass.superclass != Type::Builtin[:struct]
    ctn[Type.any, ep, env]
    return
  end
  scratch.add_ivar_read!(Type::Instance.new(struct_klass), :_keyword_init, ep) do |keyword_init, ep|
    scratch.add_ivar_read!(Type::Instance.new(struct_klass), :_members, ep) do |member_ary_ty, ep|
      next if member_ary_ty == Type.nil
      if keyword_init == Type::Instance.new(Type::Builtin[:true])
        # TODO: support kw_rest_ty
        aargs.kw_tys.each do |key, val_ty|
          found = false
          member_ary_ty.elems.lead_tys.each do |sym|
            if sym.sym == key
              found = true
              scratch.set_instance_variable(recv, sym.sym, val_ty, ep, env)
            end
          end
          unless found
            # TODO: what to do when not found?
          end
        end
      else
        member_ary_ty.elems.lead_tys.zip(aargs.lead_tys) do |sym, ty|
          ty ||= Type.nil
          scratch.set_instance_variable(recv, sym.sym, ty, ep, env)
        end
      end
    end
  end
  ctn[recv, ep, env]
end
struct_s_new(recv, mid, aargs, ep, env, scratch, &ctn) click to toggle source
# File typeprof-0.21.2/lib/typeprof/builtin.rb, line 600
def struct_s_new(recv, mid, aargs, ep, env, scratch, &ctn)
  keyword_init = false
  if aargs.kw_tys && aargs.kw_tys[:keyword_init] # XXX: more canonical way to extract keyword...
    if aargs.kw_tys[:keyword_init] == Type::Instance.new(Type::Builtin[:true])
      keyword_init = true
    end
  end

  fields = aargs.lead_tys.map {|ty| get_sym("Struct.new", ty, ep, scratch) }.compact
  struct_klass = scratch.new_struct(ep)

  scratch.set_singleton_custom_method(struct_klass, :new, Builtin.method(:object_s_new))
  scratch.set_singleton_custom_method(struct_klass, :[], Builtin.method(:object_s_new))
  fields.each do |field|
    scratch.add_attr_method(struct_klass, field, field, :accessor, true, ep)
  end
  fields = fields.map {|field| Type::Symbol.new(field, Type::Instance.new(Type::Builtin[:sym])) }
  base_ty = Type::Instance.new(Type::Builtin[:ary])
  fields = Type::Array.new(Type::Array::Elements.new(fields), base_ty)
  scratch.add_ivar_write!(Type::Instance.new(struct_klass), :_members, fields, ep)
  scratch.add_ivar_write!(Type::Instance.new(struct_klass), :_keyword_init, Type::Instance.new(Type::Builtin[:true]), ep) if keyword_init
  #set_singleton_custom_method(struct_klass, :members, Builtin.method(:...))

  ctn[struct_klass, ep, env]
end
vmcore_hash_merge_kwd(recv, mid, aargs, ep, env, scratch, &ctn) click to toggle source
# File typeprof-0.21.2/lib/typeprof/builtin.rb, line 31
def vmcore_hash_merge_kwd(recv, mid, aargs, ep, env, scratch, &ctn)
  h1 = aargs.lead_tys[0]
  h2 = aargs.lead_tys[1]
  elems = nil
  h1.each_child do |h1|
    if h1.is_a?(Type::Local) && h1.kind == Type::Hash
      h1_elems = scratch.get_container_elem_types(env, ep, h1.id)
      h2.each_child do |h2|
        if h2.is_a?(Type::Local) && h2.kind == Type::Hash
          h2_elems = scratch.get_container_elem_types(env, ep, h2.id)
          elems0 = h1_elems.union(h2_elems)
          if elems
            elems = elems.union(elems0)
          else
            elems = elems0
          end
        end
      end
    end
  end
  elems ||= Type::Hash::Elements.new({Type.any => Type.any})
  base_ty = Type::Instance.new(Type::Builtin[:hash])
  ret_ty = Type::Hash.new(elems, base_ty)
  ctn[ret_ty, ep, env]
end
vmcore_raise(recv, mid, aargs, ep, env, scratch, &ctn) click to toggle source
# File typeprof-0.21.2/lib/typeprof/builtin.rb, line 57
def vmcore_raise(recv, mid, aargs, ep, env, scratch, &ctn)
  # no-op
end
vmcore_set_method_alias(recv, mid, aargs, ep, env, scratch, &ctn) click to toggle source
# File typeprof-0.21.2/lib/typeprof/builtin.rb, line 18
def vmcore_set_method_alias(recv, mid, aargs, ep, env, scratch, &ctn)
  klass, new_mid, old_mid = aargs.lead_tys
  new_sym = get_sym("alias", new_mid, ep, scratch) or return
  old_sym = get_sym("alias", old_mid, ep, scratch) or return
  scratch.alias_method(klass, ep.ctx.cref.singleton, new_sym, old_sym, ep)
  ctn[Type.nil, ep, env]
end
vmcore_undef_method(recv, mid, aargs, ep, env, scratch, &ctn) click to toggle source
# File typeprof-0.21.2/lib/typeprof/builtin.rb, line 26
def vmcore_undef_method(recv, mid, aargs, ep, env, scratch, &ctn)
  # no-op
  ctn[Type.nil, ep, env]
end