@ng-annotate/vite-plugin 0.3.9 → 0.3.11

package/dist/index.js

@@ -1,11 +1,16 @@
 import { createWsHandler } from './ws-handler.js';
 import { createManifestPlugin } from './manifest.js';
+import { setProjectRoot, getStorePath } from './store.js';
 // Default export for Angular's angular.json `plugins` option — Angular calls it as a function.
 export default ngAnnotateMcp;
 export function ngAnnotateMcp() {
     const mainPlugin = {
         name: 'ng-annotate-mcp',
         apply: 'serve',
+        configResolved(config) {
+            setProjectRoot(config.root);
+            config.logger.info(`[ng-annotate] store → ${getStorePath()}`);
+        },
         configureServer(server) {
             createWsHandler(server);
         },

package/dist/store.d.ts

@@ -29,7 +29,9 @@ export interface Annotation {
     selectionText?: string;
 }
 export declare const STORE_DIR = ".ng-annotate";
-export declare const STORE_PATH: string;
+/** Call from the Vite `configResolved` hook to pin the store to the Vite project root. */
+export declare function setProjectRoot(root: string): void;
+export declare function getStorePath(): string;
 export declare function ensureStore(): void;
 type WatcherFn = (annotation: Annotation) => void;
 export declare function addWatcher(fn: WatcherFn): () => void;

package/dist/store.js

@@ -4,15 +4,21 @@ import { v4 as uuidv4 } from 'uuid';
 const EMPTY_STORE = { sessions: {}, annotations: {} };
 // ─── File paths ───────────────────────────────────────────────────────────────
 export const STORE_DIR = '.ng-annotate';
-const PROJECT_ROOT = process.env.NG_ANNOTATE_PROJECT_ROOT ?? process.cwd();
-export const STORE_PATH = path.join(PROJECT_ROOT, STORE_DIR, 'store.json');
+let projectRoot = process.env.NG_ANNOTATE_PROJECT_ROOT ?? process.cwd();
+/** Call from the Vite `configResolved` hook to pin the store to the Vite project root. */
+export function setProjectRoot(root) {
+    projectRoot = root;
+}
+export function getStorePath() {
+    return path.join(projectRoot, STORE_DIR, 'store.json');
+}
 // ─── Store init ───────────────────────────────────────────────────────────────
 export function ensureStore() {
-    const dir = path.join(PROJECT_ROOT, STORE_DIR);
+    const dir = path.join(projectRoot, STORE_DIR);
     if (!fs.existsSync(dir))
         fs.mkdirSync(dir, { recursive: true });
-    if (!fs.existsSync(STORE_PATH)) {
-        fs.writeFileSync(STORE_PATH, JSON.stringify(EMPTY_STORE, null, 2), 'utf8');
+    if (!fs.existsSync(getStorePath())) {
+        fs.writeFileSync(getStorePath(), JSON.stringify(EMPTY_STORE, null, 2), 'utf8');
     }
 }
 // ─── File locking ─────────────────────────────────────────────────────────────
@@ -25,10 +31,10 @@ let writeQueue = Promise.resolve();
 async function withLock(fn) {
     ensureStore();
     const result = writeQueue.then(async () => {
-        const raw = fs.readFileSync(STORE_PATH, 'utf8');
+        const raw = fs.readFileSync(getStorePath(), 'utf8');
         const data = JSON.parse(raw);
         const updated = await fn(data);
-        fs.writeFileSync(STORE_PATH, JSON.stringify(updated, null, 2), 'utf8');
+        fs.writeFileSync(getStorePath(), JSON.stringify(updated, null, 2), 'utf8');
         return updated;
     });
     // Keep the queue alive even when this task fails
@@ -38,7 +44,7 @@ async function withLock(fn) {
 }
 function readStore() {
     ensureStore();
-    return JSON.parse(fs.readFileSync(STORE_PATH, 'utf8'));
+    return JSON.parse(fs.readFileSync(getStorePath(), 'utf8'));
 }
 const watchers = new Set();
 export function addWatcher(fn) {

package/dist/ws-handler.js

@@ -22,10 +22,17 @@ export function createWsHandler(server) {
     wss.on('connection', (ws, req) => {
         void (async () => {
             const referer = req.headers.referer ?? req.headers.origin ?? '';
-            const session = await store.createSession({ active: true, url: referer });
-            const sessionId = session.id;
-            safeSend(ws, { type: 'session:created', session });
-            sessionSockets.set(sessionId, ws);
+            let sessionId;
+            try {
+                const session = await store.createSession({ active: true, url: referer });
+                sessionId = session.id;
+                safeSend(ws, { type: 'session:created', session });
+                sessionSockets.set(sessionId, ws);
+            }
+            catch (err) {
+                server.config.logger.error(`[ng-annotate] Failed to create session: ${String(err)}`);
+                return;
+            }
             ws.on('message', (raw) => {
                 void (async () => {
                     let msg;
@@ -35,25 +42,28 @@ export function createWsHandler(server) {
                     catch {
                         return;
                     }
-                    if (msg.type === 'annotation:create') {
-                        const annotation = await store.createAnnotation({
-                            ...msg.payload,
-                            sessionId,
-                        });
-                        safeSend(ws, { type: 'annotation:created', annotation });
-                    }
-                    else if (msg.type === 'annotation:reply') {
-                        const annotation = await store.addReply(msg.id, { author: 'user', message: msg.message });
-                        if (annotation) {
-                            safeSend(ws, { type: 'annotation:updated', annotation });
+                    try {
+                        if (msg.type === 'annotation:create') {
+                            const annotation = await store.createAnnotation({
+                                ...msg.payload,
+                                sessionId,
+                            });
+                            safeSend(ws, { type: 'annotation:created', annotation });
                         }
-                    }
-                    else {
-                        const annotation = await store.updateAnnotation(msg.id, { status: 'dismissed' });
-                        if (annotation) {
-                            safeSend(ws, { type: 'annotation:updated', annotation });
+                        else if (msg.type === 'annotation:reply') {
+                            const annotation = await store.addReply(msg.id, { author: 'user', message: msg.message });
+                            if (annotation)
+                                safeSend(ws, { type: 'annotation:updated', annotation });
+                        }
+                        else {
+                            const annotation = await store.updateAnnotation(msg.id, { status: 'dismissed' });
+                            if (annotation)
+                                safeSend(ws, { type: 'annotation:updated', annotation });
                         }
                     }
+                    catch (err) {
+                        server.config.logger.error(`[ng-annotate] Failed to process message: ${String(err)}`);
+                    }
                 })();
             });
             ws.on('close', () => {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ng-annotate/vite-plugin",
-  "version": "0.3.9",
+  "version": "0.3.11",
   "type": "module",
   "description": "Vite plugin for ng-annotate-mcp — WebSocket annotation bridge and component manifest for Angular dev tooling",
   "keywords": [