添加论文搜索功能

src/components/menus/toolbar/Classic.ts

@@ -67,6 +67,7 @@ import { DeleteRow } from "./table/DeleteRow.ts";
 import { DeleteTable } from "./table/DeleteTable.ts";
 
 import { Diagrams } from "./tools/Diagrams.ts";
+import { ToggleArxiv } from "./tools/ToggleArxiv.ts";
 
 import { ToggleToc } from "./page/ToggleToc.ts";
 import { BackgroundColor } from "./page/BackgroundColor.ts";
@@ -183,6 +184,7 @@ export class Classic extends HTMLElement implements UAIEditorEventListener {
 
     // 工具菜单
     toolsMenuDiagrams!: Diagrams;
+    toolsMenuToggleArxiv!: ToggleArxiv;
 
     // 页面菜单
     pageMenuToggleToc!: ToggleToc;
@@ -266,7 +268,7 @@ export class Classic extends HTMLElement implements UAIEditorEventListener {
                 this.classicScrollableContainer.appendChild(this.classicMenuAIScrollable);
             }
         })
-        
+
         this.classicScrollableContainer = document.createElement("div");
         this.classicScrollableContainer.classList.add("uai-classic-scrollable-container");
         this.classicMenu.appendChild(this.classicScrollableContainer);
@@ -450,6 +452,9 @@ export class Classic extends HTMLElement implements UAIEditorEventListener {
         this.toolsMenuDiagrams = new Diagrams({ menuType: "button", enable: true, header: "classic", hideText: false });
         this.eventComponents.push(this.toolsMenuDiagrams);
 
+        this.toolsMenuToggleArxiv = new ToggleArxiv({ menuType: "button", enable: true, header: "classic", hideText: false });
+        this.eventComponents.push(this.toolsMenuToggleArxiv);
+
         this.pageMenuToggleToc = new ToggleToc({ menuType: "button", enable: true, header: "classic", hideText: false });
         this.eventComponents.push(this.pageMenuToggleToc);
 
@@ -627,6 +632,11 @@ export class Classic extends HTMLElement implements UAIEditorEventListener {
         group1.classList.add("uai-classic-virtual-group");
         this.classicMenuToolGroup.appendChild(group1);
         group1.appendChild(this.toolsMenuDiagrams);
+
+        const group2 = document.createElement("div");
+        group2.classList.add("uai-classic-virtual-group");
+        this.classicMenuToolGroup.appendChild(group2);
+        group2.appendChild(this.toolsMenuToggleArxiv);
     }
 
     /**

src/components/menus/toolbar/Ribbon.ts

@@ -70,6 +70,7 @@ import { DeleteRow } from "./table/DeleteRow.ts";
 import { DeleteTable } from "./table/DeleteTable.ts";
 
 import { Diagrams } from "./tools/Diagrams.ts";
+import { ToggleArxiv } from "./tools/ToggleArxiv.ts";
 
 import { ToggleToc } from "./page/ToggleToc.ts";
 import { BackgroundColor } from "./page/BackgroundColor.ts";
@@ -185,6 +186,7 @@ export class Ribbon extends HTMLElement implements UAIEditorEventListener {
 
     // 工具菜单
     toolsMenuDiagrams!: Diagrams;
+    toolsMenuToggleArxiv!: ToggleArxiv;
 
     // 页面菜单
     pageMenuToggleToc!: ToggleToc;
@@ -471,6 +473,9 @@ export class Ribbon extends HTMLElement implements UAIEditorEventListener {
         this.toolsMenuDiagrams = new Diagrams({ menuType: "button", enable: true, huge: true, hideText: false });
         this.eventComponents.push(this.toolsMenuDiagrams);
 
+        this.toolsMenuToggleArxiv = new ToggleArxiv({ menuType: "button", enable: true, huge: true, hideText: false });
+        this.eventComponents.push(this.toolsMenuToggleArxiv);
+
         this.pageMenuToggleToc = new ToggleToc({ menuType: "button", enable: true, huge: true, hideText: false });
         this.eventComponents.push(this.pageMenuToggleToc);
 
@@ -734,6 +739,11 @@ export class Ribbon extends HTMLElement implements UAIEditorEventListener {
         group1.classList.add("uai-ribbon-virtual-group");
         this.ribbonMenuToolGroup.appendChild(group1);
         group1.appendChild(this.toolsMenuDiagrams);
+
+        const group2 = document.createElement("div");
+        group2.classList.add("uai-ribbon-virtual-group");
+        this.ribbonMenuToolGroup.appendChild(group2);
+        group2.appendChild(this.toolsMenuToggleArxiv);
     }
 
     /**

src/components/menus/toolbar/tools/ToggleArxiv.ts

@@ -0,0 +1,82 @@
+// Copyright (c) 2024-present AI-Labs
+
+// @ ts-nocheck
+import { EditorEvents } from "@tiptap/core";
+
+import { t } from "i18next";
+
+import { UAIEditorEventListener, UAIEditorOptions, InnerEditor } from "../../../../core/UAIEditor.ts";
+import { MenuButton, MenuButtonOptions } from "../../MenuButton.ts";
+
+import icon from "../../../../assets/icons/arxiv.svg";
+
+/**
+ * 人工智能菜单：聊天
+ */
+export class ToggleArxiv extends HTMLElement implements UAIEditorEventListener {
+    // 按钮选项
+    menuButtonOptions: MenuButtonOptions = {
+        menuType: "button",
+        enable: true,
+        icon: icon,
+        hideText: false,
+        text: t('tools.arxiv.title'),
+        tooltip: t('tools.arxiv.title'),
+    }
+
+    // 功能按钮
+    menuButton: MenuButton;
+
+    constructor(options: MenuButtonOptions) {
+        super();
+
+        // 初始化功能按钮选项
+        this.menuButtonOptions = { ...this.menuButtonOptions, ...options };
+
+        // 创建功能按钮
+        this.menuButton = new MenuButton(this.menuButtonOptions);
+    }
+
+    /**
+     * 定义创建方法
+     * @param event 
+     * @param options 
+     */
+    onCreate(event: EditorEvents["create"], options: UAIEditorOptions) {
+        this.menuButton.onCreate(event, options);
+        this.appendChild(this.menuButton);
+
+        // 定义按钮点击事件，打开人工智能聊天窗口
+        this.addEventListener("click", () => {
+            if (this.menuButtonOptions.enable) {
+                // 判断聊天窗口是否已打开
+                const display = (event.editor as InnerEditor).uaiEditor.arxivContainer.style.display;
+                if (display != "none") {
+                    // 如果窗口已打开，则关闭窗口
+                    (event.editor as InnerEditor).uaiEditor.arxivContainer.style.display = "none";
+                } else {
+                    // 如果窗口没打开，则先关闭其他窗口，再打开聊天窗口
+                    (event.editor as InnerEditor).uaiEditor.toggleContainers.forEach(toggle => {
+                        toggle.style.display = "none";
+                    });
+                    (event.editor as InnerEditor).uaiEditor.arxivContainer.style.display = "block";
+                }
+            }
+        })
+    }
+
+    /**
+     * 定义Transaction监听方法
+     * @param event 
+     * @param options 
+     */
+    onTransaction(event: EditorEvents["transaction"], options: UAIEditorOptions) {
+        this.menuButton.onTransaction(event, options);
+    }
+
+    onEditableChange(editable: boolean) {
+        this.menuButtonOptions.enable = editable;
+        this.menuButton.onEditableChange(editable);
+    }
+}
+