mstro-app 0.3.7 → 0.3.9

package/server/services/plan/types.ts

@@ -0,0 +1,164 @@
+// Copyright (c) 2025-present Mstro, Inc. All rights reserved.
+// Licensed under the MIT License. See LICENSE file for details.
+
+/**
+ * Plan Types — Project Plan Spec (PPS) data structures
+ *
+ * These types represent the parsed contents of .pm/ directory files.
+ */
+
+// ============================================================================
+// Project Config (project.md front matter)
+// ============================================================================
+
+export interface ProjectConfig {
+  name: string;
+  id: string;
+  created: string;
+  status: 'active' | 'paused' | 'completed' | 'archived';
+  estimation: 'fibonacci' | 'tshirt' | 'hours' | 'none';
+  idPrefixes: Record<string, string>;
+  workflows: WorkflowStatus[];
+  labels: string[];
+  teams: Team[];
+}
+
+export interface WorkflowStatus {
+  status: string;
+  category: 'unstarted' | 'started' | 'completed' | 'cancelled';
+  description: string;
+}
+
+export interface Team {
+  name: string;
+  description?: string;
+}
+
+// ============================================================================
+// Project State (STATE.md front matter + body)
+// ============================================================================
+
+export interface ProjectState {
+  project: string;
+  currentSprint: string | null;
+  activeMilestone: string | null;
+  paused: boolean;
+  lastSession: string | null;
+  // Parsed from markdown body sections:
+  readyToWork: IssueSummary[];
+  inProgress: IssueSummary[];
+  blocked: IssueSummary[];
+  recentlyCompleted: IssueSummary[];
+  warnings: string[];
+}
+
+export interface IssueSummary {
+  id: string;
+  title: string;
+  path: string;
+  priority: string;
+  blockedBy?: string;
+}
+
+// ============================================================================
+// Issue / Epic / Bug (backlog/*.md)
+// ============================================================================
+
+export interface Issue {
+  id: string;
+  title: string;
+  type: 'issue' | 'epic' | 'bug' | 'task';
+  status: string;
+  priority: string;
+  estimate: number | string | null;
+  labels: string[];
+  epic: string | null;
+  sprint: string | null;
+  milestone: string | null;
+  assigned: string | null;
+  created: string;
+  updated: string | null;
+  due: string | null;
+  blockedBy: string[];
+  blocks: string[];
+  relatesTo: string[];
+  // Body sections
+  description: string;
+  acceptanceCriteria: AcceptanceCriterion[];
+  technicalNotes: string | null;
+  filesToModify: string[];
+  activity: string[];
+  // Children (for epics)
+  children: string[];
+  // Progress (for epics)
+  progress: string | null;
+  // Full markdown body
+  body: string;
+  // File path relative to .pm/
+  path: string;
+}
+
+export interface AcceptanceCriterion {
+  text: string;
+  checked: boolean;
+}
+
+// ============================================================================
+// Sprint (sprints/*.md)
+// ============================================================================
+
+export interface Sprint {
+  id: string;
+  title: string;
+  status: 'planned' | 'active' | 'completed';
+  start: string;
+  end: string;
+  goal: string;
+  capacity: number | null;
+  committed: number | null;
+  completed: number | null;
+  issues: SprintIssueSummary[];
+  path: string;
+}
+
+export interface SprintIssueSummary {
+  id: string;
+  title: string;
+  path: string;
+  points: number | string | null;
+  status: string;
+}
+
+// ============================================================================
+// Milestone (milestones/*.md)
+// ============================================================================
+
+export interface Milestone {
+  id: string;
+  title: string;
+  status: 'planned' | 'in_progress' | 'completed' | 'missed';
+  targetDate: string | null;
+  progress: string | null;
+  definition: string;
+  epics: MilestoneEpicSummary[];
+  path: string;
+}
+
+export interface MilestoneEpicSummary {
+  id: string;
+  title: string;
+  path: string;
+  progress: string;
+}
+
+// ============================================================================
+// Plan full state (sent on planInit)
+// ============================================================================
+
+export interface PlanFullState {
+  project: ProjectConfig;
+  state: ProjectState;
+  issues: Issue[];
+  sprints: Sprint[];
+  milestones: Milestone[];
+}

package/server/services/plan/watcher.ts

@@ -0,0 +1,73 @@
+// Copyright (c) 2025-present Mstro, Inc. All rights reserved.
+// Licensed under the MIT License. See LICENSE file for details.
+
+/**
+ * Plan Watcher — Watches .pm/ (or legacy .plan/) directory for changes and broadcasts updates.
+ *
+ * Uses fs.watch with debouncing to batch rapid changes.
+ */
+
+import { type FSWatcher, watch } from 'node:fs';
+import type { HandlerContext } from '../websocket/handler-context.js';
+import { parsePlanDirectory, resolvePmDir } from './parser.js';
+
+export class PlanWatcher {
+  private watcher: FSWatcher | null = null;
+  private debounceTimer: ReturnType<typeof setTimeout> | null = null;
+  private readonly workingDir: string;
+  private readonly ctx: HandlerContext;
+  private started = false;
+
+  constructor(workingDir: string, ctx: HandlerContext) {
+    this.workingDir = workingDir;
+    this.ctx = ctx;
+  }
+
+  start(): void {
+    if (this.started) return;
+
+    const planDir = resolvePmDir(this.workingDir);
+    if (!planDir) return;
+
+    try {
+      this.watcher = watch(planDir, { recursive: true }, (_event, filename) => {
+        if (!filename || !filename.endsWith('.md')) return;
+        this.debounce();
+      });
+      this.started = true;
+    } catch {
+      // fs.watch recursive may not be supported on all platforms
+    }
+  }
+
+  stop(): void {
+    if (this.watcher) {
+      this.watcher.close();
+      this.watcher = null;
+    }
+    if (this.debounceTimer) {
+      clearTimeout(this.debounceTimer);
+      this.debounceTimer = null;
+    }
+    this.started = false;
+  }
+
+  private debounce(): void {
+    if (this.debounceTimer) clearTimeout(this.debounceTimer);
+    this.debounceTimer = setTimeout(() => {
+      this.handleChange();
+    }, 200);
+  }
+
+  private handleChange(): void {
+    try {
+      // Always do a full reparse — concurrent changes may affect multiple files
+      const fullState = parsePlanDirectory(this.workingDir);
+      if (fullState) {
+        this.ctx.broadcastToAll({ type: 'planStateUpdated', data: fullState });
+      }
+    } catch {
+      // Ignore parse errors from partial writes
+    }
+  }
+}

package/server/services/websocket/file-explorer-handlers.ts

@@ -542,6 +542,22 @@ function handleFindDefinition(ctx: HandlerContext, ws: WSContext, msg: WebSocket
       `(fn|struct|enum|trait|type|const|static|mod)\\s+${s}\\b`,
       `impl\\s+${s}\\b`,
     ],
+    swift: (s) => [
+      `(func|class|struct|enum|protocol|typealias|actor)\\s+${s}\\b`,
+      `(let|var)\\s+${s}\\b`,
+      `extension\\s+${s}\\b`,
+    ],
+    kotlin: (s) => [
+      `(fun|class|object|interface|typealias|enum\\s+class)\\s+${s}\\b`,
+      `(val|var)\\s+${s}\\b`,
+    ],
+    java: (s) => [
+      `(class|interface|enum)\\s+${s}\\b`,
+      `(public|private|protected|static)?\\s*(void|int|String|boolean|\\w+)\\s+${s}\\s*\\(`,
+    ],
+    ruby: (s) => [
+      `(def|class|module)\\s+${s}\\b`,
+    ],
   };
 
   const LANGUAGE_GLOBS: Record<string, string> = {
@@ -550,6 +566,10 @@ function handleFindDefinition(ctx: HandlerContext, ws: WSContext, msg: WebSocket
     python: '*.py',
     go: '*.go',
     rust: '*.rs',
+    swift: '*.swift',
+    kotlin: '*.{kt,kts}',
+    java: '*.java',
+    ruby: '*.rb',
   };
 
   const patterns = DEFINITION_PATTERNS[language] || DEFINITION_PATTERNS.typescript;

package/server/services/websocket/handler.ts

@@ -19,6 +19,8 @@ import { handleFileExplorerMessage, handleFileMessage } from './file-explorer-ha
 import { FileUploadHandler } from './file-upload-handler.js';
 import { handleGitMessage } from './git-handlers.js';
 import type { HandlerContext, UsageReporter } from './handler-context.js';
+import { handlePlanMessage } from './plan-handlers.js';
+import { handleQualityMessage } from './quality-handlers.js';
 import { handleHistoryMessage, handleSessionMessage, initializeTab, resumeHistoricalSession } from './session-handlers.js';
 import { SessionRegistry } from './session-registry.js';
 import { generateNotificationSummary, handleGetSettings, handleUpdateSettings } from './settings-handlers.js';
@@ -220,6 +222,32 @@ export class WebSocketImproviseHandler implements HandlerContext {
         return handleRemoveTab(this, ws, tabId, workingDir);
       case 'markTabViewed':
         return handleMarkTabViewed(this, ws, tabId, workingDir);
+      // Quality messages
+      case 'qualityDetectTools':
+      case 'qualityScan':
+      case 'qualityInstallTools':
+      case 'qualityCodeReview':
+      case 'qualityFixIssues':
+      case 'qualityLoadState':
+      case 'qualitySaveDirectories':
+        return handleQualityMessage(this, ws, msg, tabId, workingDir);
+      // Plan messages
+      case 'planInit':
+      case 'planGetState':
+      case 'planListIssues':
+      case 'planGetIssue':
+      case 'planGetSprint':
+      case 'planGetMilestone':
+      case 'planCreateIssue':
+      case 'planUpdateIssue':
+      case 'planDeleteIssue':
+      case 'planScaffold':
+      case 'planPrompt':
+      case 'planExecute':
+      case 'planPause':
+      case 'planStop':
+      case 'planResume':
+        return handlePlanMessage(this, ws, msg, tabId, workingDir, permission);
       // Settings messages
       case 'getSettings':
         return handleGetSettings(this, ws);
@@ -291,6 +319,4 @@ export class WebSocketImproviseHandler implements HandlerContext {
     this.sessions.delete(sessionId);
   }
 
-  cleanupStaleSessions(): void {
-  }
 }