gaschz/trilium
1
0
Fork 0
mirror of https://github.com/zadam/trilium.git synced 2025-03-01 14:22:32 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

inline file attachments when exporting single HTML file

Browse Source
This commit is contained in:
zadam 2023-05-29 23:42:08 +02:00
parent 235b779dec
commit f4b5d43899
4 changed files with 59 additions and 39 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/public/app/services/froca.js
View File

@ -360,8 +360,6 @@ class Froca {
opts.preview = !!opts.preview;
const key = `${entityType}-${entityId}-${opts.preview}`;
console.log(key);
if (!this.blobPromises[key]) {
this.blobPromises[key] = server.get(`${entityType}/${entityId}/blob?preview=${opts.preview}`)
.then(row => new FBlob(row))

47
src/services/export/single.js
View File

@ -23,13 +23,16 @@ function exportSingleNote(taskContext, branch, format, res) {
if (note.type === 'text') {
if (format === 'html') {
content = inlineAttachmentImages(content);
content = inlineAttachments(content);
if (!content.toLowerCase().includes("<html")) {
content = `<html><head><meta charset="utf-8"></head><body>${content}</body></html>`;
}
payload = html.prettyPrint(content, {indent_size: 2});
payload = content.length < 100_000
? html.prettyPrint(content, {indent_size: 2})
: content;
extension = 'html';
mime = 'text/html';
}
@ -61,30 +64,40 @@ function exportSingleNote(taskContext, branch, format, res) {
taskContext.taskSucceeded();
}
function inlineAttachmentImages(content) {
const re = /src="[^"]*api\/attachments\/([a-zA-Z0-9_]+)\/image\/?[^"]+"/g;
let match;
while (match = re.exec(content)) {
const attachment = becca.getAttachment(match[1]);
if (!attachment) {
continue;
}
if (!attachment.mime.startsWith('image/')) {
continue;
function inlineAttachments(content) {
content = content.replace(/src="[^"]*api\/attachments\/([a-zA-Z0-9_]+)\/image\/?[^"]+"/g, (match, attachmentId) => {
const attachment = becca.getAttachment(attachmentId);
if (!attachment || !attachment.mime.startsWith('image/')) {
return match;
}
const attachmentContent = attachment.getContent();
if (!Buffer.isBuffer(attachmentContent)) {
continue;
return match;
}
const base64Content = attachmentContent.toString('base64');
const srcValue = `data:${attachment.mime};base64,${base64Content}`;
content = content.replaceAll(match[0], `src="${srcValue}"`);
}
return `src="${srcValue}"`;
});
content = content.replace(/href="[^"]*#root[^"]*attachmentId=([a-zA-Z0-9_]+)\/?"/g, (match, attachmentId) => {
const attachment = becca.getAttachment(attachmentId);
if (!attachment) {
return match;
}
const attachmentContent = attachment.getContent();
if (!Buffer.isBuffer(attachmentContent)) {
return match;
}
const base64Content = attachmentContent.toString('base64');
const hrefValue = `data:${attachment.mime};base64,${base64Content}`;
return `href="${hrefValue}" download="${utils.escapeHtml(attachment.title)}"`;
});
return content;
}

39
src/services/export/zip.js
View File

@ -280,6 +280,26 @@ async function exportToZip(taskContext, branch, format, res, setHeaders = true)
});
content = content.replace(/src="[^"]*api\/attachments\/([a-zA-Z0-9_]+)\/image\/[^"]*"/g, (match, targetAttachmentId) => {
const url = findAttachment(targetAttachmentId);
return url ? `src="${url}"` : match;
});
content = content.replace(/href="[^"]*#root[^"]*attachmentId=([a-zA-Z0-9_]+)\/?"/g, (match, targetAttachmentId) => {
const url = findAttachment(targetAttachmentId);
return url ? `href="${url}"` : match;
});
content = content.replace(/href="[^"]*#root[a-zA-Z0-9_\/]*\/([a-zA-Z0-9_]+)[^"]*"/g, (match, targetNoteId) => {
const url = getNoteTargetUrl(targetNoteId, noteMeta);
return url ? `href="${url}"` : match;
});
return content;
function findAttachment(targetAttachmentId) {
let url;
const attachmentMeta = noteMeta.attachments.find(attMeta => attMeta.attachmentId === targetAttachmentId);
@ -289,17 +309,8 @@ async function exportToZip(taskContext, branch, format, res, setHeaders = true)
} else {
log.info(`Could not find attachment meta object for attachmentId '${targetAttachmentId}'`);
}
return url ? `src="${url}"` : match;
});
content = content.replace(/href="[^"]*#root[a-zA-Z0-9_\/]*\/([a-zA-Z0-9_]+)\/?"/g, (match, targetNoteId) => {
const url = getNoteTargetUrl(targetNoteId, noteMeta);
return url ? `href="${url}"` : match;
});
return content;
return url;
}
}
/**
@ -339,7 +350,7 @@ async function exportToZip(taskContext, branch, format, res, setHeaders = true)
</html>`;
}
return content.length < 100000
return content.length < 100_000
? html.prettyPrint(content, {indent_size: 2})
: content;
} else if (noteMeta.format === 'markdown') {
@ -451,7 +462,9 @@ ${markdownContent}`;
<ul>${saveNavigationInner(rootMeta)}</ul>
</body>
</html>`;
const prettyHtml = html.prettyPrint(fullHtml, {indent_size: 2});
const prettyHtml = fullHtml.length < 100_000
? html.prettyPrint(fullHtml, {indent_size: 2})
: fullHtml;
archive.append(prettyHtml, { name: navigationMeta.dataFileName });
}

10
src/services/ws.js
View File

@ -103,8 +103,8 @@ function fillInAdditionalProperties(entityChange) {
}
// fill in some extra data needed by the frontend
// first try to use becca which works for non-deleted entities
// only when that fails try to load from database
// first try to use becca, which works for non-deleted entities
// only when that fails, try to load from the database
if (entityChange.entityName === 'attributes') {
entityChange.entity = becca.getAttribute(entityChange.entityId);
@ -150,11 +150,7 @@ function fillInAdditionalProperties(entityChange) {
} else if (entityChange.entityName === 'blobs') {
entityChange.noteIds = sql.getColumn("SELECT noteId FROM notes WHERE blobId = ? AND isDeleted = 0", [entityChange.entityId]);
} else if (entityChange.entityName === 'attachments') {
entityChange.entity = sql.getRow(`
SELECT attachments.*, LENGTH(blobs.content)
FROM attachments
JOIN blobs ON blobs.blobId = attachments.blobId
WHERE attachmentId = ?`, [entityChange.entityId]);
entityChange.entity = becca.getAttachment(entityChange.entityId, {includeContentLength: true});
}
if (entityChange.entity instanceof AbstractBeccaEntity) {