93 lines
2.5 KiB
Python
93 lines
2.5 KiB
Python
#!/usr/bin/env python3
|
|
from typing import Any, Literal, Optional
|
|
from tomllib import load
|
|
from pathlib import Path
|
|
from dataclasses import dataclass, field
|
|
import questionary
|
|
|
|
TEMPLATES_DIR = Path("./_templates")
|
|
|
|
@dataclass(frozen=True)
|
|
class TemplateParameter:
|
|
prompt: str
|
|
default: Optional[Any] = None
|
|
|
|
@dataclass
|
|
class ProjectTemplate:
|
|
name: str
|
|
path: Path
|
|
templates: list[str] = field(default_factory=list)
|
|
parameters: dict[str, TemplateParameter] = field(default_factory=dict)
|
|
|
|
|
|
|
|
templates: dict[str, ProjectTemplate] = {}
|
|
|
|
|
|
for v in TEMPLATES_DIR.rglob("template.toml"):
|
|
with v.open("rb") as fp:
|
|
data = load(fp)
|
|
tpl = ProjectTemplate(
|
|
data.get('name', str(v.parent.relative_to(TEMPLATES_DIR))),
|
|
v.parent,
|
|
data.get('templates', []),
|
|
{
|
|
k: TemplateParameter(**v)
|
|
for k, v in data.get('params', {}).items()
|
|
}
|
|
)
|
|
templates[tpl.name] = tpl
|
|
|
|
|
|
name: Optional[str] = questionary.text("Project name").ask()
|
|
if not name:
|
|
exit(1)
|
|
|
|
template: Optional[ProjectTemplate] = questionary.select("Template", [
|
|
questionary.Choice(tpl.name, tpl) for tpl in templates.values()
|
|
]).ask()
|
|
if not template:
|
|
exit(1)
|
|
|
|
parameters = questionary.prompt([
|
|
{
|
|
"type": "text",
|
|
"name": k,
|
|
"message": param.prompt,
|
|
"default": param.default
|
|
}
|
|
for k, param in template.parameters.items()
|
|
])
|
|
if not parameters:
|
|
exit(1)
|
|
|
|
project_dir = Path(name)
|
|
project_dir.mkdir(parents=True, exist_ok=True)
|
|
|
|
for base, dirs, files in template.path.walk():
|
|
for dirname in dirs:
|
|
(base / dirname).mkdir(exist_ok=True, parents=True)
|
|
|
|
for filename in files:
|
|
in_path = base / filename
|
|
name = str(in_path.relative_to(template.path))
|
|
out_path = project_dir / name
|
|
if name == "template.toml":
|
|
continue
|
|
elif name in template.templates:
|
|
print("TMPL", out_path)
|
|
with in_path.open("r") as fp_in:
|
|
with out_path.open("w") as fp_out:
|
|
for line in fp_in:
|
|
line = line.replace("__PROGNAME", name)
|
|
for param in template.parameters:
|
|
line = line.replace(f"__{param}", parameters[param])
|
|
fp_out.write(line)
|
|
else:
|
|
print("COPY", out_path)
|
|
with in_path.open("rb") as fp_in:
|
|
with out_path.open("wb") as fp_out:
|
|
while (chunk := fp_in.read(32768)):
|
|
fp_out.write(chunk)
|
|
|