-- mod-version:2 -- lite-xl 2.0
local DocView = require "core.docview"
local style = require "core.style"

local draw_line_body = DocView.draw_line_body
function DocView:draw_line_body(idx, x, y, ...)
  local lh = self:get_line_height()
  for _, line1, _, line2, _ in self.doc:get_selections(true) do
    if idx >= line1 and idx < line2 and line1 ~= line2 then
      -- draw selection from the end of the line to the end of the available space
      local x1 = x + self:get_col_x_offset(idx, #self.doc.lines[idx])
      local x2 = x + self.scroll.x + self.size.x
      if x2 > x1 then
        renderer.draw_rect(x1, y, x2 - x1, lh, style.selection)
      end
    end
  end
  draw_line_body(self, idx, x, y, ...)
end