feat(context): Implement ContextPreset and FileViewPreset infrastructure
This commit is contained in:
+13
-7
@@ -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:
|
||||
"""
|
||||
|
||||
@@ -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", [])
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user