feat(context): Implement ContextPreset and FileViewPreset infrastructure

src/gui_2.py

@@ -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:
   """

src/models.py

@@ -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", [])
+  )

tests/test_context_presets_models.py

@@ -0,0 +1,45 @@
+import pytest
+from src.models import ContextPreset, FileViewPreset
+
+def test_file_view_preset_serialization():
+    p = FileViewPreset(path="test.py", view_mode="skeleton")
+    d = p.to_dict()
+    assert d == {"path": "test.py", "view_mode": "skeleton"}
+    
+    p2 = FileViewPreset.from_dict(d)
+    assert p2.path == "test.py"
+    assert p2.view_mode == "skeleton"
+
+def test_context_preset_serialization():
+    f1 = FileViewPreset(path="a.py", view_mode="full")
+    f2 = FileViewPreset(path="b.py", view_mode="summary")
+    
+    preset = ContextPreset(
+        name="test_preset",
+        files=[f1, f2],
+        screenshots=["shot1.png"]
+    )
+    
+    d = preset.to_dict()
+    assert len(d["files"]) == 2
+    assert d["files"][0]["path"] == "a.py"
+    assert d["screenshots"] == ["shot1.png"]
+    
+    p2 = ContextPreset.from_dict("test_preset", d)
+    assert p2.name == "test_preset"
+    assert len(p2.files) == 2
+    assert p2.files[0].view_mode == "full"
+    assert p2.screenshots == ["shot1.png"]
+
+def test_context_preset_from_dict_legacy():
+    # Test handling of legacy string paths in preset files
+    d = {
+        "files": ["legacy.py", {"path": "new.py", "view_mode": "skeleton"}],
+        "screenshots": []
+    }
+    preset = ContextPreset.from_dict("legacy_test", d)
+    assert len(preset.files) == 2
+    assert preset.files[0].path == "legacy.py"
+    assert preset.files[0].view_mode == "summary" # Default
+    assert preset.files[1].path == "new.py"
+    assert preset.files[1].view_mode == "skeleton"