From 6036252976a4cbe4c178849ccdb67b4e83566b60 Mon Sep 17 00:00:00 2001
From: Lucas Oliveira <lucas.edu.oli@hotmail.com>
Date: Thu, 11 Sep 2025 16:42:37 -0300
Subject: [PATCH] Added code component and markdown renderer

---
 frontend/components/code-component.tsx    |  56 +++++++++
 frontend/components/markdown-renderer.tsx | 142 ++++++++++++++++++++++
 2 files changed, 198 insertions(+)
 create mode 100644 frontend/components/code-component.tsx
 create mode 100644 frontend/components/markdown-renderer.tsx

diff --git a/frontend/components/code-component.tsx b/frontend/components/code-component.tsx
new file mode 100644
index 00000000..e612484d
--- /dev/null
+++ b/frontend/components/code-component.tsx
@@ -0,0 +1,56 @@
+import { Check, Copy } from "lucide-react";
+import { useState } from "react";
+import { Prism as SyntaxHighlighter } from "react-syntax-highlighter";
+import { tomorrow } from "react-syntax-highlighter/dist/cjs/styles/prism";
+import { Button } from "./ui/button";
+
+type CodeComponentProps = {
+  code: string;
+  language: string;
+};
+
+export default function CodeComponent({ code, language }: CodeComponentProps) {
+  const [isCopied, setIsCopied] = useState<boolean>(false);
+
+  const copyToClipboard = () => {
+    if (!navigator.clipboard || !navigator.clipboard.writeText) {
+      return;
+    }
+
+    navigator.clipboard.writeText(code).then(() => {
+      setIsCopied(true);
+
+      setTimeout(() => {
+        setIsCopied(false);
+      }, 2000);
+    });
+  };
+
+  return (
+    <div
+      className="mt-2 relative flex w-full flex-col overflow-hidden rounded-md text-left dark"
+      data-testid="chat-code-tab"
+    >
+      <Button
+        variant="ghost"
+        size="icon"
+        className="text-muted-foreground hover:bg-card absolute top-2 right-2"
+        data-testid="copy-code-button"
+        onClick={copyToClipboard}
+      >
+        {isCopied ? (
+          <Check className="h-4 w-4" />
+        ) : (
+          <Copy className="h-4 w-4" />
+        )}
+      </Button>
+      <SyntaxHighlighter
+        language={language.toLowerCase()}
+        style={tomorrow}
+        className="!mt-0 h-full w-full overflow-scroll !rounded-b-md !rounded-t-none border border-border text-left !custom-scroll !text-sm"
+      >
+        {code}
+      </SyntaxHighlighter>
+    </div>
+  );
+}
diff --git a/frontend/components/markdown-renderer.tsx b/frontend/components/markdown-renderer.tsx
new file mode 100644
index 00000000..d6c8c81a
--- /dev/null
+++ b/frontend/components/markdown-renderer.tsx
@@ -0,0 +1,142 @@
+import Markdown from "react-markdown";
+import rehypeMathjax from "rehype-mathjax";
+import rehypeRaw from "rehype-raw";
+import remarkGfm from "remark-gfm";
+import { cn } from "@/lib/utils";
+import CodeComponent from "./code-component";
+
+type MarkdownRendererProps = {
+  chatMessage: string;
+};
+
+const preprocessChatMessage = (text: string): string => {
+  // Handle <think> tags
+  let processed = text
+    .replace(/<think>/g, "`<think>`")
+    .replace(/<\/think>/g, "`</think>`");
+
+  // Clean up tables if present
+  if (isMarkdownTable(processed)) {
+    processed = cleanupTableEmptyCells(processed);
+  }
+
+  return processed;
+};
+
+export const isMarkdownTable = (text: string): boolean => {
+  if (!text?.trim()) return false;
+
+  // Single regex to detect markdown table with header separator
+  return /\|.*\|.*\n\s*\|[\s\-:]+\|/m.test(text);
+};
+
+export const cleanupTableEmptyCells = (text: string): string => {
+  return text
+    .split("\n")
+    .filter((line) => {
+      const trimmed = line.trim();
+
+      // Keep non-table lines
+      if (!trimmed.includes("|")) return true;
+
+      // Keep separator rows (contain only |, -, :, spaces)
+      if (/^\|[\s\-:]+\|$/.test(trimmed)) return true;
+
+      // For data rows, check if any cell has content
+      const cells = trimmed.split("|").slice(1, -1); // Remove delimiter cells
+      return cells.some((cell) => cell.trim() !== "");
+    })
+    .join("\n");
+};
+export const MarkdownRenderer = ({ chatMessage }: MarkdownRendererProps) => {
+  // Process the chat message to handle <think> tags and clean up tables
+  const processedChatMessage = preprocessChatMessage(chatMessage);
+
+  return (
+    <div
+      className={cn(
+        "markdown prose flex w-full max-w-full flex-col items-baseline text-base font-normal word-break-break-word dark:prose-invert",
+        !chatMessage ? "text-muted-foreground" : "text-primary",
+      )}
+    >
+      <Markdown
+        remarkPlugins={[remarkGfm]}
+        rehypePlugins={[rehypeMathjax, rehypeRaw]}
+        linkTarget="_blank"
+        components={{
+          p({ node, ...props }) {
+            return <p className="w-fit max-w-full">{props.children}</p>;
+          },
+          ol({ node, ...props }) {
+            return <ol className="max-w-full">{props.children}</ol>;
+          },
+          h1({ node, ...props }) {
+            return <h1 className="mb-6 mt-4">{props.children}</h1>;
+          },
+          h2({ node, ...props }) {
+            return <h2 className="mb-4 mt-4">{props.children}</h2>;
+          },
+          h3({ node, ...props }) {
+            return <h3 className="mb-2 mt-4">{props.children}</h3>;
+          },
+          hr({ node, ...props }) {
+            return <hr className="w-full mt-4 mb-8" />;
+          },
+          ul({ node, ...props }) {
+            return <ul className="max-w-full mb-2">{props.children}</ul>;
+          },
+          pre({ node, ...props }) {
+            return <>{props.children}</>;
+          },
+          table: ({ node, ...props }) => {
+            return (
+              <div className="max-w-full overflow-hidden rounded-md border bg-muted">
+                <div className="max-h-[600px] w-full overflow-auto p-4">
+                  <table className="!my-0 w-full">{props.children}</table>
+                </div>
+              </div>
+            );
+          },
+
+          code: ({ node, className, inline, children, ...props }) => {
+            let content = children as string;
+            if (
+              Array.isArray(children) &&
+              children.length === 1 &&
+              typeof children[0] === "string"
+            ) {
+              content = children[0] as string;
+            }
+            if (typeof content === "string") {
+              if (content.length) {
+                if (content[0] === "▍") {
+                  return <span className="form-modal-markdown-span"></span>;
+                }
+
+                // Specifically handle <think> tags that were wrapped in backticks
+                if (content === "<think>" || content === "</think>") {
+                  return <span>{content}</span>;
+                }
+              }
+
+              const match = /language-(\w+)/.exec(className || "");
+
+              return !inline ? (
+                <CodeComponent
+                  language={(match && match[1]) || ""}
+                  code={String(content).replace(/\n$/, "")}
+                />
+              ) : (
+                <code className={className} {...props}>
+                  {content}
+                </code>
+              );
+            }
+          },
+        }}
+      >
+        {processedChatMessage}
+      </Markdown>
+    </div>
+  );
+};