2024-08-06 01:11:09 +02:00
|
|
|
local utils = require('codemp.utils')
|
|
|
|
local async = require('codemp.async')
|
2024-08-08 04:34:23 +02:00
|
|
|
local state = require('codemp.state')
|
2024-08-06 01:11:09 +02:00
|
|
|
|
|
|
|
local id_buffer_map = {}
|
|
|
|
local buffer_id_map = {}
|
|
|
|
local ticks = {}
|
|
|
|
|
2024-08-08 04:34:23 +02:00
|
|
|
local function create(name, content)
|
|
|
|
state.client:get_workspace(state.workspace):create_buffer(name, content)
|
|
|
|
print(" ++ created buffer '" .. name .. "' on " .. state.workspace)
|
2024-08-06 01:11:09 +02:00
|
|
|
end
|
|
|
|
|
2024-08-08 04:48:26 +02:00
|
|
|
local function delete(name)
|
|
|
|
state.client:get_workspace(state.workspace):delete_buffer(name)
|
|
|
|
print(" -- deleted buffer " .. name)
|
|
|
|
end
|
|
|
|
|
2024-08-08 04:34:23 +02:00
|
|
|
local function attach(name, force)
|
2024-08-06 01:11:09 +02:00
|
|
|
local buffer = nil
|
|
|
|
if force then
|
|
|
|
buffer = vim.api.nvim_get_current_buf()
|
|
|
|
utils.buffer.set_content(buffer, "")
|
|
|
|
else
|
|
|
|
buffer = vim.api.nvim_create_buf(true, true)
|
2024-08-06 23:03:57 +02:00
|
|
|
vim.api.nvim_set_option_value('fileformat', 'unix', { buf = buffer })
|
2024-08-06 01:11:09 +02:00
|
|
|
-- vim.api.nvim_buf_set_option(buffer, 'filetype', 'codemp') -- TODO get from codemp?
|
|
|
|
vim.api.nvim_buf_set_name(buffer, "codemp::" .. name)
|
|
|
|
vim.api.nvim_set_current_buf(buffer)
|
|
|
|
end
|
2024-08-08 04:48:26 +02:00
|
|
|
local controller = state.client:get_workspace(state.workspace):attach(name)
|
2024-08-06 01:11:09 +02:00
|
|
|
|
|
|
|
-- TODO map name to uuid
|
|
|
|
|
|
|
|
id_buffer_map[buffer] = name
|
|
|
|
buffer_id_map[name] = buffer
|
|
|
|
ticks[buffer] = 0
|
|
|
|
|
|
|
|
-- hook serverbound callbacks
|
|
|
|
-- TODO breaks when deleting whole lines at buffer end
|
|
|
|
vim.api.nvim_buf_attach(buffer, false, {
|
|
|
|
on_bytes = function(_, buf, tick, start_row, start_col, start_offset, old_end_row, old_end_col, old_end_byte_len, new_end_row, new_end_col, new_byte_len)
|
|
|
|
if tick <= ticks[buf] then return end
|
|
|
|
if id_buffer_map[buf] == nil then return true end -- unregister callback handler
|
2024-08-14 17:38:28 +02:00
|
|
|
local content = table.concat(
|
|
|
|
vim.api.nvim_buf_get_text(buf, start_row, start_col, start_row + new_end_row, start_col + new_end_col, {}),
|
|
|
|
'\n'
|
|
|
|
)
|
2024-08-06 01:11:09 +02:00
|
|
|
-- print(string.format("%s %s %s %s -- '%s'", start_row, start_col, start_row + new_end_row, start_col + new_end_col, content))
|
2024-08-14 17:38:28 +02:00
|
|
|
controller:send(start_offset, start_offset + old_end_byte_len, content)
|
|
|
|
-- controller:send_diff(utils.buffer.get_content(buf))
|
2024-08-06 01:11:09 +02:00
|
|
|
end,
|
|
|
|
})
|
|
|
|
|
|
|
|
-- This is an ugly as hell fix: basically we receive all operations real fast at the start
|
|
|
|
-- so the buffer changes rapidly and it messes up tracking our delta/diff state and we
|
|
|
|
-- get borked translated TextChanges (the underlying CRDT is fine)
|
|
|
|
-- basically delay a bit so that it has time to sync and we can then get "normal slow" changes
|
|
|
|
-- vim.loop.sleep(200) -- moved inside poller thread to at least not block ui
|
|
|
|
|
|
|
|
-- hook clientbound callbacks
|
2024-08-08 04:34:23 +02:00
|
|
|
async.handler(name, controller, function(event)
|
2024-08-06 01:11:09 +02:00
|
|
|
ticks[buffer] = vim.api.nvim_buf_get_changedtick(buffer)
|
2024-08-14 17:38:28 +02:00
|
|
|
print(" ~~ applying change ~~ " .. event.first .. ".." .. event.last .. "::" .. event.content)
|
|
|
|
utils.buffer.set_content(buffer, event.content, event.first, event.last)
|
|
|
|
-- local before = utils.buffer.get_content(buffer)
|
|
|
|
-- local after = event:apply(before)
|
|
|
|
-- utils.buffer.set_content(buffer, after)
|
2024-08-06 01:11:09 +02:00
|
|
|
-- buffer_replace_content(buffer, event.first, event.last, event.content)
|
|
|
|
end, 20) -- wait 20ms before polling again because it overwhelms libuv?
|
|
|
|
|
|
|
|
print(" ++ attached to buffer " .. name)
|
|
|
|
end
|
|
|
|
|
2024-08-08 04:34:23 +02:00
|
|
|
local function detach(name)
|
2024-08-06 01:11:09 +02:00
|
|
|
local buffer = buffer_id_map[name]
|
|
|
|
id_buffer_map[buffer] = nil
|
|
|
|
buffer_id_map[name] = nil
|
2024-08-08 04:48:26 +02:00
|
|
|
state.client:get_workspace(state.workspace):detach(name)
|
2024-08-06 01:11:09 +02:00
|
|
|
vim.api.nvim_buf_delete(buffer, {})
|
|
|
|
|
|
|
|
print(" -- detached from buffer " .. name)
|
|
|
|
end
|
|
|
|
|
2024-08-08 04:34:23 +02:00
|
|
|
local function sync()
|
2024-08-06 01:11:09 +02:00
|
|
|
local buffer = vim.api.nvim_get_current_buf()
|
|
|
|
local name = id_buffer_map[buffer]
|
|
|
|
if name ~= nil then
|
2024-08-08 04:34:23 +02:00
|
|
|
local controller = state.client:get_workspace(state.workspace):get_buffer(name)
|
2024-08-06 01:11:09 +02:00
|
|
|
ticks[buffer] = vim.api.nvim_buf_get_changedtick(buffer)
|
2024-08-14 17:38:28 +02:00
|
|
|
utils.buffer.set_content(buffer, controller:content())
|
2024-08-06 01:11:09 +02:00
|
|
|
print(" :: synched buffer " .. name)
|
|
|
|
else
|
|
|
|
print(" !! buffer not managed")
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
|
|
return {
|
|
|
|
create = create,
|
2024-08-08 04:48:26 +02:00
|
|
|
delete = delete,
|
2024-08-06 01:11:09 +02:00
|
|
|
sync = sync,
|
|
|
|
attach = attach,
|
|
|
|
detach = detach,
|
|
|
|
map = id_buffer_map,
|
|
|
|
map_rev = buffer_id_map,
|
|
|
|
ticks = ticks,
|
|
|
|
}
|