codemp-nvim/lua/codemp/workspace.lua

123 lines
3.3 KiB
Lua
Raw Normal View History

local utils = require('codemp.utils')
2024-08-06 18:46:22 +02:00
local buffers = require('codemp.buffers')
2024-08-24 01:56:27 +02:00
local session = require('codemp.session')
local window = require('codemp.window')
local user_hl = {}
local function fetch_workspaces_list()
local new_list = {}
2024-09-01 03:06:27 +02:00
local owned = session.client:list_workspaces(true, false):await()
for _, ws in pairs(owned) do
table.insert(new_list, {
name = ws,
owned = true,
})
end
local invited = session.client:list_workspaces(false, true):await()
for _, ws in pairs(invited) do
table.insert(new_list, {
name = ws,
owned = false,
})
end
session.available = new_list
window.update()
end
2024-08-24 01:56:27 +02:00
---@param ws Workspace
local function register_cursor_callback(ws)
local controller = ws.cursor
vim.api.nvim_create_autocmd({"CursorMoved", "CursorMovedI", "ModeChanged"}, {
2024-08-24 01:56:27 +02:00
group = vim.api.nvim_create_augroup("codemp-workspace-" .. ws.name, { clear = true }),
callback = function (_)
local cur = utils.cursor.position()
2024-08-08 04:34:23 +02:00
local buf = vim.api.nvim_get_current_buf()
if buffers.map[buf] ~= nil then
local _ = controller:send(buffers.map[buf], cur[1][1], cur[1][2], cur[2][1], cur[2][2]) -- no need to await here
end
end
})
end
2024-08-24 01:56:27 +02:00
---@param ws Workspace
local function register_cursor_handler(ws)
local controller = ws.cursor
local async = vim.loop.new_async(vim.schedule_wrap(function ()
while true do
2024-08-17 01:56:09 +02:00
local event = controller:try_recv():await()
if event == nil then break end
if user_hl[event.user] == nil then
user_hl[event.user] = {
ns = vim.api.nvim_create_namespace("codemp-cursor-" .. event.user),
2024-08-17 04:26:34 +02:00
hi = utils.color(event.user),
}
end
local old_buffer = buffers.users[event.user]
if old_buffer ~= nil then
2024-08-17 04:26:49 +02:00
local old_buffer_id = buffers.map_rev[old_buffer]
if old_buffer_id ~= nil then
vim.api.nvim_buf_clear_namespace(old_buffer_id, user_hl[event.user].ns, 0, -1)
end
end
buffers.users[event.user] = event.buffer
2024-08-17 04:26:49 +02:00
local buffer_id = buffers.map_rev[event.buffer]
if buffer_id ~= nil then
utils.multiline_highlight(
2024-08-17 04:26:49 +02:00
buffer_id,
user_hl[event.user].ns,
user_hl[event.user].hi,
event.start,
event.finish
)
end
2024-08-17 04:26:49 +02:00
if old_buffer ~= event.buffer then
window.update() -- redraw user positions
end
end
end))
controller:callback(function (_controller) async:send() end)
end
---@param workspace string workspace name to join
---@return Workspace
---join a workspace and register event handlers
local function join(workspace)
2024-08-24 01:56:27 +02:00
local ws = session.client:join_workspace(workspace):await()
print(" >< joined workspace " .. ws.name)
register_cursor_callback(ws)
register_cursor_handler(ws)
-- TODO this is temporary and ad-hoc
-- ws:callback(function (event)
-- if event.type == "leave" then
-- if buffers.users[event.value] ~= nil then
-- vim.schedule(function ()
-- vim.api.nvim_buf_clear_namespace(buffers.users[event.value], user_hl[event.value].ns, 0, -1)
-- buffers.users[event.value] = nil
-- user_hl[event.value] = nil
-- end)
-- end
-- end
-- vim.schedule(function() window.update() end)
-- end)
2024-08-24 01:56:27 +02:00
session.workspace = ws
window.update()
return ws
end
local function leave()
2024-08-24 01:56:27 +02:00
session.client:leave_workspace(session.workspace.name)
print(" -- left workspace")
2024-08-24 01:56:27 +02:00
session.workspace = nil
end
return {
join = join,
leave = leave,
map = user_hl,
list = fetch_workspaces_list,
}