feat(workspace): implement WorkspaceManager and path resolution
This commit is contained in:
@@ -72,6 +72,13 @@ def get_global_personas_path() -> Path:
|
||||
def get_project_personas_path(project_root: Path) -> Path:
|
||||
return project_root / "project_personas.toml"
|
||||
|
||||
def get_global_workspace_profiles_path() -> Path:
|
||||
root_dir = Path(__file__).resolve().parent.parent
|
||||
return Path(os.environ.get("SLOP_GLOBAL_WORKSPACE_PROFILES", root_dir / "workspace_profiles.toml"))
|
||||
|
||||
def get_project_workspace_profiles_path(project_root: Path) -> Path:
|
||||
return project_root / ".ai" / "workspace_profiles.toml"
|
||||
|
||||
def _resolve_path(env_var: str, config_key: str, default: str) -> Path:
|
||||
root_dir = Path(__file__).resolve().parent.parent
|
||||
p = None
|
||||
|
||||
@@ -0,0 +1,72 @@
|
||||
import tomllib
|
||||
import tomli_w
|
||||
from pathlib import Path
|
||||
from typing import Dict, Any, Optional, Union
|
||||
from src.models import WorkspaceProfile
|
||||
from src import paths
|
||||
|
||||
class WorkspaceManager:
|
||||
"""Manages Workspace profiles across global and project-specific files."""
|
||||
|
||||
def __init__(self, project_root: Optional[Union[str, Path]] = None):
|
||||
if project_root:
|
||||
self.project_root = Path(project_root).resolve()
|
||||
else:
|
||||
self.project_root = None
|
||||
|
||||
def _get_path(self, scope: str) -> Path:
|
||||
if scope == "global":
|
||||
return paths.get_global_workspace_profiles_path()
|
||||
elif scope == "project":
|
||||
if not self.project_root:
|
||||
raise ValueError("Project root is not set, cannot resolve project scope.")
|
||||
return paths.get_project_workspace_profiles_path(self.project_root)
|
||||
else:
|
||||
raise ValueError("Invalid scope, must be 'global' or 'project'")
|
||||
|
||||
def load_all_profiles(self) -> Dict[str, WorkspaceProfile]:
|
||||
"""Merges global and project profiles into a single dictionary."""
|
||||
profiles = {}
|
||||
|
||||
global_path = paths.get_global_workspace_profiles_path()
|
||||
global_data = self._load_file(global_path)
|
||||
for name, data in global_data.get("profiles", {}).items():
|
||||
profiles[name] = WorkspaceProfile.from_dict(name, data)
|
||||
|
||||
if self.project_root:
|
||||
project_path = paths.get_project_workspace_profiles_path(self.project_root)
|
||||
project_data = self._load_file(project_path)
|
||||
for name, data in project_data.get("profiles", {}).items():
|
||||
profiles[name] = WorkspaceProfile.from_dict(name, data)
|
||||
|
||||
return profiles
|
||||
|
||||
def save_profile(self, profile: WorkspaceProfile, scope: str = "project") -> None:
|
||||
path = self._get_path(scope)
|
||||
data = self._load_file(path)
|
||||
if "profiles" not in data:
|
||||
data["profiles"] = {}
|
||||
|
||||
data["profiles"][profile.name] = profile.to_dict()
|
||||
self._save_file(path, data)
|
||||
|
||||
def delete_profile(self, name: str, scope: str = "project") -> None:
|
||||
path = self._get_path(scope)
|
||||
data = self._load_file(path)
|
||||
if "profiles" in data and name in data["profiles"]:
|
||||
del data["profiles"][name]
|
||||
self._save_file(path, data)
|
||||
|
||||
def _load_file(self, path: Path) -> Dict[str, Any]:
|
||||
if not path.exists():
|
||||
return {}
|
||||
try:
|
||||
with open(path, "rb") as f:
|
||||
return tomllib.load(f)
|
||||
except Exception:
|
||||
return {}
|
||||
|
||||
def _save_file(self, path: Path, data: Dict[str, Any]) -> None:
|
||||
path.parent.mkdir(parents=True, exist_ok=True)
|
||||
with open(path, "wb") as f:
|
||||
tomli_w.dump(data, f)
|
||||
Reference in New Issue
Block a user