From f24bd38ec78d385ecfd7242fe2bbedbd98f86cf7 Mon Sep 17 00:00:00 2001
From: Roe-xin <xiang200308037300@qq.com>
Date: Fri, 6 Mar 2026 15:40:59 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E4=BC=98=E5=8C=96=E4=B8=8A=E4=B8=8B?=
 =?UTF-8?q?=E6=96=87=E9=A1=B9=E6=98=BE=E7=A4=BA=E5=92=8C=E8=AF=86=E5=88=AB?=
 =?UTF-8?q?=E9=80=BB=E8=BE=91?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

   - 支持显示所有类型的上下文项（文件和代码片段）
   - 增强路径识别，支持代码片段格式（文件名:行号-行号）
---
 src/views/inputArea.ts   | 9 ++++-----
 src/views/messageArea.ts | 4 ++--
 2 files changed, 6 insertions(+), 7 deletions(-)

diff --git a/src/views/inputArea.ts b/src/views/inputArea.ts
index b26fa71..e99985e 100644
--- a/src/views/inputArea.ts
+++ b/src/views/inputArea.ts
@@ -432,15 +432,14 @@ export function getInputAreaScript(): string {
       // 获取上下文项
       const contextItems = window.getContextItems ? window.getContextItems() : [];
 
-      // 构建显示消息：如果有上下文文件，添加文件路径前缀
+      // 构建显示消息：如果有上下文项，添加路径前缀
       let displayText = text;
       if (contextItems.length > 0) {
-        const filePaths = contextItems
-          .filter(item => item.type === 'file')
+        const contextPaths = contextItems
           .map(item => item.displayPath || item.path)
           .join(' ');
-        if (filePaths) {
-          displayText = filePaths + ' ' + text;
+        if (contextPaths) {
+          displayText = contextPaths + ' ' + text;
         }
       }
 
diff --git a/src/views/messageArea.ts b/src/views/messageArea.ts
index c27739a..f6e2afd 100644
--- a/src/views/messageArea.ts
+++ b/src/views/messageArea.ts
@@ -863,8 +863,8 @@ export function getMessageAreaScript(): string {
           const textParts = [];
 
           parts.forEach(part => {
-            // 判断是否为文件路径：包含路径分隔符或文件扩展名
-            if (part.includes('/') || part.includes('\\\\') || /\\.[a-zA-Z0-9]+$/.test(part)) {
+            // 判断是否为文件路径或代码片段：包含路径分隔符、文件扩展名或代码片段格式（文件名:行号-行号）
+            if (part.includes('/') || part.includes('\\\\') || /\\.[a-zA-Z0-9]+$/.test(part) || /:[0-9]+-[0-9]+$/.test(part)) {
               filePaths.push(part);
             } else {
               textParts.push(part);