steep-relaxed 1.9.3.3

data/lib/steep/services/type_check_service.rb

@@ -0,0 +1,394 @@
+module Steep
+  module Services
+    class TypeCheckService
+      attr_reader :project
+      attr_reader :signature_validation_diagnostics
+      attr_reader :source_files
+      attr_reader :signature_services
+
+      class SourceFile
+        attr_reader :path
+        attr_reader :target
+        attr_reader :content
+        attr_reader :node
+        attr_reader :typing
+        attr_reader :errors
+        attr_reader :ignores
+
+        def initialize(path:, node:, content:, typing:, ignores:, errors:)
+          @path = path
+          @node = node
+          @content = content
+          @typing = typing
+          @ignores = ignores
+          @errors = errors
+        end
+
+        def self.with_syntax_error(path:, content:, error:)
+          new(path: path, node: false, content: content, errors: [error], typing: nil, ignores: nil)
+        end
+
+        def self.with_typing(path:, content:, typing:, node:, ignores:)
+          new(path: path, node: node, content: content, errors: nil, typing: typing, ignores: ignores)
+        end
+
+        def self.no_data(path:, content:)
+          new(path: path, content: content, node: false, errors: nil, typing: nil, ignores: nil)
+        end
+
+        def update_content(content)
+          self.class.new(
+            path: path,
+            content: content,
+            node: node,
+            errors: errors,
+            typing: typing,
+            ignores: ignores
+          )
+        end
+
+        def diagnostics
+          case
+          when errors
+            errors
+          when typing && ignores
+            errors = [] #: Array[Diagnostic::Ruby::Base]
+
+            errors.concat(
+              typing.errors.delete_if do |diagnostic|
+                case diagnostic.location
+                when ::Parser::Source::Range
+                  ignores.ignore?(diagnostic.location.first_line, diagnostic.location.last_line, diagnostic.diagnostic_code)
+                when RBS::Location
+                  ignores.ignore?(diagnostic.location.start_line, diagnostic.location.end_line, diagnostic.diagnostic_code)
+                end
+              end
+            )
+
+            ignores.error_ignores.each do |ignore|
+              errors << Diagnostic::Ruby::InvalidIgnoreComment.new(comment: ignore.comment)
+            end
+
+            errors
+          else
+            []
+          end
+        end
+      end
+
+      def initialize(project:)
+        @project = project
+
+        @source_files = {}
+        @signature_services = project.targets.each.with_object({}) do |target, hash| #$ Hash[Symbol, SignatureService]
+          loader = Project::Target.construct_env_loader(options: target.options, project: project)
+          hash[target.name] = SignatureService.load_from(loader, implicitly_returns_nil: target.implicitly_returns_nil)
+        end
+        @signature_validation_diagnostics = project.targets.each.with_object({}) do |target, hash| #$ Hash[Symbol, Hash[Pathname, Array[Diagnostic::Signature::Base]]]
+          hash[target.name] = {}
+        end
+      end
+
+      def signature_diagnostics
+        # @type var signature_diagnostics: Hash[Pathname, Array[Diagnostic::Signature::Base]]
+        signature_diagnostics = {}
+
+        project.targets.each do |target|
+          service = signature_services.fetch(target.name)
+
+          service.each_rbs_path do |path|
+            signature_diagnostics[path] ||= []
+          end
+
+          case service.status
+          when SignatureService::SyntaxErrorStatus, SignatureService::AncestorErrorStatus
+            service.status.diagnostics.group_by {|diag| diag.location&.buffer&.name&.to_s }.each do |path_string, diagnostics|
+              if path_string
+                path = Pathname(path_string)
+                signature_diagnostics.fetch(path).push(*diagnostics)
+              end
+            end
+          when SignatureService::LoadedStatus
+            validation_diagnostics = signature_validation_diagnostics[target.name] || {}
+            validation_diagnostics.each do |path, diagnostics|
+              signature_diagnostics.fetch(path).push(*diagnostics)
+            end
+          end
+        end
+
+        signature_diagnostics
+      end
+
+      def has_diagnostics?
+        each_diagnostics.count > 0
+      end
+
+      def diagnostics
+        each_diagnostics.to_h
+      end
+
+      def each_diagnostics(&block)
+        if block
+          signature_diagnostics.each do |path, diagnostics|
+            yield [path, diagnostics]
+          end
+
+          source_files.each_value do |file|
+            yield [file.path, file.diagnostics]
+          end
+        else
+          enum_for :each_diagnostics
+        end
+      end
+
+      def update(changes:)
+        Steep.measure "#update_signature" do
+          update_signature(changes: changes)
+        end
+
+        Steep.measure "#update_sources" do
+          update_sources(changes: changes)
+        end
+      end
+
+      def validate_signature(path:, target:)
+        Steep.logger.tagged "#validate_signature(path=#{path})" do
+          Steep.measure "validation" do
+            service = signature_services.fetch(target.name)
+
+            raise "#{path} is not library nor signature of #{target.name}" unless target.possible_signature_file?(path) || service.env_rbs_paths.include?(path)
+
+            case service.status
+            when SignatureService::SyntaxErrorStatus
+              diagnostics = service.status.diagnostics.select do |diag|
+                diag.location or raise
+                Pathname(diag.location.buffer.name) == path &&
+                  (diag.is_a?(Diagnostic::Signature::SyntaxError) || diag.is_a?(Diagnostic::Signature::UnexpectedError))
+              end
+
+            when SignatureService::AncestorErrorStatus
+              diagnostics = service.status.diagnostics.select do |diag|
+                diag.location or raise
+                Pathname(diag.location.buffer.name) == path
+              end
+
+            when SignatureService::LoadedStatus
+              validator = Signature::Validator.new(checker: service.current_subtyping || raise)
+              type_names = service.type_names(paths: Set[path], env: service.latest_env).to_set
+
+              unless type_names.empty?
+                Steep.measure2 "Validating #{type_names.size} types" do |sampler|
+                  type_names.each do |type_name|
+                    sampler.sample(type_name.to_s) do
+                      case
+                      when type_name.class?
+                        validator.validate_one_class(type_name)
+                      when type_name.interface?
+                        validator.validate_one_interface(type_name)
+                      when type_name.alias?
+                        validator.validate_one_alias(type_name)
+                      end
+                    end
+                  end
+                end
+              end
+
+              const_decls = service.const_decls(paths: Set[path], env: service.latest_env)
+              unless const_decls.empty?
+                Steep.measure2 "Validating #{const_decls.size} constants" do |sampler|
+                  const_decls.each do |name, entry|
+                    sampler.sample(name.to_s) do
+                      validator.validate_one_constant(name, entry)
+                    end
+                  end
+                end
+              end
+
+              global_decls = service.global_decls(paths: Set[path])
+              unless global_decls.empty?
+                Steep.measure2 "Validating #{global_decls.size} globals" do |sampler|
+                  global_decls.each do |name, entry|
+                    sampler.sample(name.to_s) do
+                      validator.validate_one_global(name, entry)
+                    end
+                  end
+                end
+              end
+
+              diagnostics = validator.each_error.select do |error|
+                error.location or raise
+                Pathname(error.location.buffer.name) == path
+              end
+            end
+
+            signature_validation_diagnostics.fetch(target.name)[path] = diagnostics
+          end
+        end
+      end
+
+      def typecheck_source(path:, target:)
+        return unless target
+
+        Steep.logger.tagged "#typecheck_source(path=#{path})" do
+          Steep.measure "typecheck" do
+            signature_service = signature_services.fetch(target.name)
+            subtyping = signature_service.current_subtyping
+
+            if subtyping
+              text = source_files.fetch(path).content
+              file = type_check_file(target: target, subtyping: subtyping, path: path, text: text) { signature_service.latest_constant_resolver }
+              source_files[path] = file
+
+              file.diagnostics
+            end
+          end
+        end
+      end
+
+      def update_signature(changes:)
+        Steep.logger.tagged "#update_signature" do
+          signature_targets = {} #: Hash[Pathname, Project::Target]
+          changes.each do |path, changes|
+            target = project.targets.find { _1.possible_signature_file?(path) } or next
+            signature_targets[path] = target
+          end
+
+          project.targets.each do |target|
+            Steep.logger.tagged "#{target.name}" do
+              # Load signatures from all project targets but `#unreferenced` ones
+              target_changes = changes.select do |path, _|
+                signature_target = signature_targets.fetch(path, nil) or next
+                signature_target == target || !signature_target.unreferenced
+              end
+
+              unless target_changes.empty?
+                signature_services.fetch(target.name).update(target_changes)
+              end
+            end
+          end
+        end
+      end
+
+      def update_sources(changes:)
+        changes.each do |path, changes|
+          if source_file?(path)
+            file = source_files[path] || SourceFile.no_data(path: path, content: "")
+            content = changes.inject(file.content) {|text, change| change.apply_to(text) }
+            source_files[path] = file.update_content(content)
+          end
+        end
+      end
+
+      def type_check_file(target:, subtyping:, path:, text:)
+        Steep.logger.tagged "#type_check_file(#{path}@#{target.name})" do
+          source = Source.parse(text, path: path, factory: subtyping.factory)
+          typing = TypeCheckService.type_check(source: source, subtyping: subtyping, constant_resolver: yield, cursor: nil)
+          ignores = Source::IgnoreRanges.new(ignores: source.ignores)
+          SourceFile.with_typing(path: path, content: text, node: source.node, typing: typing, ignores: ignores)
+        end
+      rescue AnnotationParser::SyntaxError => exn
+        error = Diagnostic::Ruby::AnnotationSyntaxError.new(message: exn.message, location: exn.location)
+        SourceFile.with_syntax_error(path: path, content: text, error: error)
+      rescue ::Parser::SyntaxError => exn
+        error = Diagnostic::Ruby::SyntaxError.new(message: exn.message, location: (_ = exn).diagnostic.location)
+        SourceFile.with_syntax_error(path: path, content: text, error: error)
+      rescue EncodingError => exn
+        SourceFile.no_data(path: path, content: "")
+      rescue RuntimeError => exn
+        Steep.log_error(exn)
+        SourceFile.no_data(path: path, content: text)
+      end
+
+      def self.type_check(source:, subtyping:, constant_resolver:, cursor:)
+        annotations = source.annotations(block: source.node, factory: subtyping.factory, context: nil)
+
+        case annotations.self_type
+        when AST::Types::Name::Instance
+          module_name = annotations.self_type.name
+          module_type = AST::Types::Name::Singleton.new(name: module_name)
+          instance_type = annotations.self_type
+        when AST::Types::Name::Singleton
+          module_name = annotations.self_type.name
+          module_type = annotations.self_type
+          instance_type = annotations.self_type
+        else
+          module_name = AST::Builtin::Object.module_name
+          module_type = AST::Builtin::Object.module_type
+          instance_type = AST::Builtin::Object.instance_type
+        end
+
+        definition = subtyping.factory.definition_builder.build_instance(module_name)
+
+        const_env = TypeInference::ConstantEnv.new(
+          factory: subtyping.factory,
+          context: nil,
+          resolver: constant_resolver
+        )
+        type_env = TypeInference::TypeEnv.new(const_env)
+        type_env = TypeInference::TypeEnvBuilder.new(
+          TypeInference::TypeEnvBuilder::Command::ImportConstantAnnotations.new(annotations),
+          TypeInference::TypeEnvBuilder::Command::ImportGlobalDeclarations.new(subtyping.factory),
+          TypeInference::TypeEnvBuilder::Command::ImportInstanceVariableDefinition.new(definition, subtyping.factory),
+          TypeInference::TypeEnvBuilder::Command::ImportInstanceVariableAnnotations.new(annotations),
+          TypeInference::TypeEnvBuilder::Command::ImportLocalVariableAnnotations.new(annotations)
+        ).build(type_env)
+
+        context = TypeInference::Context.new(
+          block_context: nil,
+          module_context: TypeInference::Context::ModuleContext.new(
+            instance_type: instance_type,
+            module_type: module_type,
+            implement_name: nil,
+            nesting: nil,
+            class_name: module_name,
+            instance_definition: subtyping.factory.definition_builder.build_instance(module_name),
+            module_definition: subtyping.factory.definition_builder.build_singleton(module_name)
+          ),
+          method_context: nil,
+          break_context: nil,
+          self_type: instance_type,
+          type_env: type_env,
+          call_context: TypeInference::MethodCall::TopLevelContext.new,
+          variable_context: TypeInference::Context::TypeVariableContext.empty
+        )
+
+        typing = Typing.new(source: source, root_context: context, cursor: cursor)
+
+        construction = TypeConstruction.new(
+          checker: subtyping,
+          annotations: annotations,
+          source: source,
+          context: context,
+          typing: typing
+        )
+
+        construction.synthesize(source.node) if source.node
+
+        typing
+      end
+
+      def source_file?(path)
+        source_files.key?(path) || (project.target_for_source_path(path) ? true : false)
+      end
+
+      def signature_file?(path)
+        relative_path = project.relative_path(path)
+        targets = signature_services.select {|_, sig| sig.files.key?(relative_path) || sig.env_rbs_paths.include?(path) }
+        unless targets.empty?
+          targets.keys
+        end
+      end
+
+      def app_signature_file?(path)
+        target_names = signature_services.select {|_, sig| sig.files.key?(path) }.keys
+        unless target_names.empty?
+          target_names
+        end
+      end
+
+      def lib_signature_file?(path)
+        signature_services.each_value.any? {|sig| sig.env_rbs_paths.include?(path) }
+      end
+    end
+  end
+end

data/lib/steep/services/type_name_completion.rb

@@ -0,0 +1,236 @@
+module Steep
+  module Services
+    class TypeNameCompletion
+      module Prefix
+        RawIdentPrefix = _ = Struct.new(:ident) do
+          # @implements RawIdentPrefix
+
+          def const_name?
+            ident.start_with?(/[A-Z]/)
+          end
+
+          def size
+            ident.length
+          end
+        end
+
+        NamespacedIdentPrefix = _ = Struct.new(:namespace, :ident, :size) do
+          # @implements NamespacedIdentPrefix
+
+          def const_name?
+            ident.start_with?(/[A-Z]/)
+          end
+        end
+
+        NamespacePrefix = _ = Struct.new(:namespace, :size)
+
+        def self.parse(buffer, line:, column:)
+          pos = buffer.loc_to_pos([line, column])
+          prefix = buffer.content[0...pos] or raise
+          prefix.reverse!
+
+          case prefix
+          when /\A((::\w+[A-Z])+(::)?)/
+            namespace = $1 or raise
+            NamespacePrefix.new(RBS::Namespace.parse(namespace.reverse), namespace.size)
+          when /\A::/
+            NamespacePrefix.new(RBS::Namespace.root, 2)
+          when /\A(\w*[A-Za-z_])((::\w+[A-Z])+(::)?)/
+            namespace = $1 or raise
+            identifier = $2 or raise
+            NamespacedIdentPrefix.new(RBS::Namespace.parse(identifier.reverse), namespace.reverse, namespace.size + identifier.size)
+          when /\A(\w*[A-Za-z_])::/
+            namespace = $1 or raise
+            NamespacedIdentPrefix.new(RBS::Namespace.root, namespace.reverse, namespace.size + 2)
+          when /\A(\w*[A-Za-z_])/
+            identifier = $1 or raise
+            RawIdentPrefix.new(identifier.reverse)
+          end
+        end
+      end
+
+      attr_reader :env, :context, :type_name_resolver, :map
+
+      def initialize(env:, context:, dirs:)
+        @env = env
+        @context = context
+
+        table = RBS::Environment::UseMap::Table.new()
+        table.known_types.merge(env.class_decls.keys)
+        table.known_types.merge(env.class_alias_decls.keys)
+        table.known_types.merge(env.type_alias_decls.keys)
+        table.known_types.merge(env.interface_decls.keys)
+        table.compute_children
+
+        @map = RBS::Environment::UseMap.new(table: table)
+        dirs.each do |dir|
+          dir.clauses.each do |clause|
+            @map.build_map(clause)
+          end
+        end
+
+        @type_name_resolver = RBS::Resolver::TypeNameResolver.new(env)
+      end
+
+      def each_outer_module(context = self.context, &block)
+        if block
+          if (parent, con = context)
+            namespace = each_outer_module(parent, &block)
+            case con
+            when false
+              namespace
+            when RBS::TypeName
+              ns = con.with_prefix(namespace).to_namespace
+              yield(ns)
+              ns
+            end
+          else
+            yield(RBS::Namespace.root)
+            RBS::Namespace.root
+          end
+        else
+          enum_for :each_outer_module
+        end
+      end
+
+      def each_type_name(&block)
+        if block
+          env = self.env
+
+          table = {} #: Hash[RBS::Namespace, Array[RBS::TypeName]]
+          env.class_decls.each_key do |type_name|
+            yield(type_name)
+            (table[type_name.namespace] ||= []) << type_name
+          end
+          env.type_alias_decls.each_key do |type_name|
+            yield(type_name)
+            (table[type_name.namespace] ||= []) << type_name
+          end
+          env.interface_decls.each_key do |type_name|
+            yield(type_name)
+            (table[type_name.namespace] ||= []) << type_name
+          end
+          env.class_alias_decls.each_key do |type_name|
+            yield(type_name)
+            (table[type_name.namespace] ||= []) << type_name
+          end
+
+          env.class_alias_decls.each_key do |alias_name|
+            normalized_name = env.normalize_module_name?(alias_name) or next
+            each_type_name_under(alias_name, normalized_name, table: table, &block)
+          end
+
+          resolve_pairs = [] #: Array[[RBS::TypeName, RBS::TypeName]]
+
+          map.instance_eval do
+            @map.each_key do |name|
+              relative_name = RBS::TypeName.new(name: name, namespace: RBS::Namespace.empty)
+              if absolute_name = resolve?(relative_name)
+                if env.type_name?(absolute_name)
+                  # Yields only if the relative type name resolves to existing absolute type name
+                  resolve_pairs << [relative_name, absolute_name]
+                end
+              end
+            end
+          end
+
+          resolve_pairs.each do |use_name, absolute_name|
+            yield use_name
+            each_type_name_under(use_name, absolute_name, table: table, &block)
+          end
+        else
+          enum_for :each_type_name
+        end
+      end
+
+      def each_type_name_under(module_name, normalized_name, table:, &block)
+        if children = table.fetch(normalized_name.to_namespace, nil)
+          module_namespace = module_name.to_namespace
+
+          children.each do |normalized_child_name|
+            child_name = RBS::TypeName.new(namespace: module_namespace, name: normalized_child_name.name)
+
+            yield child_name
+
+            if normalized_child_name.class?
+              each_type_name_under(child_name, env.normalize_module_name(normalized_child_name), table: table, &block)
+            end
+          end
+        end
+      end
+
+      def resolve_used_name(name)
+        return nil if name.absolute?
+
+        case
+        when resolved = map.resolve?(name)
+          resolved
+        when name.namespace.empty?
+          nil
+        else
+          if resolved_parent = resolve_used_name(name.namespace.to_type_name)
+            resolved_name = RBS::TypeName.new(namespace: resolved_parent.to_namespace, name: name.name)
+            if env.normalize_type_name?(resolved_name)
+              resolved_name
+            end
+          end
+        end
+      end
+
+      def resolve_name_in_context(name)
+        if resolved_name = resolve_used_name(name)
+          return [resolved_name, name]
+        end
+
+        name.absolute? or raise
+        normalized_name = env.normalize_type_name?(name) or raise "Cannot normalize given type name `#{name}`"
+
+        name.namespace.path.reverse_each.inject(RBS::TypeName.new(namespace: RBS::Namespace.empty, name: name.name)) do |relative_name, component|
+          if type_name_resolver.resolve(relative_name, context: context) == name
+            return [normalized_name, relative_name]
+          end
+
+          RBS::TypeName.new(
+            namespace: RBS::Namespace.new(path: [component, *relative_name.namespace.path], absolute: false),
+            name: name.name
+          )
+        end
+
+        if type_name_resolver.resolve(name.relative!, context: context) == name && !resolve_used_name(name.relative!)
+          [normalized_name, name.relative!]
+        else
+          [normalized_name, name]
+        end
+      end
+
+      def find_type_names(prefix)
+        case prefix
+        when Prefix::RawIdentPrefix
+          each_type_name.filter do |type_name|
+            type_name.split.any? {|sym| sym.to_s.downcase.include?(prefix.ident.downcase) }
+          end
+        when Prefix::NamespacedIdentPrefix
+          absolute_namespace =
+            if prefix.namespace.empty?
+              RBS::Namespace.root
+            else
+              type_name_resolver.resolve(prefix.namespace.to_type_name, context: context)&.to_namespace || prefix.namespace
+            end
+
+          each_type_name.filter do|name|
+            name.namespace == absolute_namespace &&
+              name.name.to_s.downcase.include?(prefix.ident.downcase)
+          end
+        when Prefix::NamespacePrefix
+          absolute_namespace = type_name_resolver.resolve(prefix.namespace.to_type_name, context: context)&.to_namespace || prefix.namespace
+          each_type_name.filter {|name| name.namespace == absolute_namespace }
+        else
+          # Returns all of the accessible type names from the context
+          namespaces = each_outer_module.to_set
+          # Relative type name means a *use*d type name
+          each_type_name.filter {|name| namespaces.include?(name.namespace) || !name.absolute? }
+        end
+      end
+    end
+  end
+end