refac: pdf export

This commit is contained in:
Timothy Jaeryang Baek
2025-08-26 17:34:33 +04:00
parent 07357afcf6
commit d3a952877a
11 changed files with 86 additions and 165 deletions
+30 -2
View File
@@ -1,7 +1,7 @@
<script lang="ts">
import { toast } from 'svelte-sonner';
import { DropdownMenu } from 'bits-ui';
import { getContext } from 'svelte';
import { getContext, tick } from 'svelte';
import fileSaver from 'file-saver';
const { saveAs } = fileSaver;
@@ -35,6 +35,7 @@
import Folder from '$lib/components/icons/Folder.svelte';
import Share from '$lib/components/icons/Share.svelte';
import ArchiveBox from '$lib/components/icons/ArchiveBox.svelte';
import Messages from '$lib/components/chat/Messages.svelte';
const i18n = getContext('i18n');
@@ -50,6 +51,8 @@
export let chat;
export let onClose: Function = () => {};
let showFullMessages = false;
const getChatAsText = async () => {
const history = chat.chat.history;
const messages = createMessagesList(history, history.currentId);
@@ -72,8 +75,10 @@
const downloadPdf = async () => {
if ($settings?.stylizedPdfExport ?? true) {
const containerElement = document.getElementById('messages-container');
showFullMessages = true;
await tick();
const containerElement = document.getElementById('full-messages-container');
if (containerElement) {
try {
const isDarkMode = document.documentElement.classList.contains('dark');
@@ -164,6 +169,8 @@
}
pdf.save(`chat-${chat.chat.title}.pdf`);
showFullMessages = false;
} catch (error) {
console.error('Error generating PDF', error);
}
@@ -235,6 +242,27 @@
};
</script>
{#if showFullMessages}
<div class="hidden w-full h-full flex-col">
<div id="full-messages-container">
<Messages
className="h-full flex pt-4 pb-8 w-full"
chatId={`chat-preview-${chat?.id ?? ''}`}
user={$user}
readOnly={true}
history={chat.chat.history}
messages={chat.chat.messages}
autoScroll={true}
sendMessage={() => {}}
continueResponse={() => {}}
regenerateResponse={() => {}}
messagesCount={null}
editCodeBlock={false}
/>
</div>
</div>
{/if}
<Dropdown
on:change={(e) => {
if (e.detail === false) {
@@ -81,155 +81,6 @@
saveAs(blob, `chat-${chat.chat.title}.txt`);
};
const downloadPdf = async () => {
const chat = await getChatById(localStorage.token, chatId);
if ($settings?.stylizedPdfExport ?? true) {
const containerElement = document.getElementById('messages-container');
if (containerElement) {
try {
const isDarkMode = document.documentElement.classList.contains('dark');
const virtualWidth = 800; // px, fixed width for cloned element
// Clone and style
const clonedElement = containerElement.cloneNode(true);
clonedElement.classList.add('text-black');
clonedElement.classList.add('dark:text-white');
clonedElement.style.width = `${virtualWidth}px`;
clonedElement.style.position = 'absolute';
clonedElement.style.left = '-9999px';
clonedElement.style.height = 'auto';
document.body.appendChild(clonedElement);
// Wait for DOM update/layout
await new Promise((r) => setTimeout(r, 100));
// Render entire content once
const canvas = await html2canvas(clonedElement, {
backgroundColor: isDarkMode ? '#000' : '#fff',
useCORS: true,
scale: 2, // increase resolution
width: virtualWidth
});
document.body.removeChild(clonedElement);
const pdf = new jsPDF('p', 'mm', 'a4');
const pageWidthMM = 210;
const pageHeightMM = 297;
// Convert page height in mm to px on canvas scale for cropping
// Get canvas DPI scale:
const pxPerMM = canvas.width / virtualWidth; // width in px / width in px?
// Since 1 page width is 210 mm, but canvas width is 800 px at scale 2
// Assume 1 mm = px / (pageWidthMM scaled)
// Actually better: Calculate scale factor from px/mm:
// virtualWidth px corresponds directly to 210mm in PDF, so pxPerMM:
const pxPerPDFMM = canvas.width / pageWidthMM; // canvas px per PDF mm
// Height in px for one page slice:
const pagePixelHeight = Math.floor(pxPerPDFMM * pageHeightMM);
let offsetY = 0;
let page = 0;
while (offsetY < canvas.height) {
// Height of slice
const sliceHeight = Math.min(pagePixelHeight, canvas.height - offsetY);
// Create temp canvas for slice
const pageCanvas = document.createElement('canvas');
pageCanvas.width = canvas.width;
pageCanvas.height = sliceHeight;
const ctx = pageCanvas.getContext('2d');
// Draw the slice of original canvas onto pageCanvas
ctx.drawImage(
canvas,
0,
offsetY,
canvas.width,
sliceHeight,
0,
0,
canvas.width,
sliceHeight
);
const imgData = pageCanvas.toDataURL('image/jpeg', 0.7);
// Calculate image height in PDF units keeping aspect ratio
const imgHeightMM = (sliceHeight * pageWidthMM) / canvas.width;
if (page > 0) pdf.addPage();
if (isDarkMode) {
pdf.setFillColor(0, 0, 0);
pdf.rect(0, 0, pageWidthMM, pageHeightMM, 'F'); // black bg
}
pdf.addImage(imgData, 'JPEG', 0, 0, pageWidthMM, imgHeightMM);
offsetY += sliceHeight;
page++;
}
pdf.save(`chat-${chat.chat.title}.pdf`);
} catch (error) {
console.error('Error generating PDF', error);
}
}
} else {
console.log('Downloading PDF');
const chatText = await getChatAsText(chat);
const doc = new jsPDF();
// Margins
const left = 15;
const top = 20;
const right = 15;
const bottom = 20;
const pageWidth = doc.internal.pageSize.getWidth();
const pageHeight = doc.internal.pageSize.getHeight();
const usableWidth = pageWidth - left - right;
const usableHeight = pageHeight - top - bottom;
// Font size and line height
const fontSize = 8;
doc.setFontSize(fontSize);
const lineHeight = fontSize * 1; // adjust if needed
// Split the markdown into lines (handles \n)
const paragraphs = chatText.split('\n');
let y = top;
for (let paragraph of paragraphs) {
// Wrap each paragraph to fit the width
const lines = doc.splitTextToSize(paragraph, usableWidth);
for (let line of lines) {
// If the line would overflow the bottom, add a new page
if (y + lineHeight > pageHeight - bottom) {
doc.addPage();
y = top;
}
doc.text(line, left, y);
y += lineHeight;
}
// Add empty line at paragraph breaks
y += lineHeight * 0.5;
}
doc.save(`chat-${chat.chat.title}.pdf`);
}
};
const downloadJSONExport = async () => {
const chat = await getChatById(localStorage.token, chatId);