Update init.lua for automatically loading language based on filetype

This commit is contained in:
DarthMooMancer
2025-03-13 23:44:58 -04:00
parent 293c2df3b4
commit 91e69efe03
3 changed files with 338 additions and 317 deletions
+122 -122
View File
@@ -7,111 +7,111 @@ M.new_c_header_file = nil
M.new_c_project = nil M.new_c_project = nil
M.config = { M.config = {
project_root = "~/Projects/C", project_root = "~/Projects/C",
keybinds = { keybinds = {
["<Esc>"] = "CloseTerminal", ["<Esc>"] = "CloseTerminal",
["<leader>cb"] = "CBuild", ["<leader>cb"] = "CBuild",
["<leader>cr"] = "CRun", ["<leader>cr"] = "CRun",
["<leader>cnf"] = "NewCFile", ["<leader>cnf"] = "NewCFile",
["<leader>cnp"] = "NewCProject", ["<leader>cnp"] = "NewCProject",
["<leader>cnh"] = "NewCHeaderFile", ["<leader>cnh"] = "NewCHeaderFile",
}, },
terminal = { terminal = {
right_padding = 0, right_padding = 0,
bottom_padding = 0, bottom_padding = 0,
left_padding = 0, left_padding = 0,
top_padding = 0, top_padding = 0,
border = true, border = true,
number = true, number = true,
relativenumber = true, relativenumber = true,
scroll = true, scroll = true,
} }
} }
function M.setup(opts) function M.setup(opts)
M.config = vim.tbl_deep_extend("force", M.config, opts or {}) M.config = vim.tbl_deep_extend("force", M.config, opts or {})
for key, command in pairs(M.config.keybinds) do for key, command in pairs(M.config.keybinds) do
if command == "CloseTerminal" then M.close_key = key end if command == "CloseTerminal" then M.close_key = key end
if command == "CBuild" then M.c_build = key end if command == "CBuild" then M.c_build = key end
if command == "CRun" then M.c_run = key end if command == "CRun" then M.c_run = key end
if command == "NewCFile" then M.new_c_file = key end if command == "NewCFile" then M.new_c_file = key end
if command == "NewCProject" then M.new_c_project = key end if command == "NewCProject" then M.new_c_project = key end
if command == "NewCHeaderFile" then M.new_c_header_file = key end if command == "NewCHeaderFile" then M.new_c_header_file = key end
end end
vim.api.nvim_create_user_command("NewCProject", M.create_project, {}) vim.api.nvim_create_user_command("NewCProject", M.create_project, {})
vim.api.nvim_create_user_command("NewCHeaderFile", M.create_new_header_file, {}) 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("NewCFile", M.create_new_file, {})
vim.api.nvim_create_user_command("CBuild", M.build, {}) vim.api.nvim_create_user_command("CBuild", M.build, {})
vim.api.nvim_create_user_command("CRun", M.run, {}) 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_build, ":CBuild<CR>", { silent = true })
vim.keymap.set("n", M.c_run, ":CRun<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_file, ":NewCFile<CR>", { silent = true })
vim.keymap.set("n", M.new_c_project, ":NewCProject<CR>", { silent = true }) vim.keymap.set("n", M.new_c_project, ":NewCProject<CR>", { silent = true })
vim.keymap.set("n", M.new_c_header_file, ":NewCHeaderFile<CR>", { silent = true }) vim.keymap.set("n", M.new_c_header_file, ":NewCHeaderFile<CR>", { silent = true })
end end
function M.open_float_terminal(cmd) function M.open_float_terminal(cmd)
local ui = vim.api.nvim_list_uis()[1] local ui = vim.api.nvim_list_uis()[1]
local term_cfg = M.config.terminal 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 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 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 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 col = math.max(1, math.floor((ui.width - width) / 2) + term_cfg.left_padding)
local buf = vim.api.nvim_create_buf(false, true) local buf = vim.api.nvim_create_buf(false, true)
local win = vim.api.nvim_open_win(buf, true, { local win = vim.api.nvim_open_win(buf, true, {
relative = "editor", relative = "editor",
width = width, width = width,
height = height, height = height,
row = row, row = row,
col = col, col = col,
style = "minimal", style = "minimal",
border = term_cfg.border and "rounded" or "none", 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, "winblend", vim.o.pumblend)
vim.api.nvim_win_set_option(win, "winhighlight", "Normal:Pmenu,FloatBorder:Pmenu") vim.api.nvim_win_set_option(win, "winhighlight", "Normal:Pmenu,FloatBorder:Pmenu")
vim.api.nvim_win_set_option(win, "cursorline", true) vim.api.nvim_win_set_option(win, "cursorline", true)
if(term_cfg.number == true) then if(term_cfg.number == true) then
vim.api.nvim_win_set_option(win, "number", true) vim.api.nvim_win_set_option(win, "number", true)
if(term_cfg.relativenumber == true) then if(term_cfg.relativenumber == true) then
vim.api.nvim_win_set_option(win, "relativenumber", true) 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 end
end
if(term_cfg.scroll == true) then
vim.api.nvim_win_set_option(win, "scrolloff", 5)
end
vim.fn.termopen(cmd) 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 }) vim.api.nvim_buf_set_keymap(buf, "n", M.close_key, "i<C-\\><C-n>:q<CR>", { noremap = true, silent = true })
return buf, win return buf, win
end end
local function write_file(path, content) local function write_file(path, content)
local file = assert(io.open(path, "w"), "Error creating file: " .. path) local file = assert(io.open(path, "w"), "Error creating file: " .. path)
file:write(content) file:write(content)
file:close() file:close()
vim.cmd("edit " .. path) vim.cmd("edit " .. path)
print(path .. " created successfully!") print(path .. " created successfully!")
end end
function M.create_project() function M.create_project()
vim.ui.input({ prompt = "Enter project name: " }, function(project_name) vim.ui.input({ prompt = "Enter project name: " }, function(project_name)
if not project_name or project_name == "" then if not project_name or project_name == "" then
print("Project creation canceled.") print("Project creation canceled.")
return return
end end
local project_root = vim.fn.expand(M.config.project_root) .. "/" .. project_name local project_root = vim.fn.expand(M.config.project_root) .. "/" .. project_name
for _, path in ipairs({ "/src", "/build", "/include" }) do for _, path in ipairs({ "/src", "/build", "/include" }) do
vim.fn.mkdir(project_root .. path, "p") vim.fn.mkdir(project_root .. path, "p")
end end
write_file(project_root .. "/build/" .. project_name .. ".polydev", project_name) write_file(project_root .. "/build/" .. project_name .. ".polydev", project_name)
write_file(project_root .. "/src/main.c", [[ write_file(project_root .. "/src/main.c", [[
#include <stdio.h> #include <stdio.h>
int main() { int main() {
@@ -120,7 +120,7 @@ int main() {
} }
]]) ]])
write_file(project_root .. "/CMakeLists.txt", string.format([[ write_file(project_root .. "/CMakeLists.txt", string.format([[
cmake_minimum_required(VERSION 3.10) cmake_minimum_required(VERSION 3.10)
project(%s C) project(%s C)
set(CMAKE_C_STANDARD 11) set(CMAKE_C_STANDARD 11)
@@ -130,26 +130,26 @@ set(SOURCES src/main.c)
add_executable(%s ${SOURCES}) add_executable(%s ${SOURCES})
]], project_name, project_name)) ]], project_name, project_name))
vim.cmd("edit " .. project_root .. "/src/main.c") vim.cmd("edit " .. project_root .. "/src/main.c")
end) end)
end end
function M.create_new_file() function M.create_new_file()
vim.ui.input({ prompt = "Enter file name: " }, function(class_name) vim.ui.input({ prompt = "Enter file name: " }, function(class_name)
if not class_name or class_name == "" then return print("File creation canceled.") end if not class_name or class_name == "" then return print("File creation canceled.") end
local root_dir = M.get_project_root() local root_dir = M.get_project_root()
if not root_dir then return print("Error: Project root not found.") end if not root_dir then return print("Error: Project root not found.") end
write_file(root_dir .. "/src/" .. class_name .. ".c", "") write_file(root_dir .. "/src/" .. class_name .. ".c", "")
end) end)
end end
function M.create_new_header_file() function M.create_new_header_file()
vim.ui.input({ prompt = "Enter header file name: " }, function(header_name) vim.ui.input({ prompt = "Enter header file name: " }, function(header_name)
if not header_name then return print("Header file creation canceled.") end if not header_name then return print("Header file creation canceled.") end
local root_dir = M.get_project_root() local root_dir = M.get_project_root()
if not root_dir then return print("Error: Project root not found.") end if not root_dir then return print("Error: Project root not found.") end
local guard_macro = header_name:upper():gsub("[^A-Z0-9]", "_") .. "_H" local guard_macro = header_name:upper():gsub("[^A-Z0-9]", "_") .. "_H"
local content = string.format([[ local content = string.format([[
#ifndef %s #ifndef %s
#define %s #define %s
@@ -160,39 +160,39 @@ void example_function();
#endif // %s #endif // %s
]], guard_macro, guard_macro, guard_macro) ]], guard_macro, guard_macro, guard_macro)
write_file(root_dir .. "/include/" .. header_name .. ".h", content) write_file(root_dir .. "/include/" .. header_name .. ".h", content)
end) end)
end end
function M.get_project_root() function M.get_project_root()
local dir = vim.fn.expand("%:p:h") local dir = vim.fn.expand("%:p:h")
while dir ~= "/" do while dir ~= "/" do
if vim.fn.isdirectory(dir .. "/src") == 1 or vim.fn.filereadable(dir .. "/CMakeLists.txt") == 1 then if vim.fn.isdirectory(dir .. "/src") == 1 or vim.fn.filereadable(dir .. "/CMakeLists.txt") == 1 then
return dir return dir
end
dir = vim.fn.fnamemodify(dir, ":h")
end end
dir = vim.fn.fnamemodify(dir, ":h")
end
end end
function M.build() function M.build()
local root = M.get_project_root() local root = M.get_project_root()
if not root then return print("Error: Project root not found.") end if not root then return print("Error: Project root not found.") end
local cmd = "cd " .. root .. "/build && cmake .. && make" local cmd = "cd " .. root .. "/build && cmake .. && make"
local term_buf = M.open_float_terminal(cmd) local term_buf = M.open_float_terminal(cmd)
vim.api.nvim_buf_set_option(term_buf, "modifiable", true) vim.api.nvim_buf_set_option(term_buf, "modifiable", true)
local output = vim.fn.system(cmd) local output = vim.fn.system(cmd)
local success = vim.v.shell_error == 0 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_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) vim.api.nvim_buf_set_option(term_buf, "modifiable", false)
end end
function M.run() function M.run()
local root = M.get_project_root() local root = M.get_project_root()
if not root then return print("Error: Project root not found.") end if not root then return print("Error: Project root not found.") end
local build_dir = root .. "/build" local build_dir = root .. "/build"
local files = vim.fn.glob(build_dir .. "/*.polydev", true, true) local files = vim.fn.glob(build_dir .. "/*.polydev", true, true)
if #files == 0 then return print("Error: No .polydev file found in build directory.") end if #files == 0 then return print("Error: No .polydev file found in build directory.") end
M.open_float_terminal("cd " .. build_dir .. " && ./" .. files[1]:match("([^/]+)%.polydev$")) M.open_float_terminal("cd " .. build_dir .. " && ./" .. files[1]:match("([^/]+)%.polydev$"))
end end
return M return M
+25 -4
View File
@@ -1,8 +1,29 @@
-- init.lua
local M = {} local M = {}
M.java = require("Polydev.java") -- Store loaded language modules
M.c = require("Polydev.c") M.loaded_languages = {}
-- Function to load the appropriate language module
function M.load_language_module()
local filetype = vim.bo.filetype
if M.loaded_languages[filetype] then return end -- Prevent redundant loading
local ok, lang_module = pcall(require, "Polydev." .. filetype)
if ok and lang_module.setup then
lang_module.setup()
M.loaded_languages[filetype] = true
end
end
-- Setup function to initialize the plugin
function M.setup()
-- Auto-load the correct module when opening a file
vim.api.nvim_create_autocmd("FileType", {
pattern = "*",
callback = function()
M.load_language_module()
end,
})
end
return M return M
+191 -191
View File
@@ -7,260 +7,260 @@ M.new_java_project = nil
-- Default config -- Default config
M.config = { M.config = {
project_root = "~/Projects/Java", project_root = "~/Projects/Java",
keybinds = { keybinds = {
["<Esc>"] = "CloseTerminal", ["<Esc>"] = "CloseTerminal",
["<leader>jb"] = "JavaBuild", ["<leader>jb"] = "JavaBuild",
["<leader>jr"] = "JavaRun", ["<leader>jr"] = "JavaRun",
["<leader>jnf"] = "NewJavaFile", ["<leader>jnf"] = "NewJavaFile",
["<leader>jnp"] = "NewJavaProject", ["<leader>jnp"] = "NewJavaProject",
}, },
terminal = { terminal = {
right_padding = 0, right_padding = 0,
bottom_padding = 0, bottom_padding = 0,
left_padding = 0, left_padding = 0,
top_padding = 0, top_padding = 0,
border = true, border = true,
number = true, number = true,
relativenumber = true, relativenumber = true,
scroll = true, scroll = true,
} }
} }
function M.setup(opts) function M.setup(opts)
M.config = vim.tbl_deep_extend("force", M.config, opts or {}) M.config = vim.tbl_deep_extend("force", M.config, opts or {})
-- Apply keybindings manually -- Apply keybindings manually
for key, command in pairs(M.config.keybinds) do for key, command in pairs(M.config.keybinds) do
if command == "CloseTerminal" then if command == "CloseTerminal" then
M.close_key = key M.close_key = key
end
if command == "JavaBuild" then
M.java_build = key
end
if command == "JavaRun" then
M.java_run = key
end
if command == "NewJavaFile" then
M.new_java_file = key
end
if command == "NewJavaProject" then
M.new_java_project = key
end
end end
if command == "JavaBuild" then
M.java_build = key
end
if command == "JavaRun" then
M.java_run = key
end
if command == "NewJavaFile" then
M.new_java_file = key
end
if command == "NewJavaProject" then
M.new_java_project = key
end
end
-- Register Keybinds -- Register Keybinds
vim.keymap.set("n", M.java_build, ":JavaBuild<CR>", { silent = true }) vim.keymap.set("n", M.java_build, ":JavaBuild<CR>", { silent = true })
vim.keymap.set("n", M.java_run, ":JavaRun<CR>", { silent = true }) vim.keymap.set("n", M.java_run, ":JavaRun<CR>", { silent = true })
vim.keymap.set("n", M.new_java_file, ":NewJavaFile<CR>", { silent = true }) vim.keymap.set("n", M.new_java_file, ":NewJavaFile<CR>", { silent = true })
vim.keymap.set("n", M.new_java_project, ":NewJavaProject<CR>", { silent = true }) vim.keymap.set("n", M.new_java_project, ":NewJavaProject<CR>", { silent = true })
-- Register user commands -- Register user commands
vim.api.nvim_create_user_command("NewJavaProject", M.create_project, {}) vim.api.nvim_create_user_command("NewJavaProject", M.create_project, {})
vim.api.nvim_create_user_command("NewJavaFile", M.create_new_file, {}) vim.api.nvim_create_user_command("NewJavaFile", M.create_new_file, {})
vim.api.nvim_create_user_command("JavaBuild", M.build, {}) vim.api.nvim_create_user_command("JavaBuild", M.build, {})
vim.api.nvim_create_user_command("JavaRun", M.run, {}) vim.api.nvim_create_user_command("JavaRun", M.run, {})
vim.api.nvim_create_user_command("Migrate", M.migrate, {}) vim.api.nvim_create_user_command("Migrate", M.migrate, {})
end end
-- Open floating terminal -- Open floating terminal
function M.open_float_terminal(cmd) function M.open_float_terminal(cmd)
local ui = vim.api.nvim_list_uis()[1] local ui = vim.api.nvim_list_uis()[1]
local term_cfg = M.config.terminal local term_cfg = M.config.terminal
-- Default window size (90% of the editor) -- Default window size (90% of the editor)
local default_width = math.floor(ui.width * 0.9) local default_width = math.floor(ui.width * 0.9)
local default_height = math.floor(ui.height * 0.9) local default_height = math.floor(ui.height * 0.9)
-- Default centered position -- Default centered position
local default_row = math.floor((ui.height - default_height) / 2) local default_row = math.floor((ui.height - default_height) / 2)
local default_col = math.floor((ui.width - default_width) / 2) local default_col = math.floor((ui.width - default_width) / 2)
-- Adjust size based on padding -- Adjust size based on padding
local width = default_width - term_cfg.left_padding - term_cfg.right_padding local width = default_width - term_cfg.left_padding - term_cfg.right_padding
local height = default_height - term_cfg.top_padding - (term_cfg.bottom_padding + 1) local height = default_height - term_cfg.top_padding - (term_cfg.bottom_padding + 1)
-- Adjust position based on padding -- Adjust position based on padding
local row = default_row + term_cfg.top_padding local row = default_row + term_cfg.top_padding
local col = default_col + term_cfg.left_padding local col = default_col + term_cfg.left_padding
-- Ensure window size doesn't go negative -- Ensure window size doesn't go negative
width = math.max(1, width) width = math.max(1, width)
height = math.max(1, height) height = math.max(1, height)
-- Ensure position doesn't move off-screen -- Ensure position doesn't move off-screen
row = math.max(0, row) row = math.max(0, row)
col = math.max(0, col) col = math.max(0, col)
-- Create buffer and window -- Create buffer and window
local buf = vim.api.nvim_create_buf(false, true) local buf = vim.api.nvim_create_buf(false, true)
local win = vim.api.nvim_open_win(buf, true, { local win = vim.api.nvim_open_win(buf, true, {
relative = "editor", relative = "editor",
width = width, width = width,
height = height, height = height,
row = row, row = row,
col = col, col = col,
style = "minimal", style = "minimal",
border = term_cfg.border and "rounded" or "none", border = term_cfg.border and "rounded" or "none",
}) })
-- Enable scrolling, relative line numbers, and prevent closing on click -- Enable scrolling, relative line numbers, and prevent closing on click
vim.api.nvim_win_set_option(win, "winblend", vim.o.pumblend) 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, "winhighlight", "Normal:Pmenu,FloatBorder:Pmenu")
if(M.config.terminal.number == true) then if(M.config.terminal.number == true) then
vim.api.nvim_win_set_option(win, "number", true) vim.api.nvim_win_set_option(win, "number", true)
end end
if(M.config.terminal.number == true and M.config.terminal.relativenumber == true) then if(M.config.terminal.number == true and M.config.terminal.relativenumber == true) then
vim.api.nvim_win_set_option(win, "relativenumber", true) vim.api.nvim_win_set_option(win, "relativenumber", true)
end end
if(M.config.terminal.scroll == true) then if(M.config.terminal.scroll == true) then
vim.api.nvim_win_set_option(win, "scrolloff", 5) vim.api.nvim_win_set_option(win, "scrolloff", 5)
end end
vim.api.nvim_win_set_option(win, "cursorline", true) vim.api.nvim_win_set_option(win, "cursorline", true)
vim.fn.termopen(cmd) 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 }) vim.api.nvim_buf_set_keymap(buf, "n", M.close_key, "i<C-\\><C-n>:q<CR>", { noremap = true, silent = true })
return buf, win return buf, win
end end
-- Migrate old folder to new method -- Migrate old folder to new method
function M.migrate() function M.migrate()
vim.ui.input({ prompt = "Enter project name to migrate: "}, function(project_name) vim.ui.input({ prompt = "Enter project name to migrate: "}, function(project_name)
if not project_name or project_name == "" then if not project_name or project_name == "" then
print("Migration failed") print("Migration failed")
return return
end end
local project_root = vim.fn.expand(M.config.project_root) .. "/" .. project_name local project_root = vim.fn.expand(M.config.project_root) .. "/" .. project_name
local legacy_src_dir = project_root .. "/root/src" local legacy_src_dir = project_root .. "/root/src"
local legacy_out_dir = project_root .. "/root/out" local legacy_out_dir = project_root .. "/root/out"
local src_dir = project_root .. "/src" local src_dir = project_root .. "/src"
local build_dir = project_root .. "/build" local build_dir = project_root .. "/build"
vim.fn.mkdir(src_dir, "p") vim.fn.mkdir(src_dir, "p")
vim.fn.mkdir(build_dir, "p") vim.fn.mkdir(build_dir, "p")
local compile_command = "mv " .. legacy_src_dir .. "/* " .. src_dir .. local compile_command = "mv " .. legacy_src_dir .. "/* " .. src_dir ..
" && mv " .. legacy_out_dir .. "/* " .. build_dir .. " && mv " .. legacy_out_dir .. "/* " .. build_dir ..
" && rm -rd " .. project_root .. "/" .. "root/" " && rm -rd " .. project_root .. "/" .. "root/"
local compile_status = vim.fn.system(compile_command) local compile_status = vim.fn.system(compile_command)
if vim.v.shell_error ~= 0 then if vim.v.shell_error ~= 0 then
print("Error during migration:\n" .. compile_status) print("Error during migration:\n" .. compile_status)
else else
print("Migration successful!") print("Migration successful!")
end end
end) end)
end end
-- Create a new Java project -- Create a new Java project
function M.create_project() function M.create_project()
-- Use vim.ui.input for better handling of input -- Use vim.ui.input for better handling of input
vim.ui.input({ prompt = "Enter project name: " }, function(project_name) vim.ui.input({ prompt = "Enter project name: " }, function(project_name)
if not project_name or project_name == "" then if not project_name or project_name == "" then
print("Project creation canceled.") print("Project creation canceled.")
return return
end end
-- Paths for the project -- Paths for the project
local project_root = vim.fn.expand(M.config.project_root) .. "/" .. project_name local project_root = vim.fn.expand(M.config.project_root) .. "/" .. project_name
local src_dir = project_root .. "/src" local src_dir = project_root .. "/src"
local build_dir = project_root .. "/build" local build_dir = project_root .. "/build"
-- Create directories -- Create directories
vim.fn.mkdir(src_dir, "p") vim.fn.mkdir(src_dir, "p")
vim.fn.mkdir(build_dir, "p") vim.fn.mkdir(build_dir, "p")
-- Write the Main.java file -- Write the Main.java file
local main_java_path = src_dir .. "/Main.java" local main_java_path = src_dir .. "/Main.java"
local main_java_content = [[ local main_java_content = [[
public class Main { public class Main {
public static void main(String[] args) { public static void main(String[] args) {
System.out.println("Hello, World!"); System.out.println("Hello, World!");
} }
} }
]] ]]
-- Create the file and write content -- Create the file and write content
local file = io.open(main_java_path, "w") local file = io.open(main_java_path, "w")
if file then if file then
file:write(main_java_content) file:write(main_java_content)
file:close() file:close()
vim.cmd("edit " .. main_java_path) vim.cmd("edit " .. main_java_path)
print(" Project '" .. project_name .. "' created at " .. project_root) print(" Project '" .. project_name .. "' created at " .. project_root)
else else
print("Error creating Main.java") print("Error creating Main.java")
end end
end) end)
end end
-- Create a new Java file -- Create a new Java file
function M.create_new_file() function M.create_new_file()
vim.ui.input({ prompt = "Enter class name: " }, function(class_name) vim.ui.input({ prompt = "Enter class name: " }, function(class_name)
if not class_name or class_name == "" then if not class_name or class_name == "" then
print("Class creation canceled.") print("Class creation canceled.")
return return
end end
local current_dir = vim.fn.expand("%:p:h") local current_dir = vim.fn.expand("%:p:h")
local root_dir = current_dir:match("(.*)/src") local root_dir = current_dir:match("(.*)/src")
if not root_dir then if not root_dir then
print("Error: src directory not found.") print("Error: src directory not found.")
return return
end end
local java_file_path = root_dir .. "/src/" .. class_name .. ".java" local java_file_path = root_dir .. "/src/" .. class_name .. ".java"
local java_content = string.format([[ local java_content = string.format([[
public class %s { public class %s {
// New File // New File
} }
]], class_name) ]], class_name)
local file = io.open(java_file_path, "w") local file = io.open(java_file_path, "w")
if file then if file then
file:write(java_content) file:write(java_content)
file:close() file:close()
vim.cmd("edit " .. java_file_path) vim.cmd("edit " .. java_file_path)
print( class_name .. ".java created successfully!") print( class_name .. ".java created successfully!")
else else
print("Error creating " .. class_name .. ".java") print("Error creating " .. class_name .. ".java")
end end
end) end)
end end
-- Compile Java files -- Compile Java files
function M.build() function M.build()
local current_dir = vim.fn.expand("%:p:h") local current_dir = vim.fn.expand("%:p:h")
local project_root = current_dir:match("(.*)/src") local project_root = current_dir:match("(.*)/src")
if not project_root then if not project_root then
print("Error: Must be inside the 'src' directory.") print("Error: Must be inside the 'src' directory.")
return return
end end
local src_dir = project_root .. "/src" local src_dir = project_root .. "/src"
local out_dir = project_root .. "/build" local out_dir = project_root .. "/build"
vim.fn.mkdir(out_dir, "p") vim.fn.mkdir(out_dir, "p")
local compile_command = string.format("javac -d %s %s/*.java", out_dir, src_dir) local compile_command = string.format("javac -d %s %s/*.java", out_dir, src_dir)
local compile_status = vim.fn.system(compile_command) local compile_status = vim.fn.system(compile_command)
if vim.v.shell_error ~= 0 then if vim.v.shell_error ~= 0 then
print("Error during compilation:\n" .. compile_status) print("Error during compilation:\n" .. compile_status)
else else
print("Compilation successful!") print("Compilation successful!")
end end
end end
-- Run Java program in floating terminal -- Run Java program in floating terminal
function M.run() function M.run()
local current_dir = vim.fn.expand("%:p:h") local current_dir = vim.fn.expand("%:p:h")
local project_root = current_dir:match("(.*)/src") local project_root = current_dir:match("(.*)/src")
if not project_root then if not project_root then
print("Error: Must be inside the 'src' directory.") print("Error: Must be inside the 'src' directory.")
return return
end end
local out_dir = project_root .. "/build" local out_dir = project_root .. "/build"
M.open_float_terminal("java -cp " .. out_dir .. " Main") M.open_float_terminal("java -cp " .. out_dir .. " Main")
end end
return M return M