Class: RBS::Buffer

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

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(name: nil, content:, parent: nil) ⇒ Buffer

Returns a new instance of Buffer.



9
10
11
12
13
14
15
16
17
18
19
20
 
# File 'lib/rbs/buffer.rb', line 9

def initialize(name: nil, content:, parent: nil)
  case
  when name && content
    @name = name
    @content = content
    @parent = nil
  when parent && content
    @name = parent[0].name
    @content = content
    @parent = parent
  end
end

Instance Attribute Details

#content ⇒ Object (readonly)

Returns the value of attribute content.



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

def content
  @content
end

#name ⇒ Object (readonly)

Returns the value of attribute name.



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

def name
  @name
end

#parent ⇒ Object (readonly)

Returns the value of attribute parent.



7
8
9
 
# File 'lib/rbs/buffer.rb', line 7

def parent
  @parent
end

Instance Method Details

#absolute_position(position) ⇒ Object 



168
169
170
171
172
173
174
175
 
# File 'lib/rbs/buffer.rb', line 168

def absolute_position(position)
  if parent_buffer
    pos = parent_position(position) or return
    parent_buffer.absolute_position(pos)
  else
    position
  end
end

#character_offset(byte_offset) ⇒ Object 



90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
 
# File 'lib/rbs/buffer.rb', line 90

def character_offset(byte_offset)
  top = top_buffer
  return top.character_offset(byte_offset) unless top.equal?(self)

  keys, vals = (@character_offset_cache ||= [[0], [0]])

  idx = keys.bsearch_index { |k| k > byte_offset }
  lo = idx ? idx - 1 : keys.size - 1

  base_byte = keys[lo]
  base_char = vals[lo]
  delta = byte_offset - base_byte
  return base_char if delta == 0

  result = base_char + (content.byteslice(base_byte, delta) or raise).length

  if base_byte == keys[-1]
    keys << byte_offset
    vals << result
  end

  result
end

#detach ⇒ Object 



185
186
187
 
# File 'lib/rbs/buffer.rb', line 185

def detach
  Buffer.new(name: name, content: content)
end

#inspect ⇒ Object 



86
87
88
 
# File 'lib/rbs/buffer.rb', line 86

def inspect
  "#<RBS::Buffer:#{__id__} @name=#{name}, @content=#{content.bytesize} bytes, @lines=#{ranges.size} lines,>"
end

#last_position ⇒ Object 



78
79
80
81
82
83
84
 
# File 'lib/rbs/buffer.rb', line 78

def last_position
  if ranges.empty?
    0
  else
    ranges[-1].end
  end
end

#line_count ⇒ Object 



26
27
28
 
# File 'lib/rbs/buffer.rb', line 26

def line_count
  ranges.size
end

#lines ⇒ Object 



22
23
24
 
# File 'lib/rbs/buffer.rb', line 22

def lines
  ranges.map { self.content[_1] || raise } #$ String
end

#loc_to_pos(loc) ⇒ Object 



68
69
70
71
72
73
74
75
76
 
# File 'lib/rbs/buffer.rb', line 68

def loc_to_pos(loc)
  line, column = loc

  if range = ranges.fetch(line - 1, nil)
    range.begin + column
  else
    last_position
  end
end

#parent_buffer ⇒ Object 



153
154
155
156
157
 
# File 'lib/rbs/buffer.rb', line 153

def parent_buffer
  if parent
    parent[0]
  end
end

#parent_position(position) ⇒ Object 



159
160
161
162
163
164
165
166
 
# File 'lib/rbs/buffer.rb', line 159

def parent_position(position)
  parent or raise "#parent_position is unavailable with buffer without parent"
  return nil unless position <= last_position

  line, column = pos_to_loc(position)
  parent_range = parent[1][line - 1]
  parent_range.begin + column
end

#pos_to_loc(pos) ⇒ Object 



56
57
58
59
60
61
62
63
64
65
66
 
# File 'lib/rbs/buffer.rb', line 56

def pos_to_loc(pos)
  index = ranges.bsearch_index do |range|
    pos <= range.end ? true : false
  end

  if index
    [index + 1, pos - ranges[index].begin]
  else
    [ranges.size + 1, 0]
  end
end

#ranges ⇒ Object 



30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
 
# File 'lib/rbs/buffer.rb', line 30

def ranges
  @ranges ||= begin
    if content.empty?
      ranges = [0...0] #: Array[Range[Integer]]
      lines = [""]
    else
      lines = content.lines
      lines << "" if content.end_with?("\n")

      ranges = [] #: Array[Range[Integer]]
      offset = 0

      lines.each do |line|
        size0 = line.size
        line = line.chomp
        range = offset...(offset+line.size)
        ranges << range

        offset += size0
      end
    end

    ranges
  end
end

#rbs_location(location, loc2 = nil) ⇒ Object 



114
115
116
117
118
119
120
121
 
# File 'lib/rbs/buffer.rb', line 114

def rbs_location(location, loc2=nil)
  top = top_buffer
  if loc2
    Location.new(top, character_offset(location.start_offset), character_offset(loc2.end_offset))
  else
    Location.new(top, character_offset(location.start_offset), character_offset(location.end_offset))
  end
end

#sub_buffer(lines:, byte_lines_hint: nil) ⇒ Object 



123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
 
# File 'lib/rbs/buffer.rb', line 123

def sub_buffer(lines:, byte_lines_hint: nil)
  buf = +""

  if byte_lines_hint
    byte_lines_hint.each_with_index do |range, index|
      slice = content.byteslice(range.begin, range.end - range.begin) or raise
      if slice.include?("\n")
        raise "Line #{index + 1} cannot contain newline character."
      end
      buf << slice
      buf << "\n"
    end
  else
    lines.each_with_index do |range, index|
      start_pos = range.begin
      end_pos = range.end
      slice = content[start_pos...end_pos] or raise
      if slice.include?("\n")
        raise "Line #{index + 1} cannot contain newline character."
      end
      buf << slice
      buf << "\n"
    end
  end

  buf.chomp!

  Buffer.new(content: buf, parent: [self, lines])
end

#top_buffer ⇒ Object 



177
178
179
180
181
182
183
 
# File 'lib/rbs/buffer.rb', line 177

def top_buffer
  if parent_buffer
    parent_buffer.top_buffer
  else
    self
  end
end