const toolDefinitions = [
  defineTool({
    id: "recompute-policy",
    name: "Recompute Policy",
    schema: z.object({
      policyId: z.string().min(1, "Policy ID is required"),
    }),
    handler: async (input, context) => {
      const result = await db.$transaction(async (tx) => {
        return recomputePolicyService({
          tx,
          organizationId: context.organizationId,
          policyId: input.policyId,
        });
      });

      return {
        message: `Recomputed balances for policy ${result.policyNum} (${result.invoiceCount} invoices)`,
      };
    },
  }),
] as const;

type ToolDefinition = (typeof toolDefinitions)[number];

function getToolDefinition(toolId: string): ToolDefinition | undefined {
  return toolDefinitions.find((t) => t.id === toolId);
}

export const callToolAction = createServerFn({ method: "POST" })
  .middleware([actionTracingMiddleware, actionAuth])
  .inputValidator(schema)
  .handler(async ({ data, context }) => {
    const tool = getToolDefinition(data.toolId);

    if (!tool) {
      throw new Error(`Unknown tool: ${data.toolId}`);
    }

    const validatedInput = tool.schema.parse(data.input);

    logger.info({
      msg: "Calling tool",
      toolId: data.toolId,
      input: validatedInput,
      userId: context.user.id,
      organizationId: context.activeOrganizationId,
    });

    const result = await tool.handler(validatedInput, {
      organizationId: context.activeOrganizationId,
      userId: context.user.id,
    });

    return {
      success: true as const,
      message: result.message,
      toolId: data.toolId,
    };
  });