read lines

lua/tabnine/chat/binary.lua

@@ -76,28 +76,26 @@ function ChatBinary:start()
 		end)
 	)
 
-	self.stdout:read_start(vim.schedule_wrap(function(error, chunk)
-		if chunk then
-			for _, line in pairs(utils.str_to_lines(chunk)) do
-				local message = vim.json.decode(line, { luanil = { object = true, array = true } })
-				local handler = self.registry[message.command]
-				if handler then
-					handler(message.data, function(payload)
-						-- if payload then
-						self:post_message({
-							id = message.id,
-							payload = payload,
-						})
-						-- end
-					end)
-				else
-					self:post_message({ id = message.id, error = "not_implemented" })
-				end
+	utils.read_lines_start(
+		self.stdout,
+		vim.schedule_wrap(function(line)
+			local message = vim.json.decode(line, { luanil = { object = true, array = true } })
+			local handler = self.registry[message.command]
+			if handler then
+				handler(message.data, function(payload)
+					self:post_message({
+						id = message.id,
+						payload = payload,
+					})
+				end)
+			else
+				self:post_message({ id = message.id, error = "not_implemented" })
 			end
-		elseif error then
-			print("chat binary read_start error", error)
-		end
-	end))
+		end),
+		vim.schedule_wrap(function(error)
+			print("error reading chat binary", error)
+		end)
+	)
 end
 
 function ChatBinary:new(o)

lua/tabnine/utils.lua

@@ -157,4 +157,21 @@ function M.read_file_into_buffer(file_path)
 	return bufnr
 end
 
+function M.read_lines_start(stream, on_line, on_error)
+	local buffer = ""
+	stream:read_start(function(error, chunk)
+		buffer = buffer .. chunk
+		if error then
+			on_error(error)
+			return
+		end
+		while true do
+			local start_pos = buffer:find("\n")
+			if not start_pos then break end
+			on_line(buffer:sub(1, start_pos - 1))
+			buffer = buffer:sub(start_pos + 1)
+		end
+	end)
+end
+
 return M