Class: RBS::Location

# File 'lib/rbs/wasm/location.rb', line 12

def initialize(buffer, start_pos, end_pos)
  @buffer = buffer
  @start_pos = start_pos
  @end_pos = end_pos
  @required_children = {} #: Hash[Symbol, [ Integer, Integer ]]
  @optional_children = {} #: Hash[Symbol, [ Integer, Integer ]?]
end

# File 'lib/rbs/wasm/location.rb', line 48

def [](name)
  if (range = @required_children[name])
    return Location.new(@buffer, range[0], range[1])
  end

  if @optional_children.key?(name)
    range = @optional_children[name]
    return range && Location.new(@buffer, range[0], range[1])
  end

  raise "Unknown child name given: #{name}"
end

# File 'lib/rbs/wasm/location.rb', line 32

def _add_optional_child(name, start_pos, end_pos)
  @optional_children[name] = [start_pos, end_pos]
end

# File 'lib/rbs/wasm/location.rb', line 36

def _add_optional_no_child(name)
  @optional_children[name] = nil
end

# File 'lib/rbs/wasm/location.rb', line 28

def _add_required_child(name, start_pos, end_pos)
  @required_children[name] = [start_pos, end_pos]
end

# File 'lib/rbs/wasm/location.rb', line 24

def _end_pos
  @end_pos
end

# File 'lib/rbs/wasm/location.rb', line 44

def _optional_keys
  @optional_children.keys
end

# File 'lib/rbs/wasm/location.rb', line 40

def _required_keys
  @required_children.keys
end

# File 'lib/rbs/wasm/location.rb', line 20

def _start_pos
  @start_pos
end

# File 'lib/rbs/location_aux.rb', line 110

def add_optional_child(name, range)
  if range
    _add_optional_child(name, range.begin, range.end)
  else
    _add_optional_no_child(name);
  end
end

# File 'lib/rbs/location_aux.rb', line 106

def add_required_child(name, range)
  _add_required_child(name, range.begin, range.end)
end

# File 'lib/rbs/location_aux.rb', line 27

alias aref []

# File 'lib/rbs/location_aux.rb', line 118

def each_optional_key(&block)
  if block
    _optional_keys.uniq.each(&block)
  else
    enum_for(:each_optional_key)
  end
end

# File 'lib/rbs/location_aux.rb', line 126

def each_required_key(&block)
  if block
    _required_keys.uniq.each(&block)
  else
    enum_for(:each_required_key)
  end
end

# File 'lib/rbs/location_aux.rb', line 55

def end_column
  end_loc[1]
end

# File 'lib/rbs/location_aux.rb', line 51

def end_line
  end_loc[0]
end

# File 'lib/rbs/location_aux.rb', line 63

def end_loc
  @end_loc ||= buffer.top_buffer.pos_to_loc(end_pos)
end

# File 'lib/rbs/location_aux.rb', line 35

def end_pos
  buffer.absolute_position(_end_pos) || raise
end

# File 'ext/rbs_extension/legacy_location.c', line 126

static VALUE location_initialize_copy(VALUE self, VALUE other) {
    rbs_loc *self_loc = rbs_check_location(self);
    rbs_loc *other_loc = rbs_check_location(other);

    *self_loc = (rbs_loc) {
        .buffer = Qnil,
        .rg = other_loc->rg,
        .children = NULL,
    };
    RB_OBJ_WRITE(self, &self_loc->buffer, other_loc->buffer);

    if (other_loc->children != NULL) {
        rbs_loc_legacy_alloc_children(self_loc, other_loc->children->cap);
        memcpy(self_loc->children, other_loc->children, RBS_LOC_CHILDREN_SIZE(other_loc->children->cap));
    }

    return Qnil;
}

# File 'lib/rbs/location_aux.rb', line 5

def inspect
  rks = each_required_key.to_a
  ops = each_optional_key.to_a.map {|x| "?#{x}" }
  src = if source.length <= 1
    source.inspect
  else
    source.each_line.first&.chomp&.inspect
  end
  "#<#{self.class}:#{self.__id__} buffer=#{buffer.name}, start=#{start_line}:#{start_column}, pos=#{start_pos}...#{end_pos}, children=#{(rks + ops).join(",")} source=#{src}>"
end

# File 'lib/rbs/location_aux.rb', line 134

def key?(name)
  optional_key?(name) || required_key?(name)
end

# File 'lib/rbs/location_aux.rb', line 146

def local_location
  loc = Location.new(buffer.detach, _start_pos, _end_pos)

  each_optional_key do |key|
    value = self[key]
    if value
      loc.add_optional_child(key, value._start_pos...value._end_pos)
    else
      loc.add_optional_child(key, nil)
    end
  end

  each_required_key do |key|
    value = self[key] or raise
    loc.add_required_child(key, value._start_pos...value._end_pos)
  end

  loc #: self
end

# File 'lib/rbs/location_aux.rb', line 166

def local_source
  local_location.source
end

# File 'lib/rbs/location_aux.rb', line 39

def name
  buffer.name
end

# File 'lib/rbs/location_aux.rb', line 138

def optional_key?(name)
  _optional_keys.include?(name)
end

# File 'lib/rbs/location_aux.rb', line 67

def range
  @range ||= start_pos...end_pos
end

# File 'lib/rbs/location_aux.rb', line 142

def required_key?(name)
  _required_keys.include?(name)
end

# File 'lib/rbs/location_aux.rb', line 71

def source
  @source ||= (buffer.top_buffer.content[range] || raise)
end

# File 'lib/rbs/location_aux.rb', line 47

def start_column
  start_loc[1]
end

# File 'lib/rbs/location_aux.rb', line 43

def start_line
  start_loc[0]
end

# File 'lib/rbs/location_aux.rb', line 59

def start_loc
  @start_loc ||= buffer.top_buffer.pos_to_loc(start_pos)
end

# File 'lib/rbs/location_aux.rb', line 31

def start_pos
  buffer.absolute_position(_start_pos) || raise
end

# File 'lib/rbs/location_aux.rb', line 86

def to_json(state = nil)
  {
    start: {
      line: start_line,
      column: start_column
    },
    end: {
      line: end_line,
      column: end_column
    },
    buffer: {
      name: name&.to_s
    }
  }.to_json(state)
end

# File 'lib/rbs/location_aux.rb', line 75

def to_s
  "#{name || "-"}:#{start_line}:#{start_column}...#{end_line}:#{end_column}"
end