30 lines
829 B
TypeScript
30 lines
829 B
TypeScript
import { memo } from "react";
|
|
import { initialArtifactData, useArtifact } from "@/hooks/use-artifact";
|
|
import { CrossIcon } from "./icons";
|
|
import { Button } from "./ui/button";
|
|
|
|
function PureArtifactCloseButton() {
|
|
const { setArtifact } = useArtifact();
|
|
|
|
return (
|
|
<Button
|
|
className="h-fit p-2 dark:hover:bg-neutral-700"
|
|
data-testid="artifact-close-button"
|
|
onClick={() => {
|
|
setArtifact((currentArtifact) =>
|
|
currentArtifact.status === "streaming"
|
|
? {
|
|
...currentArtifact,
|
|
isVisible: false,
|
|
}
|
|
: { ...initialArtifactData, status: "idle" }
|
|
);
|
|
}}
|
|
variant="outline"
|
|
>
|
|
<CrossIcon size={18} />
|
|
</Button>
|
|
);
|
|
}
|
|
|
|
export const ArtifactCloseButton = memo(PureArtifactCloseButton, () => true);
|