Repository navigation
Expand file tree
/
Copy pathsetup_plan.py
More file actions
117 lines (102 loc) · 3.66 KB
/
Copy pathsetup_plan.py
File metadata and controls
117 lines (102 loc) · 3.66 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
#!/usr/bin/env python3
"""Setup implementation plan for a feature."""
from __future__ import annotations
import json
import sys
from pathlib import Path
try:
from common import (
TemplateResolutionError,
get_feature_paths,
resolve_template_content,
)
except ImportError: # pragma: no cover - direct execution from unusual cwd
sys.path.insert(0, str(Path(__file__).resolve().parent))
from common import (
TemplateResolutionError,
get_feature_paths,
resolve_template_content,
)
def _json_line(payload: object) -> str:
# ASCII escapes keep the JSON intact on a legacy Windows stdout code page.
return json.dumps(payload, ensure_ascii=True, separators=(",", ":")) + "\n"
def _help_text(argv0: str) -> str:
return f"""Usage: {argv0} [--json]
--json Output results in JSON format
--help Show this help message
"""
def main(argv: list[str] | None = None) -> int:
args = list(argv if argv is not None else sys.argv[1:])
json_mode = False
for arg in args:
if arg == "--json":
json_mode = True
elif arg in {"--help", "-h"}:
sys.stdout.write(_help_text(sys.argv[0]))
return 0
else:
print(f"ERROR: Unknown option '{arg}'", file=sys.stderr)
return 1
try:
paths = get_feature_paths(script_file=Path(__file__))
except SystemExit as exc:
if exc.code == 0:
return 0
print("ERROR: Failed to resolve feature paths", file=sys.stderr)
return int(exc.code) if isinstance(exc.code, int) else 1
paths.feature_dir.mkdir(parents=True, exist_ok=True)
# Status messages go to stderr in JSON mode so stdout stays pure JSON.
status_stream = sys.stderr if json_mode else sys.stdout
if paths.impl_plan.is_file():
print(
f"Plan already exists at {paths.impl_plan}, skipping template copy",
file=status_stream,
)
else:
try:
template_content = resolve_template_content("plan-template", paths.repo_root)
except TemplateResolutionError as exc:
print(f"Error: {exc}", file=sys.stderr)
return 1
if template_content is not None:
if template_content == "":
print(
"ERROR: Plan template is empty; no new plan.md was created.",
file=sys.stderr,
)
print(
"Restore a valid plan template and rerun plan setup.",
file=sys.stderr,
)
return 1
paths.impl_plan.write_bytes(template_content.encode("utf-8"))
print(f"Copied plan template to {paths.impl_plan}", file=status_stream)
else:
print(
"ERROR: Plan template not found; no new plan.md was created.",
file=sys.stderr,
)
print(
"Restore a valid plan template and rerun plan setup.",
file=sys.stderr,
)
return 1
if json_mode:
sys.stdout.write(
_json_line(
{
"FEATURE_SPEC": str(paths.feature_spec),
"IMPL_PLAN": str(paths.impl_plan),
"FEATURE_DIR": str(paths.feature_dir),
"BRANCH": paths.current_branch,
}
)
)
else:
print(f"FEATURE_SPEC: {paths.feature_spec}")
print(f"IMPL_PLAN: {paths.impl_plan}")
print(f"FEATURE_DIR: {paths.feature_dir}")
print(f"BRANCH: {paths.current_branch}")
return 0
if __name__ == "__main__":
raise SystemExit(main())