mirror of
https://github.com/hexedtech/codemp-nvim.git
synced 2024-11-22 15:34:53 +01:00
114 lines
3.4 KiB
Lua
114 lines
3.4 KiB
Lua
local cc = require("neo-tree.sources.common.commands")
|
|
local utils = require("neo-tree.utils")
|
|
local manager = require("neo-tree.sources.manager")
|
|
local session = require("codemp.session")
|
|
local buf_manager = require("codemp.buffers")
|
|
local ws_manager = require("codemp.workspace")
|
|
local client_manager = require("codemp.client")
|
|
|
|
local M = {}
|
|
|
|
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 == "title" then return end
|
|
if selected.type == "root" then
|
|
if selected.name == "[connect]" and session.client == nil then
|
|
client_manager.connect()
|
|
end
|
|
return
|
|
end
|
|
if selected.type == "workspace" then
|
|
if selected:is_expanded() then
|
|
vim.ui.input({ prompt = "disconnect from workspace?" }, function (input)
|
|
if input == nil then return end
|
|
if input ~= "y" then return end
|
|
ws_manager.leave()
|
|
selected:collapse()
|
|
manager.refresh("codemp")
|
|
end)
|
|
else
|
|
if session.workspace ~= nil and session.workspace.name ~= selected.name then
|
|
error("must leave current workspace first")
|
|
end
|
|
if session.workspace == nil then
|
|
ws_manager.join(selected.name)
|
|
end
|
|
selected:expand()
|
|
manager.refresh("codemp")
|
|
end
|
|
return
|
|
end
|
|
if selected.type == "buffer" then
|
|
local window = utils.get_appropriate_window(state)
|
|
vim.api.nvim_set_current_win(window)
|
|
if buf_manager.map_rev[selected.name] ~= nil then
|
|
vim.api.nvim_win_set_buf(window, buf_manager.map_rev[selected.name])
|
|
return
|
|
end
|
|
local buf = vim.api.nvim_create_buf(true, false)
|
|
vim.api.nvim_win_set_buf(window, buf)
|
|
buf_manager.attach(selected.name, buf)
|
|
return
|
|
end
|
|
if selected.type == "user" then
|
|
print("another remote user")
|
|
return
|
|
end
|
|
error("unrecognized node type")
|
|
end
|
|
|
|
M.move = function(state, path, extra)
|
|
local selected = state.tree:get_node()
|
|
if selected.type == "buffer" then
|
|
local window = utils.get_appropriate_window(state)
|
|
local buf = vim.api.nvim_win_get_buf(window)
|
|
buf_manager.attach(selected.name, buf)
|
|
return
|
|
end
|
|
error("only buffers can be moved to current file")
|
|
end
|
|
|
|
M.delete = function(state, path, extra)
|
|
local selected = state.tree:get_node()
|
|
if selected.type == "buffer" then
|
|
if session.workspace == nil then error("join a workspace first") end
|
|
session.workspace:delete_buffer(selected.name):await()
|
|
print("deleted buffer " .. selected.name)
|
|
manager.refresh("codemp")
|
|
elseif selected.type == "workspace" then
|
|
if session.client == nil then error("connect to server first") end
|
|
session.client:delete_workspace(selected.name):await()
|
|
print("deleted workspace " .. selected.name)
|
|
manager.refresh("codemp")
|
|
else
|
|
print("/!\\ can only delete buffers and workspaces")
|
|
end
|
|
end
|
|
|
|
M.add = function(_state)
|
|
if session.client == nil then
|
|
vim.ui.input({ prompt = "server address" }, function(input)
|
|
if input == nil or input == "" then return end
|
|
client_manager.connect(input)
|
|
end)
|
|
elseif session.workspace == nil then
|
|
vim.ui.input({ prompt = "workspace name" }, function(input)
|
|
if input == nil or input == "" then return end
|
|
session.client:create_workspace(input):await()
|
|
manager.refresh("codemp")
|
|
end)
|
|
else
|
|
vim.ui.input({ prompt = "buffer path" }, function(input)
|
|
if input == nil or input == "" then return end
|
|
session.workspace:create_buffer(input):await()
|
|
manager.refresh("codemp")
|
|
end)
|
|
end
|
|
manager.refresh("codemp")
|
|
end
|
|
|
|
cc._add_common_commands(M)
|
|
return M
|