import { Agent } from "../agent/Agents";
import { Actions } from "../actions";
import { Action } from "../types/action.interface";
import { merkleValidator } from "./merkleValidator";

/**
 * Find an action by its name or one of its similes
 */
export function findAction(query: string): Action | undefined {
    const normalizedQuery = query.toLowerCase().trim();
    return Object.values(Actions.map((n) => n.action)).find(
        (action) =>
            action.name.toLowerCase() === normalizedQuery ||
            action.similes.some((simile) => simile.toLowerCase() === normalizedQuery),
    );
}

/**
 * Execute an action with the given input
 */
export async function executeAction(
    action: Action,
    agent: Agent,
    input: Record<string, any>,
): Promise<Record<string, any>> {
    try {
        // Validate input using Zod schema
        const validatedInput = action.schema.parse(input);

        // Execute the action with validated input
        const result = await action.handler(agent, validatedInput);

        return {
            status: "success",
            ...result,
        };
    } catch (error: any) {
        // Handle Zod validation errors specially
        if (error.errors) {
            return {
                status: "error",
                message: "Validation error",
                details: error.errors,
                code: "VALIDATION_ERROR",
            };
        }

        return {
            status: "error",
            message: error.message,
            code: error.code || "EXECUTION_ERROR",
        };
    }
}

/**
 * Get examples for an action
 */
export function getActionExamples(action: Action): string {
    return action.examples
        .flat()
        .map((example: any) => {
            return `Input: ${JSON.stringify(example.input, null, 2)}
Output: ${JSON.stringify(example.output, null, 2)}
Explanation: ${example.explanation}
---`;
        })
        .join("\n");
}