diff --git a/Polydev.polydev b/Polydev.polydev new file mode 100644 index 0000000..e69de29 diff --git a/lua/Polydev/init.lua b/lua/Polydev/init.lua index 74ec558..4b912a0 100644 --- a/lua/Polydev/init.lua +++ b/lua/Polydev/init.lua @@ -4,6 +4,7 @@ M.loaded_languages = {} M.java = require("Polydev.java") M.c = require("Polydev.c") M.lua = require("Polydev.lua") +M.python = require("Polydev.python") function M.load_language_module(lang) if M.loaded_languages[lang] then return true end diff --git a/lua/Polydev/lua.lua b/lua/Polydev/lua.lua index 8c78969..25ca670 100644 --- a/lua/Polydev/lua.lua +++ b/lua/Polydev/lua.lua @@ -4,7 +4,7 @@ M.lua_run = nil M.new_lua_file = nil M.config = { - project_root = "~/Projects/Lua", -- Modify this path as needed + project_root = "~/Projects/Lua", keybinds = { [""] = "CloseTerminal", ["pr"] = "LuaRun", @@ -39,7 +39,6 @@ function M.get_project_name() return polydev_file:match("([^/]+)%.polydev$") end --- Function to setup the configuration function M.setup(opts) M.config = vim.tbl_deep_extend("force", M.config, opts or {}) for key, command in pairs(M.config.keybinds) do @@ -48,15 +47,12 @@ function M.setup(opts) if command == "NewLuaFile" then M.new_lua_file = key end end - -- Define user commands and key mappings vim.api.nvim_create_user_command("NewLuaFile", M.create_new_file, {}) vim.api.nvim_create_user_command("LuaRun", M.run, {}) - vim.keymap.set("n", M.lua_run, ":LuaRun", { silent = true }) vim.keymap.set("n", M.new_lua_file, ":NewLuaFile", { silent = true }) end --- Function to open a floating terminal with command function M.open_float_terminal(cmd) local ui = vim.api.nvim_list_uis()[1] local term_cfg = M.config.terminal diff --git a/lua/Polydev/python.lua b/lua/Polydev/python.lua new file mode 100644 index 0000000..58e5795 --- /dev/null +++ b/lua/Polydev/python.lua @@ -0,0 +1,196 @@ +local M = {} +M.close_key = nil +M.c_run = nil +M.new_c_file = nil +M.new_c_header_file = nil + +M.config = { + project_root = "~/Projects/Python", + keybinds = { + [""] = "CloseTerminal", + ["pr"] = "CRun", + ["nf"] = "NewCFile", + ["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 == "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.fn.system("source ./venv/bin/activate") + + 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("CRun", M.run, {}) + + vim.keymap.set("n", M.c_run, ":CRun", { silent = true }) + vim.keymap.set("n", M.new_c_file, ":NewCFile", { silent = true }) + vim.keymap.set("n", M.new_c_header_file, ":NewCHeaderFile", { 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:q", { 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 + print("Project creation canceled.") + return + end + + local project_root = vim.fn.expand(M.config.project_root) .. "/" .. project_name + for _, path in ipairs({ "/" .. project_name, "/tests" }) do + vim.fn.mkdir(project_root .. path, "p") + end + + write_file(project_root .. "/" .. project_name .. "/__.init__.py", "") + write_file(project_root .. project_name .. ".polydev", project_name) + write_file(project_root .. "/main.py", [[ +def main(): + print("Hello, World!") + +if __name__ == "__main__": + main() +]]) + + vim.fn.system("python -m venv " .. project_root .. "/venv && source " .. project_root .. "/venv/bin/activate") + + write_file(project_root .. "/requirements.txt", "") + write_file(project_root .. "/setup.py", string.format([[ +from setuptools import setup, find_packages + +setup( + name="%s", # Name of the project + version="0.1.0", # Version of the project + packages=find_packages(), # Automatically find packages + install_requires=[ # List of dependencies + "requests>=2.25.1", # Example dependency + "numpy>=1.20.0", # Another example dependency + ], + author="Your Name", # Author's name + author_email="your.email@example.com", # Author's email + description="A simple Python project", # Short description + long_description=open('README.md').read(), # Long description from README + long_description_content_type="text/markdown", # Format of long description + url="https://github.com/yourname/%s", # URL for the project repository + classifiers=[ # Some additional metadata + "Programming Language :: Python :: 3", + "License :: OSI Approved :: MIT License", + "Operating System :: OS Independent", + ], + python_requires=">=3.6", # Minimum Python version required +) +]], project_name, project_name)) + + vim.cmd("edit " .. project_root .. "/main.py") + 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 .. ".c", "") + end) +end + +function M.create_new_header_file() + vim.ui.input({ prompt = "Enter header file name: " }, function(header_name) + if not header_name then return print("Header file creation canceled.") end + local root_dir = M.get_project_root() + 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 content = string.format([[ +#ifndef %s +#define %s + +#include + +// Function prototypes +void example_function(); + +#endif // %s +]], guard_macro, guard_macro, guard_macro) + write_file(root_dir .. "/include/" .. header_name .. ".h", content) + 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.run() + local root = M.get_project_root() + if not root then return print("Error: Project root not found.") end + local build_dir = root .. "/build" + 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 + M.open_float_terminal("cd " .. build_dir .. " && ./" .. files[1]:match("([^/]+)%.polydev$")) +end + +return M