mirror of
https://github.com/forcedotcom/afv-library.git
synced 2026-07-30 19:50:42 +08:00
60 lines
2.0 KiB
Python
60 lines
2.0 KiB
Python
|
"""Runtime override hook for DATA_ROOT / CACHE_ROOT.
|
||
|
|
|
||
|
|
Default layout is ``~/.vibe/data/<skill>`` and
|
||
|
|
``~/.vibe/cache/<skill>`` — runtime-agnostic branding that
|
||
|
|
follows the repo's "tool-agnostic language" guideline.
|
||
|
|
|
||
|
|
Other runtimes (AFV OOTB, Codex, Cursor, OpenCode) can override via
|
||
|
|
``--data-dir`` / ``--cache-dir`` CLI flags on entry scripts. Those flags
|
||
|
|
call ``set_*_override()`` BEFORE any pipeline module imports ``paths.py``,
|
||
|
|
so the resolution helpers below pick up the override.
|
||
|
|
|
||
|
|
Three-level rebind contract
|
||
|
|
---------------------------
|
||
|
|
Python's ``from X import Y`` snapshots ``Y`` into the importer's local
|
||
|
|
namespace. Mutating ``X.Y`` later does NOT update the importer's local
|
||
|
|
binding. Entry scripts that capture ``DATA_ROOT`` / ``CACHE_ROOT`` at
|
||
|
|
module top must therefore rebind in THREE places after override:
|
||
|
|
|
||
|
|
1. ``paths.DATA_ROOT`` — the source of truth
|
||
|
|
2. ``config.DATA_ROOT`` — the re-export
|
||
|
|
3. The entry script's own local ``DATA_ROOT`` (via ``global``)
|
||
|
|
"""
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from pathlib import Path
|
||
|
|
from typing import Optional
|
||
|
|
|
||
|
|
_DATA_OVERRIDE: Optional[Path] = None
|
||
|
|
_CACHE_OVERRIDE: Optional[Path] = None
|
||
|
|
|
||
|
|
|
||
|
|
def set_data_root_override(p: Optional[Path]) -> None:
|
||
|
|
"""Set the data-root override. Pass ``None`` to clear."""
|
||
|
|
global _DATA_OVERRIDE
|
||
|
|
_DATA_OVERRIDE = p
|
||
|
|
|
||
|
|
|
||
|
|
def set_cache_root_override(p: Optional[Path]) -> None:
|
||
|
|
"""Set the cache-root override. Pass ``None`` to clear."""
|
||
|
|
global _CACHE_OVERRIDE
|
||
|
|
_CACHE_OVERRIDE = p
|
||
|
|
|
||
|
|
|
||
|
|
def resolve_data_root(skill_name: str) -> Path:
|
||
|
|
"""Resolve the data root for ``skill_name``.
|
||
|
|
|
||
|
|
Returns the override if set, else the runtime-agnostic default
|
||
|
|
``~/.vibe/data/<skill_name>``.
|
||
|
|
"""
|
||
|
|
return _DATA_OVERRIDE or (Path.home() / ".vibe" / "data" / skill_name)
|
||
|
|
|
||
|
|
|
||
|
|
def resolve_cache_root(skill_name: str) -> Path:
|
||
|
|
"""Resolve the cache root for ``skill_name``.
|
||
|
|
|
||
|
|
Returns the override if set, else the runtime-agnostic default
|
||
|
|
``~/.vibe/cache/<skill_name>``.
|
||
|
|
"""
|
||
|
|
return _CACHE_OVERRIDE or (Path.home() / ".vibe" / "cache" / skill_name)
|