steep-relaxed 1.9.3.3

data/lib/steep/ast/node/type_assertion.rb

@@ -0,0 +1,81 @@
+module Steep
+  module AST
+    module Node
+      class TypeAssertion
+        attr_reader :location
+
+        def initialize(location)
+          @location = location
+        end
+
+        def source
+          location.source
+        end
+
+        def line
+          location.start_line
+        end
+
+        def type(context, subtyping, type_vars)
+          if ty = RBS::Parser.parse_type(type_location.buffer, range: type_location.range, variables: type_vars, require_eof: true)
+            resolver = RBS::Resolver::TypeNameResolver.new(subtyping.factory.env)
+            ty = ty.map_type_name {|name| resolver.resolve(name, context: context) || name.absolute! }
+
+            validator = Signature::Validator.new(checker: subtyping)
+            validator.rescue_validation_errors do
+              validator.validate_type(ty)
+            end
+
+            unless validator.has_error?
+              subtyping.factory.type(ty)
+            else
+              validator.each_error.to_a
+            end
+          else
+            nil
+          end
+        rescue ::RBS::ParsingError => exn
+          exn
+        end
+
+        def type_syntax?
+          RBS::Parser.parse_type(type_location.buffer, range: type_location.range, variables: [], require_eof: true)
+          true
+        rescue ::RBS::ParsingError
+          false
+        end
+
+        def type?(context, subtyping, type_vars)
+          type = type(context, subtyping, type_vars)
+
+          case type
+          when RBS::ParsingError, nil, Array
+            nil
+          else
+            type
+          end
+        end
+
+        def type_str
+          @type_str ||= source.delete_prefix(":").lstrip
+        end
+
+        def type_location
+          offset = source.size - type_str.size
+          RBS::Location.new(location.buffer, location.start_pos + offset, location.end_pos)
+        end
+
+        def self.parse(location)
+          source = location.source.strip
+
+          if source =~/\A:\s*(.+)/
+            assertion = TypeAssertion.new(location)
+            if assertion.type_syntax?
+              assertion
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/steep/ast/types/any.rb

@@ -0,0 +1,35 @@
+module Steep
+  module AST
+    module Types
+      class Any
+        extend SharedInstance
+
+        def ==(other)
+          other.is_a?(Any)
+        end
+
+        def hash
+          self.class.hash
+        end
+
+        alias eql? ==
+
+        def subst(s)
+          self
+        end
+
+        def to_s
+          "untyped"
+        end
+
+        include Helper::NoFreeVariables
+
+        include Helper::NoChild
+
+        def level
+          [1]
+        end
+      end
+    end
+  end
+end

data/lib/steep/ast/types/boolean.rb

@@ -0,0 +1,45 @@
+module Steep
+  module AST
+    module Types
+      class Boolean
+        extend SharedInstance
+        
+        def ==(other)
+          other.is_a?(Boolean)
+        end
+
+        def hash
+          self.class.hash
+        end
+
+        alias eql? ==
+
+        def subst(s)
+          self
+        end
+
+        def to_s
+          "bool"
+        end
+
+        include Helper::NoFreeVariables
+
+        include Helper::NoChild
+
+        def level
+          [0]
+        end
+
+        def back_type
+          Union.build(
+            types: [
+              Builtin::TrueClass.instance_type,
+              Builtin::FalseClass.instance_type
+            ]
+          )
+        end
+      end
+    end
+  end
+end
+

data/lib/steep/ast/types/bot.rb

@@ -0,0 +1,35 @@
+module Steep
+  module AST
+    module Types
+      class Bot
+        extend SharedInstance
+        
+        def ==(other)
+          other.is_a?(Bot)
+        end
+
+        def hash
+          self.class.hash
+        end
+
+        alias eql? ==
+
+        def subst(s)
+          self
+        end
+
+        def to_s
+          "bot"
+        end
+
+        include Helper::NoFreeVariables
+
+        include Helper::NoChild
+
+        def level
+          [2]
+        end
+      end
+    end
+  end
+end

data/lib/steep/ast/types/class.rb

@@ -0,0 +1,43 @@
+module Steep
+  module AST
+    module Types
+      class Class
+        extend SharedInstance
+
+        def to_s
+          "class"
+        end
+
+        def ==(other)
+          other.is_a?(Class)
+        end
+
+        def hash
+          self.class.hash
+        end
+
+        alias eql? ==
+
+        def subst(s)
+          if s.module_type
+            s.module_type
+          else
+            self
+          end
+        end
+
+        @@fvs = Set[instance]
+
+        def free_variables
+          @@fvs
+        end
+
+        include Helper::NoChild
+
+        def level
+          [0]
+        end
+      end
+    end
+  end
+end