fix #180: Added copy button

Used ref to point to parent div and got inner text of the div and write
it to the clipboard.
This commit is contained in:
ritikprajapat21 2025-07-09 19:42:41 +05:30
parent 2d6fb98130
commit 8530226edb
2 changed files with 361 additions and 245 deletions

View file

@ -0,0 +1,33 @@
"use client";
import { useState } from "react";
import type { RefObject } from "react";
import { Button } from "./ui/button";
import { Copy, CopyCheck } from "lucide-react";
export default function CopyButton({
ref,
}: {
ref: RefObject<HTMLDivElement | null>;
}) {
const [copy, setCopy] = useState(false);
const handleClick = () => {
if (ref.current) {
const text = ref.current.innerText;
navigator.clipboard.writeText(text);
setCopy(true);
setTimeout(() => {
setCopy(false);
}, 500);
}
};
return (
<div className="w-full flex justify-end">
<Button variant="ghost" onClick={handleClick}>
{copy ? <CopyCheck /> : <Copy />}
</Button>
</div>
);
}

View file

@ -1,4 +1,4 @@
import React, { useMemo, useState, useEffect } from "react"; import React, { useMemo, useState, useEffect, useRef } from "react";
import ReactMarkdown from "react-markdown"; import ReactMarkdown from "react-markdown";
import rehypeRaw from "rehype-raw"; import rehypeRaw from "rehype-raw";
import rehypeSanitize from "rehype-sanitize"; import rehypeSanitize from "rehype-sanitize";
@ -7,267 +7,350 @@ import { cn } from "@/lib/utils";
import { Citation } from "./chat/Citation"; import { Citation } from "./chat/Citation";
import { Source } from "./chat/types"; import { Source } from "./chat/types";
import { Prism as SyntaxHighlighter } from "react-syntax-highlighter"; import { Prism as SyntaxHighlighter } from "react-syntax-highlighter";
import { oneLight, oneDark } from "react-syntax-highlighter/dist/cjs/styles/prism"; import {
oneLight,
oneDark,
} from "react-syntax-highlighter/dist/cjs/styles/prism";
import { Check, Copy } from "lucide-react"; import { Check, Copy } from "lucide-react";
import { useTheme } from "next-themes"; import { useTheme } from "next-themes";
import CopyButton from "./copy-button";
interface MarkdownViewerProps { interface MarkdownViewerProps {
content: string; content: string;
className?: string; className?: string;
getCitationSource?: (id: number) => Source | null; getCitationSource?: (id: number) => Source | null;
type?: "user" | "ai";
} }
export function MarkdownViewer({ content, className, getCitationSource }: MarkdownViewerProps) { export function MarkdownViewer({
// Memoize the markdown components to prevent unnecessary re-renders content,
const components = useMemo(() => { className,
return { getCitationSource,
// Define custom components for markdown elements type = "user",
p: ({node, children, ...props}: any) => { }: MarkdownViewerProps) {
// If there's no getCitationSource function, just render normally const ref = useRef<HTMLDivElement>(null);
if (!getCitationSource) { // Memoize the markdown components to prevent unnecessary re-renders
return <p className="my-2" {...props}>{children}</p>; const components = useMemo(() => {
} return {
// Define custom components for markdown elements
p: ({ node, children, ...props }: any) => {
// If there's no getCitationSource function, just render normally
if (!getCitationSource) {
return (
<p className="my-2" {...props}>
{children}
</p>
);
}
// Process citations within paragraph content // Process citations within paragraph content
return <p className="my-2" {...props}>{processCitationsInReactChildren(children, getCitationSource)}</p>; return (
}, <p className="my-2" {...props}>
a: ({node, children, ...props}: any) => { {processCitationsInReactChildren(children, getCitationSource)}
// Process citations within link content if needed </p>
const processedChildren = getCitationSource );
? processCitationsInReactChildren(children, getCitationSource) },
: children; a: ({ node, children, ...props }: any) => {
return <a className="text-primary hover:underline" {...props}>{processedChildren}</a>; // Process citations within link content if needed
}, const processedChildren = getCitationSource
li: ({node, children, ...props}: any) => { ? processCitationsInReactChildren(children, getCitationSource)
// Process citations within list item content : children;
const processedChildren = getCitationSource return (
? processCitationsInReactChildren(children, getCitationSource) <a className="text-primary hover:underline" {...props}>
: children; {processedChildren}
return <li {...props}>{processedChildren}</li>; </a>
}, );
ul: ({node, ...props}: any) => <ul className="list-disc pl-5 my-2" {...props} />, },
ol: ({node, ...props}: any) => <ol className="list-decimal pl-5 my-2" {...props} />, li: ({ node, children, ...props }: any) => {
h1: ({node, children, ...props}: any) => { // Process citations within list item content
const processedChildren = getCitationSource const processedChildren = getCitationSource
? processCitationsInReactChildren(children, getCitationSource) ? processCitationsInReactChildren(children, getCitationSource)
: children; : children;
return <h1 className="text-2xl font-bold mt-6 mb-2" {...props}>{processedChildren}</h1>; return <li {...props}>{processedChildren}</li>;
}, },
h2: ({node, children, ...props}: any) => { ul: ({ node, ...props }: any) => (
const processedChildren = getCitationSource <ul className="list-disc pl-5 my-2" {...props} />
? processCitationsInReactChildren(children, getCitationSource) ),
: children; ol: ({ node, ...props }: any) => (
return <h2 className="text-xl font-bold mt-5 mb-2" {...props}>{processedChildren}</h2>; <ol className="list-decimal pl-5 my-2" {...props} />
}, ),
h3: ({node, children, ...props}: any) => { h1: ({ node, children, ...props }: any) => {
const processedChildren = getCitationSource const processedChildren = getCitationSource
? processCitationsInReactChildren(children, getCitationSource) ? processCitationsInReactChildren(children, getCitationSource)
: children; : children;
return <h3 className="text-lg font-bold mt-4 mb-2" {...props}>{processedChildren}</h3>; return (
}, <h1 className="text-2xl font-bold mt-6 mb-2" {...props}>
h4: ({node, children, ...props}: any) => { {processedChildren}
const processedChildren = getCitationSource </h1>
? processCitationsInReactChildren(children, getCitationSource) );
: children; },
return <h4 className="text-base font-bold mt-3 mb-1" {...props}>{processedChildren}</h4>; h2: ({ node, children, ...props }: any) => {
}, const processedChildren = getCitationSource
blockquote: ({node, ...props}: any) => <blockquote className="border-l-4 border-muted pl-4 italic my-2" {...props} />, ? processCitationsInReactChildren(children, getCitationSource)
hr: ({node, ...props}: any) => <hr className="my-4 border-muted" {...props} />, : children;
img: ({node, ...props}: any) => <img className="max-w-full h-auto my-4 rounded" {...props} />, return (
table: ({node, ...props}: any) => <div className="overflow-x-auto my-4"><table className="min-w-full divide-y divide-border" {...props} /></div>, <h2 className="text-xl font-bold mt-5 mb-2" {...props}>
th: ({node, ...props}: any) => <th className="px-3 py-2 text-left font-medium bg-muted" {...props} />, {processedChildren}
td: ({node, ...props}: any) => <td className="px-3 py-2 border-t border-border" {...props} />, </h2>
code: ({node, className, children, ...props}: any) => { );
const match = /language-(\w+)/.exec(className || ''); },
const language = match ? match[1] : ''; h3: ({ node, children, ...props }: any) => {
const isInline = !match; const processedChildren = getCitationSource
? processCitationsInReactChildren(children, getCitationSource)
: children;
return (
<h3 className="text-lg font-bold mt-4 mb-2" {...props}>
{processedChildren}
</h3>
);
},
h4: ({ node, children, ...props }: any) => {
const processedChildren = getCitationSource
? processCitationsInReactChildren(children, getCitationSource)
: children;
return (
<h4 className="text-base font-bold mt-3 mb-1" {...props}>
{processedChildren}
</h4>
);
},
blockquote: ({ node, ...props }: any) => (
<blockquote
className="border-l-4 border-muted pl-4 italic my-2"
{...props}
/>
),
hr: ({ node, ...props }: any) => (
<hr className="my-4 border-muted" {...props} />
),
img: ({ node, ...props }: any) => (
<img className="max-w-full h-auto my-4 rounded" {...props} />
),
table: ({ node, ...props }: any) => (
<div className="overflow-x-auto my-4">
<table className="min-w-full divide-y divide-border" {...props} />
</div>
),
th: ({ node, ...props }: any) => (
<th className="px-3 py-2 text-left font-medium bg-muted" {...props} />
),
td: ({ node, ...props }: any) => (
<td className="px-3 py-2 border-t border-border" {...props} />
),
code: ({ node, className, children, ...props }: any) => {
const match = /language-(\w+)/.exec(className || "");
const language = match ? match[1] : "";
const isInline = !match;
if (isInline) { if (isInline) {
return <code className="bg-muted px-1 py-0.5 rounded text-xs" {...props}>{children}</code>; return (
} <code className="bg-muted px-1 py-0.5 rounded text-xs" {...props}>
{children}
</code>
);
}
// For code blocks, add syntax highlighting and copy functionality // For code blocks, add syntax highlighting and copy functionality
return ( return (
<CodeBlock language={language} {...props}> <CodeBlock language={language} {...props}>
{String(children).replace(/\n$/, '')} {String(children).replace(/\n$/, "")}
</CodeBlock> </CodeBlock>
); );
} },
}; };
}, [getCitationSource]); }, [getCitationSource]);
return ( return (
<div className={cn("prose prose-sm dark:prose-invert max-w-none", className)}> <div
<ReactMarkdown className={cn("prose prose-sm dark:prose-invert max-w-none", className)}
rehypePlugins={[rehypeRaw, rehypeSanitize]} ref={ref}
remarkPlugins={[remarkGfm]} >
components={components} <ReactMarkdown
> rehypePlugins={[rehypeRaw, rehypeSanitize]}
{content} remarkPlugins={[remarkGfm]}
</ReactMarkdown> components={components}
</div> >
); {content}
</ReactMarkdown>
{type === "ai" && <CopyButton ref={ref} />}
</div>
);
} }
// Code block component with syntax highlighting and copy functionality // Code block component with syntax highlighting and copy functionality
const CodeBlock = ({ children, language }: { children: string, language: string }) => { const CodeBlock = ({
const [copied, setCopied] = useState(false); children,
const { resolvedTheme, theme } = useTheme(); language,
const [mounted, setMounted] = useState(false); }: {
children: string;
language: string;
}) => {
const [copied, setCopied] = useState(false);
const { resolvedTheme, theme } = useTheme();
const [mounted, setMounted] = useState(false);
// Prevent hydration issues // Prevent hydration issues
useEffect(() => { useEffect(() => {
setMounted(true); setMounted(true);
}, []); }, []);
const handleCopy = async () => { const handleCopy = async () => {
await navigator.clipboard.writeText(children); await navigator.clipboard.writeText(children);
setCopied(true); setCopied(true);
setTimeout(() => setCopied(false), 2000); setTimeout(() => setCopied(false), 2000);
}; };
// Choose theme based on current system/user preference // Choose theme based on current system/user preference
const isDarkTheme = mounted && (resolvedTheme === 'dark' || theme === 'dark'); const isDarkTheme = mounted && (resolvedTheme === "dark" || theme === "dark");
const syntaxTheme = isDarkTheme ? oneDark : oneLight; const syntaxTheme = isDarkTheme ? oneDark : oneLight;
return ( return (
<div className="relative my-4 group"> <div className="relative my-4 group">
<div className="absolute right-2 top-2 z-10"> <div className="absolute right-2 top-2 z-10">
<button <button
onClick={handleCopy} onClick={handleCopy}
className="p-1.5 rounded-md bg-background/80 hover:bg-background border border-border flex items-center justify-center transition-colors" className="p-1.5 rounded-md bg-background/80 hover:bg-background border border-border flex items-center justify-center transition-colors"
aria-label="Copy code" aria-label="Copy code"
> >
{copied ? {copied ? (
<Check size={14} className="text-green-500" /> : <Check size={14} className="text-green-500" />
<Copy size={14} className="text-muted-foreground" /> ) : (
} <Copy size={14} className="text-muted-foreground" />
</button> )}
</div> </button>
{mounted ? ( </div>
<SyntaxHighlighter {mounted ? (
language={language || 'text'} <SyntaxHighlighter
style={{ language={language || "text"}
...syntaxTheme, style={{
'pre[class*="language-"]': { ...syntaxTheme,
...syntaxTheme['pre[class*="language-"]'], 'pre[class*="language-"]': {
margin: 0, ...syntaxTheme['pre[class*="language-"]'],
border: 'none', margin: 0,
borderRadius: '0.375rem', border: "none",
background: 'var(--syntax-bg)' borderRadius: "0.375rem",
}, background: "var(--syntax-bg)",
'code[class*="language-"]': { },
...syntaxTheme['code[class*="language-"]'], 'code[class*="language-"]': {
border: 'none', ...syntaxTheme['code[class*="language-"]'],
background: 'var(--syntax-bg)' border: "none",
} background: "var(--syntax-bg)",
}} },
customStyle={{ }}
margin: 0, customStyle={{
borderRadius: '0.375rem', margin: 0,
fontSize: '0.75rem', borderRadius: "0.375rem",
lineHeight: '1.5rem', fontSize: "0.75rem",
backgroundColor: 'var(--syntax-bg)', lineHeight: "1.5rem",
border: 'none', backgroundColor: "var(--syntax-bg)",
}} border: "none",
codeTagProps={{ }}
className: "font-mono", codeTagProps={{
style: { className: "font-mono",
border: 'none', style: {
background: 'var(--syntax-bg)' border: "none",
} background: "var(--syntax-bg)",
}} },
showLineNumbers={false} }}
wrapLines={false} showLineNumbers={false}
lineProps={{ wrapLines={false}
style: { lineProps={{
wordBreak: 'break-all', style: {
whiteSpace: 'pre-wrap', wordBreak: "break-all",
border: 'none', whiteSpace: "pre-wrap",
borderBottom: 'none', border: "none",
paddingLeft: 0, borderBottom: "none",
paddingRight: 0, paddingLeft: 0,
margin: '0.25rem 0' paddingRight: 0,
} margin: "0.25rem 0",
}} },
PreTag="div" }}
> PreTag="div"
{children} >
</SyntaxHighlighter> {children}
) : ( </SyntaxHighlighter>
<div className="bg-muted p-4 rounded-md"> ) : (
<pre className="m-0 p-0 border-0"> <div className="bg-muted p-4 rounded-md">
<code className="text-xs font-mono border-0 leading-6">{children}</code> <pre className="m-0 p-0 border-0">
</pre> <code className="text-xs font-mono border-0 leading-6">
</div> {children}
)} </code>
</div> </pre>
); </div>
)}
</div>
);
}; };
// Helper function to process citations within React children // Helper function to process citations within React children
const processCitationsInReactChildren = (children: React.ReactNode, getCitationSource: (id: number) => Source | null): React.ReactNode => { const processCitationsInReactChildren = (
// If children is not an array or string, just return it children: React.ReactNode,
if (!children || (typeof children !== 'string' && !Array.isArray(children))) { getCitationSource: (id: number) => Source | null,
return children; ): React.ReactNode => {
} // If children is not an array or string, just return it
if (!children || (typeof children !== "string" && !Array.isArray(children))) {
return children;
}
// Handle string content directly - this is where we process citation references // Handle string content directly - this is where we process citation references
if (typeof children === 'string') { if (typeof children === "string") {
return processCitationsInText(children, getCitationSource); return processCitationsInText(children, getCitationSource);
} }
// Handle arrays of children recursively // Handle arrays of children recursively
if (Array.isArray(children)) { if (Array.isArray(children)) {
return React.Children.map(children, child => { return React.Children.map(children, (child) => {
if (typeof child === 'string') { if (typeof child === "string") {
return processCitationsInText(child, getCitationSource); return processCitationsInText(child, getCitationSource);
} }
return child; return child;
}); });
} }
return children; return children;
}; };
// Process citation references in text content // Process citation references in text content
const processCitationsInText = (text: string, getCitationSource: (id: number) => Source | null): React.ReactNode[] => { const processCitationsInText = (
// Use improved regex to catch citation numbers more reliably text: string,
// This will match patterns like [1], [42], etc. including when they appear at the end of a line or sentence getCitationSource: (id: number) => Source | null,
const citationRegex = /\[(\d+)\]/g; ): React.ReactNode[] => {
const parts: React.ReactNode[] = []; // Use improved regex to catch citation numbers more reliably
let lastIndex = 0; // This will match patterns like [1], [42], etc. including when they appear at the end of a line or sentence
let match; const citationRegex = /\[(\d+)\]/g;
let position = 0; const parts: React.ReactNode[] = [];
let lastIndex = 0;
let match;
let position = 0;
while ((match = citationRegex.exec(text)) !== null) { while ((match = citationRegex.exec(text)) !== null) {
// Add text before the citation // Add text before the citation
if (match.index > lastIndex) { if (match.index > lastIndex) {
parts.push(text.substring(lastIndex, match.index)); parts.push(text.substring(lastIndex, match.index));
} }
// Add the citation component // Add the citation component
const citationId = parseInt(match[1], 10); const citationId = parseInt(match[1], 10);
const source = getCitationSource(citationId); const source = getCitationSource(citationId);
parts.push( parts.push(
<Citation <Citation
key={`citation-${citationId}-${position}`} key={`citation-${citationId}-${position}`}
citationId={citationId} citationId={citationId}
citationText={match[0]} citationText={match[0]}
position={position} position={position}
source={source} source={source}
/> />,
); );
lastIndex = match.index + match[0].length; lastIndex = match.index + match[0].length;
position++; position++;
} }
// Add any remaining text after the last citation // Add any remaining text after the last citation
if (lastIndex < text.length) { if (lastIndex < text.length) {
parts.push(text.substring(lastIndex)); parts.push(text.substring(lastIndex));
} }
return parts; return parts;
}; };