local cc = require("neo-tree.sources.common.commands") local utils = require("neo-tree.utils") local codemp_utils = require("codemp.utils") local manager = require("neo-tree.sources.manager") local buf_manager = require("codemp.buffers") local ws_manager = require("codemp.workspace") local client_manager = require("codemp.client") local M = {} local function toggle(node) if node:is_expanded() then node:collapse() else node:expand() end manager.refresh("codemp") end M.refresh = require("neo-tree.utils").wrap(manager.refresh, "codemp") M.open = function(state, path, extra) local selected = state.tree:get_node() if selected.type == "spacer" then return end if selected.type == "entry" then return end if selected.type == "root" then return toggle(selected) end if selected.type == "title" then if CODEMP.client ~= nil then print(" :: refreshing workspace list") ws_manager.list() end return end if selected.type == "button" then if selected.name == "[connect]" and CODEMP.client == nil then client_manager.connect() end return end if selected.type == "workspace" then if CODEMP.workspace ~= nil and CODEMP.workspace:id() ~= selected.name then error("must leave current workspace first") end if CODEMP.workspace == nil then ws_manager.join(selected.name) end return end if selected.type == "buffer" then local window = utils.get_appropriate_window(state) if buf_manager.map_rev[selected.name] ~= nil then vim.api.nvim_set_current_win(window) vim.api.nvim_win_set_buf(window, buf_manager.map_rev[selected.name]) else local buf = vim.api.nvim_create_buf(true, false) buf_manager.attach(selected.name, { buffer = buf, window = window }) end return end if selected.type == "user" then if CODEMP.workspace == nil then error("cannot follow while not in a workspace") end local usr = ws_manager.map[selected.name] print(" /\\/ following " .. selected.name) CODEMP.following = selected.name local _ = CODEMP.workspace:cursor():send({ buffer = "", start_row = 0, start_col = 0, end_row = 0, end_col = 0, }) -- clear current cursor if usr ~= nil then local buf_name = buf_manager.users[selected.name] local buf_id = buf_manager.map_rev[buf_name] if buf_id ~= nil then local win = utils.get_appropriate_window(state) CODEMP.ignore_following_action = true vim.api.nvim_set_current_win(win) vim.api.nvim_win_set_buf(win, buf_id) vim.api.nvim_win_set_cursor(win, { usr.pos[1] + 1, usr.pos[2] }) end end manager.refresh("codemp") return end error("unrecognized node type") end M.move = function(state, path, extra) local selected = state.tree:get_node() if selected.type == "buffer" then return vim.ui.input({ prompt = "move content into open buffer?" }, function (input) if not input or not vim.startswith(string.lower(input), "y") then return end local window = utils.get_appropriate_window(state) local buf = vim.api.nvim_win_get_buf(window) buf_manager.attach(selected.name, { buffer = buf, skip_exists_check = true }) end) end error("only buffers can be moved to current file") end M.copy = function(state, path, extra) local selected = state.tree:get_node() if selected.type == "buffer" then return vim.ui.input({ prompt = "copy content to remote buffer?" }, function (input) if not input or not vim.startswith(string.lower(input), "y") then return end local window = utils.get_appropriate_window(state) local buf = vim.api.nvim_win_get_buf(window) local content = codemp_utils.buffer.get_content(buf) buf_manager.attach(selected.name, { buffer = buf, content = content, skip_exists_check = true }) end) end error("current file can only be copied into buffers") end M.delete = function(state, path, extra) local selected = state.tree:get_node() if selected.type == "title" then vim.ui.input({ prompt = "disconnect from server?" }, function(choice) if not choice or not vim.startswith(string.lower(choice), "y") then return end if CODEMP.workspace ~= nil then ws_manager.leave() end CODEMP.client = nil manager.refresh("codemp") collectgarbage("collect") -- to make sure we drop the reference and disconnect end) elseif selected.type == "root" and vim.startswith(selected.name, "#") then vim.ui.input({ prompt = "disconnect from workspace?" }, function (input) if not input or not vim.startswith(string.lower(input), "y") then return end ws_manager.leave() end) elseif selected.type == "buffer" then if CODEMP.workspace == nil then error("join a workspace first") end vim.ui.input({ prompt = "delete buffer '" .. selected.name .. "'?" }, function (input) if not input or not vim.startswith(string.lower(input), "y") then return end if buf_manager.map_rev[selected.name] ~= nil then buf_manager.detach(selected.name) end CODEMP.workspace:delete_buffer(selected.name):and_then(function () print("deleted buffer " .. selected.name) manager.refresh("codemp") end) end) elseif selected.type == "workspace" then if CODEMP.client == nil then error("connect to server first") end vim.ui.input({ prompt = "delete buffer '" .. selected.name .. "'?" }, function (input) if not input or not vim.startswith(string.lower(input), "y") then return end CODEMP.client:delete_workspace(selected.name):and_then(function () print("deleted workspace " .. selected.name) manager.refresh("codemp") end) end) end end M.add = function(state, path, extra) local selected = state.tree:get_node() if selected.type == "root" then if vim.startswith(selected.name, "#") then vim.ui.input({ prompt = "new buffer path" }, function(input) if input == nil or input == "" then return end CODEMP.workspace:create_buffer(input):and_then(function () manager.refresh("codemp") end) end) elseif selected.name == "workspaces" then vim.ui.input({ prompt = "new workspace name" }, function(input) if input == nil or input == "" then return end CODEMP.client:create_workspace(input):and_then(function () manager.refresh("codemp") require('codemp.workspace').list() end) end) end elseif selected.type == "workspace" then vim.ui.input({ prompt = "user name to invite" }, function(input) if input == nil or input == "" then return end CODEMP.client:invite_to_workspace(selected.name, input):and_then(function () print("invited user " .. input .. " to workspace " .. selected.name) end) end) elseif selected.type == "buffer" then if buf_manager.map_rev[selected.name] ~= nil then vim.ui.input({ prompt = "detach from '" .. selected.name .. "'?" }, function (choice) if not choice or not vim.startswith(string.lower(choice), "y") then return end buf_manager.detach(selected.name) end) end end end cc._add_common_commands(M) return M