117 lines
4.3 KiB
Lua
117 lines
4.3 KiB
Lua
local M = {}
|
|
|
|
local utils = require("vern.utils")
|
|
local templates = require("vern.templates")
|
|
|
|
local join = vim.fs.joinpath
|
|
|
|
---@param opts VernPythonConfig
|
|
function M.create_project(opts)
|
|
vim.ui.input({ prompt = "Enter project name: " }, function(pname)
|
|
pname = pname and vim.trim(pname)
|
|
if not pname or pname == "" then
|
|
return
|
|
end
|
|
|
|
local configs = require("vern.configs.python")
|
|
---@type VernPythonConfig
|
|
local cfg = vim.tbl_deep_extend("force", {}, configs, opts or {})
|
|
local ctx = { project_name = pname }
|
|
|
|
local root = join(vim.fn.expand(cfg.workspace_root), cfg.python.root, pname)
|
|
vim.fn.mkdir(root, "p")
|
|
local bin
|
|
for tbl_name, tbl in pairs(cfg.python.templates.project) do
|
|
local raw_path = join(tbl.path, tbl.name)
|
|
local path = utils.resolve_path(raw_path, cfg.python.dirs, ctx)
|
|
if tbl_name == "primary" then
|
|
bin = join(root, path)
|
|
end
|
|
local abs_file_path = join(root, path)
|
|
local abs_dir_path = vim.fs.dirname(abs_file_path)
|
|
if abs_dir_path ~= root and not vim.startswith(root, abs_dir_path) then
|
|
vim.fn.mkdir(abs_dir_path, "p")
|
|
end
|
|
utils.dump(abs_file_path, templates.fetch(tbl.kind_path, tbl.kind, pname))
|
|
end
|
|
for _, tbl in pairs(cfg.python.templates.extras) do
|
|
if tbl.enabled then
|
|
local raw_path = join(tbl.path, tbl.name)
|
|
local path = utils.resolve_path(raw_path, cfg.python.dirs, ctx)
|
|
local abs_file_path = join(root, path)
|
|
local abs_dir_path = vim.fs.dirname(abs_file_path)
|
|
if abs_dir_path ~= root and not vim.startswith(root, abs_dir_path) then
|
|
vim.fn.mkdir(abs_dir_path, "p")
|
|
end
|
|
utils.dump(abs_file_path, templates.fetch(tbl.kind_path, tbl.kind, pname))
|
|
end
|
|
end
|
|
|
|
utils.init_git(root, "python")
|
|
|
|
local venv_path = root .. "/venv"
|
|
vim.fn.system("python3 -m venv " .. venv_path)
|
|
|
|
vim.notify("Project: " .. pname .. " has been created!")
|
|
vim.cmd.edit(vim.fn.fnameescape(bin))
|
|
end)
|
|
end
|
|
|
|
---@param opts VernPythonConfig
|
|
function M.create_file(opts)
|
|
opts = opts or {}
|
|
vim.ui.input({ prompt = "Enter file name: " }, function(fname)
|
|
fname = fname and vim.trim(fname)
|
|
if not fname or fname == "" then
|
|
return
|
|
end
|
|
local configs = require("vern.configs.python")
|
|
---@type VernPythonConfig
|
|
local cfg = vim.tbl_deep_extend("force", {}, configs, opts or {})
|
|
local mod = cfg.python.templates.file.mod or {}
|
|
|
|
local raw_path = join(mod.path, mod.name)
|
|
local context = { project_name = utils.proj_name(), name = fname }
|
|
local path = utils.resolve_path(raw_path, cfg.python.dirs or {}, context)
|
|
|
|
utils.dump(path, templates.fetch(mod.kind_path, mod.kind, fname))
|
|
vim.notify("File: " .. fname .. " has been created!")
|
|
vim.cmd.edit(vim.fn.expand(path))
|
|
end)
|
|
end
|
|
|
|
---@param opts VernPythonConfig Takes a table for the runtime
|
|
function M.run(opts)
|
|
opts = opts or {}
|
|
local root = utils.proj_root() or "."
|
|
local pname = vim.fs.basename(root)
|
|
local bin = opts.python.templates.project.primary
|
|
|
|
local exe = join(bin.path, bin.name) or "src/{project_name}/main.py"
|
|
local exe_path = utils.resolve_path(exe, opts.python.dirs, { project_name = pname })
|
|
local full_path = join(vim.fn.expand(root), exe_path)
|
|
utils.term({ "python3 " .. full_path }, opts.terminal or {})
|
|
end
|
|
|
|
function M.pip_install()
|
|
vim.ui.input({ prompt = "Enter pip module: " }, function(mod)
|
|
mod = mod and vim.trim(mod)
|
|
if not mod or mod == "" then
|
|
vim.notify("Pip Cancelled", vim.log.levels.WARN)
|
|
return
|
|
end
|
|
vim.notify("Installing " .. mod, vim.log.levels.INFO)
|
|
vim.system({ "pip", "install", mod }, { text = true }, function(obj)
|
|
vim.schedule(function()
|
|
if obj.code ~= 0 then
|
|
vim.notify("Pip failed: " .. obj.stderr, vim.log.levels.WARN)
|
|
else
|
|
vim.notify("Pip installed " .. mod, vim.log.levels.INFO)
|
|
end
|
|
end)
|
|
end)
|
|
end)
|
|
end
|
|
|
|
return M
|