mirror of
https://github.com/TriliumNext/Notes.git
synced 2025-08-19 09:02:27 +08:00
104 lines
3.0 KiB
TypeScript
104 lines
3.0 KiB
TypeScript
"use strict";
|
|
|
|
import { getContentDisposition, stripTags } from "../utils.js";
|
|
import becca from "../../becca/becca.js";
|
|
import TaskContext from "../task_context.js";
|
|
import BBranch from "../../becca/entities/bbranch.js";
|
|
import { Response } from 'express';
|
|
|
|
function exportToOpml(taskContext: TaskContext, branch: BBranch, version: string, res: Response) {
|
|
if (!['1.0', '2.0'].includes(version)) {
|
|
throw new Error(`Unrecognized OPML version ${version}`);
|
|
}
|
|
|
|
const opmlVersion = parseInt(version);
|
|
|
|
const note = branch.getNote();
|
|
|
|
function exportNoteInner(branchId: string) {
|
|
const branch = becca.getBranch(branchId);
|
|
if (!branch) { throw new Error("Unable to find branch."); }
|
|
|
|
const note = branch.getNote();
|
|
if (!note) { throw new Error("Unable to find note."); }
|
|
|
|
if (note.hasOwnedLabel('excludeFromExport')) {
|
|
return;
|
|
}
|
|
|
|
const title = `${branch.prefix ? (`${branch.prefix} - `) : ''}${note.title}`;
|
|
|
|
if (opmlVersion === 1) {
|
|
const preparedTitle = escapeXmlAttribute(title);
|
|
const preparedContent = note.hasStringContent() ? prepareText(note.getContent() as string) : '';
|
|
|
|
res.write(`<outline title="${preparedTitle}" text="${preparedContent}">\n`);
|
|
}
|
|
else if (opmlVersion === 2) {
|
|
const preparedTitle = escapeXmlAttribute(title);
|
|
const preparedContent = note.hasStringContent() ? escapeXmlAttribute(note.getContent() as string) : '';
|
|
|
|
res.write(`<outline text="${preparedTitle}" _note="${preparedContent}">\n`);
|
|
}
|
|
else {
|
|
throw new Error(`Unrecognized OPML version ${opmlVersion}`);
|
|
}
|
|
|
|
taskContext.increaseProgressCount();
|
|
|
|
for (const child of note.getChildBranches()) {
|
|
if (child?.branchId) {
|
|
exportNoteInner(child.branchId);
|
|
}
|
|
}
|
|
|
|
res.write('</outline>');
|
|
}
|
|
|
|
|
|
const filename = `${branch.prefix ? (`${branch.prefix} - `) : ''}${note.title}.opml`;
|
|
|
|
res.setHeader('Content-Disposition', getContentDisposition(filename));
|
|
res.setHeader('Content-Type', 'text/x-opml');
|
|
|
|
res.write(`<?xml version="1.0" encoding="UTF-8"?>
|
|
<opml version="${version}">
|
|
<head>
|
|
<title>Trilium export</title>
|
|
</head>
|
|
<body>`);
|
|
|
|
if (branch.branchId) {
|
|
exportNoteInner(branch.branchId);
|
|
}
|
|
|
|
res.write(`</body>
|
|
</opml>`);
|
|
res.end();
|
|
|
|
taskContext.taskSucceeded();
|
|
}
|
|
|
|
function prepareText(text: string) {
|
|
const newLines = text.replace(/(<p[^>]*>|<br\s*\/?>)/g, '\n')
|
|
.replace(/ /g, ' '); // nbsp isn't in XML standard (only HTML)
|
|
|
|
const stripped = stripTags(newLines);
|
|
|
|
const escaped = escapeXmlAttribute(stripped);
|
|
|
|
return escaped.replace(/\n/g, ' ');
|
|
}
|
|
|
|
function escapeXmlAttribute(text: string) {
|
|
return text.replace(/&/g, '&')
|
|
.replace(/</g, '<')
|
|
.replace(/>/g, '>')
|
|
.replace(/"/g, '"')
|
|
.replace(/'/g, ''');
|
|
}
|
|
|
|
export default {
|
|
exportToOpml
|
|
};
|