feat(context): Implement ContextPreset and FileViewPreset infrastructure

This commit is contained in:
2026-05-11 13:13:37 -04:00
parent f1a93264ce
commit 78c009fc26
3 changed files with 97 additions and 7 deletions
+13 -7
View File
@@ -436,10 +436,15 @@ class App:
sys.stderr.flush()
if 'context_presets' not in self.controller.project:
self.controller.project['context_presets'] = {}
self.controller.project['context_presets'][name] = {
'files': [f.to_dict() if hasattr(f, 'to_dict') else {'path': str(f)} for f in self.files],
'screenshots': list(self.screenshots)
}
preset_files = []
for f in self.context_files:
path = f.path if hasattr(f, 'path') else str(f)
view_mode = f.view_mode if hasattr(f, 'view_mode') else 'summary'
preset_files.append(models.FileViewPreset(path=path, view_mode=view_mode))
preset = models.ContextPreset(name=name, files=preset_files, screenshots=list(self.screenshots))
self.controller.project['context_presets'][name] = preset.to_dict()
self.controller._save_active_project()
sys.stderr.write(f"[DEBUG] save_context_preset finished. Project keys: {list(self.controller.project.keys())}\n")
sys.stderr.flush()
@@ -450,9 +455,10 @@ class App:
"""
presets = self.controller.project.get('context_presets', {})
if name in presets:
preset = presets[name]
self.files = [models.FileItem.from_dict(f) if isinstance(f, dict) else models.FileItem(path=str(f)) for f in preset.get('files', [])]
self.screenshots = list(preset.get('screenshots', []))
preset_data = presets[name]
preset = models.ContextPreset.from_dict(name, preset_data)
self.context_files = [models.FileItem(path=f.path, view_mode=f.view_mode) for f in preset.files]
self.screenshots = list(preset.screenshots)
def delete_context_preset(self, name: str) -> None:
"""
+39
View File
@@ -907,3 +907,42 @@ def load_mcp_config(path: str) -> MCPConfiguration:
return MCPConfiguration.from_dict(data)
except Exception:
return MCPConfiguration()
@dataclass
class FileViewPreset:
path: str
view_mode: str = "summary"
def to_dict(self) -> dict[str, Any]:
return {
"path": self.path,
"view_mode": self.view_mode
}
@classmethod
def from_dict(cls, data: dict[str, Any]) -> "FileViewPreset":
return cls(
path=data.get("path", ""),
view_mode=data.get("view_mode", "summary")
)
@dataclass
class ContextPreset:
name: str
files: list[FileViewPreset] = field(default_factory=list)
screenshots: list[str] = field(default_factory=list)
def to_dict(self) -> dict[str, Any]:
return {
"files": [f.to_dict() for f in self.files],
"screenshots": self.screenshots
}
@classmethod
def from_dict(cls, name: str, data: dict[str, Any]) -> "ContextPreset":
files_data = data.get("files", [])
return cls(
name=name,
files=[FileViewPreset.from_dict(f) if isinstance(f, dict) else FileViewPreset(path=str(f)) for f in files_data],
screenshots=data.get("screenshots", [])
)