Import AITURK IDE 1.0.0-beta.1 from Hermes 63279301; preserve MIT license

tools/project_tools.py

@@ -0,0 +1,181 @@
+#!/usr/bin/env python3
+"""Project tools — the agent's INTENTIONAL handle on first-class Projects.
+
+Projects (per-profile ``projects.db``) are the named workspaces the desktop
+sidebar groups sessions into. Creating / switching a project is a deliberate act
+expressed as explicit tools — never a side effect of a terminal ``cd``.
+
+Exposed only on GUI sessions: the tools live in the `project` toolset (kept off
+``_HERMES_CORE_TOOLS``) which the desktop/TUI gateway folds into its resolved
+toolsets, so no CLI/messaging/cron schema carries them. The GUI also wires
+``set_project_workspace_callback`` so a create/switch re-anchors the live
+session's cwd and the sidebar follows the move; the DB write is the durable part.
+"""
+
+import json
+import os
+from typing import Callable, Optional
+
+from tools.registry import registry
+
+# Set by the GUI gateway (tui_gateway) at session wiring. Receives
+# ``(task_id, primary_path, project_name)`` and re-anchors that session's
+# workspace + refreshes the sidebar. ``None`` in CLI / messaging contexts — the
+# DB write still happens; there's just no live GUI session to move.
+_workspace_callback: Optional[Callable[[str, str, str], None]] = None
+
+
+def set_project_workspace_callback(fn: Optional[Callable[[str, str, str], None]]) -> None:
+    global _workspace_callback
+    _workspace_callback = fn
+
+
+def _primary_path(proj) -> Optional[str]:
+    if getattr(proj, "primary_path", None):
+        return proj.primary_path
+    for folder in proj.folders:
+        if folder.is_primary:
+            return folder.path
+    return proj.folders[0].path if proj.folders else None
+
+
+def _apply_workspace(task_id: Optional[str], path: Optional[str], name: str) -> None:
+    cb = _workspace_callback
+    if cb and task_id and path:
+        try:
+            cb(task_id, path, name)
+        except Exception:
+            pass
+
+
+def _resolve(conn, token: str):
+    from hermes_cli import projects_db as pdb
+
+    token = (token or "").strip()
+    if not token:
+        return None
+    projects = pdb.list_projects(conn, include_archived=True)
+    # Exact id / slug / name first, then case-insensitive slug / name.
+    for proj in projects:
+        if token in (proj.id, proj.slug) or proj.name == token:
+            return proj
+    low = token.lower()
+    for proj in projects:
+        if proj.slug.lower() == low or proj.name.lower() == low:
+            return proj
+    return None
+
+
+def project_list(task_id: Optional[str] = None) -> str:
+    from hermes_cli import projects_db as pdb
+
+    with pdb.connect_closing() as conn:
+        active = pdb.get_active_id(conn)
+        projects = pdb.list_projects(conn)
+
+    return json.dumps({
+        "active_id": active,
+        "projects": [
+            {
+                "id": p.id,
+                "slug": p.slug,
+                "name": p.name,
+                "primary_path": _primary_path(p),
+                "active": p.id == active,
+            }
+            for p in projects
+        ],
+    })
+
+
+def project_create(name: str, path: Optional[str] = None, task_id: Optional[str] = None) -> str:
+    name = (name or "").strip()
+    if not name:
+        return json.dumps({"success": False, "error": "name is required"})
+
+    from hermes_cli import projects_db as pdb
+
+    folder = (path or "").strip()
+    if folder:
+        folder = os.path.abspath(os.path.expanduser(folder))
+
+    try:
+        with pdb.connect_closing() as conn:
+            existing = pdb.find_by_primary_path(conn, folder) if folder else None
+            if existing is not None:
+                # Idempotent create: the folder already belongs to a project.
+                # Re-activating it beats minting a duplicate — duplicated
+                # projects render N identical sidebar subtrees (#75820).
+                pdb.set_active(conn, existing.id)
+                proj = existing
+            else:
+                pid = pdb.create_project(conn, name=name, folders=[folder] if folder else [], primary_path=folder or None)
+                pdb.set_active(conn, pid)
+                proj = pdb.get_project(conn, pid)
+    except ValueError as exc:
+        return json.dumps({"success": False, "error": str(exc)})
+
+    if proj is None:
+        return json.dumps({"success": False, "error": "project vanished after create"})
+
+    primary = _primary_path(proj)
+    _apply_workspace(task_id, primary, proj.name)
+
+    return json.dumps({"success": True, "id": proj.id, "slug": proj.slug, "name": proj.name, "primary_path": primary})
+
+
+def project_switch(project: str, task_id: Optional[str] = None) -> str:
+    from hermes_cli import projects_db as pdb
+
+    with pdb.connect_closing() as conn:
+        proj = _resolve(conn, project)
+        if proj is None:
+            return json.dumps({"success": False, "error": f"no project matching '{project}'"})
+        pdb.set_active(conn, proj.id)
+
+    primary = _primary_path(proj)
+    _apply_workspace(task_id, primary, proj.name)
+
+    return json.dumps({"success": True, "id": proj.id, "slug": proj.slug, "name": proj.name, "primary_path": primary})
+
+
+def _handle_project(args, **kw):
+    action = (args.get("action") or "").strip()
+    tid = kw.get("task_id")
+    if action == "list":
+        return project_list(task_id=tid)
+    if action == "create":
+        return project_create(name=args.get("name", ""), path=args.get("path"), task_id=tid)
+    if action == "switch":
+        return project_switch(project=args.get("name", ""), task_id=tid)
+    return json.dumps({"success": False, "error": "action must be one of: create, switch, list."})
+
+
+# Consolidated (#95681, maintainer-directed): project_list/create/switch each
+# re-taught "desktop Projects (named workspaces)"; one action enum says it
+# once (244 -> ~145 tok).
+registry.register(
+    name="desktop_project",
+    toolset="project",
+    schema={
+        "name": "desktop_project",
+        "description": (
+            "Create or switch desktop Projects (named workspaces). create: one and switch "
+            "this chat into it — pass path to anchor it to a repo/folder (the "
+            "chat's workspace moves there, the sidebar follows). switch: move "
+            "this chat into an existing project by name/slug/id — the "
+            "intentional way to move the session, not `cd`. list: all "
+            "projects + which is active."
+        ),
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "action": {"type": "string", "enum": ["create", "switch", "list"]},
+                "name": {"type": "string", "description": "create: human name. switch: name, slug, or id."},
+                "path": {"type": "string", "description": "create: repo/folder to anchor to."},
+            },
+            "required": ["action"],
+        },
+    },
+    handler=_handle_project,
+)