DarthMooMancer/vern-nvim
1
1
Fork 0
Code Issues 3 Pull Requests Activity

refactor: fixed deprecated write_file to use table instead of string for

Browse Source 
each language in templates.projects
This commit is contained in:
DarthMooMancer
2025-05-12 00:41:05 -04:00
parent 677e5bbc67
commit 4ecaf54f82
7 changed files with 103 additions and 106 deletions
Download Patch File Download Diff File Expand all files Collapse all files
lua/Polydev/templates.lua
+55 -68
View File
@@ -2,19 +2,6 @@ local utils = require("Polydev.utils")
local M = {}
---@class TemplateLanguage
---@field run fun(project_name: string, project_root: string): string
---@class Projects
---@field java TemplateLanguage
---@field python TemplateLanguage
---@field lua TemplateLanguage
---@field c TemplateLanguage
---@field cpp TemplateLanguage
---@field rust TemplateLanguage
---@field html TemplateLanguage
---@alias TemplatesLanguageName "java" | "python" | "lua" | "c" | "cpp" | "rust" | "html"
local function init_git(path, gitignore_lines)
vim.fn.mkdir(path, "p")
@@ -29,41 +16,39 @@ local function init_git(path, gitignore_lines)
end
end
---@type Projects
M.projects = {
java = {
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 = [[
local content = [[
public class Main {
public static void main(String[] args) {
System.out.println("Hello, World!");
}
}
]]
local java_gitignore = {
local gitignore = {
"build/"
}
init_git(full_project_root, java_gitignore)
utils.write_file({full_project_root, "src", "Main.java"}, main_java_content)
return full_project_root
init_git(full_project_root, gitignore)
utils.write_file({ full_project_root, "src", "Main.java" }, content)
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", [[
utils.write_file({ full_project_root, "main.py" }, [[
def main():
print("Hello, World!")
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([[
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([[
from setuptools import setup, find_packages
setup(
@@ -74,6 +59,10 @@ setup(
python_requires=">=3.6",
)
]], project_name))
local gitignore = {
"venv/"
}
init_git(full_project_root, gitignore)
vim.cmd("edit " .. full_project_root .. "/main.py")
local venv_path = full_project_root .. "/venv"
vim.fn.system("python3 -m venv " .. venv_path)
@@ -81,38 +70,35 @@ setup(
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 full_project_root
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", [[
vim.fn.mkdir(full_project_root .. "/lua/" .. project_name, "p")
utils.write_file({ full_project_root, "lua", project_name, "init.lua" }, [[
local M = {}
return M
]])
utils.write_file(full_project_root .. "/" .. project_name .. ".polydev", "[[DO NOT REMOVE THIS FILE]]")
local gitignore = { "" }
init_git(full_project_root, gitignore)
vim.cmd("edit " .. full_project_root .. "/lua/" .. project_name .. "/init.lua")
return full_project_root
end
},
c = {
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.c", [[
utils.write_file({ full_project_root, "src/main.c" }, [[
#include <stdio.h>
int
main() {
int main() {
printf("%s", "Hello World");
return 0;
}
]])
utils.write_file(full_project_root .. "/CMakeLists.txt", string.format([[
utils.write_file({ full_project_root, "CMakeLists.txt" }, string.format([[
cmake_minimum_required(VERSION 3.10)
project(%s)
include_directories(include)
@@ -125,9 +111,10 @@ 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))
vim.cmd("edit " .. full_project_root .. "/src/main.c")
local gitignore = { "build/" }
init_git(full_project_root, gitignore)
vim.fn.system(string.format("cd %s/build/ && cmake .. && cmake --build .", full_project_root))
return full_project_root
vim.cmd("edit " .. full_project_root .. "/src/main.c")
end
},
cpp = {
@@ -136,8 +123,7 @@ set(CMAKE_C_STANDARD_REQUIRED ON)
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", [[
utils.write_file({ full_project_root, "src/main.cpp" }, [[
#include <iostream>
int main() {
@@ -145,7 +131,7 @@ int main() {
return 0;
}
]])
utils.write_file(full_project_root .. "/CMakeLists.txt", string.format([[
utils.write_file({ full_project_root, "CMakeLists.txt" }, string.format([[
cmake_minimum_required(VERSION 3.10)
project(%s)
include_directories(include)
@@ -159,12 +145,10 @@ set(CMAKE_C_STANDARD_REQUIRED ON)
# target_link_libraries(Game PRIVATE SDL3::SDL3)
]], project_name, project_name))
utils.write_file(full_project_root .. "/.gitignore", string.format([[
build/
]]))
local gitignore = { "build/" }
init_git(full_project_root, gitignore)
vim.fn.system(string.format("cd %s/build/ && cmake .. && cmake --build .", full_project_root))
vim.cmd("edit " .. full_project_root .. "/src/main.cpp")
return full_project_root
end
},
rust = {
@@ -172,7 +156,6 @@ build/
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")
return full_project_root
end
},
html = {
@@ -192,26 +175,11 @@ build/
</body>
</html>
]]
utils.write_file(full_project_root .. "index.html", main_html_content)
return full_project_root
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)
@@ -324,25 +292,44 @@ public class %s {
}
}
---@type string
M.dir = ""
M.aux_files = {
c = {
run = function (file_name)
if not file_name then return print("Header file creation canceled.") end
local root_dir = utils.get_project_root()
local guard_macro = file_name:upper():gsub("[^A-Z0-9]", "_") .. "_H"
local content = string.format([[
#ifndef %s
#define %s
// Add code here
#endif
]], guard_macro, guard_macro)
utils.write_file({ root_dir, "include", file_name .. ".h" }, content)
end
}
}
---@param lang TemplatesLanguageName
---@param project_root string
function M.create_project(lang, project_root)
vim.ui.input({ prompt = "Enter project name: " }, function(project_name)
if not project_name or project_name == "" then return print("Project creation canceled.") end
M.dir = M.projects[lang].run(project_name, project_root)
if not project_name or project_name == "" then return print("Project creation canceled") end
M.projects[lang].run(project_name, 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
if not file_name or file_name == "" then return print("File creation canceled") end
M.files[lang].run(file_name)
end)
end
---@return any
function M.create_aux_file(lang)
vim.ui.input({ promt = "Enter file name: "}, function(file_name)
if not file_name or file_name == "" then return print("File creation canceled") end
M.aux_files[lang].run(file_name)
end)
end
return M