wave-agent-sdk 0.0.5 → 0.0.7

package/src/types/messaging.ts

@@ -0,0 +1,102 @@
+/**
+ * Message and communication block types
+ * Dependencies: Core (Usage)
+ */
+
+import type { Usage } from "./core.js";
+
+export enum MessageSource {
+  USER = "user",
+  HOOK = "hook",
+}
+
+export interface Message {
+  role: "user" | "assistant";
+  blocks: MessageBlock[];
+  usage?: Usage; // Usage data for this message's AI operation (assistant messages only)
+}
+
+export type MessageBlock =
+  | TextBlock
+  | ErrorBlock
+  | ToolBlock
+  | ImageBlock
+  | DiffBlock
+  | CommandOutputBlock
+  | CompressBlock
+  | MemoryBlock
+  | SubagentBlock;
+
+export interface TextBlock {
+  type: "text";
+  content: string;
+  customCommandContent?: string;
+  source?: MessageSource;
+}
+
+export interface ErrorBlock {
+  type: "error";
+  content: string;
+}
+
+export interface ToolBlock {
+  type: "tool";
+  parameters?: string;
+  result?: string;
+  shortResult?: string; // Add shortResult field
+  images?: Array<{
+    // Add image data support
+    data: string; // Base64 encoded image data
+    mediaType?: string; // Media type of the image
+  }>;
+  id?: string;
+  name?: string;
+  isRunning?: boolean; // Mark if tool is actually executing
+  success?: boolean;
+  error?: string | Error;
+  compactParams?: string; // Compact parameter display
+}
+
+export interface ImageBlock {
+  type: "image";
+  imageUrls?: string[];
+}
+
+export interface DiffBlock {
+  type: "diff";
+  path: string;
+  diffResult: Array<{
+    value: string;
+    added?: boolean;
+    removed?: boolean;
+  }>;
+}
+
+export interface CommandOutputBlock {
+  type: "command_output";
+  command: string;
+  output: string;
+  isRunning: boolean;
+  exitCode: number | null;
+}
+
+export interface CompressBlock {
+  type: "compress";
+  content: string;
+}
+
+export interface MemoryBlock {
+  type: "memory";
+  content: string;
+  isSuccess: boolean;
+  memoryType?: "project" | "user"; // Memory type
+  storagePath?: string; // Storage path text
+}
+
+export interface SubagentBlock {
+  type: "subagent";
+  subagentId: string;
+  subagentName: string;
+  status: "active" | "completed" | "error" | "aborted";
+  messages: Message[];
+}

package/src/types/processes.ts

@@ -0,0 +1,18 @@
+/**
+ * Background process and shell management types
+ * Dependencies: None
+ */
+
+import type { ChildProcess } from "child_process";
+
+export interface BackgroundShell {
+  id: string;
+  process: ChildProcess;
+  command: string;
+  startTime: number;
+  status: "running" | "completed" | "killed";
+  stdout: string;
+  stderr: string;
+  exitCode?: number;
+  runtime?: number;
+}

package/src/types/skills.ts

@@ -0,0 +1,91 @@
+/**
+ * Skill system types and constants
+ * Dependencies: Core (Logger)
+ */
+
+import type { Logger } from "./core.js";
+
+export interface SkillMetadata {
+  name: string;
+  description: string;
+  type: "personal" | "project";
+  skillPath: string;
+}
+
+export interface Skill extends SkillMetadata {
+  content: string;
+  frontmatter: SkillFrontmatter;
+  isValid: boolean;
+  errors: string[];
+}
+
+export interface SkillFrontmatter {
+  name: string;
+  description: string;
+  [key: string]: unknown;
+}
+
+export interface SkillCollection {
+  type: "personal" | "project";
+  basePath: string;
+  skills: Map<string, SkillMetadata>;
+  errors: SkillError[];
+}
+
+export interface SkillError {
+  skillPath: string;
+  message: string;
+}
+
+export interface SkillValidationResult {
+  isValid: boolean;
+  skill?: Skill;
+  errors: string[];
+}
+
+export interface SkillDiscoveryResult {
+  personalSkills: Map<string, SkillMetadata>;
+  projectSkills: Map<string, SkillMetadata>;
+  errors: SkillError[];
+}
+
+export interface SkillInvocationContext {
+  skillName: string;
+}
+
+export interface SkillToolArgs {
+  skill_name: string;
+}
+
+export interface SkillManagerOptions {
+  personalSkillsPath?: string;
+  scanTimeout?: number;
+  logger?: Logger;
+}
+
+export interface ParsedSkillFile {
+  frontmatter: SkillFrontmatter;
+  content: string;
+  skillMetadata: SkillMetadata;
+  validationErrors: string[];
+  isValid: boolean;
+}
+
+export interface SkillParseOptions {
+  validateMetadata?: boolean;
+  basePath?: string;
+}
+
+export const SKILL_DEFAULTS = {
+  PERSONAL_SKILLS_DIR: ".wave/skills",
+  PROJECT_SKILLS_DIR: ".wave/skills",
+  SKILL_FILE_NAME: "SKILL.md",
+  MAX_NAME_LENGTH: 64,
+  MAX_DESCRIPTION_LENGTH: 1024,
+  MIN_DESCRIPTION_LENGTH: 1,
+  NAME_PATTERN: /^[a-z0-9-]+$/,
+  MAX_METADATA_CACHE: 1000,
+  MAX_CONTENT_CACHE: 100,
+  SCAN_TIMEOUT: 5000,
+  LOAD_TIMEOUT: 2000,
+} as const;

package/src/utils/configResolver.ts

@@ -8,7 +8,7 @@ import {
   ModelConfig,
   ConfigurationError,
   CONFIG_ERRORS,
-} from "../types.js";
+} from "../types/index.js";
 
 export class ConfigResolver {
   /**

package/src/utils/configValidator.ts

@@ -3,7 +3,11 @@
  * Validates configuration values for correctness and security
  */
 
-import { GatewayConfig, ConfigurationError, CONFIG_ERRORS } from "../types.js";
+import {
+  GatewayConfig,
+  ConfigurationError,
+  CONFIG_ERRORS,
+} from "../types/index.js";
 
 export class ConfigValidator {
   /**

package/src/utils/convertMessagesForAPI.ts

@@ -1,4 +1,4 @@
-import type { Message } from "../types.js";
+import type { Message } from "../types/index.js";
 import { convertImageToBase64 } from "./messageOperations.js";
 import { ChatCompletionMessageToolCall } from "openai/resources";
 import { stripAnsiColors } from "./stringUtils.js";
@@ -176,15 +176,7 @@ export function convertMessagesForAPI(
         if (block.type === "text" && block.content) {
           contentParts.push({
             type: "text",
-            text: block.content,
-          });
-        }
-
-        // Handle custom command blocks - pass full content as text to AI
-        if (block.type === "custom_command" && block.content) {
-          contentParts.push({
-            type: "text",
-            text: block.content,
+            text: block.customCommandContent || block.content,
           });
         }
 

package/src/utils/customCommands.ts

@@ -1,7 +1,7 @@
 import { existsSync, readdirSync } from "fs";
 import { join, extname, basename } from "path";
 import { homedir } from "os";
-import type { CustomSlashCommand } from "../types.js";
+import type { CustomSlashCommand } from "../types/index.js";
 import { parseMarkdownFile } from "./markdownParser.js";
 
 /**

package/src/{hooks/matcher.ts → utils/hookMatcher.ts}

@@ -7,18 +7,7 @@
 
 import { minimatch } from "minimatch";
 
-export interface IHookMatcher {
-  // Test if pattern matches tool name
-  matches(pattern: string, toolName: string): boolean;
-
-  // Validate pattern syntax
-  isValidPattern(pattern: string): boolean;
-
-  // Get pattern type for optimization
-  getPatternType(pattern: string): "exact" | "glob" | "regex" | "alternatives";
-}
-
-export class HookMatcher implements IHookMatcher {
+export class HookMatcher {
   /**
    * Test if pattern matches tool name
    * Supports multiple matching strategies:

package/src/utils/markdownParser.ts

@@ -1,5 +1,5 @@
 import { readFileSync } from "fs";
-import type { CustomSlashCommandConfig } from "../types.js";
+import type { CustomSlashCommandConfig } from "../types/index.js";
 
 interface ParsedMarkdownFile {
   content: string;

package/src/utils/mcpUtils.ts

@@ -1,6 +1,6 @@
 import { ChatCompletionFunctionTool } from "openai/resources.js";
 import type { ToolPlugin, ToolResult, ToolContext } from "../tools/types.js";
-import type { McpTool, McpServerStatus } from "../types.js";
+import type { McpTool, McpServerStatus } from "../types/index.js";
 
 /**
  * Convert MCP tool to OpenAI function tool format

package/src/utils/messageOperations.ts

@@ -1,19 +1,19 @@
-import type { Message, Usage } from "../types.js";
+import type { Message, Usage, MessageSource } from "../types/index.js";
 import { readFileSync } from "fs";
 import { extname } from "path";
 import { ChatCompletionMessageFunctionToolCall } from "openai/resources.js";
 
-// Parameter interfaces for message operations
-export interface AddUserMessageParams {
-  messages: Message[];
+// Base user message parameters interface
+export interface UserMessageParams {
   content: string;
   images?: Array<{ path: string; mimeType: string }>;
-  customCommandBlock?: {
-    type: "custom_command";
-    commandName: string;
-    content: string;
-    originalInput?: string;
-  };
+  customCommandContent?: string;
+  source?: MessageSource;
+}
+
+// Parameter interfaces for message operations
+export interface AddUserMessageParams extends UserMessageParams {
+  messages: Message[];
 }
 
 export interface UpdateToolBlockParams {
@@ -31,17 +31,10 @@ export interface UpdateToolBlockParams {
 }
 
 // Agent specific interfaces (without messages parameter)
-export interface AgentToolBlockUpdateParams {
-  toolId: string;
-  args?: string;
-  result?: string;
-  success?: boolean;
-  error?: string;
-  isRunning?: boolean;
-  name?: string;
-  shortResult?: string;
-  compactParams?: string;
-}
+export type AgentToolBlockUpdateParams = Omit<
+  UpdateToolBlockParams,
+  "messages"
+>;
 
 export interface AddDiffBlockParams {
   messages: Message[];
@@ -145,16 +138,19 @@ export const addUserMessageToMessages = ({
   messages,
   content,
   images,
-  customCommandBlock,
+  customCommandContent,
+  source,
 }: AddUserMessageParams): Message[] => {
   const blocks: Message["blocks"] = [];
 
-  // If there's a custom command block, use it instead of text content
-  if (customCommandBlock) {
-    blocks.push(customCommandBlock);
-  } else {
-    blocks.push({ type: "text", content });
-  }
+  // Create text block with optional customCommandContent and source
+  const textBlock = {
+    type: "text" as const,
+    content,
+    ...(customCommandContent && { customCommandContent }),
+    ...(source && { source }),
+  };
+  blocks.push(textBlock);
 
   // If there are images, add image block
   if (images && images.length > 0) {
@@ -294,24 +290,23 @@ export const addErrorBlockToMessage = ({
   error,
 }: AddErrorBlockParams): Message[] => {
   const newMessages = [...messages];
-  // Find the last assistant message
-  let assistantMessageFound = false;
-  for (let i = newMessages.length - 1; i >= 0; i--) {
-    if (newMessages[i].role === "assistant") {
-      newMessages[i].blocks = [
-        ...newMessages[i].blocks,
+
+  // Check if the last message is an assistant message
+  const lastMessage = newMessages[newMessages.length - 1];
+  if (lastMessage && lastMessage.role === "assistant") {
+    // Create a new message object with the error block added
+    newMessages[newMessages.length - 1] = {
+      ...lastMessage,
+      blocks: [
+        ...lastMessage.blocks,
         {
           type: "error",
           content: error,
         },
-      ];
-      assistantMessageFound = true;
-      break;
-    }
-  }
-
-  // If no assistant message found, create a new assistant message with only error block
-  if (!assistantMessageFound) {
+      ],
+    };
+  } else {
+    // If the last message is not an assistant message, create a new assistant message
     newMessages.push({
       role: "assistant",
       blocks: [
@@ -586,3 +581,22 @@ export const updateSubagentBlockInMessage = (
 
   return newMessages;
 };
+
+/**
+ * Removes the last user message from the messages array
+ * Used for hook error handling when the user prompt needs to be erased
+ */
+export const removeLastUserMessage = (messages: Message[]): Message[] => {
+  const newMessages = [...messages];
+
+  // Find the index of the last user message
+  for (let i = newMessages.length - 1; i >= 0; i--) {
+    if (newMessages[i].role === "user") {
+      // Remove the user message at index i
+      newMessages.splice(i, 1);
+      break;
+    }
+  }
+
+  return newMessages;
+};

package/src/utils/skillParser.ts

@@ -5,7 +5,7 @@ import type {
   SkillParseOptions,
   SkillFrontmatter,
   SkillMetadata,
-} from "../types.js";
+} from "../types/index.js";
 
 /**
  * Parse a SKILL.md file and validate its contents

package/dist/hooks/executor.d.ts

@@ -1,56 +0,0 @@
-/**
- * Hook Command Executor
- *
- * Handles the execution of hook commands in isolated processes with proper
- * timeout handling, environment variable injection, and cross-platform support.
- */
-import { type HookExecutionContext, type HookExecutionResult, type HookExecutionOptions, type ExtendedHookExecutionContext } from "./types.js";
-import type { Logger } from "../types.js";
-export interface IHookExecutor {
-    executeCommand(command: string, context: HookExecutionContext | ExtendedHookExecutionContext, options?: HookExecutionOptions): Promise<HookExecutionResult>;
-    executeCommands(commands: string[], context: HookExecutionContext | ExtendedHookExecutionContext, options?: HookExecutionOptions): Promise<HookExecutionResult[]>;
-    isCommandSafe(command: string): boolean;
-}
-export declare class HookExecutor implements IHookExecutor {
-    private readonly defaultTimeout;
-    private readonly maxTimeout;
-    private readonly logger?;
-    private readonly skipExecution;
-    constructor(logger?: Logger, options?: {
-        skipExecution?: boolean;
-    });
-    /**
-     * Execute a single hook command in an isolated process
-     */
-    executeCommand(command: string, context: HookExecutionContext | ExtendedHookExecutionContext, options?: HookExecutionOptions): Promise<HookExecutionResult>;
-    /**
-     * Execute multiple commands in sequence
-     * Stops on first failure unless configured otherwise
-     */
-    executeCommands(commands: string[], context: HookExecutionContext | ExtendedHookExecutionContext, options?: HookExecutionOptions): Promise<HookExecutionResult[]>;
-    /**
-     * Validate command safety to prevent injection attacks
-     */
-    isCommandSafe(command: string): boolean;
-    /**
-     * Build environment variables for hook execution
-     */
-    private buildEnvironment;
-    /**
-     * Resolve environment variables in command string
-     */
-    private resolveEnvironmentVariables;
-    /**
-     * Get process statistics for monitoring
-     */
-    getExecutionStats(): {
-        platform: NodeJS.Platform;
-        defaultTimeout: number;
-        maxTimeout: number;
-    };
-    /**
-     * Test if the executor can run commands on this platform
-     */
-    isSupported(): boolean;
-}
-//# sourceMappingURL=executor.d.ts.map

package/dist/hooks/executor.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"executor.d.ts","sourceRoot":"","sources":["../../src/hooks/executor.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,OAAO,EACL,KAAK,oBAAoB,EACzB,KAAK,mBAAmB,EACxB,KAAK,oBAAoB,EACzB,KAAK,4BAA4B,EAGlC,MAAM,YAAY,CAAC;AACpB,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAyC1C,MAAM,WAAW,aAAa;IAE5B,cAAc,CACZ,OAAO,EAAE,MAAM,EACf,OAAO,EAAE,oBAAoB,GAAG,4BAA4B,EAC5D,OAAO,CAAC,EAAE,oBAAoB,GAC7B,OAAO,CAAC,mBAAmB,CAAC,CAAC;IAGhC,eAAe,CACb,QAAQ,EAAE,MAAM,EAAE,EAClB,OAAO,EAAE,oBAAoB,GAAG,4BAA4B,EAC5D,OAAO,CAAC,EAAE,oBAAoB,GAC7B,OAAO,CAAC,mBAAmB,EAAE,CAAC,CAAC;IAGlC,aAAa,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC;CACzC;AAED,qBAAa,YAAa,YAAW,aAAa;IAChD,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAS;IACxC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAU;IACrC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAS;IACjC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAU;gBAE5B,MAAM,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE;QAAE,aAAa,CAAC,EAAE,OAAO,CAAA;KAAE;IASlE;;OAEG;IACG,cAAc,CAClB,OAAO,EAAE,MAAM,EACf,OAAO,EAAE,oBAAoB,GAAG,4BAA4B,EAC5D,OAAO,GAAE,oBAAyB,GACjC,OAAO,CAAC,mBAAmB,CAAC;IAmM/B;;;OAGG;IACG,eAAe,CACnB,QAAQ,EAAE,MAAM,EAAE,EAClB,OAAO,EAAE,oBAAoB,GAAG,4BAA4B,EAC5D,OAAO,GAAE,oBAAyB,GACjC,OAAO,CAAC,mBAAmB,EAAE,CAAC;IAmDjC;;OAEG;IACH,aAAa,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAwBvC;;OAEG;IACH,OAAO,CAAC,gBAAgB;IAOxB;;OAEG;IACH,OAAO,CAAC,2BAA2B;IAqBnC;;OAEG;IACH,iBAAiB,IAAI;QACnB,QAAQ,EAAE,MAAM,CAAC,QAAQ,CAAC;QAC1B,cAAc,EAAE,MAAM,CAAC;QACvB,UAAU,EAAE,MAAM,CAAC;KACpB;IAQD;;OAEG;IACH,WAAW,IAAI,OAAO;CAQvB"}