if vim.g.loaded_quickcomment then
    return
end
vim.g.loaded_quickcomment = true

vim.g.quickcomment_escapestring = function(string_to_escape)
    local escaped_string = ''

    string_to_escape:gsub(".", function(c)
        if c:find('[-?+*%]%[.%%()$^]') == nil then
            escaped_string = escaped_string .. c
        else
            escaped_string = escaped_string .. '%' .. c
        end
    end)

    return escaped_string
end

vim.g.quickcomment_togglecommentlines = function (line_start, line_end)
    local comment_string = vim.api.nvim_buf_get_option(0, 'commentstring')
    if comment_string == nil then
        print('QuickComment: ERROR: Unable to get comment string, "commentstring" is not defined!')
        return
    end
    local sub_find, sub_end = comment_string:find('%%s')
    local escaped_string = ''
    if sub_find ~= nil and sub_find > 1 then
        escaped_string = vim.g.quickcomment_escapestring(comment_string:sub(1, sub_find - 1))
    end
    escaped_string = escaped_string .. '(.*)'
    if sub_end < comment_string:len() then
        escaped_string = escaped_string .. vim.g.quickcomment_escapestring(comment_string:sub(sub_end + 1))
    end

    local lines = vim.api.nvim_buf_get_lines(0, line_start, line_end, false)
    for i, line in ipairs(lines) do
        if line:find(escaped_string) == nil then
            lines[i] = comment_string:format(line)
        else
            local match_cap = line:gmatch(escaped_string)
            local captured = match_cap()
            if captured ~= nil then
                lines[i] = captured
            else
                print('QuickComment: ERROR: Failed to uncomment!')
                return
            end
        end
    end
    vim.api.nvim_buf_set_lines(0, line_start, line_end, false, lines)
end

vim.g.quickcomment_togglecommentline = function (linen)
    if linen == nil then
        linen = vim.api.nvim_win_get_cursor(0)[1]
    end
    vim.g.quickcomment_togglecommentlines(linen - 1, linen)
end