Class: RBS::MethodType

Inherits:
Object
  • Object
show all
Defined in:
lib/rbs/method_type.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(type_params:, type:, block:, location:) ⇒ MethodType

Returns a new instance of MethodType.



8
9
10
11
12
13
 
# File 'lib/rbs/method_type.rb', line 8

def initialize(type_params:, type:, block:, location:)
  @type_params = type_params
  @type = type
  @block = block
  @location = location
end

Instance Attribute Details

#block ⇒ Object (readonly)

Returns the value of attribute block.



5
6
7
 
# File 'lib/rbs/method_type.rb', line 5

def block
  @block
end

#location ⇒ Object (readonly)

Returns the value of attribute location.



6
7
8
 
# File 'lib/rbs/method_type.rb', line 6

def location
  @location
end

#type ⇒ Object (readonly)

Returns the value of attribute type.



4
5
6
 
# File 'lib/rbs/method_type.rb', line 4

def type
  @type
end

#type_params ⇒ Object (readonly)

Returns the value of attribute type_params.



3
4
5
 
# File 'lib/rbs/method_type.rb', line 3

def type_params
  @type_params
end

Instance Method Details

#==(other) ⇒ Object 



15
16
17
18
19
20
 
# File 'lib/rbs/method_type.rb', line 15

def ==(other)
  other.is_a?(MethodType) &&
    other.type_params == type_params &&
    other.type == type &&
    other.block == block
end

#each_type(&block) ⇒ Object 



84
85
86
87
88
89
90
91
92
93
 
# File 'lib/rbs/method_type.rb', line 84

def each_type(&block)
  if block
    type.each_type(&block)
    self.block&.yield_self do |b|
      b.type.each_type(&block)
    end
  else
    enum_for :each_type
  end
end

#free_variables(set = Set.new) ⇒ Object 



55
56
57
58
59
 
# File 'lib/rbs/method_type.rb', line 55

def free_variables(set = Set.new)
  type.free_variables(set)
  block&.type&.free_variables(set)
  set.subtract(type_param_names)
end

#map_type(&block) ⇒ Object 



61
62
63
64
65
66
67
68
69
70
 
# File 'lib/rbs/method_type.rb', line 61

def map_type(&block)
  self.class.new(
    type_params: type_params,
    type: type.map_type(&block),
    block: self.block&.yield_self do |b|
      Types::Block.new(type: b.type.map_type(&block), required: b.required)
    end,
    location: location
  )
end

#map_type_bound(&block) ⇒ Object 



72
73
74
75
76
77
78
79
80
81
82
 
# File 'lib/rbs/method_type.rb', line 72

def map_type_bound(&block)
  if type_params.empty?
    self
  else
    self.update(
      type_params: type_params.map {|param|
        param.map_type(&block)
      }
    )
  end
end

#sub(s) ⇒ Object 



31
32
33
34
35
36
37
38
39
40
41
42
43
44
 
# File 'lib/rbs/method_type.rb', line 31

def sub(s)
  sub = s.without(*type_param_names)

  self.class.new(
    type_params: type_params.map do |param|
      param.map_type do |bound|
        bound.map_type {|ty| ty.sub(sub) }
      end
    end,
    type: type.sub(sub),
    block: block&.sub(sub),
    location: location
  )
end

#to_json(state = _ = nil) ⇒ Object 



22
23
24
25
26
27
28
29
 
# File 'lib/rbs/method_type.rb', line 22

def to_json(state = _ = nil)
  {
    type_params: type_params,
    type: type,
    block: block,
    location: location
  }.to_json(state)
end

#to_s ⇒ Object 



95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
 
# File 'lib/rbs/method_type.rb', line 95

def to_s
  s = case
      when (b = block) && b.required
        "(#{type.param_to_s}) { (#{b.type.param_to_s}) -> #{b.type.return_to_s} } -> #{type.return_to_s}"
      when b = block
        "(#{type.param_to_s}) ?{ (#{b.type.param_to_s}) -> #{b.type.return_to_s} } -> #{type.return_to_s}"
      else
        "(#{type.param_to_s}) -> #{type.return_to_s}"
      end

  if type_params.empty?
    s
  else
    "[#{type_params.join(", ")}] #{s}"
  end
end

#type_param_names ⇒ Object 



112
113
114
 
# File 'lib/rbs/method_type.rb', line 112

def type_param_names
  type_params.map(&:name)
end

#update(type_params: self.type_params, type: self.type, block: self.block, location: self.location) ⇒ Object 



46
47
48
49
50
51
52
53
 
# File 'lib/rbs/method_type.rb', line 46

def update(type_params: self.type_params, type: self.type, block: self.block, location: self.location)
  self.class.new(
    type_params: type_params,
    type: type,
    block: block,
    location: location
  )
end