siyuan/app/src/protyle/preview/index.ts

271 lines
12 KiB
TypeScript

import {getEventName, openByMobile, writeText} from "../util/compatibility";
import {hasClosestByTag} from "../util/hasClosest";
import {focusByRange} from "../util/selection";
import {showMessage} from "../../dialog/message";
import {isLocalPath, pathPosix} from "../../util/pathName";
import {previewImage} from "./image";
import {needSubscribe} from "../../util/needSubscribe";
import {Constants} from "../../constants";
import {getSearch, isMobile} from "../../util/functions";
/// #if !BROWSER
import {shell} from "electron";
/// #endif
/// #if !MOBILE
import {openAsset, openBy} from "../../editor/util";
/// #endif
import {fetchPost} from "../../util/fetch";
import {processRender} from "../util/processCode";
import {highlightRender} from "../markdown/highlightRender";
import {speechRender} from "../markdown/speechRender";
export class Preview {
public element: HTMLElement;
public previewElement: HTMLElement;
private mdTimeoutId: number;
constructor(protyle: IProtyle) {
this.element = document.createElement("div");
this.element.className = "protyle-preview fn__none";
const previewElement = document.createElement("div");
previewElement.className = "b3-typography";
if (protyle.options.classes.preview) {
previewElement.classList.add(protyle.options.classes.preview);
}
previewElement.style.padding = protyle.wysiwyg.element.style.padding;
previewElement.addEventListener("click", (event: MouseEvent & { target: HTMLElement }) => {
if (event.target.tagName === "A") {
const linkAddress = event.target.getAttribute("href");
if (linkAddress.startsWith("#")) {
// 导出预览模式点击块引转换后的脚注跳转不正确 https://github.com/siyuan-note/siyuan/issues/5700
previewElement.querySelector(linkAddress).scrollIntoView();
event.stopPropagation();
event.preventDefault();
return;
}
if (isMobile()) {
openByMobile(linkAddress);
event.stopPropagation();
event.preventDefault();
return;
}
event.stopPropagation();
event.preventDefault();
if (isLocalPath(linkAddress)) {
/// #if !MOBILE
if (Constants.SIYUAN_ASSETS_EXTS.includes(pathPosix().extname((linkAddress.split("?page")[0])))) {
openAsset(linkAddress.split("?page")[0], parseInt(getSearch("page", linkAddress)));
} else {
/// #if !BROWSER
openBy(linkAddress, "folder");
/// #endif
}
/// #endif
} else {
/// #if !BROWSER
shell.openExternal(linkAddress).catch((e) => {
showMessage(e);
});
/// #else
window.open(linkAddress);
/// #endif
}
return;
}
if (event.target.tagName === "IMG") {
previewImage((event.target as HTMLImageElement).src, protyle.block.rootID);
}
});
const actions = protyle.options.preview.actions;
const actionElement = document.createElement("div");
actionElement.className = "protyle-preview__action";
const actionHtml: string[] = [];
for (let i = 0; i < actions.length; i++) {
const action = actions[i];
if (typeof action === "object") {
actionHtml.push(`<button type="button" data-type="${action.key}" class="${action.className}">${action.text}</button>`);
continue;
}
switch (action) {
case "desktop":
actionHtml.push('<button type="button" class="protyle-preview__action--current" data-type="desktop">Desktop</button>');
break;
case "tablet":
actionHtml.push('<button type="button" data-type="tablet">Tablet</button>');
break;
case "mobile":
actionHtml.push('<button type="button" data-type="mobile">Mobile/Wechat</button>');
break;
case "mp-wechat":
actionHtml.push('<button type="button" data-type="mp-wechat" class="b3-tooltips b3-tooltips__w" aria-label="复制到公众号"><svg><use xlink:href="#iconMp"></use></svg></button>');
break;
case "zhihu":
actionHtml.push('<button type="button" data-type="zhihu" class="b3-tooltips b3-tooltips__w" aria-label="复制到知乎"><svg><use xlink:href="#iconZhihu"></use></svg></button>');
break;
case "yuque":
actionHtml.push('<button type="button" data-type="yuque" class="b3-tooltips b3-tooltips__w" aria-label="复制到语雀"><svg><use xlink:href="#iconYuque"></use></svg></button>');
break;
}
}
actionElement.innerHTML = actionHtml.join("");
this.element.appendChild(actionElement);
this.element.appendChild(previewElement);
actionElement.addEventListener(getEventName(), (event) => {
const btn = hasClosestByTag(event.target as HTMLElement, "BUTTON");
if (!btn) {
return;
}
const type = btn.getAttribute("data-type");
const actionCustom = actions.find((w: IPreviewActionCustom) => w?.key === type) as IPreviewActionCustom;
if (actionCustom) {
actionCustom.click(type);
return;
}
if ((type === "mp-wechat" || type === "zhihu" || type === "yuque")) {
this.copyToX(this.element.lastElementChild.cloneNode(true) as HTMLElement, protyle, type);
return;
}
if (type === "desktop") {
previewElement.style.width = "";
previewElement.style.padding = protyle.wysiwyg.element.style.padding;
} else if (type === "tablet") {
previewElement.style.width = "1024px";
previewElement.style.padding = "8px 16px";
} else {
previewElement.style.width = "360px";
previewElement.style.padding = "8px";
}
this.render(protyle);
actionElement.querySelectorAll("button").forEach((item) => {
item.classList.remove("protyle-preview__action--current");
});
btn.classList.add("protyle-preview__action--current");
});
this.previewElement = previewElement;
}
public render(protyle: IProtyle) {
if (this.element.style.display === "none") {
return;
}
this.mdTimeoutId = window.setTimeout(() => {
fetchPost("/api/export/preview", {
id: protyle.block.parentID || protyle.options.blockId,
}, response => {
protyle.preview.previewElement.innerHTML = response.data.html;
processRender(protyle.preview.previewElement);
highlightRender(protyle.preview.previewElement);
speechRender(protyle.preview.previewElement, protyle.options.lang);
});
}, protyle.options.preview.delay);
}
private link2online(copyElement: HTMLElement) {
if (needSubscribe("")) {
return;
}
copyElement.querySelectorAll("[href],[src]").forEach(item => {
const oldLink = item.getAttribute("href") || item.getAttribute("src");
if (oldLink && oldLink.startsWith("assets/")) {
const newLink = Constants.ASSETS_ADDRESS + window.siyuan.user.userId + "/" + oldLink;
if (item.getAttribute("href")) {
item.setAttribute("href", newLink);
} else {
item.setAttribute("src", newLink);
}
}
});
}
private copyToX(copyElement: HTMLElement, protyle: IProtyle, type?: string) {
// fix math render
if (type === "mp-wechat") {
this.link2online(copyElement);
copyElement.querySelectorAll(".katex-html .base").forEach((item: HTMLElement) => {
item.style.display = "initial";
});
copyElement.querySelectorAll("mjx-container > svg").forEach((item) => {
item.setAttribute("width", (parseInt(item.getAttribute("width")) * 8) + "px");
});
} else if (type === "zhihu") {
this.link2online(copyElement);
copyElement.querySelectorAll('[data-subtype="math"]').forEach((item: HTMLElement) => {
item.outerHTML = `<img class="Formula-image" data-eeimg="true" src="//www.zhihu.com/equation?tex=" alt="${item.getAttribute("data-content")}\\" style="display: block; margin: 0 auto; max-width: 100%;">`;
});
copyElement.querySelectorAll("blockquote").forEach((item) => {
const elements: HTMLElement[] = [];
this.processZHBlockquote(item, elements);
elements.reverse().forEach(newItem => {
item.insertAdjacentElement("afterend", newItem);
});
item.remove();
});
this.processZHTable(copyElement);
} else if (type === "yuque") {
fetchPost("/api/lute/copyStdMarkdown", {
id: protyle.block.rootID,
}, (response) => {
writeText(response.data);
showMessage("已复制,可到语雀进行粘贴");
});
return;
}
// 防止背景色被粘贴到公众号中
copyElement.style.backgroundColor = "#fff";
// 代码背景
copyElement.querySelectorAll("code").forEach((item) => {
item.style.backgroundImage = "none";
});
this.element.append(copyElement);
const cloneRange = getSelection().getRangeAt(0).cloneRange();
const range = copyElement.ownerDocument.createRange();
range.selectNodeContents(copyElement);
focusByRange(range);
document.execCommand("copy");
this.element.lastElementChild.remove();
focusByRange(cloneRange);
if (type) {
showMessage(`已复制,可到${type === "zhihu" ? "知乎" : "微信公众号平台"}进行粘贴`);
}
}
private processZHBlockquote(element: HTMLElement, elements: HTMLElement[]) {
Array.from(element.children).forEach((item: HTMLElement) => {
if (item.tagName === "BLOCKQUOTE") {
this.processZHBlockquote(item, elements);
} else if (item.tagName !== "P" || item.querySelector("img")) {
elements.push(item);
} else {
const lastElement = elements[elements.length - 1];
if (!lastElement || (lastElement && lastElement.tagName !== "BLOCKQUOTE")) {
elements.push(document.createElement("blockquote"));
}
elements[elements.length - 1].append(item);
}
});
}
private processZHTable(element: HTMLElement) {
element.querySelectorAll("table").forEach(item => {
const headElement = item.querySelector("thead");
if (!headElement) {
return;
}
const tbodyElement = item.querySelector("tbody");
if (tbodyElement) {
tbodyElement.insertAdjacentElement("afterbegin", headElement.firstElementChild);
} else {
item.innerHTML = `<tbody>${headElement.innerHTML}</tbody>`;
}
headElement.remove();
});
}
}