138 lines
4.7 KiB
Lua
138 lines
4.7 KiB
Lua
local M = {}
|
|
M.close_key = nil
|
|
M.rust_build = nil
|
|
M.rust_run = nil
|
|
M.new_rust_file = nil
|
|
|
|
M.config = {
|
|
project_root = "~/Projects/Rust",
|
|
keybinds = {
|
|
["<Esc>"] = "CloseTerminal",
|
|
["<leader>pb"] = "RustBuild",
|
|
["<leader>pr"] = "RustRun",
|
|
["<leader>nf"] = "NewRustFile",
|
|
},
|
|
terminal = {
|
|
right_padding = 0,
|
|
bottom_padding = 0,
|
|
left_padding = 0,
|
|
top_padding = 0,
|
|
border = true,
|
|
number = true,
|
|
relativenumber = true,
|
|
scroll = true,
|
|
}
|
|
}
|
|
|
|
function M.setup(opts)
|
|
M.config = vim.tbl_deep_extend("force", M.config, opts or {})
|
|
for key, command in pairs(M.config.keybinds) do
|
|
if command == "CloseTerminal" then M.close_key = key end
|
|
if command == "RustBuild" then M.rust_build = key end
|
|
if command == "RustRun" then M.rust_run = key end
|
|
if command == "NewRustFile" then M.new_rust_file = key end
|
|
end
|
|
|
|
vim.api.nvim_create_user_command("NewRustFile", M.create_new_file, {})
|
|
vim.api.nvim_create_user_command("RustBuild", M.build, {})
|
|
vim.api.nvim_create_user_command("RustRun", M.run, {})
|
|
|
|
vim.keymap.set("n", M.rust_build, ":RustBuild<CR>", { silent = true })
|
|
vim.keymap.set("n", M.rust_run, ":RustRun<CR>", { silent = true })
|
|
vim.keymap.set("n", M.new_rust_file, ":NewRustFile<CR>", { silent = true })
|
|
end
|
|
|
|
function M.open_float_terminal(cmd)
|
|
local ui = vim.api.nvim_list_uis()[1]
|
|
local term_cfg = M.config.terminal
|
|
|
|
local width = math.max(1, math.floor(ui.width * 0.9) - term_cfg.left_padding - term_cfg.right_padding)
|
|
local height = math.max(1, math.floor(ui.height * 0.9) - term_cfg.top_padding - term_cfg.bottom_padding)
|
|
local row = math.max(1, math.floor((ui.height - height) / 2) + term_cfg.top_padding)
|
|
local col = math.max(1, math.floor((ui.width - width) / 2) + term_cfg.left_padding)
|
|
|
|
local buf = vim.api.nvim_create_buf(false, true)
|
|
local win = vim.api.nvim_open_win(buf, true, {
|
|
relative = "editor",
|
|
width = width,
|
|
height = height,
|
|
row = row,
|
|
col = col,
|
|
style = "minimal",
|
|
border = term_cfg.border and "rounded" or "none",
|
|
})
|
|
|
|
vim.api.nvim_win_set_option(win, "winblend", vim.o.pumblend)
|
|
vim.api.nvim_win_set_option(win, "winhighlight", "Normal:Pmenu,FloatBorder:Pmenu")
|
|
vim.api.nvim_win_set_option(win, "cursorline", true)
|
|
if(term_cfg.number == true) then
|
|
vim.api.nvim_win_set_option(win, "number", true)
|
|
if(term_cfg.relativenumber == true) then
|
|
vim.api.nvim_win_set_option(win, "relativenumber", true)
|
|
end
|
|
end
|
|
if(term_cfg.scroll == true) then
|
|
vim.api.nvim_win_set_option(win, "scrolloff", 5)
|
|
end
|
|
|
|
vim.fn.termopen(cmd)
|
|
vim.api.nvim_buf_set_keymap(buf, "n", M.close_key, "i<C-\\><C-n>:q<CR>", { noremap = true, silent = true })
|
|
return buf, win
|
|
end
|
|
|
|
local function write_file(path, content)
|
|
local file = assert(io.open(path, "w"), "Error creating file: " .. path)
|
|
file:write(content)
|
|
file:close()
|
|
vim.cmd("edit " .. path)
|
|
print(path .. " created successfully!")
|
|
end
|
|
|
|
function M.create_project()
|
|
vim.ui.input({ prompt = "Enter project name: " }, function(project_name)
|
|
if not project_name or project_name == "" then return print("Project creation canceled.") end
|
|
local project_root = vim.fn.expand(M.config.project_root) .. "/" .. project_name
|
|
vim.fn.system("cargo new " .. project_root)
|
|
vim.cmd("edit " .. project_root .. "/src/main.rs")
|
|
end)
|
|
end
|
|
|
|
function M.create_new_file()
|
|
vim.ui.input({ prompt = "Enter file name: " }, function(class_name)
|
|
if not class_name or class_name == "" then return print("File creation canceled.") end
|
|
local root_dir = M.get_project_root()
|
|
if not root_dir then return print("Error: Project root not found.") end
|
|
write_file(root_dir .. "/src/" .. class_name .. ".rs", "")
|
|
end)
|
|
end
|
|
|
|
function M.get_project_root()
|
|
local dir = vim.fn.expand("%:p:h")
|
|
while dir ~= "/" do
|
|
if vim.fn.isdirectory(dir .. "/src") == 1 or vim.fn.filereadable(dir .. "/CMakeLists.txt") == 1 then
|
|
return dir
|
|
end
|
|
dir = vim.fn.fnamemodify(dir, ":h")
|
|
end
|
|
end
|
|
|
|
function M.build()
|
|
local root = M.get_project_root()
|
|
if not root then return print("Error: Project root not found.") end
|
|
local cmd = "cargo build"
|
|
local term_buf = M.open_float_terminal(cmd)
|
|
vim.api.nvim_buf_set_option(term_buf, "modifiable", true)
|
|
local output = vim.fn.system(cmd)
|
|
local success = vim.v.shell_error == 0
|
|
vim.api.nvim_buf_set_lines(term_buf, 0, -1, false, vim.list_extend({ success and "Compilation successful!" or "Error during compilation:" }, vim.split(output, "\n", { trimempty = true })))
|
|
vim.api.nvim_buf_set_option(term_buf, "modifiable", false)
|
|
end
|
|
|
|
function M.run()
|
|
local root = M.get_project_root()
|
|
if not root then return print("Error: Project root not found.") end
|
|
M.open_float_terminal("cargo run")
|
|
end
|
|
|
|
return M
|