codemp-nvim/src/workspace.lua

local native = require('codemp.loader').load()

local utils = require('codemp.utils')
local buffers = require('codemp.buffers')
local async = require('codemp.async')
local state = require('codemp.state')
local window = require('codemp.window')

local user_hl = {}
local user_buffer = {}
local tree_buf = nil
local available_colors = { -- TODO these are definitely not portable!
	"ErrorMsg",
	"WarningMsg",
	"MatchParen",
	"SpecialMode",
	"CmpItemKindFunction",
	"CmpItemKindValue",
	"CmpItemKindInterface",
}

local function register_cursor_callback(controller)
	vim.api.nvim_create_autocmd({"CursorMoved", "CursorMovedI", "ModeChanged"}, {
		group = vim.api.nvim_create_augroup("codemp-workspace-" .. state.workspace, { clear = true }),
		callback = function (_)
			local cur = utils.cursor.position()
			local buf = vim.api.nvim_get_current_buf()
			if buffers.map[buf] ~= nil then
				controller:send(buffers.map[buf], cur[1][1], cur[1][2], cur[2][1], cur[2][2])
			end
		end
	})
end

local function register_cursor_handler(controller)
	async.handler(nil, controller, function(event)
		if user_hl[event.user] == nil then
			user_hl[event.user] = {
				ns = vim.api.nvim_create_namespace("codemp-cursor-" .. event.user),
				hi = available_colors[ math.random( #available_colors ) ],
			}
		end
		user_buffer[event.user] = event.buffer
		local buffer = buffers.map_rev[event.buffer]
		if buffer ~= nil then
			vim.api.nvim_buf_clear_namespace(buffer, user_hl[event.user].ns, 0, -1)
			utils.multiline_highlight(
				buffer,
				user_hl[event.user].ns,
				user_hl[event.user].hi,
				event.start,
				event.finish
			)
		end
	end, 20)
end

local function join(workspace)
	local controller = state.client:join_workspace(workspace)
	register_cursor_callback(controller)
	register_cursor_handler(controller)

	-- TODO nvim docs say that we should stop all threads before exiting nvim
	--  but we like to live dangerously (:
	local refresher = vim.loop.new_async(function () vim.schedule(function() window.update() end) end)
	vim.loop.new_thread({}, function(_id, _ws, _refresher)
		local _codemp = require('codemp.loader').load()
		local _workspace = _codemp.get_client(_id):get_workspace(_ws)
		while true do
			local success, res = pcall(_workspace.event, _workspace)
			if success then
				print("workspace event!")
				_refresher:send()
			else
				print("error waiting for workspace event: " .. res)
				break
			end
		end
	end, state.client.id, state.workspace, refresher)

	window.update()
end

local function leave()
	native.leave_workspace()
	print(" -- left workspace")
end

local function open_buffer_tree()
	local tree = state.client:get_workspace(state.workspace).filetree
	if tree_buf == nil then
		tree_buf = vim.api.nvim_create_buf(false, true)
		vim.api.nvim_buf_set_name(tree_buf, "codemp::" .. state.workspace)
		vim.api.nvim_set_option_value('buftype', 'nofile', { buf = tree_buf })
	end
	vim.api.nvim_set_option_value('modifiable', true, { buf = tree_buf })
	utils.buffer.set_content(tree_buf, "codemp::" .. state.workspace .. "\n\n- " .. vim.fn.join(tree, "\n- "))
	vim.api.nvim_set_option_value('modifiable', false, { buf = tree_buf })
	vim.api.nvim_open_win(tree_buf, true, {
		win = 0,
		split = 'left',
		width = 20,
	})
	vim.api.nvim_set_option_value('relativenumber', false, {})
	vim.api.nvim_set_option_value('number', false, {})
	vim.api.nvim_set_option_value('cursorlineopt', 'line', {})
end

return {
	join = join,
	leave = leave,
	map = user_hl,
	colors = available_colors,
	positions = user_buffer,
	open_buffer_tree = open_buffer_tree,
	buffer_tree = tree_buf,
}