This commit is contained in:
Romuald Wandji 2026-04-05 16:42:04 -05:00 committed by GitHub
commit 9b72c6c171
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 103 additions and 1 deletions

View file

@ -1,10 +1,11 @@
import { memo, useMemo, ReactElement } from 'react';
import { memo, useMemo, ReactElement, useRef } from 'react';
import { useRecoilValue } from 'recoil';
import MarkdownLite from '~/components/Chat/Messages/Content/MarkdownLite';
import Markdown from '~/components/Chat/Messages/Content/Markdown';
import { useMessageContext } from '~/Providers';
import { cn } from '~/utils';
import store from '~/store';
import useManualCopyToClipboard from '~/hooks/Messages/useManualCopyToClipboard';
type TextPartProps = {
text: string;
@ -22,6 +23,9 @@ const TextPart = memo(function TextPart({ text, isCreatedByUser, showCursor }: T
const enableUserMsgMarkdown = useRecoilValue(store.enableUserMsgMarkdown);
const showCursorState = useMemo(() => showCursor && isSubmitting, [showCursor, isSubmitting]);
const contentRef = useRef<HTMLDivElement>(null);
useManualCopyToClipboard(contentRef, { text });
const content: ContentType = useMemo(() => {
if (!isCreatedByUser) {
return <Markdown content={text} isLatestMessage={isLatestMessage} />;
@ -34,6 +38,7 @@ const TextPart = memo(function TextPart({ text, isCreatedByUser, showCursor }: T
return (
<div
ref={contentRef}
className={cn(
isSubmitting ? 'submitting' : '',
showCursorState && !!text.length ? 'result-streaming' : '',

View file

@ -0,0 +1,97 @@
import { useEffect } from 'react';
import { CLEANUP_REGEX, INVALID_CITATION_REGEX } from '~/utils/citations';
import type { TMessage, SearchResultData } from 'librechat-data-provider';
export default function useManualCopyToClipboard(
containerRef: React.RefObject<HTMLElement>,
messageData: Partial<Pick<TMessage, 'text' | 'content'>> & {
searchResults?: { [key: string]: SearchResultData };
},
) {
useEffect(() => {
const container = containerRef.current;
if (!container) return;
const handleManualCopy = (e: ClipboardEvent) => {
// Get the selected content
const selection = window.getSelection();
if (!selection || selection.toString().length === 0) return;
// Create a temporary container for the selected content
const tempDiv = document.createElement('div');
const range = selection.rangeCount > 0 ? selection.getRangeAt(0) : null;
const clonedSelection = range ? range.cloneContents() : null;
if (clonedSelection) {
tempDiv.appendChild(clonedSelection);
}
const stripInlineStyles = (element: Element) => {
// Remove style attribute (the main culprit for bloat)
element.removeAttribute('style');
// Keep essential classes, remove styling classes
const classList = element.getAttribute('class');
if (classList) {
const essentialClasses = classList
.split(' ')
.filter(
(cls) =>
!cls.includes('prose') &&
!cls.includes('dark:') &&
!cls.includes('light') &&
!cls.startsWith('text-') &&
!cls.startsWith('bg-') &&
!cls.startsWith('border-') &&
!cls.startsWith('shadow-') &&
!cls.startsWith('rounded-') &&
cls.trim().length > 0,
)
.join(' ');
if (essentialClasses.length > 0) {
element.setAttribute('class', essentialClasses);
} else {
element.removeAttribute('class');
}
}
// Recursively process all child elements
Array.from(element.children).forEach((child) => {
stripInlineStyles(child as Element);
});
};
// Strip inline styles from cloned content
stripInlineStyles(tempDiv);
const cleanHtml = tempDiv.innerHTML;
// === STEP 2: PLAIN TEXT VERSION ===
// Get the plain text from selection
const selectedPlainText = selection.toString();
// Apply the same cleanup as useCopyToClipboard
const cleanedText = selectedPlainText
.replace(INVALID_CITATION_REGEX, '')
.replace(CLEANUP_REGEX, '');
// Prevent default copy behavior
e.preventDefault();
// Set BOTH formats to clipboard
const clipboardData = e.clipboardData;
if (clipboardData) {
// Primary format: Clean HTML (for pasting into rich text editors)
clipboardData.setData('text/html', cleanHtml);
// Secondary format: Clean plain text (for pasting into plain text editors)
clipboardData.setData('text/plain', cleanedText);
}
};
container.addEventListener('copy', handleManualCopy);
return () => {
container.removeEventListener('copy', handleManualCopy);
};
}, [containerRef]);
}