@didim365/agent-cli-core 0.1.0

package/dist/src/hooks/types.js

@@ -0,0 +1,363 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { defaultHookTranslator } from './hookTranslator.js';
+/**
+ * Configuration source levels in precedence order (highest to lowest)
+ */
+export var ConfigSource;
+(function (ConfigSource) {
+    ConfigSource["Project"] = "project";
+    ConfigSource["User"] = "user";
+    ConfigSource["System"] = "system";
+    ConfigSource["Extensions"] = "extensions";
+})(ConfigSource || (ConfigSource = {}));
+/**
+ * Event names for the hook system
+ */
+export var HookEventName;
+(function (HookEventName) {
+    HookEventName["BeforeTool"] = "BeforeTool";
+    HookEventName["AfterTool"] = "AfterTool";
+    HookEventName["BeforeAgent"] = "BeforeAgent";
+    HookEventName["Notification"] = "Notification";
+    HookEventName["AfterAgent"] = "AfterAgent";
+    HookEventName["SessionStart"] = "SessionStart";
+    HookEventName["SessionEnd"] = "SessionEnd";
+    HookEventName["PreCompress"] = "PreCompress";
+    HookEventName["BeforeModel"] = "BeforeModel";
+    HookEventName["AfterModel"] = "AfterModel";
+    HookEventName["BeforeToolSelection"] = "BeforeToolSelection";
+})(HookEventName || (HookEventName = {}));
+/**
+ * Fields in the hooks configuration that are not hook event names
+ */
+export const HOOKS_CONFIG_FIELDS = ['enabled', 'disabled', 'notifications'];
+/**
+ * Hook implementation types
+ */
+export var HookType;
+(function (HookType) {
+    HookType["Command"] = "command";
+})(HookType || (HookType = {}));
+/**
+ * Generate a unique key for a hook configuration
+ */
+export function getHookKey(hook) {
+    const name = hook.name || '';
+    const command = hook.command || '';
+    return `${name}:${command}`;
+}
+/**
+ * Factory function to create the appropriate hook output class based on event name
+ * Returns DefaultHookOutput for all events since it contains all necessary methods
+ */
+export function createHookOutput(eventName, data) {
+    switch (eventName) {
+        case 'BeforeModel':
+            return new BeforeModelHookOutput(data);
+        case 'AfterModel':
+            return new AfterModelHookOutput(data);
+        case 'BeforeToolSelection':
+            return new BeforeToolSelectionHookOutput(data);
+        case 'BeforeTool':
+            return new BeforeToolHookOutput(data);
+        case 'AfterAgent':
+            return new AfterAgentHookOutput(data);
+        default:
+            return new DefaultHookOutput(data);
+    }
+}
+/**
+ * Default implementation of HookOutput with utility methods
+ */
+export class DefaultHookOutput {
+    continue;
+    stopReason;
+    suppressOutput;
+    systemMessage;
+    decision;
+    reason;
+    hookSpecificOutput;
+    constructor(data = {}) {
+        this.continue = data.continue;
+        this.stopReason = data.stopReason;
+        this.suppressOutput = data.suppressOutput;
+        this.systemMessage = data.systemMessage;
+        this.decision = data.decision;
+        this.reason = data.reason;
+        this.hookSpecificOutput = data.hookSpecificOutput;
+    }
+    /**
+     * Check if this output represents a blocking decision
+     */
+    isBlockingDecision() {
+        return this.decision === 'block' || this.decision === 'deny';
+    }
+    /**
+     * Check if this output requests to stop execution
+     */
+    shouldStopExecution() {
+        return this.continue === false;
+    }
+    /**
+     * Get the effective reason for blocking or stopping
+     */
+    getEffectiveReason() {
+        return this.stopReason || this.reason || 'No reason provided';
+    }
+    /**
+     * Apply LLM request modifications (specific method for BeforeModel hooks).
+     * Accepts both GenerateContentParameters (legacy) and LLMRequest (provider-independent).
+     */
+    applyLLMRequestModifications(target) {
+        // Base implementation - overridden by BeforeModelHookOutput
+        return target;
+    }
+    /**
+     * Apply tool config modifications (specific method for BeforeToolSelection hooks).
+     * Accepts both GenAI SDK types (legacy) and HookToolConfig (provider-independent).
+     */
+    applyToolConfigModifications(target) {
+        // Base implementation - overridden by BeforeToolSelectionHookOutput
+        return target;
+    }
+    /**
+     * Get sanitized additional context for adding to responses.
+     */
+    getAdditionalContext() {
+        if (this.hookSpecificOutput &&
+            'additionalContext' in this.hookSpecificOutput) {
+            const context = this.hookSpecificOutput['additionalContext'];
+            if (typeof context !== 'string') {
+                return undefined;
+            }
+            // Sanitize by escaping < and > to prevent tag injection
+            return context.replace(/</g, '&lt;').replace(/>/g, '&gt;');
+        }
+        return undefined;
+    }
+    /**
+     * Check if execution should be blocked and return error info
+     */
+    getBlockingError() {
+        if (this.isBlockingDecision()) {
+            return {
+                blocked: true,
+                reason: this.getEffectiveReason(),
+            };
+        }
+        return { blocked: false, reason: '' };
+    }
+    /**
+     * Check if context clearing was requested by hook.
+     */
+    shouldClearContext() {
+        return false;
+    }
+}
+/**
+ * Specific hook output class for BeforeTool events.
+ */
+export class BeforeToolHookOutput extends DefaultHookOutput {
+    /**
+     * Get modified tool input if provided by hook
+     */
+    getModifiedToolInput() {
+        if (this.hookSpecificOutput && 'tool_input' in this.hookSpecificOutput) {
+            const input = this.hookSpecificOutput['tool_input'];
+            if (typeof input === 'object' &&
+                input !== null &&
+                !Array.isArray(input)) {
+                return input;
+            }
+        }
+        return undefined;
+    }
+}
+/**
+ * Type guard: checks if the target is a GenerateContentParameters (has 'contents' field).
+ */
+function isGenerateContentParameters(obj) {
+    return typeof obj === 'object' && obj !== null && 'contents' in obj;
+}
+/**
+ * Specific hook output class for BeforeModel events
+ */
+export class BeforeModelHookOutput extends DefaultHookOutput {
+    /**
+     * Get synthetic LLM response if provided by hook.
+     * @deprecated Use getSyntheticLLMResponse() for provider-independent code.
+     */
+    getSyntheticResponse() {
+        if (this.hookSpecificOutput && 'llm_response' in this.hookSpecificOutput) {
+            const hookResponse = this.hookSpecificOutput['llm_response'];
+            if (hookResponse) {
+                // Convert hook format to SDK format
+                return defaultHookTranslator.fromHookLLMResponse(hookResponse);
+            }
+        }
+        return undefined;
+    }
+    /**
+     * Get synthetic LLM response in provider-independent format.
+     * Returns the raw LLMResponse without SDK conversion.
+     */
+    getSyntheticLLMResponse() {
+        if (this.hookSpecificOutput && 'llm_response' in this.hookSpecificOutput) {
+            const hookResponse = this.hookSpecificOutput['llm_response'];
+            if (hookResponse) {
+                return hookResponse;
+            }
+        }
+        return undefined;
+    }
+    /**
+     * Apply modifications to LLM request.
+     * Accepts both GenerateContentParameters (legacy) and LLMRequest (provider-independent).
+     */
+    applyLLMRequestModifications(target) {
+        if (this.hookSpecificOutput && 'llm_request' in this.hookSpecificOutput) {
+            const hookRequest = this.hookSpecificOutput['llm_request'];
+            if (hookRequest) {
+                if (isGenerateContentParameters(target)) {
+                    // Legacy path: convert hook format to SDK format
+                    const sdkRequest = defaultHookTranslator.fromHookLLMRequest(hookRequest, target);
+                    return {
+                        ...target,
+                        ...sdkRequest,
+                    };
+                }
+                else {
+                    // Provider-independent path: merge LLMRequest directly
+                    return {
+                        ...target,
+                        ...hookRequest,
+                    };
+                }
+            }
+        }
+        return target;
+    }
+}
+/**
+ * Specific hook output class for BeforeToolSelection events
+ */
+export class BeforeToolSelectionHookOutput extends DefaultHookOutput {
+    /**
+     * Apply tool configuration modifications.
+     * Accepts both GenAI SDK types (legacy) and HookToolConfig (provider-independent).
+     */
+    applyToolConfigModifications(target) {
+        if (this.hookSpecificOutput && 'toolConfig' in this.hookSpecificOutput) {
+            const hookToolConfig = this.hookSpecificOutput['toolConfig'];
+            if (hookToolConfig) {
+                // Convert hook format to SDK format
+                const sdkToolConfig = defaultHookTranslator.fromHookToolConfig(hookToolConfig);
+                return {
+                    ...target,
+                    tools: target.tools || [],
+                    toolConfig: sdkToolConfig,
+                };
+            }
+        }
+        return target;
+    }
+    /**
+     * Get tool config in provider-independent format.
+     * Returns the raw HookToolConfig without SDK conversion.
+     */
+    getHookToolConfig() {
+        if (this.hookSpecificOutput && 'toolConfig' in this.hookSpecificOutput) {
+            const hookToolConfig = this.hookSpecificOutput['toolConfig'];
+            if (hookToolConfig) {
+                return hookToolConfig;
+            }
+        }
+        return undefined;
+    }
+}
+/**
+ * Specific hook output class for AfterModel events
+ */
+export class AfterModelHookOutput extends DefaultHookOutput {
+    /**
+     * Get modified LLM response if provided by hook.
+     * @deprecated Use getModifiedLLMResponse() for provider-independent code.
+     */
+    getModifiedResponse() {
+        if (this.hookSpecificOutput && 'llm_response' in this.hookSpecificOutput) {
+            const hookResponse = this.hookSpecificOutput['llm_response'];
+            if (hookResponse?.candidates?.[0]?.content?.parts?.length) {
+                // Convert hook format to SDK format
+                return defaultHookTranslator.fromHookLLMResponse(hookResponse);
+            }
+        }
+        return undefined;
+    }
+    /**
+     * Get modified LLM response in provider-independent format.
+     * Returns the raw LLMResponse without SDK conversion.
+     */
+    getModifiedLLMResponse() {
+        if (this.hookSpecificOutput && 'llm_response' in this.hookSpecificOutput) {
+            const hookResponse = this.hookSpecificOutput['llm_response'];
+            if (hookResponse?.candidates?.[0]?.content?.parts?.length) {
+                return hookResponse;
+            }
+        }
+        return undefined;
+    }
+}
+/**
+ * Specific hook output class for AfterAgent events
+ */
+export class AfterAgentHookOutput extends DefaultHookOutput {
+    /**
+     * Check if context clearing was requested by hook
+     */
+    shouldClearContext() {
+        if (this.hookSpecificOutput && 'clearContext' in this.hookSpecificOutput) {
+            return this.hookSpecificOutput['clearContext'] === true;
+        }
+        return false;
+    }
+}
+/**
+ * Notification types
+ */
+export var NotificationType;
+(function (NotificationType) {
+    NotificationType["ToolPermission"] = "ToolPermission";
+})(NotificationType || (NotificationType = {}));
+/**
+ * SessionStart source types
+ */
+export var SessionStartSource;
+(function (SessionStartSource) {
+    SessionStartSource["Startup"] = "startup";
+    SessionStartSource["Resume"] = "resume";
+    SessionStartSource["Clear"] = "clear";
+})(SessionStartSource || (SessionStartSource = {}));
+/**
+ * SessionEnd reason types
+ */
+export var SessionEndReason;
+(function (SessionEndReason) {
+    SessionEndReason["Exit"] = "exit";
+    SessionEndReason["Clear"] = "clear";
+    SessionEndReason["Logout"] = "logout";
+    SessionEndReason["PromptInputExit"] = "prompt_input_exit";
+    SessionEndReason["Other"] = "other";
+})(SessionEndReason || (SessionEndReason = {}));
+/**
+ * PreCompress trigger types
+ */
+export var PreCompressTrigger;
+(function (PreCompressTrigger) {
+    PreCompressTrigger["Manual"] = "manual";
+    PreCompressTrigger["Auto"] = "auto";
+})(PreCompressTrigger || (PreCompressTrigger = {}));
+//# sourceMappingURL=types.js.map

package/dist/src/hooks/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/hooks/types.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAaH,OAAO,EAAE,qBAAqB,EAAE,MAAM,qBAAqB,CAAC;AAE5D;;GAEG;AACH,MAAM,CAAN,IAAY,YAKX;AALD,WAAY,YAAY;IACtB,mCAAmB,CAAA;IACnB,6BAAa,CAAA;IACb,iCAAiB,CAAA;IACjB,yCAAyB,CAAA;AAC3B,CAAC,EALW,YAAY,KAAZ,YAAY,QAKvB;AAED;;GAEG;AACH,MAAM,CAAN,IAAY,aAYX;AAZD,WAAY,aAAa;IACvB,0CAAyB,CAAA;IACzB,wCAAuB,CAAA;IACvB,4CAA2B,CAAA;IAC3B,8CAA6B,CAAA;IAC7B,0CAAyB,CAAA;IACzB,8CAA6B,CAAA;IAC7B,0CAAyB,CAAA;IACzB,4CAA2B,CAAA;IAC3B,4CAA2B,CAAA;IAC3B,0CAAyB,CAAA;IACzB,4DAA2C,CAAA;AAC7C,CAAC,EAZW,aAAa,KAAb,aAAa,QAYxB;AAED;;GAEG;AACH,MAAM,CAAC,MAAM,mBAAmB,GAAG,CAAC,SAAS,EAAE,UAAU,EAAE,eAAe,CAAC,CAAC;AAyB5E;;GAEG;AACH,MAAM,CAAN,IAAY,QAEX;AAFD,WAAY,QAAQ;IAClB,+BAAmB,CAAA;AACrB,CAAC,EAFW,QAAQ,KAAR,QAAQ,QAEnB;AAED;;GAEG;AACH,MAAM,UAAU,UAAU,CAAC,IAAgB;IACzC,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC;IAC7B,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC;IACnC,OAAO,GAAG,IAAI,IAAI,OAAO,EAAE,CAAC;AAC9B,CAAC;AAqCD;;;GAGG;AACH,MAAM,UAAU,gBAAgB,CAC9B,SAAiB,EACjB,IAAyB;IAEzB,QAAQ,SAAS,EAAE,CAAC;QAClB,KAAK,aAAa;YAChB,OAAO,IAAI,qBAAqB,CAAC,IAAI,CAAC,CAAC;QACzC,KAAK,YAAY;YACf,OAAO,IAAI,oBAAoB,CAAC,IAAI,CAAC,CAAC;QACxC,KAAK,qBAAqB;YACxB,OAAO,IAAI,6BAA6B,CAAC,IAAI,CAAC,CAAC;QACjD,KAAK,YAAY;YACf,OAAO,IAAI,oBAAoB,CAAC,IAAI,CAAC,CAAC;QACxC,KAAK,YAAY;YACf,OAAO,IAAI,oBAAoB,CAAC,IAAI,CAAC,CAAC;QACxC;YACE,OAAO,IAAI,iBAAiB,CAAC,IAAI,CAAC,CAAC;IACvC,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,OAAO,iBAAiB;IAC5B,QAAQ,CAAW;IACnB,UAAU,CAAU;IACpB,cAAc,CAAW;IACzB,aAAa,CAAU;IACvB,QAAQ,CAAgB;IACxB,MAAM,CAAU;IAChB,kBAAkB,CAA2B;IAE7C,YAAY,OAA4B,EAAE;QACxC,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC;QAC9B,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC;QAClC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,cAAc,CAAC;QAC1C,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,aAAa,CAAC;QACxC,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC;QAC9B,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QAC1B,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC,kBAAkB,CAAC;IACpD,CAAC;IAED;;OAEG;IACH,kBAAkB;QAChB,OAAO,IAAI,CAAC,QAAQ,KAAK,OAAO,IAAI,IAAI,CAAC,QAAQ,KAAK,MAAM,CAAC;IAC/D,CAAC;IAED;;OAEG;IACH,mBAAmB;QACjB,OAAO,IAAI,CAAC,QAAQ,KAAK,KAAK,CAAC;IACjC,CAAC;IAED;;OAEG;IACH,kBAAkB;QAChB,OAAO,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,MAAM,IAAI,oBAAoB,CAAC;IAChE,CAAC;IAED;;;OAGG;IACH,4BAA4B,CAC1B,MAA8C;QAE9C,4DAA4D;QAC5D,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;OAGG;IACH,4BAA4B,CAAC,MAG5B;QAIC,oEAAoE;QACpE,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACH,oBAAoB;QAClB,IACE,IAAI,CAAC,kBAAkB;YACvB,mBAAmB,IAAI,IAAI,CAAC,kBAAkB,EAC9C,CAAC;YACD,MAAM,OAAO,GAAG,IAAI,CAAC,kBAAkB,CAAC,mBAAmB,CAAC,CAAC;YAC7D,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;gBAChC,OAAO,SAAS,CAAC;YACnB,CAAC;YAED,wDAAwD;YACxD,OAAO,OAAO,CAAC,OAAO,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAC7D,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;OAEG;IACH,gBAAgB;QACd,IAAI,IAAI,CAAC,kBAAkB,EAAE,EAAE,CAAC;YAC9B,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,MAAM,EAAE,IAAI,CAAC,kBAAkB,EAAE;aAClC,CAAC;QACJ,CAAC;QACD,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC;IACxC,CAAC;IAED;;OAEG;IACH,kBAAkB;QAChB,OAAO,KAAK,CAAC;IACf,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,oBAAqB,SAAQ,iBAAiB;IACzD;;OAEG;IACH,oBAAoB;QAClB,IAAI,IAAI,CAAC,kBAAkB,IAAI,YAAY,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;YACvE,MAAM,KAAK,GAAG,IAAI,CAAC,kBAAkB,CAAC,YAAY,CAAC,CAAC;YACpD,IACE,OAAO,KAAK,KAAK,QAAQ;gBACzB,KAAK,KAAK,IAAI;gBACd,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EACrB,CAAC;gBACD,OAAO,KAAgC,CAAC;YAC1C,CAAC;QACH,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;CACF;AAED;;GAEG;AACH,SAAS,2BAA2B,CAClC,GAAY;IAEZ,OAAO,OAAO,GAAG,KAAK,QAAQ,IAAI,GAAG,KAAK,IAAI,IAAI,UAAU,IAAI,GAAG,CAAC;AACtE,CAAC;AAED;;GAEG;AACH,MAAM,OAAO,qBAAsB,SAAQ,iBAAiB;IAC1D;;;OAGG;IACH,oBAAoB;QAClB,IAAI,IAAI,CAAC,kBAAkB,IAAI,cAAc,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;YACzE,MAAM,YAAY,GAAG,IAAI,CAAC,kBAAkB,CAC1C,cAAc,CACA,CAAC;YACjB,IAAI,YAAY,EAAE,CAAC;gBACjB,oCAAoC;gBACpC,OAAO,qBAAqB,CAAC,mBAAmB,CAAC,YAAY,CAAC,CAAC;YACjE,CAAC;QACH,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;;OAGG;IACH,uBAAuB;QACrB,IAAI,IAAI,CAAC,kBAAkB,IAAI,cAAc,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;YACzE,MAAM,YAAY,GAAG,IAAI,CAAC,kBAAkB,CAC1C,cAAc,CACA,CAAC;YACjB,IAAI,YAAY,EAAE,CAAC;gBACjB,OAAO,YAAY,CAAC;YACtB,CAAC;QACH,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;;OAGG;IACM,4BAA4B,CACnC,MAA8C;QAE9C,IAAI,IAAI,CAAC,kBAAkB,IAAI,aAAa,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;YACxE,MAAM,WAAW,GAAG,IAAI,CAAC,kBAAkB,CACzC,aAAa,CACS,CAAC;YACzB,IAAI,WAAW,EAAE,CAAC;gBAChB,IAAI,2BAA2B,CAAC,MAAM,CAAC,EAAE,CAAC;oBACxC,iDAAiD;oBACjD,MAAM,UAAU,GAAG,qBAAqB,CAAC,kBAAkB,CACzD,WAAyB,EACzB,MAAM,CACP,CAAC;oBACF,OAAO;wBACL,GAAG,MAAM;wBACT,GAAG,UAAU;qBACd,CAAC;gBACJ,CAAC;qBAAM,CAAC;oBACN,uDAAuD;oBACvD,OAAO;wBACL,GAAG,MAAM;wBACT,GAAG,WAAW;qBACf,CAAC;gBACJ,CAAC;YACH,CAAC;QACH,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,6BAA8B,SAAQ,iBAAiB;IAClE;;;OAGG;IACM,4BAA4B,CAAC,MAGrC;QAIC,IAAI,IAAI,CAAC,kBAAkB,IAAI,YAAY,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;YACvE,MAAM,cAAc,GAAG,IAAI,CAAC,kBAAkB,CAC5C,YAAY,CACK,CAAC;YACpB,IAAI,cAAc,EAAE,CAAC;gBACnB,oCAAoC;gBACpC,MAAM,aAAa,GACjB,qBAAqB,CAAC,kBAAkB,CAAC,cAAc,CAAC,CAAC;gBAC3D,OAAO;oBACL,GAAG,MAAM;oBACT,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,EAAE;oBACzB,UAAU,EAAE,aAAa;iBAC1B,CAAC;YACJ,CAAC;QACH,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;OAGG;IACH,iBAAiB;QACf,IAAI,IAAI,CAAC,kBAAkB,IAAI,YAAY,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;YACvE,MAAM,cAAc,GAAG,IAAI,CAAC,kBAAkB,CAC5C,YAAY,CACK,CAAC;YACpB,IAAI,cAAc,EAAE,CAAC;gBACnB,OAAO,cAAc,CAAC;YACxB,CAAC;QACH,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,oBAAqB,SAAQ,iBAAiB;IACzD;;;OAGG;IACH,mBAAmB;QACjB,IAAI,IAAI,CAAC,kBAAkB,IAAI,cAAc,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;YACzE,MAAM,YAAY,GAAG,IAAI,CAAC,kBAAkB,CAC1C,cAAc,CACS,CAAC;YAC1B,IAAI,YAAY,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;gBAC1D,oCAAoC;gBACpC,OAAO,qBAAqB,CAAC,mBAAmB,CAC9C,YAA2B,CAC5B,CAAC;YACJ,CAAC;QACH,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;;OAGG;IACH,sBAAsB;QACpB,IAAI,IAAI,CAAC,kBAAkB,IAAI,cAAc,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;YACzE,MAAM,YAAY,GAAG,IAAI,CAAC,kBAAkB,CAC1C,cAAc,CACS,CAAC;YAC1B,IAAI,YAAY,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;gBAC1D,OAAO,YAA2B,CAAC;YACrC,CAAC;QACH,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,oBAAqB,SAAQ,iBAAiB;IACzD;;OAEG;IACM,kBAAkB;QACzB,IAAI,IAAI,CAAC,kBAAkB,IAAI,cAAc,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;YACzE,OAAO,IAAI,CAAC,kBAAkB,CAAC,cAAc,CAAC,KAAK,IAAI,CAAC;QAC1D,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;CACF;AAkFD;;GAEG;AACH,MAAM,CAAN,IAAY,gBAEX;AAFD,WAAY,gBAAgB;IAC1B,qDAAiC,CAAA;AACnC,CAAC,EAFW,gBAAgB,KAAhB,gBAAgB,QAE3B;AAsCD;;GAEG;AACH,MAAM,CAAN,IAAY,kBAIX;AAJD,WAAY,kBAAkB;IAC5B,yCAAmB,CAAA;IACnB,uCAAiB,CAAA;IACjB,qCAAe,CAAA;AACjB,CAAC,EAJW,kBAAkB,KAAlB,kBAAkB,QAI7B;AAmBD;;GAEG;AACH,MAAM,CAAN,IAAY,gBAMX;AAND,WAAY,gBAAgB;IAC1B,iCAAa,CAAA;IACb,mCAAe,CAAA;IACf,qCAAiB,CAAA;IACjB,yDAAqC,CAAA;IACrC,mCAAe,CAAA;AACjB,CAAC,EANW,gBAAgB,KAAhB,gBAAgB,QAM3B;AASD;;GAEG;AACH,MAAM,CAAN,IAAY,kBAGX;AAHD,WAAY,kBAAkB;IAC5B,uCAAiB,CAAA;IACjB,mCAAa,CAAA;AACf,CAAC,EAHW,kBAAkB,KAAlB,kBAAkB,QAG7B"}

package/dist/src/hooks/types.test.d.ts

@@ -0,0 +1,6 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export {};

package/dist/src/hooks/types.test.js

@@ -0,0 +1,278 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi } from 'vitest';
+import { createHookOutput, DefaultHookOutput, BeforeModelHookOutput, BeforeToolSelectionHookOutput, AfterModelHookOutput, HookEventName, HookType, BeforeToolHookOutput, } from './types.js';
+import { defaultHookTranslator } from './hookTranslator.js';
+vi.mock('./hookTranslator.js', () => ({
+    defaultHookTranslator: {
+        fromHookLLMResponse: vi.fn((response) => response),
+        fromHookLLMRequest: vi.fn((request, target) => ({
+            ...target,
+            ...request,
+        })),
+        fromHookToolConfig: vi.fn((config) => config),
+    },
+}));
+describe('Hook Types', () => {
+    describe('HookEventName', () => {
+        it('should contain all required event names', () => {
+            const expectedEvents = [
+                'BeforeTool',
+                'AfterTool',
+                'BeforeAgent',
+                'Notification',
+                'AfterAgent',
+                'SessionStart',
+                'SessionEnd',
+                'PreCompress',
+                'BeforeModel',
+                'AfterModel',
+                'BeforeToolSelection',
+            ];
+            for (const event of expectedEvents) {
+                expect(Object.values(HookEventName)).toContain(event);
+            }
+        });
+    });
+    describe('HookType', () => {
+        it('should contain command type', () => {
+            expect(HookType.Command).toBe('command');
+        });
+    });
+});
+describe('Hook Output Classes', () => {
+    describe('createHookOutput', () => {
+        it('should return DefaultHookOutput for unknown event names', () => {
+            const output = createHookOutput('UnknownEvent', {});
+            expect(output).toBeInstanceOf(DefaultHookOutput);
+            expect(output).not.toBeInstanceOf(BeforeModelHookOutput);
+            expect(output).not.toBeInstanceOf(AfterModelHookOutput);
+            expect(output).not.toBeInstanceOf(BeforeToolSelectionHookOutput);
+        });
+        it('should return BeforeModelHookOutput for BeforeModel event', () => {
+            const output = createHookOutput(HookEventName.BeforeModel, {});
+            expect(output).toBeInstanceOf(BeforeModelHookOutput);
+        });
+        it('should return AfterModelHookOutput for AfterModel event', () => {
+            const output = createHookOutput(HookEventName.AfterModel, {});
+            expect(output).toBeInstanceOf(AfterModelHookOutput);
+        });
+        it('should return BeforeToolSelectionHookOutput for BeforeToolSelection event', () => {
+            const output = createHookOutput(HookEventName.BeforeToolSelection, {});
+            expect(output).toBeInstanceOf(BeforeToolSelectionHookOutput);
+        });
+        it('should return BeforeToolHookOutput for BeforeTool event', () => {
+            const output = createHookOutput(HookEventName.BeforeTool, {});
+            expect(output).toBeInstanceOf(BeforeToolHookOutput);
+        });
+    });
+    describe('DefaultHookOutput', () => {
+        it('should construct with provided data', () => {
+            const data = {
+                continue: false,
+                stopReason: 'test stop',
+                suppressOutput: true,
+                systemMessage: 'test system message',
+                decision: 'block',
+                reason: 'test reason',
+                hookSpecificOutput: { key: 'value' },
+            };
+            const output = new DefaultHookOutput(data);
+            expect(output.continue).toBe(data.continue);
+            expect(output.stopReason).toBe(data.stopReason);
+            expect(output.suppressOutput).toBe(data.suppressOutput);
+            expect(output.systemMessage).toBe(data.systemMessage);
+            expect(output.decision).toBe(data.decision);
+            expect(output.reason).toBe(data.reason);
+            expect(output.hookSpecificOutput).toEqual(data.hookSpecificOutput);
+        });
+        it('should return false for isBlockingDecision if decision is not block or deny', () => {
+            const output1 = new DefaultHookOutput({ decision: 'approve' });
+            expect(output1.isBlockingDecision()).toBe(false);
+            const output2 = new DefaultHookOutput({ decision: undefined });
+            expect(output2.isBlockingDecision()).toBe(false);
+        });
+        it('should return true for isBlockingDecision if decision is block or deny', () => {
+            const output1 = new DefaultHookOutput({ decision: 'block' });
+            expect(output1.isBlockingDecision()).toBe(true);
+            const output2 = new DefaultHookOutput({ decision: 'deny' });
+            expect(output2.isBlockingDecision()).toBe(true);
+        });
+        it('should return true for shouldStopExecution if continue is false', () => {
+            const output = new DefaultHookOutput({ continue: false });
+            expect(output.shouldStopExecution()).toBe(true);
+        });
+        it('should return false for shouldStopExecution if continue is true or undefined', () => {
+            const output1 = new DefaultHookOutput({ continue: true });
+            expect(output1.shouldStopExecution()).toBe(false);
+            const output2 = new DefaultHookOutput({});
+            expect(output2.shouldStopExecution()).toBe(false);
+        });
+        it('should return reason if available', () => {
+            const output = new DefaultHookOutput({ reason: 'specific reason' });
+            expect(output.getEffectiveReason()).toBe('specific reason');
+        });
+        it('should return stopReason if reason is not available', () => {
+            const output = new DefaultHookOutput({ stopReason: 'stop reason' });
+            expect(output.getEffectiveReason()).toBe('stop reason');
+        });
+        it('should return "No reason provided" if neither reason nor stopReason are available', () => {
+            const output = new DefaultHookOutput({});
+            expect(output.getEffectiveReason()).toBe('No reason provided');
+        });
+        it('applyLLMRequestModifications should return target unchanged', () => {
+            const target = {
+                model: 'gemini-pro',
+                contents: [],
+            };
+            const output = new DefaultHookOutput({});
+            expect(output.applyLLMRequestModifications(target)).toBe(target);
+        });
+        it('applyToolConfigModifications should return target unchanged', () => {
+            const target = { toolConfig: {}, tools: [] };
+            const output = new DefaultHookOutput({});
+            expect(output.applyToolConfigModifications(target)).toBe(target);
+        });
+        it('getAdditionalContext should return additional context if present', () => {
+            const output = new DefaultHookOutput({
+                hookSpecificOutput: { additionalContext: 'some context' },
+            });
+            expect(output.getAdditionalContext()).toBe('some context');
+        });
+        it('getAdditionalContext should sanitize context by escaping <', () => {
+            const output = new DefaultHookOutput({
+                hookSpecificOutput: {
+                    additionalContext: 'context with <tag> and </hook_context>',
+                },
+            });
+            expect(output.getAdditionalContext()).toBe('context with &lt;tag&gt; and &lt;/hook_context&gt;');
+        });
+        it('getAdditionalContext should return undefined if additionalContext is not present', () => {
+            const output = new DefaultHookOutput({
+                hookSpecificOutput: { other: 'value' },
+            });
+            expect(output.getAdditionalContext()).toBeUndefined();
+        });
+        it('getAdditionalContext should return undefined if hookSpecificOutput is undefined', () => {
+            const output = new DefaultHookOutput({});
+            expect(output.getAdditionalContext()).toBeUndefined();
+        });
+        it('getBlockingError should return blocked: true and reason if blocking decision', () => {
+            const output = new DefaultHookOutput({
+                decision: 'block',
+                reason: 'blocked by hook',
+            });
+            expect(output.getBlockingError()).toEqual({
+                blocked: true,
+                reason: 'blocked by hook',
+            });
+        });
+        it('getBlockingError should return blocked: false if not blocking decision', () => {
+            const output = new DefaultHookOutput({ decision: 'approve' });
+            expect(output.getBlockingError()).toEqual({ blocked: false, reason: '' });
+        });
+    });
+    describe('BeforeModelHookOutput', () => {
+        it('getSyntheticResponse should return synthetic response if llm_response is present', () => {
+            const mockResponse = { candidates: [] };
+            const output = new BeforeModelHookOutput({
+                hookSpecificOutput: { llm_response: mockResponse },
+            });
+            expect(output.getSyntheticResponse()).toEqual(mockResponse);
+            expect(defaultHookTranslator.fromHookLLMResponse).toHaveBeenCalledWith(mockResponse);
+        });
+        it('getSyntheticResponse should return undefined if llm_response is not present', () => {
+            const output = new BeforeModelHookOutput({});
+            expect(output.getSyntheticResponse()).toBeUndefined();
+        });
+        it('applyLLMRequestModifications should apply modifications if llm_request is present', () => {
+            const target = {
+                model: 'gemini-pro',
+                contents: [{ parts: [{ text: 'original' }] }],
+            };
+            const mockRequest = {
+                messages: [{ role: 'user', content: 'modified' }],
+            };
+            const output = new BeforeModelHookOutput({
+                hookSpecificOutput: { llm_request: mockRequest },
+            });
+            const result = output.applyLLMRequestModifications(target);
+            expect(result).toEqual({ ...target, ...mockRequest });
+            expect(defaultHookTranslator.fromHookLLMRequest).toHaveBeenCalledWith(mockRequest, target);
+        });
+        it('applyLLMRequestModifications should return target unchanged if llm_request is not present', () => {
+            const target = {
+                model: 'gemini-pro',
+                contents: [],
+            };
+            const output = new BeforeModelHookOutput({});
+            expect(output.applyLLMRequestModifications(target)).toBe(target);
+        });
+    });
+    describe('BeforeToolSelectionHookOutput', () => {
+        it('applyToolConfigModifications should apply modifications if toolConfig is present', () => {
+            const target = { tools: [{ functionDeclarations: [] }] };
+            const mockToolConfig = { functionCallingConfig: { mode: 'ANY' } };
+            const output = new BeforeToolSelectionHookOutput({
+                hookSpecificOutput: { toolConfig: mockToolConfig },
+            });
+            const result = output.applyToolConfigModifications(target);
+            expect(result).toEqual({ ...target, toolConfig: mockToolConfig });
+            expect(defaultHookTranslator.fromHookToolConfig).toHaveBeenCalledWith(mockToolConfig);
+        });
+        it('applyToolConfigModifications should return target unchanged if toolConfig is not present', () => {
+            const target = { toolConfig: {}, tools: [] };
+            const output = new BeforeToolSelectionHookOutput({});
+            expect(output.applyToolConfigModifications(target)).toBe(target);
+        });
+        it('applyToolConfigModifications should initialize tools array if not present', () => {
+            const target = {};
+            const mockToolConfig = { functionCallingConfig: { mode: 'ANY' } };
+            const output = new BeforeToolSelectionHookOutput({
+                hookSpecificOutput: { toolConfig: mockToolConfig },
+            });
+            const result = output.applyToolConfigModifications(target);
+            expect(result).toEqual({ tools: [], toolConfig: mockToolConfig });
+        });
+    });
+    describe('AfterModelHookOutput', () => {
+        it('getModifiedResponse should return modified response if llm_response is present and has content', () => {
+            const mockResponse = {
+                candidates: [{ content: { role: 'model', parts: ['modified'] } }],
+            };
+            const output = new AfterModelHookOutput({
+                hookSpecificOutput: { llm_response: mockResponse },
+            });
+            expect(output.getModifiedResponse()).toEqual(mockResponse);
+            expect(defaultHookTranslator.fromHookLLMResponse).toHaveBeenCalledWith(mockResponse);
+        });
+        it('getModifiedResponse should return undefined if llm_response is present but no content', () => {
+            const mockResponse = {
+                candidates: [{ content: { role: 'model', parts: [] } }],
+            };
+            const output = new AfterModelHookOutput({
+                hookSpecificOutput: { llm_response: mockResponse },
+            });
+            expect(output.getModifiedResponse()).toBeUndefined();
+        });
+        it('getModifiedResponse should return undefined if llm_response is not present', () => {
+            const output = new AfterModelHookOutput({});
+            expect(output.getModifiedResponse()).toBeUndefined();
+        });
+        it('getModifiedResponse should return undefined if shouldStopExecution is true', () => {
+            const output = new AfterModelHookOutput({
+                continue: false,
+                stopReason: 'stopped by hook',
+            });
+            expect(output.getModifiedResponse()).toBeUndefined();
+        });
+        it('getModifiedResponse should return undefined if shouldStopExecution is true and no stopReason', () => {
+            const output = new AfterModelHookOutput({ continue: false });
+            expect(output.getModifiedResponse()).toBeUndefined();
+        });
+    });
+});
+//# sourceMappingURL=types.test.js.map