kandev/apps/web/components/task/chat/implement-plan-button.tsx

"use client";

import type { ReactNode } from "react";
import { IconChevronDown, IconPlus, IconRocket } from "@tabler/icons-react";
import { Button } from "@kandev/ui/button";
import {
  DropdownMenu,
  DropdownMenuContent,
  DropdownMenuItem,
  DropdownMenuTrigger,
} from "@kandev/ui/dropdown-menu";
import { Tooltip, TooltipContent, TooltipTrigger } from "@kandev/ui/tooltip";
import { cn } from "@kandev/ui/lib/utils";

/**
 * Split-button: primary "Implement" runs the plan inline in the current
 * session; the dropdown caret offers "Implement in fresh agent" which starts
 * a new session with a clean context window. `onClick(fresh)` is invoked
 * with `false` for the primary path and `true` for the fresh-agent path.
 */
type ImplementPlanButtonProps = {
  onClick: (fresh: boolean) => void | Promise<unknown>;
  disabled?: boolean;
  disabledReason?: string;
  framed?: boolean;
  testIds?: {
    root?: string;
    button?: string;
    menuTrigger?: string;
    freshItem?: string;
  };
};

function legacyRootTestId(rootId: string) {
  return rootId === "implement-plan-control" ? undefined : "implement-plan-control";
}

function DisabledImplementTooltip({
  disabledReason,
  children,
}: {
  disabledReason: string;
  children: ReactNode;
}) {
  return (
    <Tooltip>
      <TooltipTrigger asChild>
        <span tabIndex={0} className="inline-flex">
          {children}
        </span>
      </TooltipTrigger>
      <TooltipContent>{disabledReason}</TooltipContent>
    </Tooltip>
  );
}

export function ImplementPlanButton({
  onClick,
  disabled,
  disabledReason,
  framed,
  testIds,
}: ImplementPlanButtonProps) {
  const ids = {
    root: testIds?.root ?? "implement-plan-control",
    button: testIds?.button ?? "implement-plan-button",
    menuTrigger: testIds?.menuTrigger ?? "implement-plan-menu-trigger",
    freshItem: testIds?.freshItem ?? "implement-fresh-menu-item",
  };
  const controlHeightClass = framed ? "h-5" : "h-7";
  const primaryButton = (
    <span className="inline-flex">
      <Button
        type="button"
        variant="ghost"
        size="sm"
        data-testid={ids.button}
        disabled={disabled}
        className={cn(
          "gap-1.5 px-2 text-violet-400 rounded-r-none pr-1.5 border-transparent",
          controlHeightClass,
          "hover:bg-muted/40 focus-visible:border-transparent focus-visible:ring-violet-400/30",
          disabled ? "pointer-events-none cursor-not-allowed" : "cursor-pointer",
        )}
        onClick={() => onClick(false)}
      >
        <IconRocket className="h-4 w-4" />
        <span className="text-xs">Implement</span>
      </Button>
    </span>
  );
  const primary = disabledReason ? (
    primaryButton
  ) : (
    <Tooltip>
      <TooltipTrigger asChild>{primaryButton}</TooltipTrigger>
      <TooltipContent>Implement the plan in this session</TooltipContent>
    </Tooltip>
  );
  const splitButton = (
    <div
      data-testid={ids.root}
      data-legacy-testid={legacyRootTestId(ids.root)}
      className={cn(
        "inline-flex items-center rounded-md",
        framed &&
          "border border-violet-400/35 bg-background/40 focus-within:ring-[2px] focus-within:ring-violet-400/25",
      )}
    >
      {primary}
      <DropdownMenu>
        <DropdownMenuTrigger asChild>
          <Button
            type="button"
            variant="ghost"
            size="sm"
            data-testid={ids.menuTrigger}
            aria-label="More implement options"
            aria-disabled={disabled}
            disabled={disabled}
            className={cn(
              "px-1 text-violet-400 rounded-l-none border-y-0 border-r-0 border-l border-violet-400/20",
              controlHeightClass,
              "hover:bg-muted/40 focus-visible:border-y-0 focus-visible:border-r-0 focus-visible:border-l-violet-400/20 focus-visible:ring-0",
              disabled ? "pointer-events-none cursor-not-allowed" : "cursor-pointer",
            )}
          >
            <IconChevronDown className="h-3.5 w-3.5" />
          </Button>
        </DropdownMenuTrigger>
        <DropdownMenuContent align="end" className="w-64">
          <DropdownMenuItem
            data-testid={ids.freshItem}
            onClick={() => onClick(true)}
            className="cursor-pointer"
          >
            <IconPlus className="h-4 w-4 mr-2 shrink-0 self-start mt-0.5" />
            <div>
              <div>Implement in fresh agent</div>
              <div className="text-[11px] text-muted-foreground font-normal">
                Starts a new session with a clean context window
              </div>
            </div>
          </DropdownMenuItem>
        </DropdownMenuContent>
      </DropdownMenu>
    </div>
  );

  if (!disabledReason) return splitButton;

  return (
    <DisabledImplementTooltip disabledReason={disabledReason}>
      {splitButton}
    </DisabledImplementTooltip>
  );
}