mirror of
https://github.com/siyuan-note/siyuan.git
synced 2026-03-04 11:50:15 +01:00
This commit is contained in:
Vanessa
Liang Ding
parent
e4499946c5
commit
c0d9e23dcf
9 changed files with 182 additions and 54 deletions
|
|
@ -8,8 +8,12 @@ import {getContenteditableElement} from "../protyle/wysiwyg/getBlock";
|
|||
import {blockRender} from "../protyle/markdown/blockRender";
|
||||
import {processRender} from "../protyle/util/processCode";
|
||||
import {highlightRender} from "../protyle/markdown/highlightRender";
|
||||
import {Constants} from "../constants";
|
||||
import {setStorageVal} from "../protyle/util/compatibility";
|
||||
import {hasClosestByClassName} from "../protyle/util/hasClosest";
|
||||
import {escapeHtml} from "../util/escape";
|
||||
|
||||
export const fillContent = (protyle:IProtyle, data:string, elements:Element[]) => {
|
||||
export const fillContent = (protyle: IProtyle, data: string, elements: Element[]) => {
|
||||
setLastNodeRange(getContenteditableElement(elements[elements.length - 1]), protyle.toolbar.range);
|
||||
protyle.toolbar.range.collapse(true);
|
||||
insertHTML(data, protyle, true, true);
|
||||
|
|
@ -23,11 +27,122 @@ export const AIActions = (elements: Element[], protyle: IProtyle) => {
|
|||
elements.forEach(item => {
|
||||
ids.push(item.getAttribute("data-node-id"));
|
||||
});
|
||||
const customMenu: IMenu[] = [{
|
||||
iconHTML: Constants.ZWSP,
|
||||
label: window.siyuan.languages.aiCustomAction,
|
||||
click() {
|
||||
const dialog = new Dialog({
|
||||
title: window.siyuan.languages.aiCustomAction,
|
||||
content: `<div class="b3-dialog__content"><input class="b3-text-field fn__block" value=""></div>
|
||||
<div class="b3-dialog__action">
|
||||
<button class="b3-button b3-button--cancel">${window.siyuan.languages.cancel}</button><div class="fn__space"></div>
|
||||
<button class="b3-button b3-button--text">${window.siyuan.languages.confirm}</button>
|
||||
</div>`,
|
||||
width: isMobile() ? "80vw" : "520px",
|
||||
});
|
||||
const inputElement = dialog.element.querySelector("input") as HTMLInputElement;
|
||||
const btnsElement = dialog.element.querySelectorAll(".b3-button");
|
||||
dialog.bindInput(inputElement, () => {
|
||||
(btnsElement[1] as HTMLButtonElement).click();
|
||||
});
|
||||
inputElement.focus();
|
||||
btnsElement[0].addEventListener("click", () => {
|
||||
dialog.destroy();
|
||||
});
|
||||
btnsElement[1].addEventListener("click", () => {
|
||||
fetchPost("/api/ai/chatGPTWithAction", {
|
||||
ids,
|
||||
action: inputElement.value,
|
||||
}, (response) => {
|
||||
dialog.destroy();
|
||||
let respContent = "";
|
||||
if (response.data && "" !== response.data) {
|
||||
respContent = "\n\n" + response.data;
|
||||
}
|
||||
fillContent(protyle, `${inputElement.value}${respContent}`, elements);
|
||||
});
|
||||
});
|
||||
}
|
||||
}, {
|
||||
iconHTML: Constants.ZWSP,
|
||||
label: window.siyuan.languages.save,
|
||||
click() {
|
||||
const dialog = new Dialog({
|
||||
title: window.siyuan.languages.save,
|
||||
content: `<div class="b3-dialog__content">
|
||||
<input class="b3-text-field fn__size200" value="" placeholder="${window.siyuan.languages.memo}">
|
||||
<div class="fn__hr"></div>
|
||||
<textarea class="b3-text-field fn__block" placeholder="${window.siyuan.languages.aiCustomAction}"></textarea>
|
||||
</div>
|
||||
<div class="b3-dialog__action">
|
||||
<button class="b3-button b3-button--cancel">${window.siyuan.languages.cancel}</button><div class="fn__space"></div>
|
||||
<button class="b3-button b3-button--text">${window.siyuan.languages.confirm}</button>
|
||||
</div>`,
|
||||
width: isMobile() ? "80vw" : "520px",
|
||||
});
|
||||
const inputElement = dialog.element.querySelector("input");
|
||||
const btnsElement = dialog.element.querySelectorAll(".b3-button");
|
||||
dialog.bindInput(inputElement, () => {
|
||||
(btnsElement[1] as HTMLButtonElement).click();
|
||||
});
|
||||
inputElement.focus();
|
||||
btnsElement[0].addEventListener("click", () => {
|
||||
dialog.destroy();
|
||||
});
|
||||
btnsElement[1].addEventListener("click", () => {
|
||||
window.siyuan.storage[Constants.LOCAL_AI].push({
|
||||
name: inputElement.value,
|
||||
memo: dialog.element.querySelector("textarea").value,
|
||||
})
|
||||
setStorageVal(Constants.LOCAL_AI, window.siyuan.storage[Constants.LOCAL_AI]);
|
||||
dialog.destroy();
|
||||
});
|
||||
}
|
||||
}]
|
||||
window.siyuan.storage[Constants.LOCAL_AI].forEach((item: { name: string, memo: string }) => {
|
||||
customMenu.push({
|
||||
iconHTML: Constants.ZWSP,
|
||||
action: "iconCloseRound",
|
||||
label: escapeHtml(item.name),
|
||||
bind: (element) => {
|
||||
element.addEventListener("click", (event) => {
|
||||
if (hasClosestByClassName(event.target as Element, "b3-menu__action")) {
|
||||
window.siyuan.storage[Constants.LOCAL_AI].find((subItem: {
|
||||
name: string,
|
||||
memo: string
|
||||
}, index: number) => {
|
||||
if (element.querySelector(".b3-menu__label").textContent.trim() === subItem.name) {
|
||||
window.siyuan.storage[Constants.LOCAL_AI].splice(index, 1);
|
||||
setStorageVal(Constants.LOCAL_AI, window.siyuan.storage[Constants.LOCAL_AI]);
|
||||
element.remove();
|
||||
return true;
|
||||
}
|
||||
})
|
||||
} else {
|
||||
fetchPost("/api/ai/chatGPTWithAction", {
|
||||
ids,
|
||||
action: item.memo,
|
||||
}, (response) => {
|
||||
let respContent = "";
|
||||
if (response.data && "" !== response.data) {
|
||||
respContent = "\n\n" + response.data;
|
||||
}
|
||||
fillContent(protyle, `${item.memo}${respContent}`, elements);
|
||||
});
|
||||
window.siyuan.menus.menu.remove();
|
||||
}
|
||||
event.preventDefault();
|
||||
event.stopPropagation();
|
||||
});
|
||||
}
|
||||
})
|
||||
});
|
||||
window.siyuan.menus.menu.append(new MenuItem({
|
||||
icon: "iconSparkles",
|
||||
label: window.siyuan.languages.ai,
|
||||
type: "submenu",
|
||||
submenu: [{
|
||||
iconHTML: Constants.ZWSP,
|
||||
label: window.siyuan.languages.aiContinueWrite,
|
||||
click() {
|
||||
fetchPost("/api/ai/chatGPTWithAction", {ids, action: "Continue writing"}, (response) => {
|
||||
|
|
@ -35,115 +150,126 @@ export const AIActions = (elements: Element[], protyle: IProtyle) => {
|
|||
});
|
||||
}
|
||||
}, {
|
||||
iconHTML: Constants.ZWSP,
|
||||
label: window.siyuan.languages.aiTranslate,
|
||||
type: "submenu",
|
||||
submenu: [{
|
||||
iconHTML: Constants.ZWSP,
|
||||
label: window.siyuan.languages.aiTranslate_zh_CN,
|
||||
click() {
|
||||
fetchPost("/api/ai/chatGPTWithAction", {ids, action: "Translate as follows to [zh_CN]"}, (response) => {
|
||||
fetchPost("/api/ai/chatGPTWithAction", {
|
||||
ids,
|
||||
action: "Translate as follows to [zh_CN]"
|
||||
}, (response) => {
|
||||
fillContent(protyle, response.data, elements);
|
||||
});
|
||||
}
|
||||
}, {
|
||||
iconHTML: Constants.ZWSP,
|
||||
label: window.siyuan.languages.aiTranslate_ja_JP,
|
||||
click() {
|
||||
fetchPost("/api/ai/chatGPTWithAction", {ids, action: "Translate as follows to [ja_JP]"}, (response) => {
|
||||
fetchPost("/api/ai/chatGPTWithAction", {
|
||||
ids,
|
||||
action: "Translate as follows to [ja_JP]"
|
||||
}, (response) => {
|
||||
focusByRange(protyle.toolbar.range);
|
||||
insertHTML(response.data, protyle, true);
|
||||
});
|
||||
}
|
||||
}, {
|
||||
iconHTML: Constants.ZWSP,
|
||||
label: window.siyuan.languages.aiTranslate_ko_KR,
|
||||
click() {
|
||||
fetchPost("/api/ai/chatGPTWithAction", {ids, action: "Translate as follows to [ko_KR]"}, (response) => {
|
||||
fetchPost("/api/ai/chatGPTWithAction", {
|
||||
ids,
|
||||
action: "Translate as follows to [ko_KR]"
|
||||
}, (response) => {
|
||||
fillContent(protyle, response.data, elements);
|
||||
});
|
||||
}
|
||||
}, {
|
||||
iconHTML: Constants.ZWSP,
|
||||
label: window.siyuan.languages.aiTranslate_en_US,
|
||||
click() {
|
||||
fetchPost("/api/ai/chatGPTWithAction", {ids, action: "Translate as follows to [en_US]"}, (response) => {
|
||||
fetchPost("/api/ai/chatGPTWithAction", {
|
||||
ids,
|
||||
action: "Translate as follows to [en_US]"
|
||||
}, (response) => {
|
||||
fillContent(protyle, response.data, elements);
|
||||
});
|
||||
}
|
||||
}, {
|
||||
iconHTML: Constants.ZWSP,
|
||||
label: window.siyuan.languages.aiTranslate_es_ES,
|
||||
click() {
|
||||
fetchPost("/api/ai/chatGPTWithAction", {ids, action: "Translate as follows to [es_ES]"}, (response) => {
|
||||
fetchPost("/api/ai/chatGPTWithAction", {
|
||||
ids,
|
||||
action: "Translate as follows to [es_ES]"
|
||||
}, (response) => {
|
||||
fillContent(protyle, response.data, elements);
|
||||
});
|
||||
}
|
||||
}, {
|
||||
iconHTML: Constants.ZWSP,
|
||||
label: window.siyuan.languages.aiTranslate_fr_FR,
|
||||
click() {
|
||||
fetchPost("/api/ai/chatGPTWithAction", {ids, action: "Translate as follows to [fr_FR]"}, (response) => {
|
||||
fetchPost("/api/ai/chatGPTWithAction", {
|
||||
ids,
|
||||
action: "Translate as follows to [fr_FR]"
|
||||
}, (response) => {
|
||||
fillContent(protyle, response.data, elements);
|
||||
});
|
||||
}
|
||||
}, {
|
||||
iconHTML: Constants.ZWSP,
|
||||
label: window.siyuan.languages.aiTranslate_de_DE,
|
||||
click() {
|
||||
fetchPost("/api/ai/chatGPTWithAction", {ids, action: "Translate as follows to [de_DE]"}, (response) => {
|
||||
fetchPost("/api/ai/chatGPTWithAction", {
|
||||
ids,
|
||||
action: "Translate as follows to [de_DE]"
|
||||
}, (response) => {
|
||||
fillContent(protyle, response.data, elements);
|
||||
});
|
||||
}
|
||||
}]
|
||||
}, {
|
||||
iconHTML: Constants.ZWSP,
|
||||
label: window.siyuan.languages.aiExtractSummary,
|
||||
click() {
|
||||
fetchPost("/api/ai/chatGPTWithAction", {ids, action: window.siyuan.languages.aiExtractSummary}, (response) => {
|
||||
fetchPost("/api/ai/chatGPTWithAction", {
|
||||
ids,
|
||||
action: window.siyuan.languages.aiExtractSummary
|
||||
}, (response) => {
|
||||
fillContent(protyle, response.data, elements);
|
||||
});
|
||||
}
|
||||
}, {
|
||||
iconHTML: Constants.ZWSP,
|
||||
label: window.siyuan.languages.aiBrainStorm,
|
||||
click() {
|
||||
fetchPost("/api/ai/chatGPTWithAction", {ids, action: window.siyuan.languages.aiBrainStorm}, (response) => {
|
||||
fetchPost("/api/ai/chatGPTWithAction", {
|
||||
ids,
|
||||
action: window.siyuan.languages.aiBrainStorm
|
||||
}, (response) => {
|
||||
fillContent(protyle, response.data, elements);
|
||||
});
|
||||
}
|
||||
}, {
|
||||
iconHTML: Constants.ZWSP,
|
||||
label: window.siyuan.languages.aiFixGrammarSpell,
|
||||
click() {
|
||||
fetchPost("/api/ai/chatGPTWithAction", {ids, action: window.siyuan.languages.aiFixGrammarSpell}, (response) => {
|
||||
fetchPost("/api/ai/chatGPTWithAction", {
|
||||
ids,
|
||||
action: window.siyuan.languages.aiFixGrammarSpell
|
||||
}, (response) => {
|
||||
fillContent(protyle, response.data, elements);
|
||||
});
|
||||
}
|
||||
},{
|
||||
label: window.siyuan.languages.aiCustomAction,
|
||||
click() {
|
||||
const dialog = new Dialog({
|
||||
title: window.siyuan.languages.aiCustomAction,
|
||||
content: `<div class="b3-dialog__content"><input class="b3-text-field fn__block" value=""></div>
|
||||
<div class="b3-dialog__action">
|
||||
<button class="b3-button b3-button--cancel">${window.siyuan.languages.cancel}</button><div class="fn__space"></div>
|
||||
<button class="b3-button b3-button--text">${window.siyuan.languages.confirm}</button>
|
||||
</div>`,
|
||||
width: isMobile() ? "80vw" : "520px",
|
||||
});
|
||||
const inputElement = dialog.element.querySelector("input") as HTMLInputElement;
|
||||
const btnsElement = dialog.element.querySelectorAll(".b3-button");
|
||||
dialog.bindInput(inputElement, () => {
|
||||
(btnsElement[1] as HTMLButtonElement).click();
|
||||
});
|
||||
inputElement.focus();
|
||||
btnsElement[0].addEventListener("click", () => {
|
||||
dialog.destroy();
|
||||
});
|
||||
btnsElement[1].addEventListener("click", () => {
|
||||
fetchPost("/api/ai/chatGPTWithAction", {
|
||||
ids,
|
||||
action: inputElement.value,
|
||||
}, (response) => {
|
||||
dialog.destroy();
|
||||
let respContent = "";
|
||||
if (response.data && "" !== response.data) {
|
||||
respContent = "\n\n" + response.data;
|
||||
}
|
||||
fillContent(protyle, `${inputElement.value}${respContent}`, elements);
|
||||
});
|
||||
});
|
||||
}
|
||||
}, {
|
||||
iconHTML: Constants.ZWSP,
|
||||
label: window.siyuan.languages.custom,
|
||||
type: "submenu",
|
||||
submenu: customMenu
|
||||
}]
|
||||
}).element);
|
||||
};
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue