_stub.py

skills/client-enrich/scripts/_stub.py

1. """Frontmatter helpers shared by the client-enrich write-back scripts.
3. Frontmatter is edited as text, one key at a time, so every line the scripts do
4. not own stays byte-identical. Not run directly.
5. """
7. import re
8. import sys
11. def split_frontmatter(text):
12. m = re.match(r"^---\n(.*?)\n---\n", text, re.DOTALL)
13. if not m:
14. sys.exit("ERROR: no YAML frontmatter found at top of stub.")
15. return m.group(1), text[m.end() :]
18. def fm_get(fm, key):
19. m = re.search(rf"(?m)^{re.escape(key)}:[ \t]*(.*)$", fm)
20. return m.group(1).strip() if m else None
23. def fm_is_empty(val):
24. if val is None:
25. return True
26. v = val.strip().strip('"').strip("'").strip()
27. return v == "" or v == "[TBD]"
30. def fm_set(fm, key, value, force, literal=True):
31. """Fill `key` only if it is empty or "[TBD]" (any value with force). Never adds a key.
33. re.sub reads backslashes in the replacement as escapes, so literal=True
34. doubles them and the value lands exactly as given. patch_research.py passes
35. literal=False, which keeps its established handling of a backslash.
36. """
37. cur = fm_get(fm, key)
38. if cur is None:
39. return fm, False
40. if not force and not fm_is_empty(cur):
41. return fm, False
42. if literal:
43. value = value.replace("\\", "\\\\")
44. return re.sub(
45. rf"(?m)^{re.escape(key)}:[ \t]*.*$", f'{key}: "{value}"', fm, count=1
46. ), True