diff --git a/lua/Polydev/configs.lua b/lua/Polydev/configs.lua index d47b643..342e873 100644 --- a/lua/Polydev/configs.lua +++ b/lua/Polydev/configs.lua @@ -1,14 +1,15 @@ +---@type table local M = {} ----@class Keybinds +---@class KeybindConfigs ---@field [string] string --- ----@class ConfigsLanguage +---@class LanguageConfigs ---@field project_root? string ----@field keybinds? Keybinds -- ? makes fields optional +---@field keybinds? KeybindConfigs -- ? makes fields optional ---@field build_attributes? string --- ----@class Terminal +---@class TerminalConfigs ---@field right_padding integer ---@field left_padding integer ---@field bottom_padding integer @@ -18,14 +19,14 @@ local M = {} ---@field relativenumber boolean --- ---@class Defaults ----@field html ConfigsLanguage ----@field rust ConfigsLanguage ----@field java ConfigsLanguage ----@field python ConfigsLanguage ----@field lua ConfigsLanguage ----@field c ConfigsLanguage ----@field cpp ConfigsLanguage ----@field terminal Terminal +---@field html LanguageConfigs +---@field rust LanguageConfigs +---@field java LanguageConfigs +---@field python LanguageConfigs +---@field lua LanguageConfigs +---@field c LanguageConfigs +---@field cpp LanguageConfigs +---@field terminal TerminalConfigs ---@type Defaults M.defaults = { @@ -91,13 +92,16 @@ M.defaults = { ---@type string[] M.user_config = {} +---@param user_opts table function M.setup(user_opts) M.user_config = vim.tbl_deep_extend("force", {}, M.defaults, user_opts or {}) end ---@param lang string +---@return string|table function M.get(lang) return M.user_config[lang] or {} end +---@return table return M diff --git a/lua/Polydev/init.lua b/lua/Polydev/init.lua index c8148b2..b4bc945 100644 --- a/lua/Polydev/init.lua +++ b/lua/Polydev/init.lua @@ -46,8 +46,8 @@ function M.create_new_file() vim.ui.input({ prompt = "Enter language for new file: " }, function(lang) if not lang or lang == "" then return print("File creation canceled.") end - if M.load_language_module(lang) and M.loaded_languages[lang].create_new_file then - M.loaded_languages[lang].create_new_file() + if M.load_language_module(lang) and templates.files[lang] and templates.files[lang].run then + templates.create_new_file(lang) else print("Error: No File creation method for " .. lang) end diff --git a/lua/Polydev/languages/c.lua b/lua/Polydev/languages/c.lua index 194bb02..d3397bb 100644 --- a/lua/Polydev/languages/c.lua +++ b/lua/Polydev/languages/c.lua @@ -25,94 +25,6 @@ function M.setup(opts) end 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.opts.project_root) .. "/" .. project_name - for _, path in ipairs({ "/src", "/build", "/include" }) do - vim.fn.mkdir(project_root .. path, "p") - end - - utils.write_file(project_root .. "/build/" .. project_name .. ".polydev", project_name) - utils.write_file(project_root .. "/src/main.c", [[ -/* If you have vcpkg and want to use it: -vcpkg new --application -vcpkg add port -*/ - -#include - -int main() { - printf("%s", "Hello World"); - return 0; -} -]]) - - utils.write_file(project_root .. "/CMakeLists.txt", string.format([[ -cmake_minimum_required(VERSION 3.10) -project(%s) -include_directories(include) -set(SOURCES src/main.c) -add_executable(%s ${SOURCES}) -set(CMAKE_EXPORT_COMPILE_COMMANDS ON) -set(CMAKE_C_STANDARD 23) -set(CMAKE_C_STANDARD_REQUIRED ON) - -# find_package(SDL3 CONFIG REQUIRED) # An example of how to import libraries from vcpkg -# target_link_libraries(Game PRIVATE SDL3::SDL3) -]], project_name, project_name)) - - - utils.write_file(project_root .. "/CMakePresets.json", string.format([[ -# For the purposes of vcpkg, DO NOT REMOVE, but feel free to change as needed -{ - "version": 2, - "configurePresets": [ - { - "name": "vcpkg", - "generator": "Ninja", - "binaryDir": "${sourceDir}/build", - "cacheVariables": { - "CMAKE_TOOLCHAIN_FILE": "$env{VCPKG_ROOT}/scripts/buildsystems/vcpkg.cmake" - } - } - ] -} -]])) - - utils.write_file(project_root .. "/CMakeUserPresets.json", string.format([[ -{ - "version": 2, - "configurePresets": [ - { - "name": "default", - "inherits": "vcpkg", - "environment": { - "VCPKG_ROOT": "/Users//vcpkg" - } - } - ] -} - -]])) - vim.cmd("edit " .. project_root .. "/src/main.c") - vim.fn.system(string.format("cd %s/build/ && cmake .. && cmake --build .", project_root)) - 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 - utils.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 diff --git a/lua/Polydev/languages/cpp.lua b/lua/Polydev/languages/cpp.lua index 383f282..61e5c8a 100644 --- a/lua/Polydev/languages/cpp.lua +++ b/lua/Polydev/languages/cpp.lua @@ -25,95 +25,6 @@ function M.setup(opts) end 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.opts.project_root) .. "/" .. project_name - for _, path in ipairs({ "/src", "/build", "/include" }) do - vim.fn.mkdir(project_root .. path, "p") - end - - utils.write_file(project_root .. "/build/" .. project_name .. ".polydev", project_name) - utils.write_file(project_root .. "/src/main.cpp", [[ -/* If you have vcpkg and want to use it: -vcpkg new --application -vcpkg add port -*/ - -#include - -int main() { - std::cout << "Hello World" << std::endl; - return 0; -} -]]) - - utils.write_file(project_root .. "/CMakeLists.txt", string.format([[ -cmake_minimum_required(VERSION 3.10) -project(%s) -include_directories(include) -set(SOURCES src/main.cpp) -add_executable(%s ${SOURCES}) -set(CMAKE_EXPORT_COMPILE_COMMANDS ON) -set(CMAKE_C_STANDARD 23) -set(CMAKE_C_STANDARD_REQUIRED ON) - -# find_package(SDL3 CONFIG REQUIRED) # An example of how to import libraries from vcpkg -# target_link_libraries(Game PRIVATE SDL3::SDL3) -]], project_name, project_name)) - - - utils.write_file(project_root .. "/CMakePresets.json", string.format([[ -# For the purposes of vcpkg, DO NOT REMOVE, but feel free to change as needed -{ - "version": 2, - "configurePresets": [ - { - "name": "vcpkg", - "generator": "Ninja", - "binaryDir": "${sourceDir}/build", - "cacheVariables": { - "CMAKE_TOOLCHAIN_FILE": "$env{VCPKG_ROOT}/scripts/buildsystems/vcpkg.cmake" - } - } - ] -} -]])) - - utils.write_file(project_root .. "/CMakeUserPresets.json", string.format([[ -{ - "version": 2, - "configurePresets": [ - { - "name": "default", - "inherits": "vcpkg", - "environment": { - "VCPKG_ROOT": "/Users//vcpkg" - } - } - ] -} - -]])) - vim.fn.system(string.format("cd %s/build/ && cmake .. && cmake --build .", project_root)) - - vim.cmd("edit " .. project_root .. "/src/main.cpp") - 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 - utils.write_file(root_dir .. "/src/" .. class_name .. ".cpp", "") - 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 diff --git a/lua/Polydev/languages/html.lua b/lua/Polydev/languages/html.lua index 859298c..40f526d 100644 --- a/lua/Polydev/languages/html.lua +++ b/lua/Polydev/languages/html.lua @@ -1,55 +1,13 @@ -local utils = require("Polydev.utils") +---@type table local M = {} +---@type table|nil M.opts = nil +---@param opts table function M.setup(opts) M.opts = vim.tbl_deep_extend("force", {}, require("Polydev.configs").get("html"), opts or {}) 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.opts.project_root) .. "/" .. project_name - - local main_html_content = [[ - - - - Page Title - - - -

My First Heading

-

My first paragraph.

- - - -]] - utils.write_file(project_root .. "index.html", main_html_content) - end) -end - -function M.create_new_file() - vim.ui.input({ prompt = "Enter page name: " }, function(page) - if not page or page == "" then return print("Class creation canceled.") end - local html_content = string.format([[ - - - - Page Title - - - -

My First Heading

-

My first paragraph.

- - - -]]) - - utils.write_file("./" .. page .. ".html", html_content) - end) -end - +---@return table return M diff --git a/lua/Polydev/languages/java.lua b/lua/Polydev/languages/java.lua index b38ab1b..9208221 100644 --- a/lua/Polydev/languages/java.lua +++ b/lua/Polydev/languages/java.lua @@ -21,21 +21,6 @@ function M.setup(opts) end end -function M.create_new_file() - vim.ui.input({ prompt = "Enter class name: " }, function(class_name) - if not class_name or class_name == "" then return print("Class creation canceled.") end - local root_dir = vim.fn.expand("%:p:h"):match("(.*)/src") - if not root_dir then return print("Error: src directory not found.") end - local java_content = string.format([[ -public class %s { - // New File -} -]], class_name) - - utils.write_file(root_dir .. "/src/" .. class_name .. ".java", java_content) - end) -end - function M.build() local project_root = vim.fn.expand("%:p:h"):match("(.*)/src") if not project_root then return print("Error: Could not detect project root directory.") end diff --git a/lua/Polydev/languages/lua.lua b/lua/Polydev/languages/lua.lua index f80386d..3e3db89 100644 --- a/lua/Polydev/languages/lua.lua +++ b/lua/Polydev/languages/lua.lua @@ -1,9 +1,16 @@ +---@module "Polydev.utils" local utils = require("Polydev.utils") + +---@type table local M = {} +---@type table M.keybinds = {} + +---@type table|nil M.opts = nil +---@param opts table function M.setup(opts) M.opts = vim.tbl_deep_extend("force", {}, require("Polydev.configs").get("lua"), opts or {}) for key, command in pairs(M.opts.keybinds) do @@ -17,59 +24,30 @@ function M.setup(opts) end end -function M.get_project_root() - local dir = vim.fn.expand("%:p:h") - while dir ~= "/" do - local polydev_file = vim.fn.glob(dir .. "/*.polydev") - if polydev_file ~= "" then return dir end - dir = vim.fn.fnamemodify(dir, ":h") - end - return nil -end - -function M.get_project_name() - local polydev_file = vim.fn.glob((M.get_project_root() or "") .. "/*.polydev") - return polydev_file:match("([^/]+)%.polydev$") -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.opts.project_root) .. "/" .. project_name - for _, path in ipairs({ "/lua/" .. project_name }) do vim.fn.mkdir(project_root .. path, "p") end - - utils.write_file(project_root .. "/lua/" .. project_name .. "/init.lua", [[ -local M = {} - -return M -]]) - - utils.write_file(project_root .. "/" .. project_name .. ".polydev", "[[DO NOT REMOVE THIS FILE]]") - vim.cmd("edit " .. project_root .. "/lua/" .. project_name .. "/init.lua") - 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() - local project_name = M.get_project_name() - if not root_dir or not project_name then return print("Error: Project root not found.") end - utils.write_file(root_dir .. "/lua/" .. project_name .. "/" .. class_name .. ".lua", "") - end) -end - function M.run() - local root = M.get_project_root() - local project_name = M.get_project_name() - if not root or not project_name then return print("Error: Project root not found.") end - local init_lua_path = root .. "/lua/" .. project_name .. "/init.lua" + ---@return string|nil + local function get_project_root() + local dir = vim.fn.expand("%:p:h") + while dir ~= "/" do + local polydev_file = vim.fn.glob(dir .. "/*.polydev") + if polydev_file ~= "" then return dir end + dir = vim.fn.fnamemodify(dir, ":h") + end + return nil + end + + ---@return string + local function get_project_name() + return vim.fn.glob((M.get_project_root() or "") .. "/*.polydev"):match("([^/]+)%.polydev$") + end + if not get_project_root() or not get_project_name() then return print("Error: Project root not found.") end + local init_lua_path = get_project_root() .. "/lua/" .. get_project_name() .. "/init.lua" if vim.fn.filereadable(init_lua_path) == 1 then utils.open_float_terminal("lua " .. init_lua_path) else - print("Error: " .. init_lua_path .. " not found in " .. root) + print("Error: " .. init_lua_path .. " not found in " .. get_project_root()) end end +---@return table return M diff --git a/lua/Polydev/languages/python.lua b/lua/Polydev/languages/python.lua index 30f66a9..c1b796d 100644 --- a/lua/Polydev/languages/python.lua +++ b/lua/Polydev/languages/python.lua @@ -42,26 +42,6 @@ function M.get_project_root() return nil end -function M.create_new_file() - vim.ui.input({ prompt = "Enter file name: " }, function(file_name) - if not file_name or file_name == "" then - return print("File creation canceled.") - end - - if not file_name:match("%.py$") then - file_name = file_name .. ".py" - end - - local root_dir = M.get_project_root() - if not root_dir then - return print("Error: Project root not found.") - end - - local file_path = root_dir .. "/include/" .. file_name - utils.write_file(file_path, "") - end) -end - function M.install_dependency() vim.ui.input({ prompt = "Enter pip module: " }, function(pip_module) if not pip_module or pip_module == "" then diff --git a/lua/Polydev/languages/rust.lua b/lua/Polydev/languages/rust.lua index 8cd59bb..7c3c7a9 100644 --- a/lua/Polydev/languages/rust.lua +++ b/lua/Polydev/languages/rust.lua @@ -21,24 +21,6 @@ function M.setup(opts) end 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 - utils.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 diff --git a/lua/Polydev/templates.lua b/lua/Polydev/templates.lua index f06f46b..68ed1f9 100644 --- a/lua/Polydev/templates.lua +++ b/lua/Polydev/templates.lua @@ -11,7 +11,11 @@ local M = {} ---@field java TemplateLanguage ---@field python TemplateLanguage ---@field lua TemplateLanguage ----@alias TemplatesLanguageName "java" | "python" | "lua" +---@field c TemplateLanguage +---@field cpp TemplateLanguage +---@field rust TemplateLanguage +---@field html TemplateLanguage +---@alias TemplatesLanguageName "java" | "python" | "lua" | "c" | "cpp" | "rust" | "html" ---@type Projects M.projects = { @@ -19,7 +23,6 @@ M.projects = { run = function(project_name, project_root) local full_project_root = vim.fn.expand(project_root) .. "/" .. project_name for _, path in ipairs({ "/src", "/build" }) do vim.fn.mkdir(full_project_root .. path, "p") end - local main_java_content = [[ public class Main { public static void main(String[] args) { @@ -28,14 +31,13 @@ public class Main { } ]] utils.write_file(full_project_root .. "/src/Main.java", main_java_content) - return nil + return nil end }, python = { run = function(project_name, project_root) local full_project_root = vim.fn.expand(project_root) .. "/" .. project_name vim.fn.mkdir(full_project_root .. "/include", "p") - utils.write_file(full_project_root .. "/main.py", [[ def main(): print("Hello, World!") @@ -43,7 +45,6 @@ def main(): if __name__ == "__main__": main() ]]) - utils.write_file(full_project_root .. "/include/__init__.py", "") utils.write_file(full_project_root .. "/requirements.txt", "") utils.write_file(full_project_root .. "/setup.py", string.format([[ @@ -57,23 +58,20 @@ setup( python_requires=">=3.6", ) ]], project_name)) - vim.cmd("edit " .. full_project_root .. "/main.py") local venv_path = full_project_root .. "/venv" vim.fn.system("python3 -m venv " .. venv_path) vim.g.python3_host_prog = venv_path .. "/bin/python" vim.fn.setenv("VIRTUAL_ENV", venv_path) vim.fn.setenv("PATH", venv_path .. "/bin:" .. vim.fn.getenv("PATH")) - print("Virtual environment activated for " .. project_name) - return nil + return nil end }, lua = { run = function(project_name, project_root) local full_project_root = vim.fn.expand(project_root) .. "/" .. project_name for _, path in ipairs({ "/lua/" .. project_name }) do vim.fn.mkdir(full_project_root .. path, "p") end - utils.write_file(full_project_root .. "/lua/" .. project_name .. "/init.lua", [[ local M = {} @@ -81,9 +79,297 @@ return M ]]) utils.write_file(full_project_root .. "/" .. project_name .. ".polydev", "[[DO NOT REMOVE THIS FILE]]") vim.cmd("edit " .. full_project_root .. "/lua/" .. project_name .. "/init.lua") - return nil + return nil end }, + c = { + run = function(project_name, project_root) + local full_project_root = vim.fn.expand(M.opts.project_root) .. "/" .. project_name + for _, path in ipairs({ "/src", "/build", "/include" }) do vim.fn.mkdir(full_project_root .. path, "p") end + utils.write_file(full_project_root .. "/build/" .. project_name .. ".polydev", project_name) + utils.write_file(full_project_root .. "/src/main.c", [[ +/* If you have vcpkg and want to use it: +vcpkg new --application +vcpkg add port +*/ + +#include + +int main() { + printf("%s", "Hello World"); + return 0; +} +]]) + utils.write_file(full_project_root .. "/CMakeLists.txt", string.format([[ +cmake_minimum_required(VERSION 3.10) +project(%s) +include_directories(include) +set(SOURCES src/main.c) +add_executable(%s ${SOURCES}) +set(CMAKE_EXPORT_COMPILE_COMMANDS ON) +set(CMAKE_C_STANDARD 23) +set(CMAKE_C_STANDARD_REQUIRED ON) + +# find_package(SDL3 CONFIG REQUIRED) # An example of how to import libraries from vcpkg +# target_link_libraries(Game PRIVATE SDL3::SDL3) +]], project_name, project_name)) + utils.write_file(full_project_root .. "/CMakePresets.json", string.format([[ +# For the purposes of vcpkg, DO NOT REMOVE, but feel free to change as needed +{ + "version": 2, + "configurePresets": [ + { + "name": "vcpkg", + "generator": "Ninja", + "binaryDir": "${sourceDir}/build", + "cacheVariables": { + "CMAKE_TOOLCHAIN_FILE": "$env{VCPKG_ROOT}/scripts/buildsystems/vcpkg.cmake" + } + } + ] +} +]])) + utils.write_file(full_project_root .. "/CMakeUserPresets.json", string.format([[ +{ + "version": 2, + "configurePresets": [ + { + "name": "default", + "inherits": "vcpkg", + "environment": { + "VCPKG_ROOT": "/Users//vcpkg" + } + } + ] +} + +]])) + vim.cmd("edit " .. full_project_root .. "/src/main.c") + vim.fn.system(string.format("cd %s/build/ && cmake .. && cmake --build .", full_project_root)) + end + }, + cpp = { + run = function (project_name, project_root) + local full_project_root = vim.fn.expand(project_root) .. "/" .. project_name + for _, path in ipairs({ "/src", "/build", "/include" }) do + vim.fn.mkdir(full_project_root .. path, "p") + end + utils.write_file(full_project_root .. "/build/" .. project_name .. ".polydev", project_name) + utils.write_file(full_project_root .. "/src/main.cpp", [[ +/* If you have vcpkg and want to use it: +vcpkg new --application +vcpkg add port +*/ + +#include + +int main() { + std::cout << "Hello World" << std::endl; + return 0; +} +]]) + utils.write_file(full_project_root .. "/CMakeLists.txt", string.format([[ +cmake_minimum_required(VERSION 3.10) +project(%s) +include_directories(include) +set(SOURCES src/main.cpp) +add_executable(%s ${SOURCES}) +set(CMAKE_EXPORT_COMPILE_COMMANDS ON) +set(CMAKE_C_STANDARD 23) +set(CMAKE_C_STANDARD_REQUIRED ON) + +# find_package(SDL3 CONFIG REQUIRED) # An example of how to import libraries from vcpkg +# target_link_libraries(Game PRIVATE SDL3::SDL3) +]], project_name, project_name)) + utils.write_file(full_project_root .. "/CMakePresets.json", string.format([[ +# For the purposes of vcpkg, DO NOT REMOVE, but feel free to change as needed +{ + "version": 2, + "configurePresets": [ + { + "name": "vcpkg", + "generator": "Ninja", + "binaryDir": "${sourceDir}/build", + "cacheVariables": { + "CMAKE_TOOLCHAIN_FILE": "$env{VCPKG_ROOT}/scripts/buildsystems/vcpkg.cmake" + } + } + ] +} +]])) + utils.write_file(full_project_root .. "/CMakeUserPresets.json", string.format([[ +{ + "version": 2, + "configurePresets": [ + { + "name": "default", + "inherits": "vcpkg", + "environment": { + "VCPKG_ROOT": "/Users//vcpkg" + } + } + ] +} + +]])) + vim.fn.system(string.format("cd %s/build/ && cmake .. && cmake --build .", full_project_root)) + vim.cmd("edit " .. full_project_root .. "/src/main.cpp") + return nil + end + }, + rust = { + run = function (project_name, project_root) + local full_project_root = vim.fn.expand(project_root) .. "/" .. project_name + vim.fn.system("cargo new " .. full_project_root) + vim.cmd("edit " .. full_project_root .. "/src/main.rs") + end + }, + html = { + run = function(project_name, project_root) + local full_project_root = vim.fn.expand(project_root) .. "/" .. project_name + local main_html_content = [[ + + + + Page Title + + + +

My First Heading

+

My first paragraph.

+ + + +]] + utils.write_file(full_project_root .. "index.html", main_html_content) + end + }, +} + +---@class TemplateFile +---@field run fun(file_name: string): nil + +---@class Files +---@field java TemplateFile +---@field python TemplateFile +---@field lua TemplateFile +---@field c TemplateFile +---@field cpp TemplateFile +---@field rust TemplateFile +---@field html TemplateFile +---@alias TemplatesFileName "java" | "python" | "lua" | "c" | "cpp" | "rust" | "html" + +---@type Files +M.files = { + java = { + run = function (file_name) + local root_dir = vim.fn.expand("%:p:h"):match("(.*)/src") + if not root_dir then return print("Error: src directory not found.") end + local java_content = string.format([[ +public class %s { + // New File +} +]], class_name) + + utils.write_file(root_dir .. "/src/" .. file_name .. ".java", java_content) + end + }, + python = { + run = function (file_name) + if not file_name:match("%.py$") then file_name = file_name .. ".py" end + local root_dir = M.get_project_root() + if not root_dir then return print("Error: Project root not found.") end + local file_path = root_dir .. "/include/" .. file_name + utils.write_file(file_path, "") + end + }, + lua = { + run = function (file_name) + local function get_project_root() + local dir = vim.fn.expand("%:p:h") + while dir ~= "/" do + local polydev_file = vim.fn.glob(dir .. "/*.polydev") + if polydev_file ~= "" then return dir end + dir = vim.fn.fnamemodify(dir, ":h") + end + return nil + end + local function get_project_name() return vim.fn.glob((get_project_root() or "") .. "/*.polydev"):match("([^/]+)%.polydev$") end + if not get_project_root() or not get_project_name() then return print("Error: Project root not found.") end + utils.write_file(get_project_root() .. "/lua/" .. get_project_name() .. "/" .. file_name .. ".lua", "") + end + }, + c = { + run = function (file_name) + local function 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 + + if not file_name or file_name == "" then return print("File creation canceled.") end + local root_dir = get_project_root() + if not root_dir then return print("Error: Project root not found.") end + utils.write_file(root_dir .. "/src/" .. file_name .. ".c", "") + end + + }, + cpp = { + run = function (file_name) + local function 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 + local root_dir = get_project_root() + if not root_dir then return print("Error: Project root not found.") end + utils.write_file(root_dir .. "/src/" .. file_name .. ".cpp", "") + end + }, + rust = { + run = function (file_name) + local function get_project_root() + local dir = vim.fn.expand("%:p:h") + while dir ~= "/" do + if vim.fn.isdirectory(dir .. "/src") == 1 then + return dir + end + dir = vim.fn.fnamemodify(dir, ":h") + end + end + local root_dir = get_project_root() + if not root_dir then return print("Error: Project root not found.") end + utils.write_file(root_dir .. "/src/" .. file_name .. ".rs", "") + end + }, + html = { + run = function (file_name) + local html_content = string.format([[ + + + + Page Title + + + +

My First Heading

+

My first paragraph.

+ + + +]]) + + utils.write_file("./" .. file_name .. ".html", html_content) + end + } } ---@param lang TemplatesLanguageName @@ -95,5 +381,13 @@ function M.create_project(lang, project_root) end) end +---@param lang TemplatesFileName +function M.create_new_file(lang) + vim.ui.input({ prompt = "Enter file name: " }, function(file_name) + if not file_name or file_name == "" then return print("File creation canceled.") end + M.files[lang].run(file_name) + end) +end + ---@return any return M