Class: RBS::EnvironmentWalker
Relationships & Source Files | |
Super Chains via Extension / Inclusion / Inheritance | |
Instance Chain:
self,
TSort
|
|
Inherits: | Object |
Defined in: | lib/rbs/environment_walker.rb |
Constant Summary
-
InstanceNode =
# File 'lib/rbs/environment_walker.rb', line 5_ = Struct.new(:type_name, keyword_init: true)
-
SingletonNode =
# File 'lib/rbs/environment_walker.rb', line 6_ = Struct.new(:type_name, keyword_init: true)
-
TypeNameNode =
# File 'lib/rbs/environment_walker.rb', line 7_ = Struct.new(:type_name, keyword_init: true)
Class Method Summary
- .new(env:) ⇒ EnvironmentWalker constructor
Instance Attribute Summary
- #env readonly
- #only_ancestors? ⇒ Boolean readonly
Instance Method Summary
Constructor Details
.new(env:) ⇒ EnvironmentWalker
# File 'lib/rbs/environment_walker.rb', line 11
def initialize(env:) @env = env @only_ancestors = false end
Instance Attribute Details
#env (readonly)
[ GitHub ]# File 'lib/rbs/environment_walker.rb', line 9
attr_reader :env
#only_ancestors? ⇒ Boolean
(readonly)
[ GitHub ]
# File 'lib/rbs/environment_walker.rb', line 25
def only_ancestors? @only_ancestors end
Instance Method Details
#builder
[ GitHub ]# File 'lib/rbs/environment_walker.rb', line 16
def builder @builder ||= DefinitionBuilder.new(env: env) end
#each_type_name(type, &block)
[ GitHub ]# File 'lib/rbs/environment_walker.rb', line 99
def each_type_name(type, &block) each_type_node(type) do |node| yield node.type_name end end
#each_type_node(type, &block)
[ GitHub ]# File 'lib/rbs/environment_walker.rb', line 105
def each_type_node(type, &block) case type when RBS::Types::Bases::Any when RBS::Types::Bases::Class when RBS::Types::Bases::Instance when RBS::Types::Bases::Self when RBS::Types::Bases::Top when RBS::Types::Bases::Bottom when RBS::Types::Bases::Bool when RBS::Types::Bases::Void when RBS::Types::Bases::Nil when RBS::Types::Variable when RBS::Types::ClassSingleton yield SingletonNode.new(type_name: type.name) when RBS::Types::ClassInstance yield InstanceNode.new(type_name: type.name) type.args.each do |ty| each_type_node(ty, &block) end when RBS::Types::Interface yield TypeNameNode.new(type_name: type.name) type.args.each do |ty| each_type_node(ty, &block) end when RBS::Types::Alias yield TypeNameNode.new(type_name: type.name) type.args.each do |ty| each_type_node(ty, &block) end when RBS::Types::Union, RBS::Types::Intersection, RBS::Types::Tuple type.types.each do |ty| each_type_node ty, &block end when RBS::Types::Optional each_type_node type.type, &block when RBS::Types::Literal # nop when RBS::Types::Record type.fields.each_value do |ty| each_type_node ty, &block end when RBS::Types::Proc type.each_type do |ty| each_type_node ty, &block end else raise "Unexpected type given: #{type}" end end
#only_ancestors!(only = true)
[ GitHub ]# File 'lib/rbs/environment_walker.rb', line 20
def only_ancestors!(only = true) @only_ancestors = only self end
#tsort_each_child(node, &block)
[ GitHub ]# File 'lib/rbs/environment_walker.rb', line 44
def tsort_each_child(node, &block) name = node.type_name unless name.namespace.empty? yield SingletonNode.new(type_name: name.namespace.to_type_name) end case node when TypeNameNode case when name.interface? definition = builder.build_interface(name) unless only_ancestors? definition.each_type do |type| each_type_node type, &block end end when name.alias? each_type_node builder. (name), &block else raise "Unexpected TypeNameNode with type_name=#{name}" end when InstanceNode, SingletonNode definition = if node.is_a?(InstanceNode) builder.build_instance(name) else builder.build_singleton(name) end if ancestors = definition.ancestors ancestors.ancestors.each do |ancestor| case ancestor when Definition::Ancestor::Instance yield InstanceNode.new(type_name: ancestor.name) unless only_ancestors? ancestor.args.each do |type| each_type_node type, &block end end when Definition::Ancestor::Singleton yield SingletonNode.new(type_name: ancestor.name) end end end unless only_ancestors? definition.each_type do |type| each_type_node type, &block end end end end
#tsort_each_node(&block)
[ GitHub ]# File 'lib/rbs/environment_walker.rb', line 31
def tsort_each_node(&block) env.class_decls.each_key do |type_name| yield InstanceNode.new(type_name: type_name) yield SingletonNode.new(type_name: type_name) end env.interface_decls.each_key do |type_name| yield TypeNameNode.new(type_name: type_name) end env.type_alias_decls.each_key do |type_name| yield TypeNameNode.new(type_name: type_name) end end