Gecko [ v3.cesa.co.za ]

Name	Size	Permission	Date
annotate	[ DIR ]	drwxr-xr-x	2025-06-05 11:22
ast	[ DIR ]	drwxr-xr-x	2025-06-05 11:22
collection	[ DIR ]	drwxr-xr-x	2025-06-05 11:22
definition_builder	[ DIR ]	drwxr-xr-x	2025-06-05 11:22
parser_compat	[ DIR ]	drwxr-xr-x	2025-06-05 11:22
prototype	[ DIR ]	drwxr-xr-x	2025-06-05 11:22
test	[ DIR ]	drwxr-xr-x	2025-06-05 11:22
ancestor_graph.rb	2.47 KB	-rw-r--r--	2022-04-12 11:11
annotate.rb	157 B	-rw-r--r--	2022-04-12 11:11
buffer.rb	1 KB	-rw-r--r--	2022-04-12 11:11
builtin_names.rb	1.59 KB	-rw-r--r--	2022-04-12 11:11
char_scanner.rb	292 B	-rw-r--r--	2022-04-12 11:11
cli.rb	27.19 KB	-rw-r--r--	2022-04-12 11:11
collection.rb	295 B	-rw-r--r--	2022-04-12 11:11
constant.rb	461 B	-rw-r--r--	2022-04-12 11:11
constant_table.rb	5.04 KB	-rw-r--r--	2022-04-12 11:11
definition.rb	9.47 KB	-rw-r--r--	2022-04-12 11:11
definition_builder.rb	32.63 KB	-rw-r--r--	2022-04-12 11:11
environment.rb	15.26 KB	-rw-r--r--	2022-04-12 11:11
environment_loader.rb	3.32 KB	-rw-r--r--	2022-04-12 11:11
environment_walker.rb	4.2 KB	-rw-r--r--	2022-04-12 11:11
errors.rb	11.68 KB	-rw-r--r--	2022-04-12 11:11
factory.rb	301 B	-rw-r--r--	2022-04-12 11:11
location_aux.rb	2.73 KB	-rw-r--r--	2022-04-12 11:11
locator.rb	5.39 KB	-rw-r--r--	2022-04-12 11:11
method_type.rb	2.65 KB	-rw-r--r--	2022-04-12 11:11
namespace.rb	2.12 KB	-rw-r--r--	2022-04-12 11:11
parser_aux.rb	1.3 KB	-rw-r--r--	2022-04-12 11:11
repository.rb	2.83 KB	-rw-r--r--	2022-04-12 11:11
substitution.rb	1.23 KB	-rw-r--r--	2022-04-12 11:11
test.rb	2.57 KB	-rw-r--r--	2022-04-12 11:11
type_alias_dependency.rb	2.82 KB	-rw-r--r--	2022-04-12 11:11
type_alias_regularity.rb	3.14 KB	-rw-r--r--	2022-04-12 11:11
type_name.rb	1.63 KB	-rw-r--r--	2022-04-12 11:11
type_name_resolver.rb	1.56 KB	-rw-r--r--	2022-04-12 11:11
types.rb	27.19 KB	-rw-r--r--	2022-04-12 11:11
validator.rb	4.6 KB	-rw-r--r--	2022-04-12 11:11
variance_calculator.rb	4.45 KB	-rw-r--r--	2022-04-12 11:11
vendorer.rb	2.03 KB	-rw-r--r--	2022-04-12 11:11
version.rb	35 B	-rw-r--r--	2022-04-12 11:11
writer.rb	7.91 KB	-rw-r--r--	2022-04-12 11:11

Rename

module RBS
  class EnvironmentWalker
    InstanceNode = _ = Struct.new(:type_name, keyword_init: true)
    SingletonNode = _ = Struct.new(:type_name, keyword_init: true)
    TypeNameNode = _ = Struct.new(:type_name, keyword_init: true)

attr_reader :env

def initialize(env:)
      @env = env
      @only_ancestors = false
    end

def builder
      @builder ||= DefinitionBuilder.new(env: env)
    end

def only_ancestors!(only = true)
      @only_ancestors = only
      self
    end

def only_ancestors?
      @only_ancestors
    end

include TSort

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.alias_decls.each_key do |type_name|
        yield TypeNameNode.new(type_name: type_name)
      end
    end

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.expand_alias1(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

def each_type_name(type, &block)
      each_type_node(type) do |node|
        yield node.type_name
      end
    end

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
  end
end