import * as vscode from 'vscode'; import * as fs from 'fs'; import * as path from 'path'; import * as os from 'os'; import { _getBrainDir, findBrainFiles, buildApiUrl, getActiveBrainProfile, getBrainProfiles, logError, logInfo, resolveEngine, summarizeText } from './utils'; import { getConfig } from './config'; import { AgentExecutor, ChatMessage } from './agent'; import { BridgeInterface } from './bridge'; import { buildProjectChronicleGuardContext, ProjectChronicleManager, ProjectProfile } from './features/projectChronicle'; import type { ModelLifecycleManager } from './lmstudio/lifecycleManager'; import type { IActivityTracker } from './lmstudio/activityTracker'; import { handleChatMessage } from './sidebar/chatHandlers'; import { handleBrainMessage } from './sidebar/brainHandlers'; import { handleChronicleMessage } from './sidebar/chronicleHandlers'; import { handleAgentMessage } from './sidebar/agentHandlers'; import { getOrCreateAgentEntry, resolveScopeForAgent } from './skills/agentKnowledgeMap'; import { estimateModelParamsB } from './lib/contextManager'; import { loadExternalSkills, formatSkillsAsPromptBlock } from './skills/externalSkillLoader'; export interface SidebarLmStudioDeps { lifecycle: ModelLifecycleManager; activity: IActivityTracker; /** Returns the list of model identifiers currently loaded in LM Studio (cached). */ loadedModels: () => Promise; } interface LastVisibleChatSnapshot { history: ChatMessage[]; brainProfileId: string; sessionId: string | null; timestamp: number; negativePrompt?: string; } interface ChatSession { id: string; title: string; timestamp: number; history: ChatMessage[]; brainProfileId: string; negativePrompt?: string; } /** * Sidebar UI Provider implementing BridgeInterface for BridgeServer */ export class SidebarChatProvider implements vscode.WebviewViewProvider, BridgeInterface { public static readonly viewType = 'g1nation-v2-view'; static readonly activeSessionStateKey = 'g1nation.activeSessionId'; static readonly lastVisibleChatStateKey = 'g1nation.lastVisibleChat'; static readonly blankChatStateKey = 'g1nation.blankChatActive'; static readonly lastAgentStateKey = 'g1nation.lastAgentPath'; static readonly chronicleProjectsStateKey = 'g1nation.chronicleProjects'; static readonly activeChronicleProjectStateKey = 'g1nation.activeChronicleProjectId'; static readonly lastAutoChronicleSignatureStateKey = 'g1nation.lastAutoChronicleSignature'; _view?: vscode.WebviewView; public brainEnabled = true; _currentSessionBrainId: string | null = null; _currentNegativePrompt: string = ''; readonly _chronicle = new ProjectChronicleManager(); _modelDiscoveryInFlight = false; _modelsCache: { url: string; models: string[]; online: boolean; fetchedAt: number } | null = null; static readonly MODELS_CACHE_TTL_MS = 30000; constructor( readonly _extensionUri: vscode.Uri, readonly _context: vscode.ExtensionContext, readonly _agent: AgentExecutor, readonly _lmStudio?: SidebarLmStudioDeps ) { this._agent.setHistoryChangeListener((history) => { void this._persistLastVisibleChat(history); }); } /** Surface LM Studio lifecycle errors (load/unload failures) to the chat UI as a non-fatal toast. */ public postLmStudioError(message: string): void { this._view?.webview.postMessage({ type: 'lmStudioError', value: message }); } public resolveWebviewView( webviewView: vscode.WebviewView, context: vscode.WebviewViewResolveContext, _token: vscode.CancellationToken, ) { this._view = webviewView; webviewView.webview.options = { enableScripts: true, localResourceRoots: [this._extensionUri] }; // [State Persistence Fix] 사이드바가 다시 보여질 때 세팅값 자동 복원 let _lastVisibilityRefresh = 0; webviewView.onDidChangeVisibility(() => { if (!webviewView.visible) return; const now = Date.now(); // 5초 이내에 이미 갱신했으면 건너뜀 if (now - _lastVisibilityRefresh < 5000) return; _lastVisibilityRefresh = now; logInfo('Sidebar became visible, restoring state...'); void this._sendModels(); void this._sendBrainProfiles(); void this._sendAgentsList(); void this._sendReadyStatus(); }); webviewView.webview.html = this._getHtml(webviewView.webview); this._agent.setWebview(webviewView.webview); void this._restoreActiveSessionIntoView(); void this._sendReadyStatus(); webviewView.webview.onDidReceiveMessage(async (data) => { if (await handleChatMessage(this, data)) return; if (await handleBrainMessage(this, data)) return; if (await handleChronicleMessage(this, data)) return; if (await handleAgentMessage(this, data)) return; logInfo(`Unhandled sidebar message: ${data?.type}`); }); } _currentSessionId: string | null = null; async _restoreActiveSessionIntoView() { if (!this._view) return; const blankChatActive = this._context.globalState.get(SidebarChatProvider.blankChatStateKey, false); const currentHistory = this._agent.getHistory(); if (blankChatActive && currentHistory.length === 0) { return; } const activeSessionId = this._currentSessionId || this._context.globalState.get(SidebarChatProvider.activeSessionStateKey, null); if (activeSessionId) { const loaded = await this._loadSession(activeSessionId, true); if (loaded) return; await this._context.globalState.update(SidebarChatProvider.activeSessionStateKey, null); } if (currentHistory.length > 0) { this._view.webview.postMessage({ type: 'restoreHistory', value: currentHistory }); await this._persistLastVisibleChat(currentHistory); return; } const snapshot = this._context.globalState.get(SidebarChatProvider.lastVisibleChatStateKey, null); if (snapshot?.history?.length) { this._currentSessionId = snapshot.sessionId || null; this._currentSessionBrainId = snapshot.brainProfileId || getActiveBrainProfile().id; this._currentNegativePrompt = snapshot.negativePrompt || ''; await this._setActiveBrainProfile(this._currentSessionBrainId, true); this._agent.setHistory(snapshot.history); this._view.webview.postMessage({ type: 'restoreHistory', value: snapshot.history, negativePrompt: this._currentNegativePrompt }); } } async _persistLastVisibleChat(history: ChatMessage[] = this._agent.getHistory()) { if (history.length === 0) { await this._context.globalState.update(SidebarChatProvider.lastVisibleChatStateKey, null); return; } const snapshot: LastVisibleChatSnapshot = { history, brainProfileId: this._currentSessionBrainId || getActiveBrainProfile().id, sessionId: this._currentSessionId, timestamp: Date.now(), negativePrompt: this._currentNegativePrompt }; await this._context.globalState.update(SidebarChatProvider.lastVisibleChatStateKey, snapshot); } async _saveCurrentSession() { const history = this._agent.getHistory(); if (history.length === 0) return; let sessions = this._getSessions(); const firstMsg = history.find(m => m.role === 'user')?.content; const title = typeof firstMsg === 'string' ? firstMsg.substring(0, 50).replace(/\n/g, ' ') + (firstMsg.length > 50 ? '...' : '') : 'New Chat'; const brainProfileId = this._currentSessionBrainId || getActiveBrainProfile().id; if (!this._currentSessionId) { this._currentSessionId = Date.now().toString(); sessions.unshift({ id: this._currentSessionId, title, timestamp: Date.now(), history, brainProfileId, negativePrompt: this._currentNegativePrompt }); } else { const idx = sessions.findIndex(s => s.id === this._currentSessionId); if (idx >= 0) { sessions[idx].history = history; sessions[idx].timestamp = Date.now(); sessions[idx].brainProfileId = brainProfileId; sessions[idx].negativePrompt = this._currentNegativePrompt; if (!sessions[idx].title || sessions[idx].title === 'New Chat') { sessions[idx].title = title; } } else { sessions.unshift({ id: this._currentSessionId, title, timestamp: Date.now(), history, brainProfileId, negativePrompt: this._currentNegativePrompt }); } } // Keep only last 50 sessions if (sessions.length > 50) sessions = sessions.slice(0, 50); await this._putSessions(sessions); await this._context.globalState.update(SidebarChatProvider.activeSessionStateKey, this._currentSessionId); await this._persistLastVisibleChat(history); await this._sendSessionList(); } async _sendSessionList() { if (!this._view) return; const sessions = this._getSessions(); const list = sessions.map(s => ({ id: s.id, title: s.title, timestamp: s.timestamp, brainProfileId: s.brainProfileId || '', messageCount: s.history.length, history: s.history })); this._view.webview.postMessage({ type: 'sessionList', value: list }); } async _loadSession(id: string, skipSessionListRefresh: boolean = false): Promise { if (!id) { logError('Session load requested without an id.'); this._view?.webview.postMessage({ type: 'error', value: 'Chat session id is missing.' }); return false; } const sessions = this._getSessions(); const session = sessions.find(s => s.id === id) || this._getSessionById(id); if (session) { const history = Array.isArray(session.history) ? session.history : []; if (history.length === 0) { logError('Session load failed because history is empty or invalid.', { id }); this._view?.webview.postMessage({ type: 'error', value: 'This chat session has no saved messages.' }); return false; } this._agent.stop(); this._currentSessionId = id; this._currentNegativePrompt = session.negativePrompt || ''; const sessionBrainId = session.brainProfileId || getActiveBrainProfile().id; await this._setActiveBrainProfile(sessionBrainId, true); this._agent.setHistory(history); await this._context.globalState.update(SidebarChatProvider.activeSessionStateKey, id); await this._context.globalState.update(SidebarChatProvider.blankChatStateKey, false); await this._persistLastVisibleChat(history); this._view?.webview.postMessage({ type: 'sessionLoaded', value: { id, title: session.title || 'Chat Session', history, negativePrompt: this._currentNegativePrompt } }); if (!skipSessionListRefresh) { await this._sendSessionList(); } logInfo('Chat session loaded.', { id, messages: history.length }); return true; } logError('Session load failed because id was not found.', { id, sessionCount: sessions.length }); this._view?.webview.postMessage({ type: 'error', value: 'Chat session was not found.' }); return false; } async _deleteSession(id: string) { let sessions = this._getSessions(); sessions = sessions.filter(s => s.id !== id); await this._putSessions(sessions); if (this._currentSessionId === id) { this._currentSessionId = null; this._agent.resetConversation(); await this._context.globalState.update(SidebarChatProvider.activeSessionStateKey, null); await this._context.globalState.update(SidebarChatProvider.lastVisibleChatStateKey, null); await this._context.globalState.update(SidebarChatProvider.blankChatStateKey, true); this.clearChat(); } await this._sendSessionList(); } _getSessions(): ChatSession[] { const rawSessions = this._context.globalState.get('chat_sessions', []) || []; return rawSessions .map((session, index): ChatSession | null => { const history = Array.isArray(session?.history) ? session.history.filter((message: any) => message && (message.role === 'user' || message.role === 'assistant' || message.role === 'system') && message.content !== undefined ) : []; if (!session?.id || history.length === 0) return null; const firstMsg = history.find((message: ChatMessage) => message.role === 'user')?.content; const fallbackTitle = typeof firstMsg === 'string' ? firstMsg.substring(0, 50).replace(/\n/g, ' ') + (firstMsg.length > 50 ? '...' : '') : `Chat ${index + 1}`; return { id: String(session.id), title: String(session.title || fallbackTitle), timestamp: typeof session.timestamp === 'number' ? session.timestamp : Date.now(), history, brainProfileId: String(session.brainProfileId || getActiveBrainProfile().id), negativePrompt: String(session.negativePrompt || '') }; }) .filter((session): session is ChatSession => !!session) .sort((a, b) => b.timestamp - a.timestamp) .slice(0, 50); } _getSessionById(id: string): ChatSession | null { const rawSessions = this._context.globalState.get('chat_sessions', []) || []; const raw = rawSessions.find((session: any) => String(session?.id) === String(id)); if (!raw) return null; const history = Array.isArray(raw.history) ? raw.history.filter((message: any) => message && (message.role === 'user' || message.role === 'assistant' || message.role === 'system') && message.content !== undefined ) : []; if (history.length === 0) return null; const firstMsg = history.find((message: ChatMessage) => message.role === 'user')?.content; const fallbackTitle = typeof firstMsg === 'string' ? firstMsg.substring(0, 50).replace(/\n/g, ' ') + (firstMsg.length > 50 ? '...' : '') : 'Chat Session'; return { id: String(raw.id), title: String(raw.title || fallbackTitle), timestamp: typeof raw.timestamp === 'number' ? raw.timestamp : Date.now(), history, brainProfileId: String(raw.brainProfileId || getActiveBrainProfile().id), negativePrompt: String(raw.negativePrompt || '') }; } async _putSessions(sessions: ChatSession[]) { await this._context.globalState.update('chat_sessions', sessions.slice(0, 50)); } async _sendBrainStatus() { if (!this._view) return; const activeBrain = getActiveBrainProfile(); const brainDir = activeBrain.localBrainPath; const files = findBrainFiles(brainDir); this._view.webview.postMessage({ type: 'brainStatus', value: { count: files.length, path: brainDir, name: activeBrain.name, description: activeBrain.description || '', repo: activeBrain.secondBrainRepo || '' } }); } /** * One-line "current readiness" snapshot for the sidebar's status bar: * engine online?, model loaded?, Brain file count, active Agent + mapped knowledge * folder count, memory on/off, context window. Cheap — no network calls except the * already-cached LM Studio loaded-models list and online flag. */ async _sendReadyStatus() { if (!this._view) return; let payload: any; try { const config = getConfig(); const engineKind = resolveEngine(config.ollamaUrl); const activeBrain = getActiveBrainProfile(); let brainFiles = 0; try { brainFiles = findBrainFiles(activeBrain.localBrainPath).length; } catch { /* ignore */ } const agentPath = this._context.globalState.get(SidebarChatProvider.lastAgentStateKey, 'none'); let agentName: string | null = null; let scopeFolders = 0; let mapped = false; if (agentPath && agentPath !== 'none') { agentName = path.basename(agentPath).replace(/\.md$/i, ''); try { const scope = resolveScopeForAgent(agentPath, activeBrain.localBrainPath || ''); scopeFolders = scope.folders.length; if (scope.agent?.name) agentName = scope.agent.name; mapped = scope.source !== 'none'; } catch { /* ignore */ } } let modelLoaded: boolean | null = null; if (engineKind === 'lmstudio') { try { const loaded = (await this._lmStudio?.loadedModels()) || []; modelLoaded = loaded.includes(config.defaultModel); } catch { modelLoaded = null; } } const paramB = estimateModelParamsB(config.defaultModel); const cappedForSmallModel = config.smallModelContextCap > 0 && paramB !== null && paramB <= 4 && config.contextLength > config.smallModelContextCap; const effectiveContextLength = cappedForSmallModel ? config.smallModelContextCap : config.contextLength; payload = { engine: { kind: engineKind, label: engineKind === 'lmstudio' ? 'LM Studio' : 'Ollama', online: this._modelsCache?.online ?? null, }, model: { name: config.defaultModel, loaded: modelLoaded, paramB }, brain: { name: activeBrain.name, files: brainFiles }, agent: { name: agentName, scopeFolders, mapped }, memory: config.memoryEnabled, multiAgent: config.multiAgentEnabled, contextLength: effectiveContextLength, nominalContextLength: config.contextLength, cappedForSmallModel, }; } catch (err: any) { logError('Failed to build ready status.', { error: err?.message || String(err) }); return; } this._view.webview.postMessage({ type: 'readyStatus', value: payload }); } async _sendBrainProfiles() { if (!this._view) return; const activeBrain = getActiveBrainProfile(); this._currentSessionBrainId = this._currentSessionBrainId || activeBrain.id; const profiles = getBrainProfiles().map((profile) => ({ id: profile.id, name: profile.name, path: profile.localBrainPath, description: profile.description || '', repo: profile.secondBrainRepo || '' })); this._view.webview.postMessage({ type: 'brainProfiles', value: { activeBrainId: activeBrain.id, profiles } }); void this._sendReadyStatus(); } _postBrainProfiles(profiles: any[], activeBrainId: string) { if (!this._view) return; this._view.webview.postMessage({ type: 'brainProfiles', value: { activeBrainId, profiles: profiles.map((p: any) => ({ id: p.id || '', name: p.name || '', path: p.localBrainPath || '', description: p.description || '', repo: p.secondBrainRepo || '' })) } }); } async _setActiveBrainProfile(profileId: string, silent: boolean = false) { const profiles = getBrainProfiles(); const nextProfile = profiles.find((profile) => profile.id === profileId) || profiles[0]; if (!nextProfile) return; await vscode.workspace.getConfiguration('g1nation').update('activeBrainId', nextProfile.id, vscode.ConfigurationTarget.Global); this._currentSessionBrainId = nextProfile.id; await this._sendBrainProfiles(); await this._sendBrainStatus(); if (!silent) { this.injectSystemMessage(`**[Brain Switched]** ${nextProfile.name}\n\`${nextProfile.localBrainPath}\``); } } async _manageBrains() { const activeBrain = getActiveBrainProfile(); const choice = await vscode.window.showQuickPick([ { label: 'Add Brain Folder', description: 'Create a new brain profile from a local folder' }, { label: 'Open Active Brain Folder', description: activeBrain.localBrainPath }, { label: 'Open Brain Settings', description: 'Edit names, paths, repos, and descriptions' } ], { placeHolder: `Active Brain: ${activeBrain.name} (${activeBrain.localBrainPath})` }); if (!choice) return; if (choice.label === 'Add Brain Folder') { await this._addBrainProfile(); return; } if (choice.label === 'Open Active Brain Folder') { if (!fs.existsSync(activeBrain.localBrainPath)) { fs.mkdirSync(activeBrain.localBrainPath, { recursive: true }); } await vscode.commands.executeCommand('revealFileInOS', vscode.Uri.file(activeBrain.localBrainPath)); return; } vscode.commands.executeCommand('workbench.action.openSettings', 'g1nation.brainProfiles'); } async _addBrainProfile() { const selected = await vscode.window.showOpenDialog({ canSelectFiles: false, canSelectFolders: true, canSelectMany: false, openLabel: 'Use as Brain' }); const folder = selected?.[0]?.fsPath; if (!folder) return; const defaultName = path.basename(folder) || 'New Brain'; const name = await vscode.window.showInputBox({ prompt: 'Name this brain profile', value: defaultName, validateInput: (value) => value.trim() ? null : 'Brain name is required.' }); if (!name) return; const description = await vscode.window.showInputBox({ prompt: 'Optional description shown in the Astra sidebar', value: '' }); const repo = await vscode.window.showInputBox({ prompt: 'Optional Second Brain Git repository URL', value: '' }); // Read raw settings directly to avoid virtual default-brain (injected in memory by getConfig()) // being saved into the settings file and corrupting the profile list on next load. const cfg = vscode.workspace.getConfiguration('g1nation'); const existingRaw: any[] = cfg.get('brainProfiles', []) || []; const idBase = name.toLowerCase().replace(/[^a-z0-9]+/g, '-').replace(/^-|-$/g, '') || 'brain'; let id = idBase; let suffix = 2; while (existingRaw.some((p: any) => p.id === id)) { id = `${idBase}-${suffix++}`; } const newProfile = { id, name: name.trim(), localBrainPath: folder, secondBrainRepo: (repo || '').trim(), description: (description || '').trim() }; const nextProfiles = [...existingRaw, newProfile]; await cfg.update('brainProfiles', nextProfiles, vscode.ConfigurationTarget.Global); await cfg.update('activeBrainId', id, vscode.ConfigurationTarget.Global); this._currentSessionBrainId = id; // cfg.update() is async and VSCode's config cache may not reflect the new value // immediately, so we post the freshly-built profile list directly. this._postBrainProfiles(nextProfiles, id); await this._sendBrainStatus(); this.injectSystemMessage(`**[Brain Added]** ${name.trim()}\n\`${folder}\``); } async _editBrainProfile(profileId?: string) { const currentProfiles = getBrainProfiles(); const target = currentProfiles.find((profile) => profile.id === profileId) || getActiveBrainProfile(); if (!target) return; const name = await vscode.window.showInputBox({ prompt: 'Edit brain profile name', value: target.name, validateInput: (value) => value.trim() ? null : 'Brain name is required.' }); if (!name) return; const folder = await vscode.window.showInputBox({ prompt: 'Edit local brain folder path', value: target.localBrainPath, validateInput: (value) => value.trim() ? null : 'Brain folder path is required.' }); if (!folder) return; const description = await vscode.window.showInputBox({ prompt: 'Edit optional description shown in the Astra sidebar', value: target.description || '' }); const repo = await vscode.window.showInputBox({ prompt: 'Edit optional Second Brain Git repository URL', value: target.secondBrainRepo || '' }); const nextProfiles = currentProfiles.map((profile) => profile.id === target.id ? { ...profile, name: name.trim(), localBrainPath: folder.trim(), secondBrainRepo: (repo || '').trim(), description: (description || '').trim() } : profile ); const cfg = vscode.workspace.getConfiguration('g1nation'); await cfg.update('brainProfiles', nextProfiles, vscode.ConfigurationTarget.Global); await cfg.update('activeBrainId', target.id, vscode.ConfigurationTarget.Global); this._currentSessionBrainId = target.id; this._postBrainProfiles(nextProfiles, target.id); await this._sendBrainStatus(); this.injectSystemMessage(`**[Brain Updated]** ${name.trim()}\n\`${folder.trim()}\``); } async _deleteBrainProfile(profileId?: string) { const currentProfiles = getBrainProfiles(); const target = currentProfiles.find((profile) => profile.id === profileId) || getActiveBrainProfile(); if (!target) return; if (currentProfiles.length <= 1) { vscode.window.showWarningMessage('At least one brain profile is required.'); return; } const confirm = await vscode.window.showWarningMessage( `Delete brain profile "${target.name}"? The folder itself will not be deleted.`, { modal: true }, 'Delete Profile' ); if (confirm !== 'Delete Profile') return; const nextProfiles = currentProfiles.filter((profile) => profile.id !== target.id); const nextActive = nextProfiles[0]; const cfg = vscode.workspace.getConfiguration('g1nation'); await cfg.update('brainProfiles', nextProfiles, vscode.ConfigurationTarget.Global); await cfg.update('activeBrainId', nextActive.id, vscode.ConfigurationTarget.Global); this._currentSessionBrainId = nextActive.id; this._postBrainProfiles(nextProfiles, nextActive.id); await this._sendBrainStatus(); this.injectSystemMessage(`**[Brain Deleted]** ${target.name}`); } async _saveWikiRaw() { const history = this._agent.getHistory(); if (history.length === 0) { vscode.window.showWarningMessage('There is no conversation to save as wiki raw data.'); return; } const activeBrain = getActiveBrainProfile(); const rawDir = path.join(activeBrain.localBrainPath, 'raw-data'); if (!fs.existsSync(rawDir)) { fs.mkdirSync(rawDir, { recursive: true }); } const category = await vscode.window.showInputBox({ prompt: 'Wiki raw data category', value: 'Project Notes' }); if (category === undefined) return; const expectedValue = await vscode.window.showInputBox({ prompt: 'Expected value or future use', value: 'Reusable as source material for a future wiki document.' }); if (expectedValue === undefined) return; const timestamp = new Date(); const slug = this._slugify(history.find((message) => message.role === 'user')?.content || 'conversation'); const fileName = `${this._formatTimestampForFile(timestamp)}-${slug}.md`; const filePath = path.join(rawDir, fileName); const markdown = this._buildWikiRawMarkdown(history, { category: category.trim() || 'Project Notes', expectedValue: expectedValue.trim() || 'Reusable as source material for a future wiki document.', activeBrainName: activeBrain.name, activeBrainPath: activeBrain.localBrainPath, createdAt: timestamp.toISOString() }); await vscode.workspace.fs.writeFile(vscode.Uri.file(filePath), Buffer.from(markdown, 'utf8')); vscode.window.showInformationMessage(`Wiki raw data saved: ${path.basename(filePath)}`); this.injectSystemMessage(`**[Wiki Raw Saved]** \`${filePath}\``); } _buildWikiRawMarkdown(history: ChatMessage[], meta: { category: string; expectedValue: string; activeBrainName: string; activeBrainPath: string; createdAt: string; }): string { const firstUserMessage = history.find((message) => message.role === 'user')?.content || ''; const latestUserMessage = [...history].reverse().find((message) => message.role === 'user')?.content || ''; const latestAssistantMessage = [...history].reverse().find((message) => message.role === 'assistant')?.content || ''; const rationaleNotes = history .filter((message) => message.role === 'assistant' && message.rationale) .map((message, index) => [ `### Process Note ${index + 1}`, message.rationale?.problem ? `- Problem: ${message.rationale.problem}` : '', message.rationale?.goal ? `- Goal: ${message.rationale.goal}` : '', message.rationale?.reasoning ? `- Reasoning: ${message.rationale.reasoning}` : '' ].filter(Boolean).join('\n')) .join('\n\n'); const transcript = history .map((message, index) => [ `### ${index + 1}. ${message.role.toUpperCase()}`, '', message.content.trim() || '(empty)' ].join('\n')) .join('\n\n'); return [ '---', `title: "${this._escapeYamlString(this._summarizeForTitle(firstUserMessage))}"`, `category: "${this._escapeYamlString(meta.category)}"`, `created_at: "${meta.createdAt}"`, `source: "Astra conversation"`, `brain: "${this._escapeYamlString(meta.activeBrainName)}"`, 'status: raw', '---', '', `# ${this._summarizeForTitle(firstUserMessage)}`, '', '## Category', meta.category, '', '## What This Contains', this._summarizeTextForWiki(latestAssistantMessage || firstUserMessage), '', '## Expected Value', meta.expectedValue, '', '## Discovery Process', rationaleNotes || 'No explicit rationale metadata was captured. Use the transcript below to reconstruct the reasoning path.', '', '## Conclusion', [ 'This conclusion was derived from the latest user request and assistant response.', '', `- Latest user request: ${this._summarizeTextForWiki(latestUserMessage)}`, `- Latest assistant conclusion: ${this._summarizeTextForWiki(latestAssistantMessage)}` ].join('\n'), '', '## Coding Implementation Notes', 'Use this section to turn the raw transcript into a wiki-ready implementation note. Capture which files changed, why they changed, and what verification was run.', '', '## Source Brain', `- Name: ${meta.activeBrainName}`, `- Path: ${meta.activeBrainPath}`, '', '## Raw Conversation Transcript', transcript, '' ].join('\n'); } _formatTimestampForFile(date: Date): string { return date.toISOString().replace(/[:.]/g, '-').replace('T', '_').replace('Z', ''); } _slugify(value: string): string { const slug = value .toLowerCase() .replace(/[^a-z0-9가-힣]+/g, '-') .replace(/^-|-$/g, '') .slice(0, 48); return slug || 'conversation'; } _summarizeForTitle(value: string): string { const normalized = value.replace(/\s+/g, ' ').trim(); if (!normalized) return 'Astra Conversation Raw Data'; return normalized.length > 80 ? `${normalized.slice(0, 80)}...` : normalized; } _summarizeTextForWiki(value: string): string { const normalized = value.replace(/\s+/g, ' ').trim(); if (!normalized) return 'Not captured.'; return normalized.length > 500 ? `${normalized.slice(0, 500)}...` : normalized; } _escapeYamlString(value: string): string { return value.replace(/\\/g, '\\\\').replace(/"/g, '\\"'); } // --- BridgeInterface Methods --- public injectSystemMessage(msg: string): void { this._view?.webview.postMessage({ type: 'streamStart' }); this._view?.webview.postMessage({ type: 'streamChunk', value: msg }); this._view?.webview.postMessage({ type: 'streamEnd' }); } public getHistoryText(): string { return "Conversation history placeholder for evaluation."; } public sendPromptFromExtension(prompt: string): void { if (this._view) { this._view.show?.(true); this._view.webview.postMessage({ type: 'injectPrompt', value: prompt }); } } public findBrainFiles(dir: string): string[] { return findBrainFiles(dir); } // --- End BridgeInterface --- public focusInput() { this._view?.webview.postMessage({ type: 'focusInput' }); } public clearChat() { this._view?.webview.postMessage({ type: 'clearChat' }); } public async syncBrain() { const activeBrain = getActiveBrainProfile(); const brainDir = activeBrain.localBrainPath; if (!fs.existsSync(brainDir)) { vscode.window.showErrorMessage("Second Brain directory not found."); return; } vscode.window.withProgress({ location: vscode.ProgressLocation.Notification, title: "Astra: Syncing Second Brain...", cancellable: false }, async () => { try { const { execSync } = require('child_process'); execSync(`git add .`, { cwd: brainDir }); execSync(`git commit -m "[G1-Sync] Manual knowledge update"`, { cwd: brainDir }); execSync(`git push`, { cwd: brainDir }); vscode.window.showInformationMessage("Second Brain synced successfully."); } catch (err: any) { vscode.window.showInformationMessage("Brain Synced: Local knowledge is up-to-date (no changes to push)."); } }); } _getChronicleProjects(): ProjectProfile[] { const raw = this._context.globalState.get(SidebarChatProvider.chronicleProjectsStateKey, []) || []; const valid = raw.filter((profile: ProjectProfile) => profile && typeof profile.projectId === 'string' && typeof profile.projectName === 'string' && typeof profile.recordRoot === 'string' ); if (valid.length > 0) return valid; const workspaceRoot = vscode.workspace.workspaceFolders?.[0]?.uri.fsPath; if (!workspaceRoot) return []; const now = new Date().toISOString(); const projectName = path.basename(workspaceRoot) || 'Current Project'; return [{ projectId: this._slugify(projectName), projectName, projectRoot: workspaceRoot, recordRoot: path.join(workspaceRoot, 'docs', 'records', projectName), description: 'Auto-detected current workspace project.', corePurpose: 'Capture project planning, decisions, development notes, bugs, and retrospectives as Markdown.', targetUsers: ['Project developer'], avoidDirections: ['Do not tightly couple records to chat execution internals.'], detailLevel: 'standard', createdAt: now, updatedAt: now }]; } async _putChronicleProjects(projects: ProjectProfile[]) { await this._context.globalState.update(SidebarChatProvider.chronicleProjectsStateKey, projects); } _getActiveChronicleProject(): ProjectProfile | null { const projects = this._getChronicleProjects(); if (projects.length === 0) return null; const activeId = this._context.globalState.get(SidebarChatProvider.activeChronicleProjectStateKey, ''); return projects.find(project => project.projectId === activeId) || projects[0]; } async _sendChronicleProjects() { if (!this._view) return; const projects = this._getChronicleProjects(); const active = this._getActiveChronicleProject(); this._view.webview.postMessage({ type: 'chronicleProjects', value: { activeProjectId: active?.projectId || '', projects: projects.map(project => ({ id: project.projectId, name: project.projectName, root: project.projectRoot || '', recordRoot: project.recordRoot, description: project.description || '' })) } }); } async _createChronicleProject() { const workspaceRoot = vscode.workspace.workspaceFolders?.[0]?.uri.fsPath || ''; const defaultName = workspaceRoot ? path.basename(workspaceRoot) : 'New Project'; const projectName = await vscode.window.showInputBox({ prompt: 'Project name for Chronicle records', value: defaultName, validateInput: (value) => value.trim() ? null : 'Project name is required.' }); if (!projectName) return; const description = await vscode.window.showInputBox({ prompt: 'One-line project description', value: 'Project planning, decisions, development logs, and bug records.' }); if (description === undefined) return; const projectRoot = await vscode.window.showInputBox({ prompt: 'Project root path', value: workspaceRoot, validateInput: (value) => value.trim() ? null : 'Project root is required.' }); if (!projectRoot) return; const defaultRecordRoot = path.join(projectRoot.trim(), 'docs', 'records', projectName.trim()); const recordRoot = await vscode.window.showInputBox({ prompt: 'Markdown record folder path', value: defaultRecordRoot, validateInput: (value) => value.trim() ? null : 'Record folder path is required.' }); if (!recordRoot) return; const corePurpose = await vscode.window.showInputBox({ prompt: 'Core project purpose or guardrail', value: 'Keep project knowledge traceable through Markdown records.' }); if (corePurpose === undefined) return; const detailChoice = await vscode.window.showQuickPick([ { label: 'standard', description: 'Request, question intent, decisions, planning, development, and bugs' }, { label: 'simple', description: 'Request summary, decisions, and implementation result' }, { label: 'detailed', description: 'Standard records plus alternatives, lessons, and retrospectives' } ], { placeHolder: 'Chronicle record detail level' }); if (!detailChoice) return; const now = new Date().toISOString(); const projects = this._getChronicleProjects(); const idBase = this._slugify(projectName.trim()); let projectId = idBase; let suffix = 2; while (projects.some(project => project.projectId === projectId)) { projectId = `${idBase}-${suffix++}`; } const profile: ProjectProfile = { projectId, projectName: projectName.trim(), projectRoot: projectRoot.trim(), recordRoot: recordRoot.trim(), description: description.trim(), corePurpose: corePurpose.trim(), targetUsers: ['Project developer'], avoidDirections: ['Do not mix records across projects.', 'Do not tightly couple records to core agent execution.'], detailLevel: detailChoice.label as ProjectProfile['detailLevel'], createdAt: now, updatedAt: now }; this._chronicle.ensureProject(profile); const nextProjects = [...projects.filter(project => project.projectId !== profile.projectId), profile]; await this._putChronicleProjects(nextProjects); await this._context.globalState.update(SidebarChatProvider.activeChronicleProjectStateKey, profile.projectId); await this._sendChronicleProjects(); this.injectSystemMessage(`**[Designer Project Created]** ${profile.projectName}\n\`${profile.recordRoot}\``); } async _setActiveChronicleProject(projectId: string) { if (!projectId || projectId === 'new') { await this._createChronicleProject(); return; } const target = this._getChronicleProjects().find(project => project.projectId === projectId); if (!target) return; await this._context.globalState.update(SidebarChatProvider.activeChronicleProjectStateKey, target.projectId); await this._sendChronicleProjects(); await this._sendChronicleRecords(); this.injectSystemMessage(`**[Designer Project Selected]** ${target.projectName}\n\`${target.recordRoot}\``); } async _openChronicleFolder() { const profile = this._getActiveChronicleProject(); if (!profile) { vscode.window.showWarningMessage('No Chronicle project is selected.'); return; } try { this._chronicle.ensureProject(profile); await vscode.commands.executeCommand('revealFileInOS', vscode.Uri.file(profile.recordRoot)); } catch (err: any) { vscode.window.showErrorMessage(`Failed to open Chronicle folder: ${err.message}`); } } async _sendChronicleRecords() { if (!this._view) return; const profile = this._getActiveChronicleProject(); if (!profile) { this._view.webview.postMessage({ type: 'chronicleRecords', value: [] }); return; } try { const records = this._chronicle.listRecords(profile).map(record => ({ section: record.section, fileName: record.fileName, path: record.filePath, relativePath: record.relativePath, updatedAt: record.updatedAt })); this._view.webview.postMessage({ type: 'chronicleRecords', value: records }); } catch (err: any) { vscode.window.showErrorMessage(`Failed to list Chronicle records: ${err.message}`); } } async _openChronicleRecord(recordPath: string) { const profile = this._getActiveChronicleProject(); if (!profile || !recordPath) { vscode.window.showWarningMessage('Select a Chronicle record first.'); return; } const root = path.resolve(profile.recordRoot); const target = path.resolve(recordPath); if (!target.startsWith(`${root}${path.sep}`) || path.extname(target) !== '.md') { vscode.window.showErrorMessage('Selected Chronicle record path is not valid.'); return; } if (!fs.existsSync(target)) { vscode.window.showErrorMessage('Selected Chronicle record no longer exists.'); await this._sendChronicleRecords(); return; } const doc = await vscode.workspace.openTextDocument(target); await vscode.window.showTextDocument(doc); } async _writeChroniclePlanningFromCurrentChat() { const profile = this._getActiveChronicleProject(); if (!profile) { vscode.window.showWarningMessage('Select or create a Designer project first.'); return; } const history = this._agent.getHistory(); const latestUser = [...history].reverse().find(message => message.role === 'user')?.content || ''; const latestAssistant = [...history].reverse().find(message => message.role === 'assistant')?.content || ''; const featureName = await vscode.window.showInputBox({ prompt: 'Feature name for the planning document', value: this._summarizeForTitle(latestUser || 'Project Chronicle Feature') }); if (!featureName) return; try { const createdAt = new Date().toISOString(); const result = this._chronicle.writePlanning(profile, { featureName: featureName.trim(), purpose: 'Record the reason, scope, direction, and success criteria before implementation.', background: this._summarizeTextForWiki(latestAssistant || latestUser), userIntent: this._summarizeTextForWiki(latestUser), sourceRequest: latestUser || 'No user request captured in the current chat.', scope: [ 'Create a project-specific planning record.', 'Capture user intent and implementation direction.', 'Keep the record independent from chat execution internals.' ], outOfScope: [ 'Full automatic transcript capture.', 'External database integration.', 'Git automation.' ], developmentDirection: 'Use Project Chronicle as a low-dependency Markdown record layer.', dependencyStrategy: 'Use local filesystem writes through the independent projectChronicle module.', expectedValue: 'Future work can understand why this feature exists and what decisions shaped it.', successCriteria: [ 'The planning document is created under the selected project record folder.', 'The document includes user intent, scope, out-of-scope items, and success criteria.' ], developerInstruction: 'Use this document as the implementation guardrail for the next development step.', createdAt }); this._chronicle.appendTimeline(profile, [`Planning record created: ${result.relativePath}`], createdAt); vscode.window.showInformationMessage(`Chronicle planning saved: ${result.relativePath}`); await this._sendChronicleRecords(); this.injectSystemMessage(`**[Chronicle Planning Saved]** \`${result.filePath}\``); } catch (err: any) { vscode.window.showErrorMessage(`Failed to write planning record: ${err.message}`); } } async _writeChronicleDiscussionFromCurrentChat() { const profile = this._getActiveChronicleProject(); if (!profile) { vscode.window.showWarningMessage('Select or create a Designer project first.'); return; } const history = this._agent.getHistory(); const latestUser = [...history].reverse().find(message => message.role === 'user')?.content || ''; const latestAssistant = [...history].reverse().find(message => message.role === 'assistant')?.content || ''; const title = await vscode.window.showInputBox({ prompt: 'Discussion title', value: this._summarizeForTitle(latestUser || 'Project Discussion') }); if (!title) return; const question = await vscode.window.showInputBox({ prompt: 'AI question to record (optional)', value: '' }); if (question === undefined) return; let questions: any[] = []; if (question.trim()) { const reason = await vscode.window.showInputBox({ prompt: 'Why was this question asked?', value: 'To avoid writing records to the wrong project or making an unclear design decision.' }); if (reason === undefined) return; const impact = await vscode.window.showInputBox({ prompt: 'How does this question affect the decision?', value: 'It determines the correct project context, scope, or implementation path.' }); if (impact === undefined) return; questions = [{ question: question.trim(), reason: reason.trim(), expectedInformation: 'Information needed to clarify project context, scope, or decision direction.', impactOnDecision: impact.trim() }]; } try { const createdAt = new Date().toISOString(); const result = this._chronicle.writeDiscussion(profile, { title: title.trim(), userRequest: this._summarizeTextForWiki(latestUser || 'No user request captured in the current chat.'), interpretedIntent: 'Capture the discussion as a reusable project knowledge record instead of leaving it only in chat history.', questions, discussions: [ this._summarizeTextForWiki(latestAssistant || latestUser || 'No discussion detail captured yet.') ], decisions: [], createdAt }); this._chronicle.appendTimeline(profile, [`Discussion record created: ${result.relativePath}`], createdAt); vscode.window.showInformationMessage(`Chronicle discussion saved: ${result.relativePath}`); await this._sendChronicleRecords(); this.injectSystemMessage(`**[Chronicle Discussion Saved]** \`${result.filePath}\``); } catch (err: any) { vscode.window.showErrorMessage(`Failed to write discussion record: ${err.message}`); } } async _writeChronicleDecisionFromInput() { const profile = this._getActiveChronicleProject(); if (!profile) { vscode.window.showWarningMessage('Select or create a Designer project first.'); return; } const title = await vscode.window.showInputBox({ prompt: 'Decision title', value: 'Use independent Markdown record module' }); if (!title) return; const decision = await vscode.window.showInputBox({ prompt: 'Decision', value: 'Implement this behavior as an independent Project Chronicle module.' }); if (decision === undefined) return; const reason = await vscode.window.showInputBox({ prompt: 'Decision reason', value: 'To reduce coupling and keep project records portable.' }); if (reason === undefined) return; const alternatives = await vscode.window.showInputBox({ prompt: 'Rejected alternatives (comma-separated)', value: 'Integrate with Second Brain, integrate directly into Agent execution' }); if (alternatives === undefined) return; try { const createdAt = new Date().toISOString(); const adrNumber = this._chronicle.nextAdrNumber(profile); const result = this._chronicle.writeDecision(profile, { title: title.trim(), status: 'accepted', context: 'A project record needs to capture not only what changed, but why the direction was chosen.', decision: decision.trim(), reason: reason.trim(), alternatives: alternatives.split(',').map(item => item.trim()).filter(Boolean), consequences: [ 'Records can evolve independently from chat and agent internals.', 'Future automation can emit chronicle events without owning the core execution path.' ], createdAt }, adrNumber); this._chronicle.appendTimeline(profile, [`Decision record created: ${result.relativePath}`], createdAt); vscode.window.showInformationMessage(`Chronicle decision saved: ${result.relativePath}`); await this._sendChronicleRecords(); this.injectSystemMessage(`**[Chronicle Decision Saved]** \`${result.filePath}\``); } catch (err: any) { vscode.window.showErrorMessage(`Failed to write decision record: ${err.message}`); } } async _writeChronicleDevelopmentFromCurrentChat() { const profile = this._getActiveChronicleProject(); if (!profile) { vscode.window.showWarningMessage('Select or create a Designer project first.'); return; } const history = this._agent.getHistory(); const latestUser = [...history].reverse().find(message => message.role === 'user')?.content || ''; const latestAssistant = [...history].reverse().find(message => message.role === 'assistant')?.content || ''; const featureName = await vscode.window.showInputBox({ prompt: 'Feature name for the development log', value: this._summarizeForTitle(latestUser || 'Implementation Log') }); if (!featureName) return; try { const createdAt = new Date().toISOString(); const result = this._chronicle.writeDevelopmentLog(profile, { featureName: featureName.trim(), purpose: 'Record the actual implementation outcome for later maintenance.', implementationSummary: this._summarizeTextForWiki(latestAssistant || 'Implementation summary was not captured from chat yet.'), architecture: 'Project Chronicle records are written through an independent Markdown module.', changedFiles: ['Capture exact changed files after verification.'], dependencyNotes: 'Keep dependencies limited to TypeScript, Node fs/path, and VS Code extension APIs.', bugs: [], lessons: [ 'Write implementation notes as soon as a stable development step finishes.', 'Keep generated records project-specific.' ], createdAt }); this._chronicle.appendTimeline(profile, [`Development log created: ${result.relativePath}`], createdAt); vscode.window.showInformationMessage(`Chronicle development log saved: ${result.relativePath}`); await this._sendChronicleRecords(); this.injectSystemMessage(`**[Chronicle Development Saved]** \`${result.filePath}\``); } catch (err: any) { vscode.window.showErrorMessage(`Failed to write development record: ${err.message}`); } } async _writeChronicleBugFromInput() { const profile = this._getActiveChronicleProject(); if (!profile) { vscode.window.showWarningMessage('Select or create a Designer project first.'); return; } const title = await vscode.window.showInputBox({ prompt: 'Bug title', value: 'record-generation-issue' }); if (!title) return; const symptom = await vscode.window.showInputBox({ prompt: 'Bug symptom', value: 'Describe what failed or looked wrong.' }); if (symptom === undefined) return; const cause = await vscode.window.showInputBox({ prompt: 'Bug cause', value: 'Cause is not confirmed yet.' }); if (cause === undefined) return; const fix = await vscode.window.showInputBox({ prompt: 'Fix', value: 'Describe the fix or mitigation.' }); if (fix === undefined) return; try { const createdAt = new Date().toISOString(); const bugNumber = this._chronicle.nextBugNumber(profile); const result = this._chronicle.writeBug(profile, { title: title.trim(), symptom: symptom.trim(), cause: cause.trim(), fix: fix.trim(), prevention: 'Validate project selection, record path, and write permissions before generating files.', createdAt }, bugNumber); this._chronicle.appendTimeline(profile, [`Bug record created: ${result.relativePath}`], createdAt); vscode.window.showInformationMessage(`Chronicle bug record saved: ${result.relativePath}`); await this._sendChronicleRecords(); this.injectSystemMessage(`**[Chronicle Bug Saved]** \`${result.filePath}\``); } catch (err: any) { vscode.window.showErrorMessage(`Failed to write bug record: ${err.message}`); } } async _writeChronicleRetrospectiveFromInput() { const profile = this._getActiveChronicleProject(); if (!profile) { vscode.window.showWarningMessage('Select or create a Designer project first.'); return; } const title = await vscode.window.showInputBox({ prompt: 'Retrospective title', value: 'Project Chronicle Guard iteration' }); if (!title) return; const summary = await vscode.window.showInputBox({ prompt: 'Work summary', value: 'Completed an incremental development step and recorded the outcome.' }); if (summary === undefined) return; const wentWell = await vscode.window.showInputBox({ prompt: 'What went well? (comma-separated)', value: 'Kept the feature independent, Generated Markdown records, Preserved project context' }); if (wentWell === undefined) return; const toImprove = await vscode.window.showInputBox({ prompt: 'What should improve? (comma-separated)', value: 'More automatic question intent capture, Richer record editing UI' }); if (toImprove === undefined) return; const nextActions = await vscode.window.showInputBox({ prompt: 'Next actions (comma-separated)', value: 'Add tests, Improve Designer UI, Add event-based record capture' }); if (nextActions === undefined) return; try { const createdAt = new Date().toISOString(); const result = this._chronicle.writeRetrospective(profile, { title: title.trim(), summary: summary.trim(), wentWell: wentWell.split(',').map(item => item.trim()).filter(Boolean), toImprove: toImprove.split(',').map(item => item.trim()).filter(Boolean), nextActions: nextActions.split(',').map(item => item.trim()).filter(Boolean), createdAt }); this._chronicle.appendTimeline(profile, [`Retrospective created: ${result.relativePath}`], createdAt); vscode.window.showInformationMessage(`Chronicle retrospective saved: ${result.relativePath}`); await this._sendChronicleRecords(); this.injectSystemMessage(`**[Chronicle Retrospective Saved]** \`${result.filePath}\``); } catch (err: any) { vscode.window.showErrorMessage(`Failed to write retrospective record: ${err.message}`); } } async _autoWriteChronicleAfterPrompt() { const history = this._agent.getHistory(); const latestUser = [...history].reverse().find(message => message.role === 'user')?.content || ''; const latestAssistant = [...history].reverse().find(message => message.role === 'assistant')?.content || ''; const profile = this._getChronicleProjectForConversation(latestUser) || this._getActiveChronicleProject(); if (!profile) return; const recordType = this._inferAutoChronicleRecordType(latestUser, latestAssistant); if (!recordType) return; const signature = [ profile.projectId, recordType, this._summarizeTextForWiki(latestUser).slice(0, 240), this._summarizeTextForWiki(latestAssistant).slice(0, 240) ].join('|'); const lastSignature = this._context.globalState.get(SidebarChatProvider.lastAutoChronicleSignatureStateKey, ''); if (signature === lastSignature) return; try { this._chronicle.ensureProject(profile); const createdAt = new Date().toISOString(); const title = this._summarizeForTitle(latestUser || latestAssistant || 'Project Chronicle Auto Record'); const summary = this._summarizeTextForWiki(latestAssistant || latestUser); let result; if (recordType === 'bug') { const bugNumber = this._chronicle.nextBugNumber(profile); result = this._chronicle.writeBug(profile, { title, symptom: this._summarizeTextForWiki(latestUser || 'Issue was detected during the conversation.'), cause: 'Captured automatically from the current conversation. Confirm root cause during follow-up review if needed.', fix: summary, prevention: 'Keep automatic records tied to the active project and verify the relevant test or reproduction path.', createdAt }, bugNumber); } else if (recordType === 'planning') { result = this._chronicle.writePlanning(profile, { featureName: title, purpose: 'Capture the current planning or architecture direction before implementation continues.', background: summary, userIntent: this._summarizeTextForWiki(latestUser), sourceRequest: latestUser || 'No user request captured in the current chat.', scope: ['Continue from the active project conversation.', 'Use the selected project record folder automatically.'], outOfScope: ['Manual record type selection.', 'Blocking the user with record-writing prompts.'], developmentDirection: summary, dependencyStrategy: 'Prefer existing project modules and local Markdown records.', expectedValue: 'Future work can resume with the latest project intent and reasoning preserved.', successCriteria: ['The record is saved automatically after a meaningful project turn.', 'The record stays under the active project.'], developerInstruction: 'Use this record as lightweight context for the next development or review pass.', createdAt }); } else if (recordType === 'decision') { const adrNumber = this._chronicle.nextAdrNumber(profile); result = this._chronicle.writeDecision(profile, { title, status: 'accepted', context: this._summarizeTextForWiki(latestUser), decision: summary, reason: 'Captured automatically because the conversation contained decision-oriented language.', alternatives: [], consequences: ['Future prompts should treat this as project context unless the user changes direction.'], createdAt }, adrNumber); } else if (recordType === 'development') { result = this._chronicle.writeDevelopmentLog(profile, { featureName: title, purpose: 'Record the implementation or verification outcome from the current conversation.', implementationSummary: summary, architecture: 'Captured automatically from the assistant response and active project context.', changedFiles: this._extractChangedFilesFromText(latestAssistant), dependencyNotes: 'No new dependency note was captured automatically.', bugs: [], lessons: ['Automatic project records should be generated in the background when the turn contains durable project knowledge.'], createdAt }); } else { result = this._chronicle.writeDiscussion(profile, { title, userRequest: this._summarizeTextForWiki(latestUser || 'No user request captured in the current chat.'), interpretedIntent: 'Capture a meaningful project discussion automatically instead of requiring manual record selection.', questions: [], discussions: [summary], decisions: [], createdAt }); } this._chronicle.appendTimeline(profile, [`Auto ${recordType} record created: ${result.relativePath}`], createdAt); await this._context.globalState.update(SidebarChatProvider.lastAutoChronicleSignatureStateKey, signature); await this._sendChronicleRecords(); vscode.window.setStatusBarMessage(`Astra: Chronicle auto-saved ${recordType}`, 3500); } catch (err: any) { logError('Automatic Chronicle record write failed.', { error: err?.message || String(err), recordType }); } } _getChronicleProjectForConversation(text: string): ProjectProfile | null { const projectPath = this._extractLocalProjectPath(text); if (!projectPath) return null; const projects = this._getChronicleProjects(); const resolvedPath = path.resolve(projectPath); const existing = projects.find(project => { const root = project.projectRoot ? path.resolve(project.projectRoot) : ''; const recordRoot = path.resolve(project.recordRoot); return root === resolvedPath || recordRoot.startsWith(`${resolvedPath}${path.sep}`); }); if (existing) return existing; const projectName = path.basename(resolvedPath) || 'Current Project'; const now = new Date().toISOString(); return { projectId: this._slugify(projectName), projectName, projectRoot: resolvedPath, recordRoot: path.join(resolvedPath, 'docs', 'records', projectName), description: 'Auto-detected from the local project path in the conversation.', corePurpose: 'Capture project direction, architecture discussion, decisions, and development notes as Markdown.', targetUsers: ['Project developer'], avoidDirections: ['Do not mix records across projects.'], detailLevel: 'standard', createdAt: now, updatedAt: now }; } _extractLocalProjectPath(text: string): string | null { const match = text.match(/\/Volumes\/Data\/project\/Antigravity\/[^\s`"'<>]+/i); if (!match) return null; const candidate = match[0].replace(/[.,;:)\]]+$/, ''); try { if (fs.existsSync(candidate) && fs.statSync(candidate).isDirectory()) { return candidate; } } catch { return null; } return null; } _inferAutoChronicleRecordType(userText: string, assistantText: string): 'planning' | 'discussion' | 'decision' | 'development' | 'bug' | null { const combined = `${userText}\n${assistantText}`; if (!combined.trim()) return null; if (/(기록하지마|저장하지마|no\s+record|do\s+not\s+record)/i.test(combined)) return null; if (!/(프로젝트|코드|아키텍처|설계|개선|수정|구현|테스트|검증|이슈|문제|버그|오류|끊김|결정|방향|기록|지식|review|architecture|implement|fix|bug|issue|test|decision)/i.test(combined)) { return null; } if (/(버그|오류|에러|이슈|문제|끊김|안\s*됨|실패|bug|error|issue|failed|failure)/i.test(userText)) { return 'bug'; } if (/(수정 완료|개선 완료|구현 완료|패치|테스트.*통과|검증.*완료|변경.*파일|compile|jest|tsc|passed|implemented|fixed)/i.test(assistantText)) { return 'development'; } if (/(결정|확정|채택|방향은|하기로|하지 않기로|decision|decide|accepted)/i.test(combined)) { return 'decision'; } if (/(계획|설계|아키텍처|조사|방향|로드맵|mvp|planning|architecture|roadmap|design)/i.test(userText)) { return 'planning'; } if (/(개선|수정|구현|테스트|검증|패킹|compile|jest|tsc|implement|fix|test|verify)/i.test(combined)) { return 'development'; } return 'discussion'; } _extractChangedFilesFromText(text: string): string[] { const files = new Set(); for (const match of text.matchAll(/`([^`\n]+\.(?:ts|tsx|js|jsx|json|md|css|html|py|yml|yaml))`/gi)) { files.add(match[1].trim()); } return files.size > 0 ? Array.from(files).slice(0, 12) : ['No explicit changed file list was captured automatically.']; } async _writeChronicleRecord(recordType: string) { switch (recordType) { case 'planning': await this._writeChroniclePlanningFromCurrentChat(); break; case 'discussion': await this._writeChronicleDiscussionFromCurrentChat(); break; case 'decision': await this._writeChronicleDecisionFromInput(); break; case 'development': await this._writeChronicleDevelopmentFromCurrentChat(); break; case 'bug': await this._writeChronicleBugFromInput(); break; case 'retrospective': await this._writeChronicleRetrospectiveFromInput(); break; default: vscode.window.showWarningMessage('Select a Chronicle record type first.'); } } _getAgentsDir(): string { // 1) Explicit config override (works on any OS — useful on Windows or for skills outside the workspace). const configured = (vscode.workspace.getConfiguration('g1nation').get('agentSkillsPath', '') || '').trim(); const expanded = configured.startsWith('~/') || configured === '~' ? path.join(os.homedir(), configured.slice(1).replace(/^[\\/]/, '')) : configured; if (expanded && path.isAbsolute(expanded)) { if (!fs.existsSync(expanded)) { try { fs.mkdirSync(expanded, { recursive: true }); } catch { /* fall through to workspace */ } } if (fs.existsSync(expanded)) return expanded; } // 2) Default: /.agent/skills const workspaceFolders = vscode.workspace.workspaceFolders; if (workspaceFolders) { const localPath = path.join(workspaceFolders[0].uri.fsPath, '.agent', 'skills'); if (!fs.existsSync(localPath)) { fs.mkdirSync(localPath, { recursive: true }); } return localPath; } return ''; } async _sendAgentsList() { if (!this._view) return; const dir = this._getAgentsDir(); const agents = []; if (dir && fs.existsSync(dir)) { const files = fs.readdirSync(dir); for (const f of files) { if (f.endsWith('.md')) { agents.push({ name: f.replace('.md', ''), path: path.join(dir, f) }); } } } const lastPath = this._context.globalState.get(SidebarChatProvider.lastAgentStateKey, 'none'); this._view.webview.postMessage({ type: 'agentsList', value: agents, selected: lastPath }); void this._sendReadyStatus(); } async _handleProactiveSuggestion(context: string) { if (!this._view) return; let suggestion = ''; switch (context) { case 'settings_exploration': suggestion = '💡 **Tip:** 모델 설정을 최적화하여 답변 속도를 2배 이상 높일 수 있습니다. 설정에서 `Max Context Size`를 조정해보세요!'; break; case 'brain_sync_exploration': suggestion = '🧠 **Knowledge Sync:** 최근에 수정한 파일이 지식 베이스에 반영되지 않았나요? 지금 동기화 버튼을 눌러 최신 정보를 업데이트하세요.'; break; case 'agent_selection_exploration': suggestion = '🤖 **Agent Skills:** 특정 언어나 프레임워크에 특화된 에이전트 스킬을 선택하면 더 정확한 코드를 생성할 수 있습니다.'; break; default: suggestion = '💡 새로운 기능을 발견하셨나요? 궁금한 점이 있다면 언제든 물어보세요!'; } this._view.webview.postMessage({ type: 'streamStart' }); this._view.webview.postMessage({ type: 'streamChunk', value: `\n\n> [!TIP]\n> ${suggestion}\n` }); this._view.webview.postMessage({ type: 'streamEnd' }); } async _createAgent() { const name = await vscode.window.showInputBox({ prompt: 'Name of the new Agent (e.g., frontend_expert)', placeHolder: 'Agent name...' }); if (!name) return; const safeName = name.trim().replace(/[^a-zA-Z0-9_\-\u3131-\uD79D가-힣]/g, '_'); if (!safeName) return; const dir = this._getAgentsDir(); if (!dir) { vscode.window.showErrorMessage('Agent directory could not be determined.'); return; } const filePath = path.join(dir, `${safeName}.md`); if (!fs.existsSync(filePath)) { fs.writeFileSync(filePath, `# Agent Persona: ${safeName}\n\nAdd your instructions here...\n`, 'utf8'); } const doc = await vscode.workspace.openTextDocument(filePath); await vscode.window.showTextDocument(doc); await this._sendAgentsList(); } async _sendAgentContent(agentPath: string) { if (!this._view || !agentPath || agentPath === 'none') return; if (fs.existsSync(agentPath)) { const content = fs.readFileSync(agentPath, 'utf8'); const negativePrompt = this._context.globalState.get(`negativePrompt:${agentPath}`, ''); this._view.webview.postMessage({ type: 'agentContent', value: content, negativePrompt: negativePrompt }); await this._context.globalState.update(SidebarChatProvider.lastAgentStateKey, agentPath); } } async _updateAgent(agentPath: string, content: string, negativePrompt?: string) { if (!agentPath || agentPath === 'none') return; try { fs.writeFileSync(agentPath, content, 'utf8'); if (negativePrompt !== undefined) { await this._context.globalState.update(`negativePrompt:${agentPath}`, negativePrompt); } vscode.window.showInformationMessage('Agent skill updated successfully.'); } catch (err: any) { vscode.window.showErrorMessage(`Failed to update agent: ${err.message}`); } } async _deleteAgent(agentPath: string) { if (!agentPath || agentPath === 'none') return; try { const agentsDir = path.resolve(this._getAgentsDir()); const targetPath = path.resolve(agentPath); if (!targetPath.startsWith(`${agentsDir}${path.sep}`) || path.extname(targetPath) !== '.md') { vscode.window.showErrorMessage('Selected agent skill path is not valid.'); return; } if (!fs.existsSync(targetPath)) { await this._context.globalState.update(SidebarChatProvider.lastAgentStateKey, 'none'); await this._sendAgentsList(); return; } const confirm = await vscode.window.showWarningMessage( `Delete agent skill "${path.basename(targetPath, '.md')}"?`, { modal: true }, 'Delete Skill' ); if (confirm !== 'Delete Skill') return; fs.unlinkSync(targetPath); await this._context.globalState.update(SidebarChatProvider.lastAgentStateKey, 'none'); await this._sendAgentsList(); this._view?.webview.postMessage({ type: 'agentDeleted' }); vscode.window.showInformationMessage('Agent skill deleted successfully.'); } catch (err: any) { vscode.window.showErrorMessage(`Failed to delete agent: ${err.message}`); } } async _handlePrompt(data: any) { if (!this._view) return; const { value, model, internet, files, agentFile, negativePrompt, designerGuard, secondBrainTrace, secondBrainTraceDebug, brainProfileId } = data; this._currentNegativePrompt = negativePrompt || ''; const selectedBrainId = typeof brainProfileId === 'string' && brainProfileId && brainProfileId !== 'new' ? brainProfileId : getActiveBrainProfile().id; this._currentSessionBrainId = selectedBrainId; let agentSkillContext = undefined; if (agentFile && agentFile !== 'none' && fs.existsSync(agentFile)) { const fileContent = fs.readFileSync(agentFile, 'utf8'); // Guard: a freshly-created agent still has only the placeholder template // ("# Agent Persona: …\n\nAdd your instructions here…"). Treating that as a real // agent prompt just confuses the model — fall back to normal mode and tell the user. const body = fileContent.replace(/^?#\s*Agent\s*Persona\s*:.*$/im, '').trim(); const isPlaceholder = !body || /^add your instructions here/i.test(body); if (isPlaceholder) { logInfo('Selected agent has no real instructions — running without agent mode.', { agentFile }); this._view?.webview.postMessage({ type: 'lmStudioError', value: '선택한 에이전트에 내용이 없습니다 — 에이전트 프롬프트를 작성한 뒤 다시 시도하세요. (이번 응답은 에이전트 없이 처리합니다)' }); } else { agentSkillContext = fileContent; // Merge in any external skill .md files the user has mapped to this agent. We concatenate // into the same agentSkillContext blob so the rest of the pipeline (agent.ts, agent-mode // override) treats them identically to the agent's own .md — no further changes needed. try { const entry = getOrCreateAgentEntry(agentFile); const bundle = loadExternalSkills(entry.skillFolders); const block = formatSkillsAsPromptBlock(bundle); if (block) { agentSkillContext = `${agentSkillContext.trim()}\n\n${block}`; } } catch (e: any) { logError('External skill load failed.', { error: e?.message || String(e) }); } } } const designerContext = designerGuard !== false ? this._buildDesignerGuardContext() : undefined; // [File Processing v2] 파일 타입별 분류 처리 let processedPrompt = value || ''; let imageFiles: any[] | undefined = undefined; if (files && Array.isArray(files) && files.length > 0) { const textContents: string[] = []; const images: any[] = []; for (const file of files) { const name = file.name?.toLowerCase() || ''; const type = file.type || ''; if (name.endsWith('.pdf') || type === 'application/pdf') { // PDF: 서버사이드 텍스트 추출 (pdf-parse v2 API) + Vision 폴백 let pdfTextOk = false; try { const { PDFParse } = require('pdf-parse'); const rawBuffer = Buffer.from(file.data, 'base64'); const uint8 = new Uint8Array(rawBuffer.buffer, rawBuffer.byteOffset, rawBuffer.byteLength); const parser = new PDFParse(uint8); await parser.load(); const textResult = await parser.getText(); const extracted = (typeof textResult === 'string' ? textResult : (textResult?.text || '')).trim(); const cleanText = extracted.replace(/\n*-- \d+ of \d+ --\n*/g, '\n').trim(); if (cleanText && cleanText.length > 30) { textContents.push(`\n[PDF: ${file.name}]\n${cleanText}`); logInfo(`PDF text extracted successfully.`, { fileName: file.name, chars: cleanText.length }); pdfTextOk = true; } // [Vision Fallback] 텍스트가 비어있으면 페이지 이미지 추출 -> Vision 모델에 전달 if (!pdfTextOk) { logInfo(`PDF has no text layer. Extracting page screenshots for vision analysis.`, { fileName: file.name }); const screenshots = await parser.getScreenshot({ page: 1 }); if (screenshots?.pages && screenshots.pages.length > 0) { const maxPages = Math.min(screenshots.pages.length, 8); // 메모리 보호: 최대 8페이지 for (let i = 0; i < maxPages; i++) { const page = screenshots.pages[i]; if (page?.data) { const pageBase64 = Buffer.from(page.data).toString('base64'); images.push({ name: `${file.name}_page${i + 1}.png`, type: 'image/png', data: pageBase64 }); } } textContents.push(`\n[PDF: ${file.name}]\n(이미지 기반 PDF ${screenshots.total}페이지 중 ${maxPages}페이지를 이미지로 추출하여 Vision 분석합니다. 각 페이지 이미지를 참조하여 문서의 내용을 상세히 분석하고 한국어로 정리하세요.)`); logInfo(`PDF vision fallback: extracted ${maxPages} page screenshots.`, { fileName: file.name, totalPages: screenshots.total }); pdfTextOk = true; // Vision 분석으로 처리 완료 } } } catch (pdfError: any) { logError(`PDF processing failed.`, { fileName: file.name, error: pdfError?.message || String(pdfError) }); } // 최종 폴백: 텍스트도 없고 이미지 추출도 실패한 경우 if (!pdfTextOk) { textContents.push(`\n[PDF: ${file.name}]\n(PDF 분석에 실패했습니다. 이 파일을 텍스트로 변환하여 다시 시도해주세요.)`); } } else if ( type.startsWith('text/') || type === 'application/json' || /\.(txt|md|csv|json|js|ts|jsx|tsx|html|css|py|java|rs|go|yaml|yml|xml|toml|sql|sh)$/i.test(name) ) { // 텍스트 파일: base64 디코딩 try { const decoded = Buffer.from(file.data, 'base64').toString('utf-8'); textContents.push(`\n[FILE: ${file.name}]\n\`\`\`\n${decoded}\n\`\`\``); } catch (decodeError: any) { logError(`Text file decode failed.`, { fileName: file.name, error: decodeError?.message }); textContents.push(`\n[FILE: ${file.name}]\n(디코딩 오류)`); } } else if (type.startsWith('image/')) { // 이미지: 기존 vision 방식 유지 images.push(file); } else { // 미지원 타입: 파일명만 기록 textContents.push(`\n[ATTACHMENT: ${file.name}]\n(지원하지 않는 파일 형식: ${type || 'unknown'})`); } } // 추출된 텍스트를 프롬프트에 주입 if (textContents.length > 0) { processedPrompt = `${processedPrompt}\n\n--- 첨부 파일 내용 ---${textContents.join('\n')}`; } imageFiles = images.length > 0 ? images : undefined; } try { await this._agent.handlePrompt(processedPrompt, model, { internetEnabled: internet, visionContent: imageFiles, agentSkillContext, agentSkillFile: typeof agentFile === 'string' ? agentFile : undefined, negativePrompt, designerContext, secondBrainTraceEnabled: secondBrainTrace !== false, secondBrainTraceDebug: !!secondBrainTraceDebug, brainProfileId: selectedBrainId }); } catch (error: any) { logError('Prompt handling failed in sidebar provider.', { error: error?.message || String(error), promptPreview: summarizeText(value || '', 200) }); this._view.webview.postMessage({ type: 'error', value: error.message }); } finally { void this._sendReadyStatus(); } } _buildDesignerGuardContext(): string { return buildProjectChronicleGuardContext(this._getActiveChronicleProject()); } async _sendModels(force: boolean = false) { if (!this._view) return; if (this._modelDiscoveryInFlight) { logInfo('Model discovery already in progress, skipping.'); return; } this._modelDiscoveryInFlight = true; try { const config = getConfig(); const url = config.ollamaUrl; let defaultModel = config.defaultModel; let models: string[] = []; let online = false; const cache = this._modelsCache; const cacheFresh = !!cache && cache.url === url && (Date.now() - cache.fetchedAt) < SidebarChatProvider.MODELS_CACHE_TTL_MS; if (!force && cacheFresh && cache) { models = cache.models.slice(); online = cache.online; this._view.webview.postMessage({ type: 'engineStatus', value: { online, url } }); } else { const engine = resolveEngine(url); // 단일 엔진만 const modelsUrl = buildApiUrl(url, engine, 'models'); try { logInfo('Model discovery started.', { engine, modelsUrl, force }); const res = await fetch(modelsUrl, { signal: AbortSignal.timeout(5000) }); const rawText = await res.text(); if (!res.ok) { logError('Model discovery returned non-OK status.', { engine, modelsUrl, status: res.status, body: summarizeText(rawText, 300) }); } else { const data = rawText ? JSON.parse(rawText) as any : {}; models = engine === 'lmstudio' ? (data.data || []).map((m: any) => m.id) : (data.models || []).map((m: any) => m.name); if (models.length > 0) { logInfo('Model discovery succeeded.', { engine, count: models.length, modelsPreview: models.slice(0, 5) }); } } } catch (e: any) { logError('Model discovery failed.', { engine, modelsUrl, error: e?.message || String(e) }); } online = models.length > 0; this._modelsCache = { url, models: models.slice(), online, fetchedAt: Date.now() }; this._view.webview.postMessage({ type: 'engineStatus', value: { online, url } }); } if (models.length === 0) { models = defaultModel ? [defaultModel] : []; } const baseModel = defaultModel?.replace(/:\d+$/, ''); if (baseModel && baseModel !== defaultModel && models.includes(baseModel)) { defaultModel = baseModel; await vscode.workspace.getConfiguration('g1nation').update('defaultModel', defaultModel, vscode.ConfigurationTarget.Global); } if (models.length > 0 && !defaultModel) { // [State Persistence Fix v2] defaultModel이 완전히 비어있을 때만 첫 번째 모델로 설정 defaultModel = models[0]; await vscode.workspace.getConfiguration('g1nation').update('defaultModel', defaultModel, vscode.ConfigurationTarget.Global); } else if (models.length > 0 && defaultModel && !models.includes(defaultModel)) { // [State Persistence Fix v2] 저장된 모델이 로컬 엔진 목록에 없는 경우: // 강제 리셋하지 않고, 저장된 모델을 목록 선두에 추가하여 사용자 선택을 보존 logInfo('Saved model not in local engine list. Preserving user selection.', { saved: defaultModel, localModels: models.slice(0, 3) }); models.unshift(defaultModel); } const defaultIdx = models.indexOf(defaultModel); if (defaultIdx > 0) { models.splice(defaultIdx, 1); models.unshift(defaultModel); } let loadedModels: string[] = []; if (resolveEngine(url) === 'lmstudio' && this._lmStudio) { try { loadedModels = await this._lmStudio.loadedModels(); } catch (e) { logInfo('LM Studio loadedModels probe failed (non-fatal).', { error: (e as any)?.message ?? String(e) }); } } this._view.webview.postMessage({ type: 'modelsList', value: { models, selected: defaultModel, loadedModels } }); } catch (err) { logError('Model list update failed.', err); } finally { this._modelDiscoveryInFlight = false; } void this._sendReadyStatus(); } static _htmlTemplateCache: string | undefined; _getHtml(webview: vscode.Webview): string { if (!SidebarChatProvider._htmlTemplateCache) { const tplPath = path.join(this._extensionUri.fsPath, 'media', 'sidebar.html'); SidebarChatProvider._htmlTemplateCache = fs.readFileSync(tplPath, 'utf8'); } const mediaRoot = vscode.Uri.joinPath(this._extensionUri, 'media'); const stylesUri = webview.asWebviewUri(vscode.Uri.joinPath(mediaRoot, 'sidebar.css')).toString(); const scriptUri = webview.asWebviewUri(vscode.Uri.joinPath(mediaRoot, 'sidebar.js')).toString(); return SidebarChatProvider._htmlTemplateCache .replace('__STYLES_URI__', stylesUri) .replace('__SCRIPT_URI__', scriptUri); } }