wave-agent-sdk 0.2.1 → 0.5.0

package/src/agent.ts

@@ -1,3 +1,4 @@
+import { ForegroundTaskManager } from "./managers/foregroundTaskManager.js";
 import {
   MessageManager,
   type MessageManagerCallbacks,
@@ -13,12 +14,14 @@ import { McpManager, type McpManagerCallbacks } from "./managers/mcpManager.js";
 import { LspManager } from "./managers/lspManager.js";
 import { BashManager } from "./managers/bashManager.js";
 import {
-  BackgroundBashManager,
-  type BackgroundBashManagerCallbacks,
-} from "./managers/backgroundBashManager.js";
+  BackgroundTaskManager,
+  type BackgroundTaskManagerCallbacks,
+} from "./managers/backgroundTaskManager.js";
 import { SlashCommandManager } from "./managers/slashCommandManager.js";
 import { PluginManager } from "./managers/pluginManager.js";
 import { HookManager } from "./managers/hookManager.js";
+import { ReversionManager } from "./managers/reversionManager.js";
+import { ReversionService } from "./services/reversionService.js";
 import { PermissionManager } from "./managers/permissionManager.js";
 import { PlanManager } from "./managers/planManager.js";
 import type {
@@ -36,6 +39,8 @@ import type {
   Usage,
   PermissionMode,
   PermissionCallback,
+  BackgroundTask,
+  ForegroundTask,
 } from "./types/index.js";
 import { MemoryRuleManager } from "./managers/MemoryRuleManager.js";
 import { LiveConfigManager } from "./managers/liveConfigManager.js";
@@ -70,6 +75,8 @@ export interface AgentOptions {
   fastModel?: string;
   maxInputTokens?: number;
   maxTokens?: number;
+  /** Preferred language for agent communication */
+  language?: string;
 
   // Existing options (preserved)
   callbacks?: AgentCallbacks;
@@ -96,10 +103,12 @@ export interface AgentOptions {
 
 export interface AgentCallbacks
   extends MessageManagerCallbacks,
-    BackgroundBashManagerCallbacks,
+    BackgroundTaskManagerCallbacks,
     McpManagerCallbacks,
     SubagentManagerCallbacks {
+  onTasksChange?: (tasks: BackgroundTask[]) => void;
   onPermissionModeChange?: (mode: PermissionMode) => void;
+  onBackgroundCurrentTask?: () => void;
 }
 
 export class Agent {
@@ -107,7 +116,7 @@ export class Agent {
   private aiManager: AIManager;
 
   private bashManager: BashManager | null = null;
-  private backgroundBashManager: BackgroundBashManager;
+  private backgroundTaskManager: BackgroundTaskManager;
   private logger?: Logger; // Add optional logger property
   private toolManager: ToolManager; // Add tool registry instance
   private mcpManager: McpManager; // Add MCP manager instance
@@ -119,8 +128,10 @@ export class Agent {
   private pluginManager: PluginManager; // Add plugin manager instance
   private skillManager: SkillManager; // Add skill manager instance
   private hookManager: HookManager; // Add hooks manager instance
+  private reversionManager: ReversionManager;
   private memoryRuleManager: MemoryRuleManager; // Add memory rule manager instance
   private liveConfigManager: LiveConfigManager; // Add live configuration manager
+  private foregroundTaskManager: ForegroundTaskManager;
   private configurationService: ConfigurationService; // Add configuration service
   private workdir: string; // Working directory
   private systemPrompt?: string; // Custom system prompt
@@ -160,44 +171,8 @@ export class Agent {
     );
   }
 
-  /**
-   * Update agent configuration dynamically
-   *
-   * @param config - Configuration updates for gateway, model, and token limit
-   */
-  public updateConfig(config: {
-    gateway?: Partial<GatewayConfig>;
-    model?: Partial<ModelConfig>;
-    maxInputTokens?: number;
-    maxTokens?: number;
-  }): void {
-    if (config.gateway) {
-      this.options.apiKey = config.gateway.apiKey ?? this.options.apiKey;
-      this.options.baseURL = config.gateway.baseURL ?? this.options.baseURL;
-      this.options.defaultHeaders =
-        config.gateway.defaultHeaders ?? this.options.defaultHeaders;
-      this.options.fetchOptions =
-        config.gateway.fetchOptions ?? this.options.fetchOptions;
-      this.options.fetch = config.gateway.fetch ?? this.options.fetch;
-    }
-
-    if (config.model) {
-      this.options.agentModel =
-        config.model.agentModel ?? this.options.agentModel;
-      this.options.fastModel = config.model.fastModel ?? this.options.fastModel;
-      this.options.maxTokens = config.model.maxTokens ?? this.options.maxTokens;
-    }
-
-    if (config.maxInputTokens !== undefined) {
-      this.options.maxInputTokens = config.maxInputTokens;
-    }
-
-    if (config.maxTokens !== undefined) {
-      this.options.maxTokens = config.maxTokens;
-    }
-
-    // Re-validate configuration after update
-    this.resolveAndValidateConfig();
+  public getLanguage(): string | undefined {
+    return this.configurationService.resolveLanguage(this.options.language);
   }
 
   /**
@@ -231,8 +206,15 @@ export class Agent {
     // Store options for dynamic configuration resolution
     this.options = options;
 
-    this.backgroundBashManager = new BackgroundBashManager({
-      callbacks,
+    this.foregroundTaskManager = new ForegroundTaskManager();
+
+    this.backgroundTaskManager = new BackgroundTaskManager({
+      callbacks: {
+        ...callbacks,
+        onTasksChange: (tasks) => {
+          callbacks.onTasksChange?.(tasks);
+        },
+      },
       workdir: this.workdir,
     });
     this.mcpManager = new McpManager({ callbacks, logger: this.logger }); // Initialize MCP manager
@@ -261,17 +243,23 @@ export class Agent {
       workdir: this.workdir,
     });
 
+    // Initialize memory rule manager
+    this.memoryRuleManager = new MemoryRuleManager({
+      workdir: this.workdir,
+    });
+
     // Initialize MessageManager
     this.messageManager = new MessageManager({
       callbacks,
       workdir: this.workdir,
       logger: this.logger,
+      memoryRuleManager: this.memoryRuleManager,
     });
 
-    // Initialize memory rule manager
-    this.memoryRuleManager = new MemoryRuleManager({
-      workdir: this.workdir,
-    });
+    // Initialize ReversionManager
+    this.reversionManager = new ReversionManager(
+      new ReversionService(this.messageManager.getTranscriptPath()),
+    );
 
     // Create a wrapper for canUseTool that triggers notification hooks
     const canUseToolWithNotification: PermissionCallback | undefined =
@@ -321,8 +309,11 @@ export class Agent {
       lspManager: this.lspManager,
       logger: this.logger,
       permissionManager: this.permissionManager,
+      reversionManager: this.reversionManager,
       permissionMode: options.permissionMode, // Let PermissionManager handle defaultMode resolution
       canUseToolCallback: canUseToolWithNotification,
+      backgroundTaskManager: this.backgroundTaskManager,
+      foregroundTaskManager: this.foregroundTaskManager,
     }); // Initialize tool registry with permission support
     this.liveConfigManager = new LiveConfigManager({
       workdir: this.workdir,
@@ -353,8 +344,11 @@ export class Agent {
       getGatewayConfig: () => this.getGatewayConfig(),
       getModelConfig: () => this.getModelConfig(),
       getMaxInputTokens: () => this.getMaxInputTokens(),
+      getLanguage: () => this.getLanguage(),
       hookManager: this.hookManager,
       onUsageAdded: (usage) => this.addUsage(usage),
+      backgroundTaskManager: this.backgroundTaskManager,
+      memoryRuleManager: this.memoryRuleManager,
     });
 
     // Initialize AI manager with resolved configuration
@@ -362,7 +356,7 @@ export class Agent {
       messageManager: this.messageManager,
       toolManager: this.toolManager,
       logger: this.logger,
-      backgroundBashManager: this.backgroundBashManager,
+      backgroundTaskManager: this.backgroundTaskManager,
       hookManager: this.hookManager,
       permissionManager: this.permissionManager,
       callbacks: {
@@ -374,9 +368,11 @@ export class Agent {
       workdir: this.workdir,
       systemPrompt: this.systemPrompt,
       stream: this.stream, // Pass streaming mode flag
+      reversionManager: this.reversionManager,
       getGatewayConfig: () => this.getGatewayConfig(),
       getModelConfig: () => this.getModelConfig(),
       getMaxInputTokens: () => this.getMaxInputTokens(),
+      getLanguage: () => this.getLanguage(),
       getEnvironmentVars: () => this.configurationService.getEnvironmentVars(), // Provide access to configuration environment variables
     });
 
@@ -384,6 +380,7 @@ export class Agent {
     this.slashCommandManager = new SlashCommandManager({
       messageManager: this.messageManager,
       aiManager: this.aiManager,
+      backgroundTaskManager: this.backgroundTaskManager,
       workdir: this.workdir,
       logger: this.logger,
     });
@@ -478,29 +475,8 @@ export class Agent {
   }
 
   /** Get combined memory content (project + user + modular rules) */
-  public get combinedMemory(): string {
-    let combined = "";
-    if (this._projectMemoryContent.trim()) {
-      combined += this._projectMemoryContent;
-    }
-    if (this._userMemoryContent.trim()) {
-      if (combined) {
-        combined += "\n\n";
-      }
-      combined += this._userMemoryContent;
-    }
-
-    // Add modular memory rules
-    const filesInContext = this.messageManager.getFilesInContext();
-    const activeRules = this.memoryRuleManager.getActiveRules(filesInContext);
-    if (activeRules.length > 0) {
-      if (combined) {
-        combined += "\n\n";
-      }
-      combined += activeRules.map((r) => r.content).join("\n\n");
-    }
-
-    return combined;
+  public async getCombinedMemory(): Promise<string> {
+    return this.messageManager.getCombinedMemory();
   }
 
   /** Get AI loading status */
@@ -523,12 +499,25 @@ export class Agent {
     id: string,
     filter?: string,
   ): { stdout: string; stderr: string; status: string } | null {
-    return this.backgroundBashManager.getOutput(id, filter);
+    return this.backgroundTaskManager.getOutput(id, filter);
   }
 
   /** Kill background bash shell */
   public killBackgroundShell(id: string): boolean {
-    return this.backgroundBashManager.killShell(id);
+    return this.backgroundTaskManager.stopTask(id);
+  }
+
+  /** Get background task output */
+  public getBackgroundTaskOutput(
+    id: string,
+    filter?: string,
+  ): { stdout: string; stderr: string; status: string } | null {
+    return this.backgroundTaskManager.getOutput(id, filter);
+  }
+
+  /** Stop background task */
+  public stopBackgroundTask(id: string): boolean {
+    return this.backgroundTaskManager.stopTask(id);
   }
 
   /**
@@ -667,6 +656,9 @@ export class Agent {
     // Resolve and validate configuration after loading settings.json
     this.resolveAndValidateConfig();
 
+    // Set global logger for SDK-wide access before discovering rules
+    setGlobalLogger(this.logger || null);
+
     // Discover modular memory rules
     try {
       await this.memoryRuleManager.discoverRules();
@@ -674,9 +666,6 @@ export class Agent {
       this.logger?.error("Failed to discover memory rules:", error);
     }
 
-    // Set global logger for SDK-wide access after validation
-    setGlobalLogger(this.logger || null);
-
     // Initialize live configuration reload
     try {
       this.logger?.debug("Initializing live configuration reload...");
@@ -955,14 +944,42 @@ export class Agent {
     this.slashCommandManager.abortCurrentCommand();
   }
 
+  /**
+   * Register a foreground task that can be backgrounded
+   */
+  public registerForegroundTask(task: ForegroundTask): void {
+    this.foregroundTaskManager.registerForegroundTask(task);
+  }
+
+  /**
+   * Unregister a foreground task
+   */
+  public unregisterForegroundTask(id: string): void {
+    this.foregroundTaskManager.unregisterForegroundTask(id);
+  }
+
+  /**
+   * Background the current foreground task
+   */
+  public async backgroundCurrentTask(): Promise<void> {
+    await this.foregroundTaskManager.backgroundCurrentTask();
+    this.options.callbacks?.onBackgroundCurrentTask?.();
+
+    // If there was no foreground task (e.g. a tool that doesn't register one),
+    // or even if there was, we should stop the AI recursion loop.
+    if (!this.foregroundTaskManager.hasActiveTasks()) {
+      this.aiManager.abortRecursion();
+    }
+  }
+
   /** Destroy managers, clean up resources */
   public async destroy(): Promise<void> {
     await this.messageManager.saveSession();
     this.abortAIMessage(); // This will abort tools including Task tool (subagents)
     this.abortBashCommand();
     this.abortSlashCommand();
-    // Cleanup background bash manager
-    this.backgroundBashManager.cleanup();
+    // Cleanup background task manager
+    this.backgroundTaskManager.cleanup();
     // Cleanup MCP connections
     await this.mcpManager.cleanup();
     // Cleanup LSP connections
@@ -983,6 +1000,23 @@ export class Agent {
     // Cleanup memory store
   }
 
+  /**
+   * Get a subagent instance by its ID
+   * @param subagentId - The ID of the subagent instance
+   */
+  public getSubagentInstance(
+    subagentId: string,
+  ): import("./managers/subagentManager.js").SubagentInstance | null {
+    return this.subagentManager.getInstance(subagentId);
+  }
+
+  /**
+   * Trigger the rewind UI callback
+   */
+  public triggerShowRewind(): void {
+    this.messageManager.triggerShowRewind();
+  }
+
   /**
    * Send a message to the AI agent with optional images
    *
@@ -1122,12 +1156,18 @@ export class Agent {
       const typeLabel = type === "project" ? "Project Memory" : "User Memory";
       const storagePath = "AGENTS.md";
 
-      this.messageManager.addMemoryBlock(
-        `${typeLabel}: ${memoryText}`,
-        true,
-        type,
-        storagePath,
-      );
+      const messages = this.messageManager.getMessages();
+      const lastMessage = messages[messages.length - 1];
+      if (lastMessage && lastMessage.role === "assistant") {
+        lastMessage.blocks.push({
+          type: "memory",
+          content: `${typeLabel}: ${memoryText}`,
+          isSuccess: true,
+          memoryType: type,
+          storagePath,
+        });
+        this.messageManager.setMessages(messages);
+      }
     } catch (error) {
       // Add failed MemoryBlock to the last assistant message
       const memoryText = message.substring(1).trim();
@@ -1136,12 +1176,18 @@ export class Agent {
       const errorMessage =
         error instanceof Error ? error.message : String(error);
 
-      this.messageManager.addMemoryBlock(
-        `${typeLabel}: ${memoryText} - Error: ${errorMessage}`,
-        false,
-        type,
-        storagePath,
-      );
+      const messages = this.messageManager.getMessages();
+      const lastMessage = messages[messages.length - 1];
+      if (lastMessage && lastMessage.role === "assistant") {
+        lastMessage.blocks.push({
+          type: "memory",
+          content: `${typeLabel}: ${memoryText} - Error: ${errorMessage}`,
+          isSuccess: false,
+          memoryType: type,
+          storagePath,
+        });
+        this.messageManager.setMessages(messages);
+      }
     }
   }
 
@@ -1189,6 +1235,13 @@ export class Agent {
     return this.slashCommandManager.getCustomCommands();
   }
 
+  /**
+   * Register a custom slash command
+   */
+  public registerSlashCommand(command: SlashCommand): void {
+    this.slashCommandManager.registerCommand(command);
+  }
+
   /**
    * Get the current permission mode
    */
@@ -1210,30 +1263,41 @@ export class Agent {
   }
 
   /**
-   * Handle plan mode transition, generating or clearing plan file path
-   * @param mode - The current effective permission mode
+   * Truncate history to a specific index and revert file changes.
+   * @param index - The index of the user message to truncate to.
    */
-  private handlePlanModeTransition(mode: PermissionMode): void {
-    if (mode === "plan") {
-      this.planManager
-        .getOrGeneratePlanFilePath()
-        .then(({ path }) => {
-          this.logger?.debug("Plan file path generated", { path });
-          this.permissionManager.setPlanFilePath(path);
-        })
-        .catch((error) => {
-          this.logger?.error("Failed to generate plan file path", error);
-        });
-    } else {
-      this.permissionManager.setPlanFilePath(undefined);
-    }
+  public async truncateHistory(index: number): Promise<void> {
+    await this.messageManager.truncateHistory(index, this.reversionManager);
+  }
+
+  /**
+   * Get the current plan file path (for testing and UI)
+   */
+  public getPlanFilePath(): string | undefined {
+    return this.permissionManager.getPlanFilePath();
+  }
+
+  /**
+   * Get all currently allowed rules (for testing and UI)
+   */
+  public getAllowedRules(): string[] {
+    return this.permissionManager.getAllowedRules();
+  }
+
+  /**
+   * Check permission for a tool call (for testing)
+   */
+  public async checkPermission(
+    context: import("./types/permissions.js").ToolPermissionContext,
+  ): Promise<import("./types/permissions.js").PermissionDecision> {
+    return this.permissionManager.checkPermission(context);
   }
 
   /**
    * Add a persistent permission rule
    * @param rule - The rule to add (e.g., "Bash(ls)")
    */
-  private async addPermissionRule(rule: string): Promise<void> {
+  public async addPermissionRule(rule: string): Promise<void> {
     // 1. Expand rule if it's a Bash command
     let rulesToAdd = [rule];
     const bashMatch = rule.match(/^Bash\((.*)\)$/);
@@ -1266,4 +1330,24 @@ export class Agent {
       }
     }
   }
+
+  /**
+   * Handle plan mode transition, generating or clearing plan file path
+   * @param mode - The current effective permission mode
+   */
+  private handlePlanModeTransition(mode: PermissionMode): void {
+    if (mode === "plan") {
+      this.planManager
+        .getOrGeneratePlanFilePath()
+        .then(({ path }) => {
+          this.logger?.debug("Plan file path generated", { path });
+          this.permissionManager.setPlanFilePath(path);
+        })
+        .catch((error) => {
+          this.logger?.error("Failed to generate plan file path", error);
+        });
+    } else {
+      this.permissionManager.setPlanFilePath(undefined);
+    }
+  }
 }

package/src/constants/prompts.ts

@@ -44,8 +44,11 @@ export const SUBAGENT_POLICY = `
 - You should proactively use the ${TASK_TOOL_NAME} tool with specialized agents when the task at hand matches the agent's description.
 - VERY IMPORTANT: When exploring the codebase to gather context or to answer a question that is not a needle query for a specific file/class/function, it is CRITICAL that you use the ${TASK_TOOL_NAME} tool with subagent_type=Explore instead of running search commands directly.`;
 
-export const FILE_TOOL_POLICY = `
-- Use specialized tools instead of bash commands when possible, as this provides a better user experience. For file operations, use dedicated tools: ${READ_TOOL_NAME} for reading files instead of cat/head/tail, ${EDIT_TOOL_NAME}/${MULTI_EDIT_TOOL_NAME} for editing instead of sed/awk, ${WRITE_TOOL_NAME} for creating files instead of cat with heredoc or echo redirection, and ${LS_TOOL_NAME}/${GLOB_TOOL_NAME}/${GREP_TOOL_NAME} for searching and listing files instead of find/ls/grep.`;
+export const FILE_TOOL_POLICY_PREFIX = `\n- Use specialized tools instead of bash commands when possible, as this provides a better user experience. For file operations, use dedicated tools:`;
+export const READ_FILE_POLICY = ` ${READ_TOOL_NAME} for reading files instead of cat/head/tail`;
+export const EDIT_FILE_POLICY = ` ${EDIT_TOOL_NAME}/${MULTI_EDIT_TOOL_NAME} for editing instead of sed/awk`;
+export const WRITE_FILE_POLICY = ` ${WRITE_TOOL_NAME} for creating files instead of cat with heredoc or echo redirection`;
+export const SEARCH_FILE_POLICY = ` ${LS_TOOL_NAME}/${GLOB_TOOL_NAME}/${GREP_TOOL_NAME} for searching and listing files instead of find/ls/grep`;
 
 export const BASH_POLICY = `
 - Reserve bash tools exclusively for actual system commands and terminal operations that require shell execution. NEVER use bash echo or other command-line tools to communicate thoughts, explanations, or instructions to the user. Output all communication directly in your response text instead.
@@ -53,7 +56,24 @@ export const BASH_POLICY = `
 
 export const DEFAULT_SYSTEM_PROMPT = BASE_SYSTEM_PROMPT;
 
-export const INIT_PROMPT = `Please analyze this codebase and create a AGENTS.md file, which will be given to future instances of Wave Code to operate in this repository.
+export const GENERAL_PURPOSE_SYSTEM_PROMPT = `You are an agent. Given the user's message, you should use the tools available to complete the task. Do what has been asked; nothing more, nothing less. When you complete the task simply respond with a detailed writeup.
+
+Your strengths:
+- Searching for code, configurations, and patterns across large codebases
+- Analyzing multiple files to understand system architecture
+- Investigating complex questions that require exploring many files
+- Performing multi-step research tasks
+
+Guidelines:
+- For file searches: Use Grep or Glob when you need to search broadly. Use Read when you know the specific file path.
+- For analysis: Start broad and narrow down. Use multiple search strategies if the first doesn't yield results.
+- Be thorough: Check multiple locations, consider different naming conventions, look for related files.
+- NEVER create files unless they're absolutely necessary for achieving your goal. ALWAYS prefer editing an existing file to creating a new one.
+- NEVER proactively create documentation files (*.md) or README files. Only create documentation files if explicitly requested.
+- In your final response always share relevant file names and code snippets. Any file paths you return in your response MUST be absolute. Do NOT use relative paths.
+- For clear communication, avoid using emojis.`;
+
+export const INIT_PROMPT = `Please analyze this codebase and create a AGENTS.md file, which will be given to future instances of Agent to operate in this repository.
 
 What to add:
 1. Commands that will be commonly used, such as how to build, lint, and run tests. Include the necessary commands to develop in this codebase, such as how to run a single test.
@@ -72,7 +92,7 @@ Usage notes:
 \`\`\`
 # AGENTS.md
 
-This file provides guidance to Wave Code when working with code in this repository.
+This file provides guidance to Agent when working with code in this repository.
 \`\`\``;
 
 export function buildSystemPrompt(
@@ -104,7 +124,27 @@ export function buildSystemPrompt(
     toolNames.has(GLOB_TOOL_NAME) ||
     toolNames.has(GREP_TOOL_NAME)
   ) {
-    prompt += FILE_TOOL_POLICY;
+    const parts: string[] = [];
+    if (toolNames.has(READ_TOOL_NAME)) {
+      parts.push(READ_FILE_POLICY);
+    }
+    if (toolNames.has(EDIT_TOOL_NAME) || toolNames.has(MULTI_EDIT_TOOL_NAME)) {
+      parts.push(EDIT_FILE_POLICY);
+    }
+    if (toolNames.has(WRITE_TOOL_NAME)) {
+      parts.push(WRITE_FILE_POLICY);
+    }
+    if (
+      toolNames.has(LS_TOOL_NAME) ||
+      toolNames.has(GLOB_TOOL_NAME) ||
+      toolNames.has(GREP_TOOL_NAME)
+    ) {
+      parts.push(SEARCH_FILE_POLICY);
+    }
+
+    if (parts.length > 0) {
+      prompt += FILE_TOOL_POLICY_PREFIX + parts.join(",") + ".";
+    }
   }
 
   if (toolNames.has(BASH_TOOL_NAME)) {

package/src/constants/tools.ts

@@ -1,7 +1,7 @@
 export const ASK_USER_QUESTION_TOOL_NAME = "AskUserQuestion";
 export const BASH_TOOL_NAME = "Bash";
-export const BASH_OUTPUT_TOOL_NAME = "BashOutput";
-export const KILL_BASH_TOOL_NAME = "KillBash";
+export const TASK_OUTPUT_TOOL_NAME = "TaskOutput";
+export const TASK_STOP_TOOL_NAME = "TaskStop";
 export const DELETE_FILE_TOOL_NAME = "Delete";
 export const EDIT_TOOL_NAME = "Edit";
 export const EXIT_PLAN_MODE_TOOL_NAME = "ExitPlanMode";

package/src/index.ts

@@ -18,7 +18,6 @@ export * from "./managers/pluginScopeManager.js";
 export * from "./agent.js";
 
 // Export all utilities
-export * from "./utils/bashHistory.js";
 export * from "./utils/bashParser.js";
 export * from "./utils/convertMessagesForAPI.js";
 export * from "./utils/fileFilter.js";
@@ -27,6 +26,7 @@ export * from "./utils/globalLogger.js";
 export * from "./utils/mcpUtils.js";
 export * from "./utils/messageOperations.js";
 export * from "./utils/path.js";
+export * from "./utils/promptHistory.js";
 export * from "./utils/stringUtils.js";
 export * from "./utils/customCommands.js";
 export * from "./utils/hookMatcher.js";

package/src/managers/MemoryRuleManager.ts

@@ -37,6 +37,10 @@ export class MemoryRuleManager {
     const projectRulesDir = path.join(this.workdir, ".wave", "rules");
     const userRulesDir = path.join(os.homedir(), ".wave", "rules");
 
+    logger.debug(`Scanning for modular memory rules...`);
+    logger.debug(`  User rules directory: ${userRulesDir}`);
+    logger.debug(`  Project rules directory: ${projectRulesDir}`);
+
     const newRules: Record<string, MemoryRule> = {};
 
     // Discover user rules first, then project rules so project rules can override if needed
@@ -45,7 +49,18 @@ export class MemoryRuleManager {
     await this.scanDirectory(projectRulesDir, "project", newRules);
 
     this.state.rules = newRules;
-    logger.debug(`Discovered ${Object.keys(newRules).length} memory rules`);
+    const ruleCount = Object.keys(newRules).length;
+    logger.debug(`Discovered ${ruleCount} modular memory rules`);
+
+    if (ruleCount > 0) {
+      logger.debug("Loaded memory rules:");
+      for (const [id, rule] of Object.entries(newRules)) {
+        const pathInfo = rule.metadata.paths
+          ? `paths: ${JSON.stringify(rule.metadata.paths)}`
+          : "global (no path restriction)";
+        logger.debug(`  - ${id} (${rule.source}): ${pathInfo}`);
+      }
+    }
   }
 
   private async scanDirectory(
@@ -121,6 +136,7 @@ export class MemoryRuleManager {
       const relativeId = path.relative(rulesRoot, filePath);
       rule.id = relativeId;
       registry[rule.id] = rule;
+      logger.debug(`Loaded memory rule: ${relativeId} from ${filePath}`);
     } catch (error) {
       logger.error(`Failed to parse memory rule at ${filePath}:`, error);
     }
@@ -132,7 +148,7 @@ export class MemoryRuleManager {
   getActiveRules(filesInContext: string[]): MemoryRule[] {
     const activeRules: MemoryRule[] = [];
     for (const rule of Object.values(this.state.rules)) {
-      if (this.service.isRuleActive(rule, filesInContext)) {
+      if (this.service.isRuleActive(rule, filesInContext, this.workdir)) {
         activeRules.push(rule);
       }
     }