@ragrabbit/mcp

Verified
MIT License
OverviewInspectSchema Related Servers Reviews Score
"use client";

import { CodeHeaderProps, MarkdownTextPrimitive, useIsMarkdownCodeBlock } from "@assistant-ui/react-markdown";
import remarkGfm from "remark-gfm";
import rehypeKatex from "rehype-katex";
import remarkMath from "remark-math";
import { FC, memo, useState } from "react";
import { CheckIcon, CopyIcon } from "lucide-react";

import { TooltipIconButton } from "./tooltip-icon-button";
import { SyntaxHighlighter } from "./syntax-highlighter";
import { cn } from "@repo/design/lib/utils";

import "katex/dist/katex.min.css";

export const CodeHeader: FC<CodeHeaderProps> = ({ language, code }) => {
  const { isCopied, copyToClipboard } = useCopyToClipboard();
  const onCopy = () => {
    if (!code || isCopied) return;
    copyToClipboard(code);
  };

  return (
    <div className="flex items-center justify-between gap-4 rounded-t-lg bg-zinc-900 px-4 py-2 text-sm font-semibold text-white">
      <span className="lowercase [&>span]:text-xs">{language}</span>
      <TooltipIconButton tooltip="Copy" onClick={onCopy}>
        {!isCopied && <CopyIcon />}
        {isCopied && <CheckIcon />}
      </TooltipIconButton>
    </div>
  );
};

export const markdownOpts = {
  remarkPlugins: [remarkGfm, remarkMath],
  rehypePlugins: [rehypeKatex],
  components: {
    h1: ({ node, className, ...props }) => (
      <h1 className={cn("mb-8 scroll-m-20 text-4xl font-extrabold tracking-tight last:mb-0", className)} {...props} />
    ),
    h2: ({ node, className, ...props }) => (
      <h2
        className={cn("mb-4 mt-8 scroll-m-20 text-3xl font-semibold tracking-tight first:mt-0 last:mb-0", className)}
        {...props}
      />
    ),
    h3: ({ node, className, ...props }) => (
      <h3
        className={cn("mb-4 mt-6 scroll-m-20 text-2xl font-semibold tracking-tight first:mt-0 last:mb-0", className)}
        {...props}
      />
    ),
    h4: ({ node, className, ...props }) => (
      <h4
        className={cn("mb-4 mt-6 scroll-m-20 text-xl font-semibold tracking-tight first:mt-0 last:mb-0", className)}
        {...props}
      />
    ),
    h5: ({ node, className, ...props }) => (
      <h5 className={cn("my-4 text-lg font-semibold first:mt-0 last:mb-0", className)} {...props} />
    ),
    h6: ({ node, className, ...props }) => (
      <h6 className={cn("my-4 font-semibold first:mt-0 last:mb-0", className)} {...props} />
    ),
    p: ({ node, className, ...props }) => (
      <p className={cn("mb-5 mt-5 leading-7 first:mt-0 last:mb-0", className)} {...props} />
    ),
    a: ({ node, className, ...props }) => (
      <a
        target="_blank"
        className={cn("text-primary font-medium underline underline-offset-4", className)}
        {...props}
      />
    ),
    blockquote: ({ node, className, ...props }) => (
      <blockquote className={cn("border-l-2 pl-6 italic", className)} {...props} />
    ),
    ul: ({ node, className, ...props }) => (
      <ul className={cn("my-5 ml-6 list-disc [&>li]:mt-2", className)} {...props} />
    ),
    ol: ({ node, className, ...props }) => (
      <ol className={cn("my-5 ml-6 list-decimal [&>li]:mt-2", className)} {...props} />
    ),
    hr: ({ node, className, ...props }) => <hr className={cn("my-5 border-b", className)} {...props} />,
    table: ({ node, className, ...props }) => (
      <table className={cn("my-5 w-full border-separate border-spacing-0 overflow-y-auto", className)} {...props} />
    ),
    th: ({ node, className, ...props }) => (
      <th
        className={cn(
          "bg-muted px-4 py-2 text-left font-bold first:rounded-tl-lg last:rounded-tr-lg [&[align=center]]:text-center [&[align=right]]:text-right",
          className
        )}
        {...props}
      />
    ),
    td: ({ node, className, ...props }) => (
      <td
        className={cn(
          "border-b border-l px-4 py-2 text-left last:border-r [&[align=center]]:text-center [&[align=right]]:text-right",
          className
        )}
        {...props}
      />
    ),
    tr: ({ node, className, ...props }) => (
      <tr
        className={cn(
          "m-0 border-b p-0 first:border-t [&:last-child>td:first-child]:rounded-bl-lg [&:last-child>td:last-child]:rounded-br-lg",
          className
        )}
        {...props}
      />
    ),
    sup: ({ node, className, ...props }) => (
      <sup className={cn("[&>a]:text-xs [&>a]:no-underline", className)} {...props} />
    ),
    pre: ({ node, className, ...props }) => (
      <pre className={cn("overflow-x-auto rounded-b-lg bg-black p-4 text-white", className)} {...props} />
    ),
    code: function Code({ node, className, ...props }) {
      const isCodeBlock = useIsMarkdownCodeBlock();
      return <code className={cn(!isCodeBlock && "rounded border font-semibold", className)} {...props} />;
    },
    CodeHeader,
    SyntaxHighlighter,
  },
};

const MarkdownTextImpl = () => {
  return <MarkdownTextPrimitive {...markdownOpts} />;
};

export const MarkdownText = memo(MarkdownTextImpl);

const useCopyToClipboard = ({
  copiedDuration = 3000,
}: {
  copiedDuration?: number;
} = {}) => {
  const [isCopied, setIsCopied] = useState<boolean>(false);

  const copyToClipboard = (value: string) => {
    if (!value) return;

    navigator.clipboard.writeText(value).then(() => {
      setIsCopied(true);
      setTimeout(() => setIsCopied(false), copiedDuration);
    });
  };

  return { isCopied, copyToClipboard };
};