pi2web 0.2.0 → 0.2.6

package/dist/voice.js

@@ -1,13 +1,13 @@
 import { EventEmitter } from 'node:events';
 import { randomUUID } from 'node:crypto';
 const punctuation = new Set(['。', '！', '？', '!', '?', '.']);
-const closing = new Set(['”', '’', '"', '\'', '）', ')', '】', ']']);
-const endpoint = (base, path) => `${base.replace(/\/$/, '')}${base.replace(/\/$/, '').endsWith('/v1') ? '' : '/v1'}${path}`;
-const authHeaders = (key) => key ? { authorization: `Bearer ${key}` } : {};
+const closing = new Set(['”', '’', '"', "'", '）', ')', '】', ']']);
+export const endpoint = (base, path) => `${base.replace(/\/$/, '')}${base.replace(/\/$/, '').endsWith('/v1') ? '' : '/v1'}${path}`;
+export const authHeaders = (key) => key ? { authorization: `Bearer ${key}` } : {};
 const abortError = (error) => error instanceof Error && error.name === 'AbortError';
 const timeoutError = (error) => error instanceof Error && error.name === 'TimeoutError';
 /** Combines caller cancellation with a hard timeout so a stalled speech server cannot block the announcement queue forever. */
-const deadline = (ms, signal) => signal ? AbortSignal.any([signal, AbortSignal.timeout(ms)]) : AbortSignal.timeout(ms);
+export const deadline = (ms, signal) => signal ? AbortSignal.any([signal, AbortSignal.timeout(ms)]) : AbortSignal.timeout(ms);
 export function assistantText(message) {
     if (typeof message === 'string')
         return message.trim() || undefined;
@@ -19,7 +19,11 @@ export function assistantText(message) {
     if (typeof value.content === 'string')
         return value.content.trim() || undefined;
     if (Array.isArray(value.content)) {
-        const text = value.content.filter((part) => part?.type === 'text' && typeof part.text === 'string').map((part) => part.text).join('\n').trim();
+        const text = value.content
+            .filter((part) => part?.type === 'text' && typeof part.text === 'string')
+            .map((part) => part.text)
+            .join('\n')
+            .trim();
         return text || undefined;
     }
 }
@@ -42,8 +46,11 @@ export function prepareSummaryInput(text, maxChars = 32000) {
     cleaned = `${cleaned.slice(0, head)}\n\n[中间内容已省略]\n\n${cleaned.slice(-tail)}`;
     return cleaned;
 }
-function clipSummaryField(text, maxChars) {
-    const normalized = text.replace(/[ \t]+/g, ' ').replace(/\n{3,}/g, '\n\n').trim();
+export function clipSummaryField(text, maxChars) {
+    const normalized = text
+        .replace(/[ \t]+/g, ' ')
+        .replace(/\n{3,}/g, '\n\n')
+        .trim();
     if (normalized.length <= maxChars)
         return normalized;
     const head = Math.floor(maxChars * 0.375), tail = maxChars - head;
@@ -78,14 +85,24 @@ export function parseSummaryResult(text) {
     if (!summary)
         throw new Error('Summary endpoint returned no summary');
     const rawName = typeof value?.sessionName === 'string' ? value.sessionName.trim() : '';
-    const sessionName = rawName.replace(/[\r\n\t]+/g, ' ').replace(/\s{2,}/g, ' ').slice(0, 60).trim();
+    const sessionName = rawName
+        .replace(/[\r\n\t]+/g, ' ')
+        .replace(/\s{2,}/g, ' ')
+        .slice(0, 60)
+        .trim();
     return { summary, ...(sessionName ? { sessionName } : {}) };
 }
 export function formatSpokenSummary(summary, sessionName, language = 'zh-CN') {
-    const name = sessionName?.replace(/[\r\n\t]+/g, ' ').replace(/\s{2,}/g, ' ').slice(0, 60).trim();
+    const name = sessionName
+        ?.replace(/[\r\n\t]+/g, ' ')
+        .replace(/\s{2,}/g, ' ')
+        .slice(0, 60)
+        .trim();
     if (!name)
         return summary;
-    return language.toLowerCase().startsWith('zh') ? `会话${name}已完成：${summary}` : `Session ${name} completed: ${summary}`;
+    return language.toLowerCase().startsWith('zh')
+        ? `会话${name}已完成：${summary}`
+        : `Session ${name} completed: ${summary}`;
 }
 export class SentenceChunker {
     maxChars;
@@ -93,8 +110,13 @@ export class SentenceChunker {
     constructor(maxChars = 100) {
         this.maxChars = maxChars;
     }
-    add(delta) { this.buffer += delta; return this.consume(false); }
-    close() { return this.consume(true); }
+    add(delta) {
+        this.buffer += delta;
+        return this.consume(false);
+    }
+    close() {
+        return this.consume(true);
+    }
     consume(final) {
         const chunks = [];
         while (this.buffer) {
@@ -169,16 +191,35 @@ export class VoiceManager {
     generations = new Map();
     announcementQueue = Promise.resolve();
     closed = false;
+    /** Some backends rename or reject optional sampling parameters; the working shape is learned once and reused. */
+    summaryParams = { temperature: true, tokenLimitField: 'max_tokens' };
     config;
     constructor(config, fetcher = fetch) {
         this.fetcher = fetcher;
-        this.config = { language: 'zh-CN', maxInputChars: 32000, maxOutputTokens: 160, sampleRate: 24000, ttsFormat: 'pcm', requestTimeoutMs: 120000, ...config };
+        this.config = {
+            language: 'zh-CN',
+            maxInputChars: 32000,
+            sampleRate: 24000,
+            ttsFormat: 'pcm',
+            requestTimeoutMs: 120000,
+            ...config
+        };
+    }
+    subscribe(listener) {
+        this.emitter.on('event', listener);
+        return () => this.emitter.off('event', listener);
+    }
+    capabilities() {
+        return { tts: true, stt: !!this.config.sttModel };
+    }
+    recordUserPrompt(agentId, prompt) {
+        const value = prompt.trim();
+        if (value)
+            this.lastPrompt.set(agentId, value);
+    }
+    emit(agentId, event) {
+        this.emitter.emit('event', agentId, event);
     }
-    subscribe(listener) { this.emitter.on('event', listener); return () => this.emitter.off('event', listener); }
-    capabilities() { return { tts: true, stt: !!this.config.sttModel }; }
-    recordUserPrompt(agentId, prompt) { const value = prompt.trim(); if (value)
-        this.lastPrompt.set(agentId, value); }
-    emit(agentId, event) { this.emitter.emit('event', agentId, event); }
     handleAgentEvent(agentId, event, context = {}) {
         if (event.type === 'before_agent_start' && typeof event.prompt === 'string') {
             this.lastPrompt.set(agentId, event.prompt);
@@ -200,17 +241,26 @@ export class VoiceManager {
             const finalOutput = this.lastAssistant.get(agentId), userPrompt = this.lastPrompt.get(agentId) ?? '';
             this.lastAssistant.delete(agentId);
             this.lastPrompt.delete(agentId);
-            if (finalOutput)
+            // A run the user cancelled has no result worth speaking; Pi 0.99+ reports that as `aborted`.
+            if (finalOutput && event.aborted !== true)
                 void this.announce(agentId, { userPrompt, finalOutput, ...context });
         }
     }
-    cancel(agentId) { this.generations.set(agentId, (this.generations.get(agentId) ?? 0) + 1); const controller = this.active.get(agentId); if (controller) {
-        controller.abort();
-        this.active.delete(agentId);
-        this.emit(agentId, { type: 'voice_cancelled' });
-    } }
-    close() { this.closed = true; for (const id of [...new Set([...this.active.keys(), ...this.generations.keys()])])
-        this.cancel(id); this.emitter.removeAllListeners(); }
+    cancel(agentId) {
+        this.generations.set(agentId, (this.generations.get(agentId) ?? 0) + 1);
+        const controller = this.active.get(agentId);
+        if (controller) {
+            controller.abort();
+            this.active.delete(agentId);
+            this.emit(agentId, { type: 'voice_cancelled' });
+        }
+    }
+    close() {
+        this.closed = true;
+        for (const id of [...new Set([...this.active.keys(), ...this.generations.keys()])])
+            this.cancel(id);
+        this.emitter.removeAllListeners();
+    }
     async transcribe(audio, mimeType = 'audio/webm', filename = 'recording.webm') {
         if (!this.config.sttModel)
             throw Object.assign(new Error('Voice STT model is not configured'), { code: 'VOICE_STT_DISABLED' });
@@ -220,7 +270,16 @@ export class VoiceManager {
         form.append('file', new Blob([copy.buffer], { type: mimeType }), filename);
         form.append('model', this.config.sttModel);
         form.append('response_format', 'json');
-        const response = await this.fetcher(endpoint(this.config.speechBaseUrl, '/audio/transcriptions'), { method: 'POST', signal: deadline(this.config.requestTimeoutMs), headers: authHeaders(this.config.speechApiKey), body: form }).catch(error => { throw timeoutError(error) ? Object.assign(new Error(`Speech transcription timed out after ${this.config.requestTimeoutMs} ms`), { code: 'VOICE_STT_FAILED' }) : error; });
+        const response = await this.fetcher(endpoint(this.config.speechBaseUrl, '/audio/transcriptions'), {
+            method: 'POST',
+            signal: deadline(this.config.requestTimeoutMs),
+            headers: authHeaders(this.config.speechApiKey),
+            body: form
+        }).catch(error => {
+            throw timeoutError(error)
+                ? Object.assign(new Error(`Speech transcription timed out after ${this.config.requestTimeoutMs} ms`), { code: 'VOICE_STT_FAILED' })
+                : error;
+        });
         if (!response.ok)
             throw Object.assign(new Error(`Speech transcription failed (${response.status}): ${(await response.text()).slice(0, 500)}`), { code: 'VOICE_STT_FAILED' });
         const value = await response.json();
@@ -233,20 +292,26 @@ export class VoiceManager {
         const input = typeof value === 'string' ? { userPrompt: '', finalOutput: value } : value;
         this.cancel(agentId);
         const generation = this.generations.get(agentId);
-        const job = this.announcementQueue.then(() => this.closed || this.generations.get(agentId) !== generation ? undefined : this.runAnnouncement(agentId, input, generation));
+        const job = this.announcementQueue.then(() => this.closed || this.generations.get(agentId) !== generation
+            ? undefined
+            : this.runAnnouncement(agentId, input, generation));
         this.announcementQueue = job.then(() => undefined, () => undefined);
         return job;
     }
     async runAnnouncement(agentId, input, generation) {
         const controller = new AbortController(), playbackId = randomUUID();
         this.active.set(agentId, controller);
-        this.emit(agentId, { type: 'voice_start', playbackId, encoding: this.config.ttsFormat === 'mp3' ? 'mp3' : 'pcm_s16le', sampleRate: this.config.sampleRate, channels: 1 });
+        this.emit(agentId, {
+            type: 'voice_start',
+            playbackId,
+            encoding: this.config.ttsFormat === 'mp3' ? 'mp3' : 'pcm_s16le',
+            sampleRate: this.config.sampleRate,
+            channels: 1
+        });
         let rawResult = '', spokenSummary = '';
         try {
             const summaryInput = prepareSummaryContext(input, this.config.maxInputChars);
-            const response = await this.fetcher(endpoint(this.config.summaryBaseUrl, '/chat/completions'), { method: 'POST', signal: deadline(this.config.requestTimeoutMs, controller.signal), headers: { 'content-type': 'application/json', ...authHeaders(this.config.summaryApiKey) }, body: JSON.stringify({ model: this.config.summaryModel, stream: true, temperature: 0.2, max_tokens: this.config.maxOutputTokens, messages: [{ role: 'system', content: this.summaryPrompt() }, { role: 'user', content: summaryInput }] }) });
-            if (!response.ok)
-                throw new Error(`Summary request failed (${response.status}): ${(await response.text()).slice(0, 500)}`);
+            const response = await this.requestSummary(summaryInput, controller.signal);
             for await (const delta of sseText(response))
                 rawResult += delta;
             const result = parseSummaryResult(rawResult), sessionName = input.sessionName?.trim() || result.sessionName;
@@ -256,7 +321,11 @@ export class VoiceManager {
                     await input.setSessionName(result.sessionName);
                 }
                 catch (error) {
-                    this.emit(agentId, { type: 'voice_session_name_error', playbackId, message: error.message });
+                    this.emit(agentId, {
+                        type: 'voice_session_name_error',
+                        playbackId,
+                        message: error.message
+                    });
                 }
             }
             this.emit(agentId, { type: 'voice_summary_delta', playbackId, text: spokenSummary });
@@ -264,19 +333,28 @@ export class VoiceManager {
             for (const sentence of [...chunker.add(spokenSummary), ...chunker.close()])
                 await this.speak(agentId, playbackId, sentence, controller.signal);
             if (this.active.get(agentId) === controller)
-                this.emit(agentId, { type: 'voice_end', playbackId, summary: spokenSummary, sessionName: result.sessionName });
+                this.emit(agentId, {
+                    type: 'voice_end',
+                    playbackId,
+                    summary: spokenSummary,
+                    sessionName: result.sessionName
+                });
         }
         catch (error) {
             if (!abortError(error) && this.active.get(agentId) === controller) {
                 this.emit(agentId, { type: 'voice_error', playbackId, message: error.message });
-                const fallback = this.config.language.toLowerCase().startsWith('zh') ? 'Agent 已完成，等待输入。' : 'The agent has finished and is ready for input.';
+                const fallback = this.config.language.toLowerCase().startsWith('zh')
+                    ? 'Agent 已完成，等待输入。'
+                    : 'The agent has finished and is ready for input.';
                 if (!spokenSummary)
                     try {
                         this.emit(agentId, { type: 'voice_summary_delta', playbackId, text: fallback });
                         await this.speak(agentId, playbackId, fallback, controller.signal);
                         this.emit(agentId, { type: 'voice_end', playbackId, summary: fallback });
                     }
-                    catch { /* The text notification remains available when TTS also fails. */ }
+                    catch {
+                        /* The text notification remains available when TTS also fails. */
+                    }
                 else
                     this.emit(agentId, { type: 'voice_end', playbackId, summary: spokenSummary, partial: true });
             }
@@ -288,16 +366,89 @@ export class VoiceManager {
                 controller.abort();
         }
     }
-    summaryPrompt() { const language = this.config.language.toLowerCase().startsWith('zh') ? '简体中文' : '与用户输入相同的主要语言'; return `你负责把一次 Pi 编程任务压缩成准确、自然、可直接朗读的完成通知，并在需要时为 Session 命名。\n\n输入是一个 JSON 对象：\n- userPrompt：用户本轮真正想完成的任务。\n- piFinalOutput：Pi 最后一次回复，包含完成情况、改动、验证结果和后续事项。\n- sessionNeedsName：只有为 true 时才需要生成 Session 名称。\n\n先结合 userPrompt 判断目标，再以 piFinalOutput 为事实依据总结。不要把用户的要求误说成已经完成；没有明确证据时不要声称测试通过或任务成功。摘要使用${language}，写 2 到 4 个短句，最多 120 个中文字符或 70 个英文单词。优先交代：是否完成、最重要的结果或修改、测试/验证结果、失败原因，以及用户必须采取的下一步。省略代码、命令、URL、完整路径、哈希、冗长文件清单、日志和 Markdown；不要使用标题、列表、“总结如下”等套话。\n\n如果 sessionNeedsName 为 true，请根据 userPrompt 生成一个具体、简短、便于检索的名称：中文建议 6 到 18 个字，英文建议 3 到 8 个词；使用任务主题或目标，不写“新会话”“任务总结”等空泛名称，不带句号、引号、Emoji 或路径。如果 sessionNeedsName 为 false，sessionName 必须为 null。\n\n只输出一个合法 JSON 对象，不要输出 Markdown 代码块、解释或任何额外文字。有名称时使用 {"summary":"适合直接朗读的摘要","sessionName":"简短名称"}；无需命名时使用 {"summary":"适合直接朗读的摘要","sessionName":null}。summary 必须是 JSON 字符串，sessionName 必须是 JSON 字符串或 null。`; }
+    /** Retries once per unsupported parameter so a strict model still produces a summary instead of failing the announcement. */
+    async requestSummary(summaryInput, signal) {
+        for (let attempt = 0;; attempt++) {
+            const body = {
+                model: this.config.summaryModel,
+                stream: true,
+                messages: [
+                    { role: 'system', content: this.summaryPrompt() },
+                    { role: 'user', content: summaryInput }
+                ]
+            };
+            if (this.summaryParams.temperature && typeof this.config.summaryTemperature === 'number')
+                body.temperature = this.config.summaryTemperature;
+            if (this.summaryParams.tokenLimitField && this.config.maxOutputTokens)
+                body[this.summaryParams.tokenLimitField] = this.config.maxOutputTokens;
+            const response = await this.fetcher(endpoint(this.config.summaryBaseUrl, '/chat/completions'), {
+                method: 'POST',
+                signal: deadline(this.config.requestTimeoutMs, signal),
+                headers: { 'content-type': 'application/json', ...authHeaders(this.config.summaryApiKey) },
+                body: JSON.stringify(body)
+            });
+            if (response.ok)
+                return response;
+            const detail = (await response.text()).slice(0, 500);
+            if (response.status === 400 && attempt < 2 && this.relaxSummaryParams(detail))
+                continue;
+            throw new Error(`Summary request failed (${response.status}): ${detail}`);
+        }
+    }
+    relaxSummaryParams(detail) {
+        const text = detail.toLowerCase();
+        if (!/unsupported|not supported|unrecognized|unknown|invalid/.test(text))
+            return false;
+        let changed = false;
+        if (this.summaryParams.temperature && text.includes('temperature')) {
+            this.summaryParams.temperature = false;
+            changed = true;
+        }
+        const field = this.summaryParams.tokenLimitField;
+        if (field === 'max_tokens' && text.includes('max_tokens')) {
+            this.summaryParams.tokenLimitField = text.includes('max_completion_tokens')
+                ? 'max_completion_tokens'
+                : null;
+            changed = true;
+        }
+        else if (field === 'max_completion_tokens' && text.includes('max_completion_tokens')) {
+            this.summaryParams.tokenLimitField = null;
+            changed = true;
+        }
+        return changed;
+    }
+    summaryPrompt() {
+        const language = this.config.language.toLowerCase().startsWith('zh')
+            ? '简体中文'
+            : '与用户输入相同的主要语言';
+        return `你负责把一次 Pi 编程任务压缩成准确、自然、可直接朗读的完成通知，并在需要时为 Session 命名。\n\n输入是一个 JSON 对象：\n- userPrompt：用户本轮真正想完成的任务。\n- piFinalOutput：Pi 最后一次回复，包含完成情况、改动、验证结果和后续事项。\n- sessionNeedsName：只有为 true 时才需要生成 Session 名称。\n\n先结合 userPrompt 判断目标，再以 piFinalOutput 为事实依据总结。不要把用户的要求误说成已经完成；没有明确证据时不要声称测试通过或任务成功。摘要使用${language}，写 2 到 4 个短句，最多 120 个中文字符或 70 个英文单词。优先交代：是否完成、最重要的结果或修改、测试/验证结果、失败原因，以及用户必须采取的下一步。省略代码、命令、URL、完整路径、哈希、冗长文件清单、日志和 Markdown；不要使用标题、列表、“总结如下”等套话。\n\n如果 sessionNeedsName 为 true，请根据 userPrompt 生成一个具体、简短、便于检索的名称：中文建议 6 到 18 个字，英文建议 3 到 8 个词；使用任务主题或目标，不写“新会话”“任务总结”等空泛名称，不带句号、引号、Emoji 或路径。如果 sessionNeedsName 为 false，sessionName 必须为 null。\n\n只输出一个合法 JSON 对象，不要输出 Markdown 代码块、解释或任何额外文字。有名称时使用 {"summary":"适合直接朗读的摘要","sessionName":"简短名称"}；无需命名时使用 {"summary":"适合直接朗读的摘要","sessionName":null}。summary 必须是 JSON 字符串，sessionName 必须是 JSON 字符串或 null。`;
+    }
     async speak(agentId, playbackId, text, signal) {
-        const response = await this.fetcher(endpoint(this.config.speechBaseUrl, '/audio/speech'), { method: 'POST', signal: deadline(this.config.requestTimeoutMs, signal), headers: { 'content-type': 'application/json', ...authHeaders(this.config.speechApiKey) }, body: JSON.stringify({ model: this.config.ttsModel, voice: this.config.ttsVoice, input: text, response_format: this.config.ttsFormat, sample_rate: this.config.sampleRate, stream_format: 'audio' }) });
+        const response = await this.fetcher(endpoint(this.config.speechBaseUrl, '/audio/speech'), {
+            method: 'POST',
+            signal: deadline(this.config.requestTimeoutMs, signal),
+            headers: { 'content-type': 'application/json', ...authHeaders(this.config.speechApiKey) },
+            body: JSON.stringify({
+                model: this.config.ttsModel,
+                voice: this.config.ttsVoice,
+                input: text,
+                response_format: this.config.ttsFormat,
+                sample_rate: this.config.sampleRate,
+                stream_format: 'audio'
+            })
+        });
         if (!response.ok)
             throw new Error(`Speech synthesis failed (${response.status}): ${(await response.text()).slice(0, 500)}`);
         if (this.config.ttsFormat === 'mp3') {
             const audio = Buffer.from(await response.arrayBuffer());
             if (!audio.length)
                 throw new Error('Speech synthesis returned no audio');
-            this.emit(agentId, { type: 'voice_audio_chunk', playbackId, encoding: 'mp3', audio: audio.toString('base64') });
+            this.emit(agentId, {
+                type: 'voice_audio_chunk',
+                playbackId,
+                encoding: 'mp3',
+                audio: audio.toString('base64')
+            });
             return;
         }
         if (!response.body)
@@ -313,7 +464,12 @@ export class VoiceManager {
             const bytes = pending ? Buffer.concat([pending, Buffer.from(value)]) : Buffer.from(value);
             const evenLength = bytes.length & ~1;
             if (evenLength)
-                this.emit(agentId, { type: 'voice_audio_chunk', playbackId, sampleRate: this.config.sampleRate, audio: bytes.subarray(0, evenLength).toString('base64') });
+                this.emit(agentId, {
+                    type: 'voice_audio_chunk',
+                    playbackId,
+                    sampleRate: this.config.sampleRate,
+                    audio: bytes.subarray(0, evenLength).toString('base64')
+                });
             pending = evenLength < bytes.length ? Buffer.from(bytes.subarray(evenLength)) : undefined;
         }
         if (pending)

package/dist/workspaces.js

@@ -7,31 +7,88 @@ export class WorkspaceStore {
     constructor(items = []) {
         this.items = items;
     }
-    list() { return this.items.map(x => ({ ...x })); }
-    replace(items) { this.items = items.map(item => ({ ...item, rootPath: path.resolve(item.rootPath) })); }
-    add(label, rootPath) { const root = path.resolve(rootPath); let rootStat; try {
-        rootStat = statSync(root);
-    }
-    catch (error) {
-        if (error.code === 'ENOENT' || error.code === 'ENOTDIR')
-            throw Object.assign(new Error('Workspace root does not exist'), { code: 'WORKSPACE_ROOT_NOT_FOUND' });
-        throw error;
-    } if (!rootStat.isDirectory())
-        throw Object.assign(new Error('Workspace root is not a directory'), { code: 'WORKSPACE_ROOT_NOT_DIRECTORY' }); const key = process.platform === 'win32' ? root.toLowerCase() : root; if (this.items.some(item => (process.platform === 'win32' ? item.rootPath.toLowerCase() : item.rootPath) === key))
-        throw Object.assign(new Error('Workspace root already exists'), { code: 'WORKSPACE_ALREADY_EXISTS' }); const item = { id: randomUUID(), label, rootPath: root, createdAt: new Date().toISOString() }; this.items.push(item); return item; }
-    get(id) { return this.items.find(x => x.id === id); }
-    async resolve(ws, relative = '.') { const candidate = path.resolve(ws.rootPath, relative); const root = await realpath(ws.rootPath); let actual; try {
-        actual = await realpath(candidate);
-    }
-    catch {
-        throw Object.assign(new Error('Path does not exist'), { code: 'PATH_NOT_FOUND' });
-    } if (actual !== root && !actual.startsWith(root + path.sep))
-        throw Object.assign(new Error('Path is outside the workspace root'), { code: 'WORKSPACE_PATH_OUTSIDE_ROOT' }); return actual; }
-    async tree(id, relative = '.') { const ws = this.get(id); if (!ws)
-        throw Object.assign(new Error('Workspace not found'), { code: 'WORKSPACE_NOT_FOUND' }); const dir = await this.resolve(ws, relative); if (!(await stat(dir)).isDirectory())
-        throw new Error('Not a directory'); const entries = await readdir(dir, { withFileTypes: true }); return Promise.all(entries.sort((a, b) => Number(b.isDirectory()) - Number(a.isDirectory()) || a.name.localeCompare(b.name)).map(async (e) => { const p = path.join(dir, e.name); const s = await lstat(p); return { name: e.name, type: e.isDirectory() ? 'directory' : 'file', size: s.size, modifiedAt: s.mtime.toISOString() }; })); }
-    async stat(id, relative = '.') { const ws = this.get(id); if (!ws)
-        throw Object.assign(new Error('Workspace not found'), { code: 'WORKSPACE_NOT_FOUND' }); const p = await this.resolve(ws, relative); const s = await lstat(p); return { path: relative, type: s.isDirectory() ? 'directory' : s.isFile() ? 'file' : 'other', size: s.size, modifiedAt: s.mtime.toISOString() }; }
+    list() {
+        return this.items.map(x => ({ ...x }));
+    }
+    replace(items) {
+        this.items = items.map(item => ({ ...item, rootPath: path.resolve(item.rootPath) }));
+    }
+    add(label, rootPath) {
+        const root = path.resolve(rootPath);
+        let rootStat;
+        try {
+            rootStat = statSync(root);
+        }
+        catch (error) {
+            if (error.code === 'ENOENT' ||
+                error.code === 'ENOTDIR')
+                throw Object.assign(new Error('Workspace root does not exist'), { code: 'WORKSPACE_ROOT_NOT_FOUND' });
+            throw error;
+        }
+        if (!rootStat.isDirectory())
+            throw Object.assign(new Error('Workspace root is not a directory'), {
+                code: 'WORKSPACE_ROOT_NOT_DIRECTORY'
+            });
+        const key = process.platform === 'win32' ? root.toLowerCase() : root;
+        if (this.items.some(item => (process.platform === 'win32' ? item.rootPath.toLowerCase() : item.rootPath) === key))
+            throw Object.assign(new Error('Workspace root already exists'), { code: 'WORKSPACE_ALREADY_EXISTS' });
+        const item = { id: randomUUID(), label, rootPath: root, createdAt: new Date().toISOString() };
+        this.items.push(item);
+        return item;
+    }
+    get(id) {
+        return this.items.find(x => x.id === id);
+    }
+    async resolve(ws, relative = '.') {
+        const candidate = path.resolve(ws.rootPath, relative);
+        const root = await realpath(ws.rootPath);
+        let actual;
+        try {
+            actual = await realpath(candidate);
+        }
+        catch {
+            throw Object.assign(new Error('Path does not exist'), { code: 'PATH_NOT_FOUND' });
+        }
+        if (actual !== root && !actual.startsWith(root + path.sep))
+            throw Object.assign(new Error('Path is outside the workspace root'), {
+                code: 'WORKSPACE_PATH_OUTSIDE_ROOT'
+            });
+        return actual;
+    }
+    async tree(id, relative = '.') {
+        const ws = this.get(id);
+        if (!ws)
+            throw Object.assign(new Error('Workspace not found'), { code: 'WORKSPACE_NOT_FOUND' });
+        const dir = await this.resolve(ws, relative);
+        if (!(await stat(dir)).isDirectory())
+            throw new Error('Not a directory');
+        const entries = await readdir(dir, { withFileTypes: true });
+        return Promise.all(entries
+            .sort((a, b) => Number(b.isDirectory()) - Number(a.isDirectory()) || a.name.localeCompare(b.name))
+            .map(async (e) => {
+            const p = path.join(dir, e.name);
+            const s = await lstat(p);
+            return {
+                name: e.name,
+                type: e.isDirectory() ? 'directory' : 'file',
+                size: s.size,
+                modifiedAt: s.mtime.toISOString()
+            };
+        }));
+    }
+    async stat(id, relative = '.') {
+        const ws = this.get(id);
+        if (!ws)
+            throw Object.assign(new Error('Workspace not found'), { code: 'WORKSPACE_NOT_FOUND' });
+        const p = await this.resolve(ws, relative);
+        const s = await lstat(p);
+        return {
+            path: relative,
+            type: s.isDirectory() ? 'directory' : s.isFile() ? 'file' : 'other',
+            size: s.size,
+            modifiedAt: s.mtime.toISOString()
+        };
+    }
     async file(id, relative, offset = 0, limit = 1024 * 1024) {
         const ws = this.get(id);
         if (!ws)
@@ -55,6 +112,14 @@ export class WorkspaceStore {
             await handle.close();
         }
         const binary = chunk.includes(0);
-        return { path: relative, size: s.size, modifiedAt: s.mtime.toISOString(), binary, content: binary ? undefined : chunk.toString('utf8'), offset, limit: chunk.length };
+        return {
+            path: relative,
+            size: s.size,
+            modifiedAt: s.mtime.toISOString(),
+            binary,
+            content: binary ? undefined : chunk.toString('utf8'),
+            offset,
+            limit: chunk.length
+        };
     }
 }