mstro-app 0.4.29 → 0.4.33

package/server/services/websocket/session-history.ts

@@ -2,6 +2,7 @@
 // Licensed under the MIT License. See LICENSE file for details.
 
 import { existsSync, readdirSync, readFileSync, unlinkSync } from 'node:fs';
+import { readFile } from 'node:fs/promises';
 import { join } from 'node:path';
 import type { HandlerContext } from './handler-context.js';
 import type { WebSocketMessage, WSContext } from './types.js';
@@ -9,8 +10,9 @@ import type { WebSocketMessage, WSContext } from './types.js';
 export function handleHistoryMessage(ctx: HandlerContext, ws: WSContext, msg: WebSocketMessage, tabId: string, workingDir: string): void {
   switch (msg.type) {
     case 'getSessions': {
-      const result = getSessionsList(workingDir, msg.data?.limit ?? 20, msg.data?.offset ?? 0);
-      ctx.send(ws, { type: 'sessions', tabId, data: result });
+      getSessionsList(workingDir, msg.data?.limit ?? 20, msg.data?.offset ?? 0).then(result => {
+        ctx.send(ws, { type: 'sessions', tabId, data: result });
+      });
       break;
     }
     case 'getSessionsCount':
@@ -42,7 +44,7 @@ function getSessionsCount(workingDir: string): number {
   return readdirSync(sessionsDir).filter((name: string) => name.endsWith('.json')).length;
 }
 
-function getSessionsList(workingDir: string, limit: number = 20, offset: number = 0): { sessions: Array<Record<string, unknown> | null>; total: number; hasMore: boolean } {
+async function getSessionsList(workingDir: string, limit: number = 20, offset: number = 0): Promise<{ sessions: Array<Record<string, unknown> | null>; total: number; hasMore: boolean }> {
   const sessionsDir = join(workingDir, '.mstro', 'history');
 
   if (!existsSync(sessionsDir)) {
@@ -60,17 +62,17 @@ function getSessionsList(workingDir: string, limit: number = 20, offset: number
   const total = historyFiles.length;
   const pageFiles = historyFiles.slice(offset, offset + limit);
 
-  const sessions = pageFiles.map((filename: string) => {
+  const sessions = await Promise.all(pageFiles.map(async (filename: string) => {
     const historyPath = join(sessionsDir, filename);
     try {
-      const historyData = JSON.parse(readFileSync(historyPath, 'utf-8'));
-      return buildSessionSummary(historyData);
+      const raw = await readFile(historyPath, 'utf-8');
+      return buildSessionSummary(JSON.parse(raw));
     } catch {
       return null;
     }
-  }).filter(Boolean);
+  }));
 
-  return { sessions, total, hasMore: offset + limit < total };
+  return { sessions: sessions.filter(Boolean), total, hasMore: offset + limit < total };
 }
 
 function getSessionById(workingDir: string, sessionId: string): Record<string, unknown> | null {

package/server/services/websocket/skill-handlers.ts

@@ -0,0 +1,90 @@
+// Copyright (c) 2025-present Mstro, Inc. All rights reserved.
+// Licensed under the MIT License. See LICENSE file for details.
+
+import { existsSync, readdirSync, readFileSync } from 'node:fs';
+import { dirname, join } from 'node:path';
+import { fileURLToPath } from 'node:url';
+import { findSkillsDir } from '../../utils/paths.js';
+import type { HandlerContext } from './handler-context.js';
+import type { SkillEntry, WSContext } from './types.js';
+
+const __dirname = dirname(fileURLToPath(import.meta.url));
+const SYSTEM_AGENTS_DIR = join(__dirname, '..', 'plan', 'agents');
+
+function parseFrontmatter(content: string): Record<string, string> {
+	if (!content.startsWith('---')) return {};
+	const endIdx = content.indexOf('---', 3);
+	if (endIdx === -1) return {};
+	const yaml = content.slice(3, endIdx).trim();
+	const result: Record<string, string> = {};
+	for (const line of yaml.split('\n')) {
+		const colonIdx = line.indexOf(':');
+		if (colonIdx === -1) continue;
+		const key = line.slice(0, colonIdx).trim();
+		let val = line.slice(colonIdx + 1).trim();
+		if ((val.startsWith('"') && val.endsWith('"')) || (val.startsWith("'") && val.endsWith("'"))) {
+			val = val.slice(1, -1);
+		}
+		result[key] = val;
+	}
+	return result;
+}
+
+function scanProjectSkills(skillsDir: string): SkillEntry[] {
+	if (!existsSync(skillsDir)) return [];
+	const entries: SkillEntry[] = [];
+	for (const name of readdirSync(skillsDir, { withFileTypes: true })) {
+		if (!name.isDirectory()) continue;
+		const skillFile = join(skillsDir, name.name, 'SKILL.md');
+		if (!existsSync(skillFile)) continue;
+		try {
+			const content = readFileSync(skillFile, 'utf-8');
+			const fm = parseFrontmatter(content);
+			if (fm['user-invocable'] === 'false') continue;
+			entries.push({
+				name: fm.name || name.name,
+				displayName: `/${fm.name || name.name}`,
+				description: fm.description || '',
+				source: 'project',
+			});
+		} catch { /* skip unreadable files */ }
+	}
+	return entries;
+}
+
+function scanSystemAgents(agentsDir: string, seen: Set<string>): SkillEntry[] {
+	if (!existsSync(agentsDir)) return [];
+	const entries: SkillEntry[] = [];
+	for (const file of readdirSync(agentsDir)) {
+		if (!file.endsWith('.md')) continue;
+		const name = file.replace(/\.md$/, '');
+		if (seen.has(name)) continue;
+		try {
+			const content = readFileSync(join(agentsDir, file), 'utf-8');
+			const fm = parseFrontmatter(content);
+			if (fm['user-invocable'] === 'false') continue;
+			entries.push({
+				name: fm.name || name,
+				displayName: `/${fm.name || name}`,
+				description: fm.description || '',
+				source: 'system',
+			});
+		} catch { /* skip unreadable files */ }
+	}
+	return entries;
+}
+
+export function handleListSkills(ctx: HandlerContext, ws: WSContext, workingDir: string): void {
+	const skills: SkillEntry[] = [];
+
+	const projectSkillsDir = findSkillsDir(workingDir);
+	if (projectSkillsDir) {
+		skills.push(...scanProjectSkills(projectSkillsDir));
+	}
+
+	const seen = new Set(skills.map(s => s.name));
+	skills.push(...scanSystemAgents(SYSTEM_AGENTS_DIR, seen));
+
+	skills.sort((a, b) => a.name.localeCompare(b.name));
+	ctx.send(ws, { type: 'skillsList', data: { skills } });
+}

package/server/services/websocket/types.ts

@@ -168,7 +168,9 @@ export interface WebSocketMessage {
     | 'deployHttpRequest'
     // Deploy usage/health message types (cli→server)
     | 'deployUsageReport'
-    | 'deployAiHealthUpdate';
+    | 'deployAiHealthUpdate'
+    // Skill discovery
+    | 'listSkills';
   tabId?: string;
   terminalId?: string;
   // biome-ignore lint/suspicious/noExplicitAny: message envelope carries heterogeneous payloads
@@ -346,7 +348,9 @@ export interface WebSocketResponse {
     | 'deployHttpResponseChunk'
     | 'deployStatus'
     | 'deployUsageReportAck'
-    | 'deployAiHealthAck';
+    | 'deployAiHealthAck'
+    // Skill discovery response types
+    | 'skillsList';
   tabId?: string;
   terminalId?: string;
   // biome-ignore lint/suspicious/noExplicitAny: message envelope carries heterogeneous payloads
@@ -358,6 +362,13 @@ export interface ConnectionData {
   workingDir: string;
 }
 
+export interface SkillEntry {
+  name: string;
+  displayName: string;
+  description: string;
+  source: 'project' | 'system' | 'builtin';
+}
+
 // Extended autocomplete option with metadata
 export interface AutocompleteResult {
   value: string;

package/server/utils/paths.ts

@@ -8,7 +8,8 @@
  * Works correctly whether running from source or installed globally.
  */
 
-import { dirname, resolve } from 'node:path';
+import { existsSync } from 'node:fs';
+import { dirname, join, resolve } from 'node:path';
 import { fileURLToPath } from 'node:url';
 
 // ES module equivalent of __dirname for this file
@@ -29,3 +30,18 @@ export const MSTRO_ROOT = resolve(__dirname, '../..');
  */
 export const MCP_SERVER_PATH = resolve(MSTRO_ROOT, 'server/mcp/server.ts');
 
+/**
+ * Walk up from startDir looking for `.claude/skills/`. Returns the path if found, null otherwise.
+ */
+export function findSkillsDir(startDir: string): string | null {
+  let dir = startDir;
+  for (let i = 0; i < 10; i++) {
+    const candidate = join(dir, '.claude', 'skills');
+    if (existsSync(candidate)) return candidate;
+    const parent = dirname(dir);
+    if (parent === dir) break;
+    dir = parent;
+  }
+  return null;
+}
+