Add rudimentry tracking to not return data after the last line changed

Author: alerque

lua/commonmarker/init.lua

@@ -35,10 +35,11 @@ local function get_contents (buffer)
 	return table.concat(lines)
 end
 
-local function highlight (buffer, namespace, firstline)
+local function highlight (buffer, namespace, firstline, lastline)
 	local contents = get_contents(buffer)
 	local firstbyte = call_function("line2byte", { firstline })
-	local events = rust.get_offsets(contents, firstbyte)
+	local lastbyte = call_function("line2byte", { lastline + 1 }) - 1
+	local events = rust.get_offsets(contents, firstbyte, lastbyte)
 	for _, event in ipairs(events) do
 		repeat -- Allow continue in for loop
 			local sline, scol = byte2pos(event.first)
@@ -67,13 +68,13 @@ end
 function commonmarker:attach (buffer)
 	if self._attachments[buffer] then return end
 	self._attachments[buffer] = true
-	highlight(buffer, self._namespace, 0)
+	highlight(buffer, self._namespace, 1, call_function("line", { "$" }))
 	buf_attach(buffer, false, {
-			on_lines = function (_, _, _, firstline, _, _)
-				buf_clear_namespace(buffer, self._namespace, firstline, -1)
+			on_lines = function (_, _, _, firstline, lastline, _)
+				buf_clear_namespace(buffer, self._namespace, firstline, lastline)
 				-- Returning true here detaches, we thought we should have been already
 				if not self._attachments[buffer] then return true end
-				highlight(buffer, self._namespace, firstline)
+				highlight(buffer, self._namespace, firstline, lastline)
 			end,
 			on_detach = function (_)
 				self._attachments[buffer] = nil

src/lib.rs

@@ -27,15 +27,15 @@ struct MdTag {
 
 type Events = Vec<MdTag>;
 
-fn get_offsets(buffer: String, firstbyte: usize) -> Result<Events> {
+fn get_offsets(buffer: String, firstbyte: usize, lastbyte: usize) -> Result<Events> {
     let options = Options::all();
     let parser = Parser::new_ext(buffer.as_str(), options);
     let mut events = Events::new();
     for (event, range) in parser.into_offset_iter() {
         let first = range.start + 1;
         let last = range.end + 1;
         let mut lang = None;
-        if first < firstbyte {
+        if first < firstbyte || first > lastbyte {
             continue;
         }
         let group = match event {

src/lua.rs

@@ -5,9 +5,12 @@ fn to_html(_: &Lua, buffer: String) -> LuaResult<String> {
     Ok(super::to_html(buffer).unwrap())
 }
 
-fn get_offsets(lua: &Lua, (buffer, firstbyte): (String, usize)) -> LuaResult<LuaTable> {
+fn get_offsets(
+    lua: &Lua,
+    (buffer, firstbyte, lastbyte): (String, usize, usize),
+) -> LuaResult<LuaTable> {
     let table = lua.create_table().unwrap();
-    let events = super::get_offsets(buffer, firstbyte).unwrap();
+    let events = super::get_offsets(buffer, firstbyte, lastbyte).unwrap();
     for (i, event) in events.iter().enumerate() {
         let info = lua.create_table().unwrap();
         info.set("group", event.group.as_str()).unwrap();

src/python.rs

@@ -8,8 +8,13 @@ fn to_html(_py: Python, buffer: String) -> PyResult<String> {
 }
 
 #[pyfunction]
-fn get_offsets(_py: Python, buffer: String, firstbyte: usize) -> PyResult<&PyDict> {
-    let events = super::get_offsets(buffer, firstbyte).unwrap();
+fn get_offsets(
+    _py: Python,
+    buffer: String,
+    firstbyte: usize,
+    lastbyte: usize,
+) -> PyResult<&PyDict> {
+    let events = super::get_offsets(buffer, firstbyte, lastbyte).unwrap();
     let pyevents = PyDict::new(_py);
     let mut i: u32 = 1;
     for event in events.iter() {