JOJO
f661dc6456
- Split server/chat.py, server/status.py, server/tasks.py into sub-packages
- Split utils/context_manager.py, utils/conversation_manager.py into mixin packages
- Split modules/file_manager.py, persistent_terminal.py, terminal_ops.py, mcp_client_manager.py into packages
- Split core/main_terminal_parts/context.py into base + mixins
- Split static/src/app/methods/{ui,message,upload,taskPolling,conversation} into sub-modules
- Fix flattened method exports, dynamic import paths, missing cross-module imports
- Add missing permission/network/execution mode constants and _PERMISSION_MODE_LABEL
- Update AGENTS.md, CLAUDE.md and project memories to reflect new structure
346 lines
13 KiB
Python
346 lines
13 KiB
Python
from __future__ import annotations
|
|
from server.status import status_bp
|
|
from server.status.base import (
|
|
_active_task_counts,
|
|
_close_terminal_for_key,
|
|
_is_host_mode_request,
|
|
)
|
|
import time
|
|
import re
|
|
import os
|
|
import json
|
|
import shutil
|
|
import subprocess
|
|
import sys
|
|
import tempfile
|
|
import plistlib
|
|
from pathlib import Path
|
|
from flask import Blueprint, jsonify, request, send_file, session
|
|
|
|
from server.auth_helpers import api_login_required, resolve_admin_policy
|
|
from server.context import with_terminal, attach_user_broadcast
|
|
from server.state import (
|
|
PROJECT_STORAGE_CACHE,
|
|
PROJECT_STORAGE_CACHE_TTL_SECONDS,
|
|
PROJECT_MAX_STORAGE_MB,
|
|
container_manager,
|
|
user_manager,
|
|
)
|
|
from config import AGENT_VERSION, TERMINAL_SANDBOX_MODE
|
|
from modules.host_workspace_manager import (
|
|
create_host_workspace,
|
|
delete_host_workspace,
|
|
load_host_workspace_catalog,
|
|
rename_host_workspace,
|
|
resolve_host_workspace,
|
|
set_default_host_workspace,
|
|
)
|
|
from utils.host_workspace_debug import write_host_workspace_debug
|
|
from server.utils_common import log_conn_diag
|
|
import server.state as state
|
|
def _build_host_workspaces_payload(catalog: dict, current_id: str) -> list[dict]:
|
|
running_by_workspace = _active_task_counts("host")
|
|
workspaces = []
|
|
for item in catalog.get("workspaces") or []:
|
|
ws_id = item.get("workspace_id")
|
|
workspaces.append({
|
|
"workspace_id": ws_id,
|
|
"label": item.get("label"),
|
|
"path": item.get("path"),
|
|
"is_current": ws_id == current_id,
|
|
"running_task_count": int(running_by_workspace.get(ws_id, 0)),
|
|
})
|
|
return workspaces
|
|
|
|
@status_bp.route('/api/host/workspaces')
|
|
@api_login_required
|
|
def list_host_workspaces():
|
|
if not _is_host_mode_request():
|
|
return jsonify({"success": False, "error": "仅宿主机模式可用"}), 403
|
|
|
|
catalog, current = resolve_host_workspace(session.get("host_workspace_id"))
|
|
current_id = current.get("workspace_id")
|
|
session['host_workspace_id'] = current_id
|
|
session['workspace_id'] = current_id
|
|
|
|
workspaces = _build_host_workspaces_payload(catalog, current_id)
|
|
|
|
return jsonify({
|
|
"success": True,
|
|
"data": {
|
|
"source_path": catalog.get("source_path"),
|
|
"default_workspace_id": catalog.get("default_workspace_id"),
|
|
"current_workspace_id": current_id,
|
|
"workspaces": workspaces,
|
|
}
|
|
})
|
|
|
|
@status_bp.route('/api/host/workspaces/create', methods=['GET', 'POST'])
|
|
@api_login_required
|
|
def create_host_workspace_api():
|
|
if not _is_host_mode_request():
|
|
return jsonify({"success": False, "error": "仅宿主机模式可用"}), 403
|
|
|
|
payload = request.get_json(silent=True) if request.method != "GET" else None
|
|
workspace_path = (
|
|
request.args.get("path")
|
|
or (payload or {}).get("path")
|
|
or ""
|
|
).strip()
|
|
label = (
|
|
request.args.get("label")
|
|
or (payload or {}).get("label")
|
|
or ""
|
|
).strip()
|
|
set_default_raw = (
|
|
request.args.get("set_default")
|
|
or (payload or {}).get("set_default")
|
|
or ""
|
|
)
|
|
set_default = str(set_default_raw).lower() in {"1", "true", "yes", "on"}
|
|
|
|
if not workspace_path:
|
|
return jsonify({"success": False, "error": "缺少 path"}), 400
|
|
|
|
try:
|
|
result = create_host_workspace(workspace_path, label=label, set_default=set_default)
|
|
except Exception as exc:
|
|
return jsonify({"success": False, "error": str(exc)}), 500
|
|
|
|
catalog = result.get("catalog") or {}
|
|
workspaces = []
|
|
current_id = (session.get("host_workspace_id") or "").strip()
|
|
for item in catalog.get("workspaces") or []:
|
|
workspaces.append({
|
|
"workspace_id": item.get("workspace_id"),
|
|
"label": item.get("label"),
|
|
"path": item.get("path"),
|
|
"is_current": item.get("workspace_id") == current_id,
|
|
})
|
|
|
|
return jsonify({
|
|
"success": True,
|
|
"data": {
|
|
"created": bool(result.get("created")),
|
|
"workspace": result.get("workspace") or {},
|
|
"source_path": catalog.get("source_path"),
|
|
"default_workspace_id": catalog.get("default_workspace_id"),
|
|
"current_workspace_id": current_id,
|
|
"workspaces": workspaces,
|
|
}
|
|
})
|
|
|
|
@status_bp.route('/api/host/workspaces/delete', methods=['POST'])
|
|
@api_login_required
|
|
def delete_host_workspace_api():
|
|
if not _is_host_mode_request():
|
|
return jsonify({"success": False, "error": "仅宿主机模式可用"}), 403
|
|
|
|
payload = request.get_json(silent=True) or {}
|
|
workspace_id = (payload.get("workspace_id") or "").strip()
|
|
if not workspace_id:
|
|
return jsonify({"success": False, "error": "缺少 workspace_id"}), 400
|
|
if _active_task_counts("host").get(workspace_id):
|
|
return jsonify({"success": False, "error": "该工作区有运行中的任务，暂不能删除"}), 409
|
|
try:
|
|
_close_terminal_for_key(f"host::{workspace_id}")
|
|
container_manager.release_container(f"host::{workspace_id}", reason="delete_host_workspace")
|
|
result = delete_host_workspace(workspace_id)
|
|
except ValueError as exc:
|
|
return jsonify({"success": False, "error": str(exc)}), 400
|
|
except Exception as exc:
|
|
return jsonify({"success": False, "error": str(exc)}), 500
|
|
|
|
catalog = result.get("catalog") or load_host_workspace_catalog()
|
|
current_id = (session.get("host_workspace_id") or "").strip()
|
|
if current_id == workspace_id or not any(
|
|
item.get("workspace_id") == current_id for item in (catalog.get("workspaces") or [])
|
|
):
|
|
_, current = resolve_host_workspace(catalog.get("default_workspace_id"))
|
|
current_id = current.get("workspace_id") or catalog.get("default_workspace_id") or "default"
|
|
current_path = str(Path(current.get("path") or "").expanduser().resolve())
|
|
session["host_workspace_id"] = current_id
|
|
session["workspace_id"] = current_id
|
|
with state.HOST_ACTIVE_WORKSPACE_LOCK:
|
|
state.HOST_ACTIVE_WORKSPACE_ID = current_id
|
|
state.HOST_ACTIVE_WORKSPACE_PATH = current_path
|
|
state.HOST_ACTIVE_WORKSPACE_VERSION += 1
|
|
return jsonify({
|
|
"success": True,
|
|
"data": {
|
|
"deleted_workspace_id": workspace_id,
|
|
"source_path": catalog.get("source_path"),
|
|
"default_workspace_id": catalog.get("default_workspace_id"),
|
|
"current_workspace_id": current_id,
|
|
"workspaces": _build_host_workspaces_payload(catalog, current_id),
|
|
}
|
|
})
|
|
|
|
@status_bp.route('/api/host/workspaces/rename', methods=['POST'])
|
|
@api_login_required
|
|
def rename_host_workspace_api():
|
|
if not _is_host_mode_request():
|
|
return jsonify({"success": False, "error": "仅宿主机模式可用"}), 403
|
|
|
|
payload = request.get_json(silent=True) or {}
|
|
workspace_id = (payload.get("workspace_id") or "").strip()
|
|
label = (payload.get("label") or payload.get("name") or "").strip()
|
|
if not workspace_id:
|
|
return jsonify({"success": False, "error": "缺少 workspace_id"}), 400
|
|
if not label:
|
|
return jsonify({"success": False, "error": "工作区名称不能为空"}), 400
|
|
try:
|
|
result = rename_host_workspace(workspace_id, label)
|
|
except ValueError as exc:
|
|
return jsonify({"success": False, "error": str(exc)}), 400
|
|
except Exception as exc:
|
|
return jsonify({"success": False, "error": str(exc)}), 500
|
|
catalog = result.get("catalog") or load_host_workspace_catalog()
|
|
current_id = (session.get("host_workspace_id") or "").strip()
|
|
return jsonify({
|
|
"success": True,
|
|
"data": {
|
|
"workspace": result.get("workspace") or {},
|
|
"source_path": catalog.get("source_path"),
|
|
"default_workspace_id": catalog.get("default_workspace_id"),
|
|
"current_workspace_id": current_id,
|
|
"workspaces": _build_host_workspaces_payload(catalog, current_id),
|
|
}
|
|
})
|
|
|
|
@status_bp.route('/api/host/workspaces/set-default', methods=['POST'])
|
|
@api_login_required
|
|
def set_default_host_workspace_api():
|
|
if not _is_host_mode_request():
|
|
return jsonify({"success": False, "error": "仅宿主机模式可用"}), 403
|
|
|
|
payload = request.get_json(silent=True) or {}
|
|
workspace_id = (payload.get("workspace_id") or "").strip()
|
|
if not workspace_id:
|
|
return jsonify({"success": False, "error": "缺少 workspace_id"}), 400
|
|
try:
|
|
result = set_default_host_workspace(workspace_id)
|
|
except ValueError as exc:
|
|
return jsonify({"success": False, "error": str(exc)}), 400
|
|
except Exception as exc:
|
|
return jsonify({"success": False, "error": str(exc)}), 500
|
|
|
|
catalog = result.get("catalog") or load_host_workspace_catalog()
|
|
current_id = (session.get("host_workspace_id") or "").strip()
|
|
return jsonify({
|
|
"success": True,
|
|
"data": {
|
|
"default_workspace_id": catalog.get("default_workspace_id"),
|
|
"current_workspace_id": current_id,
|
|
"workspaces": _build_host_workspaces_payload(catalog, current_id),
|
|
}
|
|
})
|
|
|
|
@status_bp.route('/api/host/workspaces/select', methods=['GET', 'POST'])
|
|
@api_login_required
|
|
def select_host_workspace():
|
|
if not _is_host_mode_request():
|
|
return jsonify({"success": False, "error": "仅宿主机模式可用"}), 403
|
|
|
|
payload = request.get_json(silent=True) if request.method != "GET" else None
|
|
workspace_id = (
|
|
request.args.get("workspace_id")
|
|
or (payload or {}).get("workspace_id")
|
|
or ""
|
|
).strip()
|
|
if not workspace_id:
|
|
return jsonify({"success": False, "error": "缺少 workspace_id"}), 400
|
|
write_host_workspace_debug(
|
|
"status.select_host_workspace.request",
|
|
workspace_id=workspace_id,
|
|
current_session_workspace_id=session.get("host_workspace_id"),
|
|
)
|
|
|
|
catalog = load_host_workspace_catalog()
|
|
current_id = (session.get("host_workspace_id") or "").strip()
|
|
target = next(
|
|
(item for item in (catalog.get("workspaces") or []) if item.get("workspace_id") == workspace_id),
|
|
None,
|
|
)
|
|
if not target:
|
|
return jsonify({"success": False, "error": "workspace_id 不存在"}), 404
|
|
target_path = str(Path(target.get("path") or "").expanduser().resolve())
|
|
write_host_workspace_debug(
|
|
"status.select_host_workspace.target",
|
|
workspace_id=workspace_id,
|
|
target_path=target_path,
|
|
current_workspace_id=current_id,
|
|
)
|
|
|
|
previous_workspace_id = session.get("workspace_id")
|
|
with state.HOST_ACTIVE_WORKSPACE_LOCK:
|
|
state.HOST_ACTIVE_WORKSPACE_ID = workspace_id
|
|
state.HOST_ACTIVE_WORKSPACE_PATH = target_path
|
|
state.HOST_ACTIVE_WORKSPACE_VERSION += 1
|
|
session['host_workspace_id'] = workspace_id
|
|
session['workspace_id'] = workspace_id
|
|
|
|
try:
|
|
container_handle = state.container_manager.ensure_container(
|
|
"host",
|
|
target_path,
|
|
container_key=f"host::{workspace_id}",
|
|
preferred_mode="host",
|
|
)
|
|
except RuntimeError as exc:
|
|
if current_id:
|
|
session['host_workspace_id'] = current_id
|
|
if previous_workspace_id is not None:
|
|
session['workspace_id'] = previous_workspace_id
|
|
elif current_id:
|
|
session['workspace_id'] = current_id
|
|
write_host_workspace_debug(
|
|
"status.select_host_workspace.ensure_container_failed",
|
|
workspace_id=workspace_id,
|
|
target_path=target_path,
|
|
error=str(exc),
|
|
)
|
|
return jsonify({"success": False, "error": str(exc)}), 503
|
|
|
|
host_terminal = state.user_terminals.get(f"host::{workspace_id}")
|
|
if host_terminal:
|
|
try:
|
|
host_terminal.update_container_session(container_handle)
|
|
attach_user_broadcast(host_terminal, "host")
|
|
host_terminal.username = "host"
|
|
host_terminal.user_role = "admin"
|
|
write_host_workspace_debug(
|
|
"status.select_host_workspace.updated_terminal",
|
|
terminal_id=id(host_terminal),
|
|
workspace_id=workspace_id,
|
|
project_path=str(getattr(host_terminal, "project_path", "")),
|
|
context_project_path=str(getattr(getattr(host_terminal, "context_manager", None), "project_path", "")),
|
|
current_conversation_id=getattr(getattr(host_terminal, "context_manager", None), "current_conversation_id", None),
|
|
)
|
|
except Exception:
|
|
try:
|
|
if getattr(host_terminal, "terminal_manager", None):
|
|
host_terminal.terminal_manager.close_all()
|
|
except Exception:
|
|
pass
|
|
state.user_terminals.pop(f"host::{workspace_id}", None)
|
|
write_host_workspace_debug(
|
|
"status.select_host_workspace.drop_terminal_after_error",
|
|
workspace_id=workspace_id,
|
|
)
|
|
|
|
write_host_workspace_debug(
|
|
"status.select_host_workspace.success",
|
|
workspace_id=workspace_id,
|
|
target_path=target_path,
|
|
)
|
|
return jsonify({
|
|
"success": True,
|
|
"data": {
|
|
"current_workspace_id": workspace_id,
|
|
"project_path": target_path,
|
|
"default_workspace_id": catalog.get("default_workspace_id"),
|
|
"reloaded": current_id != workspace_id,
|
|
}
|
|
})
|