From 359caca6729f7ac76e65b96a4a192dd7e3c75502 Mon Sep 17 00:00:00 2001
From: JOJO <1498581755@qq.com>
Date: Sat, 4 Jul 2026 22:58:58 +0800
Subject: [PATCH] feat(input): support file mentions
---
server/files.py | 144 +-
static/src/App.vue | 1 +
static/src/components/chat/ChatArea.vue | 36 +-
static/src/components/input/FileAtMenu.vue | 234 +++
static/src/components/input/InputComposer.vue | 570 +++++-
.../styles/components/chat/_chat-area.scss | 3 +-
.../styles/components/input/_composer.scss | 1773 +++++++++--------
7 files changed, 1785 insertions(+), 976 deletions(-)
create mode 100644 static/src/components/input/FileAtMenu.vue
diff --git a/server/files.py b/server/files.py
index f26a254..414b17b 100644
--- a/server/files.py
+++ b/server/files.py
@@ -1,10 +1,11 @@
"""文件与GUI文件管理相关路由。"""
from __future__ import annotations
import os
+import re
import zipfile
from io import BytesIO
from pathlib import Path
-from typing import Dict, Any
+from typing import Dict, Any, List, Optional, Tuple
from flask import Blueprint, jsonify, request, send_file
from werkzeug.utils import secure_filename
@@ -311,4 +312,145 @@ def gui_text_entry(terminal, workspace, username):
return jsonify({"success": False, "error": str(exc)}), 400
+def _score_file_match(path: str, query: str) -> Optional[Tuple[int, ...]]:
+ """为 @文件 搜索计算匹配分数；分数越低越靠前，None 表示不匹配。"""
+ path_lower = path.lower()
+ query_lower = query.lower()
+
+ if path_lower.startswith(query_lower):
+ return (0, len(path))
+
+ query_parts = query_lower.split('/')
+ path_parts = path_lower.split('/')
+
+ if len(query_parts) > 1 and len(path_parts) >= len(query_parts):
+ matched = True
+ for i, q in enumerate(query_parts):
+ p = path_parts[i]
+ if not p.startswith(q) and q not in p:
+ matched = False
+ break
+ if matched:
+ return (1, len(path))
+
+ name = path_parts[-1] if path_parts else path_lower
+ if name.startswith(query_lower):
+ return (2, len(path))
+
+ if query_lower in name:
+ return (3, len(path))
+
+ if query_lower in path_lower:
+ return (4, len(path))
+
+ return None
+
+
+def _scan_project_entries(project_path: Path, max_depth: int = 6) -> Tuple[List[Dict[str, Any]], List[Dict[str, Any]]]:
+ """扫描项目目录，返回文件和文件夹列表（包含隐藏目录）。"""
+ files: List[Dict[str, Any]] = []
+ folders: List[Dict[str, Any]] = []
+
+ for root, dirs, filenames in os.walk(project_path):
+ rel_root = Path(root).relative_to(project_path)
+ level = len(rel_root.parts) if str(rel_root) != '.' else 0
+ if level > max_depth:
+ del dirs[:]
+ continue
+
+ for d in list(dirs):
+ dir_path = rel_root / d if str(rel_root) != '.' else Path(d)
+ folders.append({
+ "name": d,
+ "path": str(dir_path).replace('\\', '/'),
+ "type": "directory"
+ })
+
+ for f in filenames:
+ file_path = rel_root / f if str(rel_root) != '.' else Path(f)
+ ext = Path(f).suffix.lower()
+ files.append({
+ "name": f,
+ "path": str(file_path).replace('\\', '/'),
+ "type": "file",
+ "extension": ext
+ })
+
+ return files, folders
+
+
+def _scan_root_entries(project_path: Path) -> Tuple[List[Dict[str, Any]], List[Dict[str, Any]]]:
+ """只扫描项目根目录的直接子项（包含隐藏目录）。"""
+ files: List[Dict[str, Any]] = []
+ folders: List[Dict[str, Any]] = []
+ if not project_path.exists() or not project_path.is_dir():
+ return files, folders
+ for entry in sorted(project_path.iterdir(), key=lambda p: (p.is_file(), p.name.lower())):
+ if entry.is_dir():
+ folders.append({
+ "name": entry.name,
+ "path": entry.name,
+ "type": "directory"
+ })
+ elif entry.is_file():
+ files.append({
+ "name": entry.name,
+ "path": entry.name,
+ "type": "file",
+ "extension": entry.suffix.lower()
+ })
+ return files, folders
+
+
+@files_bp.route('/api/project/files/search', methods=['GET'])
+@api_login_required
+@with_terminal
+def search_project_files(terminal, workspace, username):
+ """为 @文件 功能提供项目内文件/文件夹搜索（包含隐藏目录）。"""
+ policy = resolve_admin_policy(get_current_user_record())
+ if policy.get("ui_blocks", {}).get("collapse_workspace") or policy.get("ui_blocks", {}).get("block_file_manager"):
+ return jsonify({"success": False, "error": "文件浏览已被管理员禁用"}), 403
+
+ query = str(request.args.get('q') or '').strip()
+
+ try:
+ project_path = Path(getattr(workspace, 'project_path', '') or '').expanduser().resolve()
+ if not project_path.exists():
+ return jsonify({"success": False, "error": "项目路径不存在"}), 400
+ except Exception as exc:
+ return jsonify({"success": False, "error": f"项目路径无效: {exc}"}), 400
+
+ try:
+ files, folders = _scan_project_entries(project_path, max_depth=6)
+ except Exception as exc:
+ return jsonify({"success": False, "error": f"扫描项目失败: {exc}"}), 500
+
+ if not query:
+ root_files, root_folders = _scan_root_entries(project_path)
+ root_items = root_folders + root_files
+ return jsonify({
+ "success": True,
+ "data": {
+ "items": root_items[:50],
+ "total": len(root_items)
+ }
+ })
+
+ scored: List[Tuple[Tuple[int, ...], Dict[str, Any]]] = []
+ for entry in files + folders:
+ score = _score_file_match(entry["path"], query)
+ if score is not None:
+ scored.append((score, entry))
+
+ scored.sort(key=lambda x: (x[0], x[1]["path"].lower()))
+ limit = max(10, min(100, int(request.args.get('limit') or 50)))
+ return jsonify({
+ "success": True,
+ "data": {
+ "items": [entry for _, entry in scored[:limit]],
+ "total": len(scored)
+ }
+ })
+
+
__all__ = ["files_bp"]
diff --git a/static/src/App.vue b/static/src/App.vue
index 3ae5834..c5bcfbc 100644
--- a/static/src/App.vue
+++ b/static/src/App.vue
@@ -295,6 +295,7 @@
:block-tool-toggle="policyUiBlocks.block_tool_toggle"
:block-realtime-terminal="policyUiBlocks.block_realtime_terminal"
:terminal-count="terminalSessions ? Object.keys(terminalSessions).length : 0"
+ :host-mode="versioningHostMode"
:block-focus-panel="policyUiBlocks.block_focus_panel"
:block-token-panel="policyUiBlocks.block_token_panel"
:block-compress-conversation="policyUiBlocks.block_compress_conversation"
diff --git a/static/src/components/chat/ChatArea.vue b/static/src/components/chat/ChatArea.vue
index fdf408e..181e476 100644
--- a/static/src/components/chat/ChatArea.vue
+++ b/static/src/components/chat/ChatArea.vue
@@ -933,16 +933,35 @@ const escapeUserHtml = (value: string): string =>
.replace(/'/g, ''');
const USER_SKILL_LINK_RE = /\[\$([^\]\n]+)\]\(([^)\n]*\/\.agents\/skills\/[^)\n]+\/SKILL\.md)\)/g;
+const USER_FILE_LINK_RE = /\[([^\]\n]+)\]\(file:\/\/([^)\n]+)\)/g;
+const SUB_AGENT_DONE_LABEL_RE = /^子智能体\d+\s*任务完成$/;
+const SUB_AGENT_DONE_PREFIX_RE = /^(?:✅\s*)?子智能体\s*#?\s*(\d+)\s*任务摘要[:：]/;
+const BG_RUN_COMMAND_DONE_LABEL_RE = /^(?:\[)?后台\s*run_command\s*完成(?:\])?$/;
+const RENDERABLE_ACTION_TYPES = new Set([
+ 'thinking',
+ 'text',
+ 'tool',
+ 'append',
+ 'append_payload',
+ 'system'
+]);
function renderUserMessageContent(content: string): string {
const source = String(content || '');
- USER_SKILL_LINK_RE.lastIndex = 0;
+ const combinedRe = new RegExp(
+ `(?:${USER_SKILL_LINK_RE.source})|(?:${USER_FILE_LINK_RE.source})`,
+ 'g'
+ );
let html = '';
let cursor = 0;
let match: RegExpExecArray | null;
- while ((match = USER_SKILL_LINK_RE.exec(source))) {
+ while ((match = combinedRe.exec(source))) {
html += escapeUserHtml(source.slice(cursor, match.index));
- html += ` ${escapeUserHtml(match[1] || '')} `;
+ if (match[1] !== undefined) {
+ html += ` ${escapeUserHtml(match[1] || '')} `;
+ } else if (match[3] !== undefined) {
+ html += ` ${escapeUserHtml(match[3] || '')} `;
+ }
cursor = match.index + match[0].length;
}
html += escapeUserHtml(source.slice(cursor));
@@ -1040,17 +1059,6 @@ watch(
statusAvatarActiveIndex.value = 0;
}
);
-const SUB_AGENT_DONE_LABEL_RE = /^子智能体\d+\s*任务完成$/;
-const SUB_AGENT_DONE_PREFIX_RE = /^(?:✅\s*)?子智能体\s*#?\s*(\d+)\s*任务摘要[:：]/;
-const BG_RUN_COMMAND_DONE_LABEL_RE = /^(?:\[)?后台\s*run_command\s*完成(?:\])?$/;
-const RENDERABLE_ACTION_TYPES = new Set([
- 'thinking',
- 'text',
- 'tool',
- 'append',
- 'append_payload',
- 'system'
-]);
const nowMs = ref(Date.now());
let timerHandle: number | null = null;
const debugLoggedSystemActionKeys = new Set();
diff --git a/static/src/components/input/FileAtMenu.vue b/static/src/components/input/FileAtMenu.vue
new file mode 100644
index 0000000..eabf456
--- /dev/null
+++ b/static/src/components/input/FileAtMenu.vue
@@ -0,0 +1,234 @@
+
+
+
+
+
+
+
+
+
diff --git a/static/src/components/input/InputComposer.vue b/static/src/components/input/InputComposer.vue
index 3154f29..adc6059 100644
--- a/static/src/components/input/InputComposer.vue
+++ b/static/src/components/input/InputComposer.vue
@@ -47,11 +47,7 @@
role="listbox"
:aria-label="slashMenuAriaLabel"
>
-
+
-
+
@@ -108,7 +112,9 @@