From 9d0622f5bea51e5a73728caea9521d258a9dfcdd Mon Sep 17 00:00:00 2001
From: DarthMooMancer <107283522+DarthMooMancer@users.noreply.github.com>
Date: Fri, 14 Mar 2025 14:03:30 -0400
Subject: [PATCH] Adding rust

---
 lua/Polydev/rust.lua | 141 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 141 insertions(+)
 create mode 100644 lua/Polydev/rust.lua

diff --git a/lua/Polydev/rust.lua b/lua/Polydev/rust.lua
new file mode 100644
index 0000000..6d68124
--- /dev/null
+++ b/lua/Polydev/rust.lua
@@ -0,0 +1,141 @@
+local M = {}
+M.close_key = nil
+M.c_build = nil
+M.c_run = nil
+M.new_c_file = nil
+M.new_c_header_file = nil
+
+M.config = {
+    project_root = "~/Projects/C",
+    keybinds = {
+	["<Esc>"] = "CloseTerminal",
+	["<leader>pb"] = "CBuild",
+	["<leader>pr"] = "CRun",
+	["<leader>nf"] = "NewCFile",
+	["<leader>nh"] = "NewCHeaderFile",
+    },
+    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 == "CBuild" then M.c_build = key end
+	if command == "CRun" then M.c_run = key end
+	if command == "NewCFile" then M.new_c_file = key end
+	if command == "NewCHeaderFile" then M.new_c_header_file = key end
+    end
+
+    vim.api.nvim_create_user_command("NewCHeaderFile", M.create_new_header_file, {})
+    vim.api.nvim_create_user_command("NewCFile", M.create_new_file, {})
+    vim.api.nvim_create_user_command("CBuild", M.build, {})
+    vim.api.nvim_create_user_command("CRun", M.run, {})
+
+    vim.keymap.set("n", M.c_build, ":CBuild<CR>", { silent = true })
+    vim.keymap.set("n", M.c_run, ":CRun<CR>", { silent = true })
+    vim.keymap.set("n", M.new_c_file, ":NewCFile<CR>", { silent = true })
+    vim.keymap.set("n", M.new_c_header_file, ":NewCHeaderFile<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 + 1))
+    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.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