lixiyu-net/components/mdx/PreWrapper.tsx

import { useRef, useState } from "react";
import { FaCheck } from "react-icons/fa";
import { IoCopyOutline } from "react-icons/io5";
import { Button } from "../ui/button";

const PreWrapper = ({ children }: { children: JSX.Element }) => {
  const textInput = useRef(null);
  const [hovered, setHovered] = useState(false);
  const [copied, setCopied] = useState(false);

  const onEnter = () => {
    setHovered(true);
  };
  const onExit = () => {
    setHovered(false);
    setCopied(false);
  };
  const onCopy = () => {
    setCopied(true);
    //@ts-ignore
    textInput.current && navigator.clipboard.writeText(textInput.current.textContent);
  };

  return (
    <div
      className="flat-scrollbar-normal relative"
      dir="ltr"
      onMouseLeave={onExit}
      onMouseMove={onEnter}
      ref={textInput}
    >
      {hovered && (
        <Button
          aria-label="Copy code"
          className={`absolute top-2 right-2 h-8 w-8 rounded p-1 ${
            copied ? "text-green-500 hover:text-green-500" : ""
          }`}
          onClick={onCopy}
          variant={"outline"}
        >
          {copied ? <FaCheck /> : <IoCopyOutline />}
        </Button>
      )}
      <pre
        translate="no"
        className="flat-scrollbar-normal not-prose rounded-md bg-[#F6F8FA] p-2 text-sm selection:bg-gray-300 selection:text-inherit dark:bg-[#0d1117] dark:selection:bg-gray-700"
      >
        {children}
      </pre>
    </div>
  );
};

export default PreWrapper;