1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42

/**
 * Markdown renders agent text (GitHub-flavoured: tables, task lists, ...).
 * Kept as the single markdown entry point so styling and plugins stay
 * consistent everywhere agent content appears.
 *
 * @example
 * <Markdown text="**bold** and `code`" />
 */

import ReactMarkdown from "react-markdown";
import remarkGfm from "remark-gfm";
import type { Components } from "react-markdown";
import { CopyButton } from "./CopyButton";

export function Markdown({ text }: { text: string }) {
	const components: Components = {
		pre: ({ children, ...props }) => {
			// Extract code content from pre > code structure
			const codeElement = children as React.ReactElement;
			const codeContent =
				codeElement?.props?.children &&
				typeof codeElement.props.children === "string"
					? codeElement.props.children
					: "";

			return (
				<div className="code-block-wrapper">
					{codeContent && <CopyButton content={codeContent} label="Copy code" />}
					<pre {...props}>{children}</pre>
				</div>
			);
		},
	};

	return (
		<div className="markdown">
			<ReactMarkdown remarkPlugins={[remarkGfm]} components={components}>
				{text}
			</ReactMarkdown>
		</div>
	);
}