yaak/apps/yaak-proxy/components/ActionIconButton.tsx

import type { ActionInvocation } from "@yaakapp-internal/proxy-lib";
import { IconButton, type IconButtonProps } from "@yaakapp-internal/ui";
import { useCallback } from "react";
import { useRpcMutation } from "../hooks/useRpcMutation";
import { useActionMetadata } from "../hooks/useActionMetadata";

type ActionIconButtonProps = Omit<IconButtonProps, "onClick" | "title"> & {
  action: ActionInvocation;
  title?: string;
};

export function ActionIconButton({ action, ...props }: ActionIconButtonProps) {
  const meta = useActionMetadata(action);
  const { mutate, isPending } = useRpcMutation("execute_action");

  const onClick = useCallback(() => {
    mutate(action);
  }, [action, mutate]);

  return (
    <IconButton
      {...props}
      title={props.title ?? meta?.label ?? "…"}
      disabled={props.disabled || isPending}
      isLoading={isPending}
      onClick={onClick}
    />
  );
}