Made `doc:toggle-line-comments` command skip empty lines

This commit is contained in:
rxi 2020-05-08 09:07:13 +01:00
parent bf8565d2a1
commit 1d2a0aada5
1 changed files with 18 additions and 15 deletions

View File

@ -3,7 +3,6 @@ local command = require "core.command"
local common = require "core.common" local common = require "core.common"
local config = require "core.config" local config = require "core.config"
local translate = require "core.doc.translate" local translate = require "core.doc.translate"
local search = require "core.doc.search"
local DocView = require "core.docview" local DocView = require "core.docview"
@ -25,25 +24,29 @@ local function get_indent_string()
end end
local function insert_at_start_of_selected_lines(text) local function insert_at_start_of_selected_lines(text, skip_empty)
local line1, col1, line2, col2, swap = doc():get_selection(true) local line1, col1, line2, col2, swap = doc():get_selection(true)
for line = line1, line2 do for line = line1, line2 do
doc():insert(line, 1, text) local line_text = doc().lines[line]
if (not skip_empty or line_text:find("%S")) then
doc():insert(line, 1, text)
end
end end
doc():set_selection(line1, col1 + #text, line2, col2 + #text, swap) doc():set_selection(line1, col1 + #text, line2, col2 + #text, swap)
end end
local function remove_from_start_of_selected_lines(text) local function remove_from_start_of_selected_lines(text, skip_empty)
local line1, col1, line2, col2, swap = doc():get_selection(true) local line1, col1, line2, col2, swap = doc():get_selection(true)
for line = line1, line2 do for line = line1, line2 do
if doc().lines[line]:sub(1, #text) == text then local line_text = doc().lines[line]
if line_text:sub(1, #text) == text
and (not skip_empty or line_text:find("%S"))
then
doc():remove(line, 1, line, #text + 1) doc():remove(line, 1, line, #text + 1)
if line == line1 then col1 = col1 - #text end
if line == line2 then col2 = col2 - #text end
end end
end end
doc():set_selection(line1, col1, line2, col2, swap) doc():set_selection(line1, col1 - #text, line2, col2 - #text, swap)
end end
@ -218,21 +221,21 @@ local commands = {
end, end,
["doc:toggle-line-comments"] = function() ["doc:toggle-line-comments"] = function()
if not doc().highlighter.syntax.comment then return end local comment = doc().highlighter.syntax.comment
local text = doc().highlighter.syntax.comment .. " " if not comment then return end
local comment_text = comment .. " "
local line1, _, line2 = doc():get_selection(true) local line1, _, line2 = doc():get_selection(true)
local uncomment = true local uncomment = true
for line = line1, line2 do for line = line1, line2 do
local str = doc().lines[line]:match("^[ \t]*(.*)$") local text = doc().lines[line]
if str and str:sub(1, #text) ~= text then if text:find("%S") and text:find(comment_text, 1, true) ~= 1 then
uncomment = false uncomment = false
break
end end
end end
if uncomment then if uncomment then
remove_from_start_of_selected_lines(text) remove_from_start_of_selected_lines(comment_text, true)
else else
insert_at_start_of_selected_lines(text) insert_at_start_of_selected_lines(comment_text, true)
end end
end, end,