feat(vscode-ide-companion): import chat chat customEditor to vscode extension folder

This commit is contained in:
yiliang114
2025-11-17 18:53:00 +08:00
parent 0eeffc6875
commit dc40995e70
17 changed files with 2428 additions and 4 deletions

View File

@@ -0,0 +1,340 @@
/**
* @license
* Copyright 2025 Qwen Team
* SPDX-License-Identifier: Apache-2.0
*/
:root {
--vscode-font-family: var(--vscode-font-family);
--vscode-editor-background: var(--vscode-editor-background);
--vscode-editor-foreground: var(--vscode-editor-foreground);
--vscode-input-background: var(--vscode-input-background);
--vscode-input-foreground: var(--vscode-input-foreground);
--vscode-button-background: var(--vscode-button-background);
--vscode-button-foreground: var(--vscode-button-foreground);
--vscode-button-hoverBackground: var(--vscode-button-hoverBackground);
}
* {
margin: 0;
padding: 0;
box-sizing: border-box;
}
body {
font-family: var(--vscode-font-family);
background-color: var(--vscode-editor-background);
color: var(--vscode-editor-foreground);
overflow: hidden;
}
.chat-container {
display: flex;
flex-direction: column;
height: 100vh;
width: 100%;
}
.messages-container {
flex: 1;
overflow-y: auto;
padding: 20px;
display: flex;
flex-direction: column;
gap: 16px;
}
.message {
display: flex;
flex-direction: column;
gap: 4px;
max-width: 80%;
padding: 12px 16px;
border-radius: 8px;
animation: fadeIn 0.2s ease-in;
}
@keyframes fadeIn {
from {
opacity: 0;
transform: translateY(10px);
}
to {
opacity: 1;
transform: translateY(0);
}
}
.message.user {
align-self: flex-end;
background-color: var(--vscode-button-background);
color: var(--vscode-button-foreground);
}
.message.assistant {
align-self: flex-start;
background-color: var(--vscode-input-background);
color: var(--vscode-input-foreground);
border: 1px solid rgba(255, 255, 255, 0.1);
}
.message.streaming {
position: relative;
}
.streaming-indicator {
position: absolute;
right: 12px;
bottom: 12px;
animation: pulse 1.5s ease-in-out infinite;
}
@keyframes pulse {
0%,
100% {
opacity: 1;
}
50% {
opacity: 0.3;
}
}
.message-content {
white-space: pre-wrap;
word-wrap: break-word;
line-height: 1.5;
}
.message-timestamp {
font-size: 11px;
opacity: 0.6;
align-self: flex-end;
}
.input-form {
display: flex;
gap: 8px;
padding: 16px;
background-color: var(--vscode-editor-background);
border-top: 1px solid rgba(255, 255, 255, 0.1);
}
.input-field {
flex: 1;
padding: 10px 12px;
background-color: var(--vscode-input-background);
color: var(--vscode-input-foreground);
border: 1px solid rgba(255, 255, 255, 0.1);
border-radius: 4px;
font-size: 14px;
font-family: var(--vscode-font-family);
outline: none;
}
.input-field:focus {
border-color: var(--vscode-button-background);
}
.input-field:disabled {
opacity: 0.5;
cursor: not-allowed;
}
.send-button {
padding: 10px 20px;
background-color: var(--vscode-button-background);
color: var(--vscode-button-foreground);
border: none;
border-radius: 4px;
font-size: 14px;
font-weight: 500;
cursor: pointer;
transition: background-color 0.2s;
}
.send-button:hover:not(:disabled) {
background-color: var(--vscode-button-hoverBackground);
}
.send-button:disabled {
opacity: 0.5;
cursor: not-allowed;
}
/* Scrollbar styling */
.messages-container::-webkit-scrollbar {
width: 8px;
}
.messages-container::-webkit-scrollbar-track {
background: transparent;
}
.messages-container::-webkit-scrollbar-thumb {
background: rgba(255, 255, 255, 0.2);
border-radius: 4px;
}
.messages-container::-webkit-scrollbar-thumb:hover {
background: rgba(255, 255, 255, 0.3);
}
/* Session selector styles */
.chat-header {
display: flex;
justify-content: flex-end;
padding: 12px 16px;
background-color: var(--vscode-editor-background);
border-bottom: 1px solid rgba(255, 255, 255, 0.1);
}
.session-button {
padding: 6px 12px;
background-color: var(--vscode-button-background);
color: var(--vscode-button-foreground);
border: none;
border-radius: 4px;
font-size: 13px;
cursor: pointer;
transition: background-color 0.2s;
}
.session-button:hover {
background-color: var(--vscode-button-hoverBackground);
}
.session-selector-overlay {
position: fixed;
top: 0;
left: 0;
right: 0;
bottom: 0;
background-color: rgba(0, 0, 0, 0.5);
display: flex;
align-items: center;
justify-content: center;
z-index: 1000;
animation: fadeIn 0.2s ease-in;
}
.session-selector {
background-color: var(--vscode-editor-background);
border: 1px solid rgba(255, 255, 255, 0.2);
border-radius: 8px;
width: 80%;
max-width: 500px;
max-height: 70vh;
display: flex;
flex-direction: column;
box-shadow: 0 4px 12px rgba(0, 0, 0, 0.3);
}
.session-selector-header {
display: flex;
justify-content: space-between;
align-items: center;
padding: 16px 20px;
border-bottom: 1px solid rgba(255, 255, 255, 0.1);
}
.session-selector-header h3 {
margin: 0;
font-size: 16px;
font-weight: 500;
}
.session-selector-header button {
background: none;
border: none;
color: var(--vscode-editor-foreground);
font-size: 20px;
cursor: pointer;
padding: 4px 8px;
border-radius: 4px;
transition: background-color 0.2s;
}
.session-selector-header button:hover {
background-color: rgba(255, 255, 255, 0.1);
}
.session-selector-actions {
padding: 12px 20px;
border-bottom: 1px solid rgba(255, 255, 255, 0.1);
}
.new-session-button {
width: 100%;
padding: 8px 16px;
background-color: var(--vscode-button-background);
color: var(--vscode-button-foreground);
border: none;
border-radius: 4px;
font-size: 14px;
font-weight: 500;
cursor: pointer;
transition: background-color 0.2s;
}
.new-session-button:hover {
background-color: var(--vscode-button-hoverBackground);
}
.session-list {
flex: 1;
overflow-y: auto;
padding: 12px;
}
.no-sessions {
text-align: center;
padding: 40px 20px;
color: rgba(255, 255, 255, 0.5);
}
.session-item {
padding: 12px 16px;
margin-bottom: 8px;
background-color: var(--vscode-input-background);
border: 1px solid rgba(255, 255, 255, 0.1);
border-radius: 6px;
cursor: pointer;
transition: all 0.2s;
}
.session-item:hover {
background-color: rgba(255, 255, 255, 0.05);
border-color: var(--vscode-button-background);
}
.session-title {
font-size: 14px;
font-weight: 500;
margin-bottom: 4px;
white-space: nowrap;
overflow: hidden;
text-overflow: ellipsis;
}
.session-meta {
display: flex;
justify-content: space-between;
font-size: 11px;
opacity: 0.7;
margin-bottom: 4px;
}
.session-time {
color: var(--vscode-descriptionForeground);
}
.session-count {
color: var(--vscode-descriptionForeground);
}
.session-id {
font-size: 12px;
opacity: 0.6;
font-family: monospace;
}

View File

@@ -0,0 +1,276 @@
/**
* @license
* Copyright 2025 Qwen Team
* SPDX-License-Identifier: Apache-2.0
*/
import React, { useState, useEffect, useRef } from 'react';
import { useVSCode } from './hooks/useVSCode.js';
import type { ChatMessage } from '../agents/QwenAgentManager.js';
import type { Conversation } from '../storage/ConversationStore.js';
export const App: React.FC = () => {
const vscode = useVSCode();
const [messages, setMessages] = useState<ChatMessage[]>([]);
const [inputText, setInputText] = useState('');
const [isStreaming, setIsStreaming] = useState(false);
const [currentStreamContent, setCurrentStreamContent] = useState('');
const [qwenSessions, setQwenSessions] = useState<
Array<Record<string, unknown>>
>([]);
const [showSessionSelector, setShowSessionSelector] = useState(false);
const messagesEndRef = useRef<HTMLDivElement>(null);
const handlePermissionRequest = React.useCallback(
(request: {
options: Array<{ name: string; kind: string; optionId: string }>;
toolCall: { title?: string };
}) => {
const optionNames = request.options.map((opt) => opt.name).join(', ');
const confirmed = window.confirm(
`Tool permission request:\n${request.toolCall.title || 'Tool Call'}\n\nOptions: ${optionNames}\n\nAllow?`,
);
const selectedOption = confirmed
? request.options.find((opt) => opt.kind === 'allow_once')
: request.options.find((opt) => opt.kind === 'reject_once');
vscode.postMessage({
type: 'permissionResponse',
data: { optionId: selectedOption?.optionId || 'reject_once' },
});
},
[vscode],
);
useEffect(() => {
// Listen for messages from extension
const handleMessage = (event: MessageEvent) => {
const message = event.data;
switch (message.type) {
case 'conversationLoaded': {
const conversation = message.data as Conversation;
setMessages(conversation.messages);
break;
}
case 'message': {
const newMessage = message.data as ChatMessage;
setMessages((prev) => [...prev, newMessage]);
break;
}
case 'streamStart':
setIsStreaming(true);
setCurrentStreamContent('');
break;
case 'streamChunk':
setCurrentStreamContent((prev) => prev + message.data.chunk);
break;
case 'streamEnd':
// Finalize the streamed message
if (currentStreamContent) {
const assistantMessage: ChatMessage = {
role: 'assistant',
content: currentStreamContent,
timestamp: Date.now(),
};
setMessages((prev) => [...prev, assistantMessage]);
}
setIsStreaming(false);
setCurrentStreamContent('');
break;
case 'error':
console.error('Error from extension:', message.data.message);
setIsStreaming(false);
break;
case 'permissionRequest':
// Show permission dialog
handlePermissionRequest(message.data);
break;
case 'qwenSessionList':
setQwenSessions(message.data.sessions || []);
break;
case 'qwenSessionSwitched':
setShowSessionSelector(false);
// Load messages from the session
if (message.data.messages) {
setMessages(message.data.messages);
} else {
setMessages([]);
}
setCurrentStreamContent('');
break;
case 'conversationCleared':
setMessages([]);
setCurrentStreamContent('');
break;
default:
break;
}
};
window.addEventListener('message', handleMessage);
return () => window.removeEventListener('message', handleMessage);
}, [currentStreamContent, handlePermissionRequest]);
useEffect(() => {
// Auto-scroll to bottom when messages change
messagesEndRef.current?.scrollIntoView({ behavior: 'smooth' });
}, [messages, currentStreamContent]);
const handleSubmit = (e: React.FormEvent) => {
e.preventDefault();
if (!inputText.trim() || isStreaming) {
console.log('Submit blocked:', { inputText, isStreaming });
return;
}
console.log('Sending message:', inputText);
vscode.postMessage({
type: 'sendMessage',
data: { text: inputText },
});
setInputText('');
};
const handleLoadQwenSessions = () => {
vscode.postMessage({ type: 'getQwenSessions', data: {} });
setShowSessionSelector(true);
};
const handleNewQwenSession = () => {
vscode.postMessage({ type: 'newQwenSession', data: {} });
setShowSessionSelector(false);
// Clear messages in UI
setMessages([]);
setCurrentStreamContent('');
};
const handleSwitchSession = (sessionId: string) => {
vscode.postMessage({
type: 'switchQwenSession',
data: { sessionId },
});
};
return (
<div className="chat-container">
{showSessionSelector && (
<div className="session-selector-overlay">
<div className="session-selector">
<div className="session-selector-header">
<h3>Qwen Sessions</h3>
<button onClick={() => setShowSessionSelector(false)}></button>
</div>
<div className="session-selector-actions">
<button
className="new-session-button"
onClick={handleNewQwenSession}
>
New Session
</button>
</div>
<div className="session-list">
{qwenSessions.length === 0 ? (
<p className="no-sessions">No sessions available</p>
) : (
qwenSessions.map((session) => {
const sessionId =
(session.id as string) ||
(session.sessionId as string) ||
'';
const title =
(session.title as string) ||
(session.name as string) ||
'Untitled Session';
const lastUpdated =
(session.lastUpdated as string) ||
(session.startTime as string) ||
'';
const messageCount = (session.messageCount as number) || 0;
return (
<div
key={sessionId}
className="session-item"
onClick={() => handleSwitchSession(sessionId)}
>
<div className="session-title">{title}</div>
<div className="session-meta">
<span className="session-time">
{new Date(lastUpdated).toLocaleString()}
</span>
<span className="session-count">
{messageCount} messages
</span>
</div>
<div className="session-id">
{sessionId.substring(0, 8)}...
</div>
</div>
);
})
)}
</div>
</div>
</div>
)}
<div className="chat-header">
<button className="session-button" onClick={handleLoadQwenSessions}>
📋 Sessions
</button>
</div>
<div className="messages-container">
{messages.map((msg, index) => (
<div key={index} className={`message ${msg.role}`}>
<div className="message-content">{msg.content}</div>
<div className="message-timestamp">
{new Date(msg.timestamp).toLocaleTimeString()}
</div>
</div>
))}
{isStreaming && currentStreamContent && (
<div className="message assistant streaming">
<div className="message-content">{currentStreamContent}</div>
<div className="streaming-indicator"></div>
</div>
)}
<div ref={messagesEndRef} />
</div>
<form className="input-form" onSubmit={handleSubmit}>
<input
type="text"
className="input-field"
placeholder="Type your message..."
value={inputText}
onChange={(e) => setInputText((e.target as HTMLInputElement).value)}
disabled={isStreaming}
/>
<button
type="submit"
className="send-button"
disabled={isStreaming || !inputText.trim()}
>
Send
</button>
</form>
</div>
);
};

View File

@@ -0,0 +1,34 @@
/**
* @license
* Copyright 2025 Qwen Team
* SPDX-License-Identifier: Apache-2.0
*/
import { useMemo } from 'react';
export interface VSCodeAPI {
postMessage: (message: unknown) => void;
getState: () => unknown;
setState: (state: unknown) => void;
}
declare const acquireVsCodeApi: () => VSCodeAPI;
export function useVSCode(): VSCodeAPI {
return useMemo(() => {
if (typeof acquireVsCodeApi !== 'undefined') {
return acquireVsCodeApi();
}
// Fallback for development/testing
return {
postMessage: (message: unknown) => {
console.log('Mock postMessage:', message);
},
getState: () => ({}),
setState: (state: unknown) => {
console.log('Mock setState:', state);
},
};
}, []);
}

View File

@@ -0,0 +1,15 @@
/**
* @license
* Copyright 2025 Qwen Team
* SPDX-License-Identifier: Apache-2.0
*/
import ReactDOM from 'react-dom/client';
import { App } from './App.js';
import './App.css';
const container = document.getElementById('root');
if (container) {
const root = ReactDOM.createRoot(container);
root.render(<App />);
}