286 lines
10 KiB
Lua
286 lines
10 KiB
Lua
local utils = require("Polydev.utils")
|
|
local configs = require("Polydev.configs")
|
|
local M = {}
|
|
|
|
local loaded_languages = {}
|
|
|
|
---@param lang string
|
|
---@param opts? table
|
|
---@return boolean
|
|
function M.load_language_module(lang, opts)
|
|
if loaded_languages[lang] then
|
|
return true
|
|
end
|
|
|
|
local ok, mod = pcall(require, "Polydev.languages")
|
|
if not ok or type(mod.languages) ~= "table" then
|
|
return false
|
|
end
|
|
|
|
local lang_mod = mod.languages[lang]
|
|
if type(lang_mod) ~= "table" or type(lang_mod.setup) ~= "function" then
|
|
return false
|
|
end
|
|
|
|
lang_mod:setup(opts and opts[lang] or {})
|
|
loaded_languages[lang] = lang_mod
|
|
return true
|
|
end
|
|
|
|
M.languages = {
|
|
javascript = {
|
|
setup = function(self, user_opts)
|
|
local opts =
|
|
vim.tbl_deep_extend("force", {}, configs.get("javascript"), user_opts or {})
|
|
vim.api.nvim_create_user_command("JavascriptRun", function()
|
|
self:run()
|
|
end, {})
|
|
for key, _ in pairs(opts.keybinds) do
|
|
vim.keymap.set("n", key, ":JavascriptRun<CR>", { silent = true })
|
|
end
|
|
end,
|
|
run = function()
|
|
local root = utils.get_project_root()
|
|
|
|
local cmd = string.format("node " .. root .. "/main.js")
|
|
local output = vim.fn.system(cmd)
|
|
|
|
if not vim.v.shell_error == 0 then
|
|
local term_buf = utils.terminal({ cmd })
|
|
vim.api.nvim_set_option_value("modifiable", true, { buf = term_buf })
|
|
vim.api.nvim_buf_set_lines(
|
|
term_buf,
|
|
0,
|
|
-1,
|
|
false,
|
|
vim.list_extend(
|
|
{ "Error during compilation:" },
|
|
vim.split(output, "\n", { trimempty = true })
|
|
)
|
|
)
|
|
vim.api.nvim_set_option_value("modifiable", false, { buf = term_buf })
|
|
else
|
|
utils.terminal({ "node " .. root .. "/main.js" })
|
|
end
|
|
end,
|
|
},
|
|
java = {
|
|
setup = function(self, user_opts)
|
|
local opts = vim.tbl_deep_extend("force", {}, configs.get("java"), user_opts or {})
|
|
vim.api.nvim_create_user_command("JavaRun", function()
|
|
self:run()
|
|
end, {})
|
|
for key, _ in pairs(opts.keybinds) do
|
|
vim.keymap.set("n", key, ":JavaRun<CR>", { silent = true })
|
|
end
|
|
end,
|
|
run = function()
|
|
local root = utils.get_project_root()
|
|
|
|
local cmd = string.format("javac -d %s %s", root .. "/build", root .. "/src/*.java")
|
|
local output = vim.fn.system(cmd)
|
|
|
|
if not vim.v.shell_error == 0 then
|
|
local term_buf = utils.terminal({ cmd })
|
|
vim.api.nvim_set_option_value("modifiable", true, { buf = term_buf })
|
|
vim.api.nvim_buf_set_lines(
|
|
term_buf,
|
|
0,
|
|
-1,
|
|
false,
|
|
vim.list_extend(
|
|
{ "Error during compilation:" },
|
|
vim.split(output, "\n", { trimempty = true })
|
|
)
|
|
)
|
|
vim.api.nvim_set_option_value("modifiable", false, { buf = term_buf })
|
|
else
|
|
utils.terminal({ "java -cp " .. root, "build" .. " Main" })
|
|
end
|
|
end,
|
|
},
|
|
c = {
|
|
setup = function(self, user_opts)
|
|
local opts = vim.tbl_deep_extend("force", {}, configs.get("c"), user_opts or {})
|
|
vim.api.nvim_create_user_command("CRun", function()
|
|
self:run()
|
|
end, {})
|
|
for key, _ in pairs(opts.keybinds) do
|
|
vim.keymap.set("n", key, ":CRun<CR>", { silent = true })
|
|
end
|
|
end,
|
|
run = function()
|
|
local root = utils.get_project_root()
|
|
local build_dir = root .. "/build"
|
|
|
|
local cmd = "cd '" .. build_dir .. "' && cmake --build ."
|
|
local output = vim.fn.system(cmd)
|
|
local success = vim.v.shell_error == 0
|
|
|
|
if not success then
|
|
local term_buf = utils.terminal({ cmd })
|
|
vim.api.nvim_set_option_value("modifiable", true, { buf = term_buf })
|
|
vim.api.nvim_buf_set_lines(
|
|
term_buf,
|
|
0,
|
|
-1,
|
|
false,
|
|
vim.list_extend(
|
|
{ "Error during compilation:" },
|
|
vim.split(output, "\n", { trimempty = true })
|
|
)
|
|
)
|
|
vim.api.nvim_set_option_value("modifiable", false, { buf = term_buf })
|
|
end
|
|
if success then
|
|
local run_cmd = "cd '" .. build_dir .. "' && ./" .. utils.get_project_name()
|
|
utils.terminal({ run_cmd })
|
|
end
|
|
end,
|
|
},
|
|
cpp = {
|
|
setup = function(self, user_opts)
|
|
local opts = vim.tbl_deep_extend("force", {}, configs.get("cpp"), user_opts or {})
|
|
vim.api.nvim_create_user_command("CppRun", function()
|
|
self:run()
|
|
end, {})
|
|
for key, _ in pairs(opts.keybinds) do
|
|
vim.keymap.set("n", key, ":CppRun<CR>", { silent = true })
|
|
end
|
|
end,
|
|
run = function()
|
|
local root = utils.get_project_root()
|
|
local build_dir = root .. "/build"
|
|
|
|
local cmd = "cd '" .. build_dir .. "' && cmake --build ."
|
|
local output = vim.fn.system(cmd)
|
|
local success = vim.v.shell_error == 0
|
|
|
|
if not success then
|
|
local term_buf = utils.terminal({ cmd })
|
|
|
|
vim.api.nvim_set_option_value("modifiable", true, { buf = term_buf })
|
|
vim.api.nvim_buf_set_lines(
|
|
term_buf,
|
|
0,
|
|
-1,
|
|
false,
|
|
vim.list_extend(
|
|
{ "Error during compilation:" },
|
|
vim.split(output, "\n", { trimempty = true })
|
|
)
|
|
)
|
|
vim.api.nvim_set_option_value("modifiable", false, { buf = term_buf })
|
|
end
|
|
if success then
|
|
local run_cmd = "cd '" .. build_dir .. "' && ./" .. utils.get_project_name()
|
|
utils.terminal({ run_cmd })
|
|
end
|
|
end,
|
|
},
|
|
html = {
|
|
setup = function(user_opts)
|
|
vim.tbl_deep_extend("force", {}, configs.get("html"), user_opts or {})
|
|
end,
|
|
},
|
|
rust = {
|
|
setup = function(self, user_opts)
|
|
local opts = vim.tbl_deep_extend("force", {}, configs.get("rust"), user_opts or {})
|
|
vim.api.nvim_create_user_command("RustRun", function()
|
|
self:run()
|
|
end, {})
|
|
for key, _ in pairs(opts.keybinds) do
|
|
vim.keymap.set("n", key, ":RustRun<CR>", { silent = true })
|
|
end
|
|
end,
|
|
run = function()
|
|
local root = utils.get_project_root()
|
|
if not root then
|
|
return print("Error: Project root not found.")
|
|
end
|
|
|
|
local cmd = { "cargo build" }
|
|
local output = vim.fn.system(table.concat(cmd))
|
|
|
|
if not vim.v.shell_error == 0 then
|
|
local term_buf = utils.terminal(cmd)
|
|
vim.api.nvim_set_option_value("modifiable", true, { buf = term_buf })
|
|
vim.api.nvim_buf_set_lines(
|
|
term_buf,
|
|
0,
|
|
-1,
|
|
false,
|
|
vim.list_extend(
|
|
{ "Error during compilation:" },
|
|
vim.split(output, "\n", { trimempty = true })
|
|
)
|
|
)
|
|
vim.api.nvim_set_option_value("modifiable", false, { buf = term_buf })
|
|
else
|
|
utils.terminal({ "cargo run" })
|
|
end
|
|
end,
|
|
},
|
|
python = {
|
|
setup = function(self, user_opts)
|
|
local opts = vim.tbl_deep_extend("force", {}, configs.get("python"), user_opts or {})
|
|
vim.api.nvim_create_user_command("PythonPip", function()
|
|
self:install_dependency()
|
|
end, {})
|
|
vim.api.nvim_create_user_command("PythonRun", function()
|
|
self:run()
|
|
end, {})
|
|
for key, command in pairs(opts.keybinds) do
|
|
if command == "PythonRun" then
|
|
vim.keymap.set("n", key, ":PythonRun<CR>", { silent = true })
|
|
end
|
|
if command == "PythonPip" then
|
|
vim.keymap.set("n", key, ":PythonPip<CR>", { silent = true })
|
|
end
|
|
end
|
|
|
|
local root = utils.get_project_root()
|
|
if root then
|
|
local venv_path = root .. "/venv/bin/activate"
|
|
if vim.fn.filereadable(venv_path) == 1 then
|
|
vim.env.VIRTUAL_ENV = venv_path
|
|
vim.env.PATH = root .. "/venv/bin:" .. vim.env.PATH
|
|
end
|
|
end
|
|
end,
|
|
install_dependency = function()
|
|
vim.ui.input({ prompt = "Enter pip module: " }, function(pip_module)
|
|
if not pip_module or pip_module == "" then
|
|
return print("Pip canceled")
|
|
end
|
|
utils.terminal({ "pip install " .. pip_module })
|
|
end)
|
|
end,
|
|
run = function()
|
|
utils.terminal({ "python3 " .. utils.get_project_root(), "main.py" })
|
|
end,
|
|
},
|
|
lua = {
|
|
setup = function(self, user_opts)
|
|
local opts = vim.tbl_deep_extend("force", {}, configs.get("lua"), user_opts or {})
|
|
vim.api.nvim_create_user_command("LuaRun", function()
|
|
self:run()
|
|
end, {})
|
|
for key, _ in pairs(opts.keybinds) do
|
|
vim.keymap.set("n", key, ":LuaRun<CR>", { silent = true })
|
|
end
|
|
end,
|
|
run = function()
|
|
local init_lua_path = utils.get_project_root()
|
|
.. "/lua/"
|
|
.. utils.get_project_name()
|
|
.. "/init.lua"
|
|
if vim.fn.filereadable(init_lua_path) == 1 then
|
|
utils.terminal({ "lua " .. init_lua_path })
|
|
end
|
|
end,
|
|
},
|
|
}
|
|
|
|
return M
|