je-editor 0.0.202__py3-none-any.whl → 0.0.228__py3-none-any.whl

je_editor/pyside_ui/git_ui/git_client/graph_view.py

@@ -0,0 +1,218 @@
+from __future__ import annotations
+
+from typing import Optional
+
+from PySide6.QtCore import Qt, QRectF, QPointF
+from PySide6.QtGui import QPainter, QPainterPath, QPen, QColor, QBrush, QTransform
+from PySide6.QtWidgets import (
+    QGraphicsView,
+    QGraphicsScene,
+    QGraphicsEllipseItem,
+    QGraphicsPathItem, QGraphicsSimpleTextItem,
+)
+
+from je_editor.git_client.commit_graph import CommitGraph
+from je_editor.utils.multi_language.multi_language_wrapper import language_wrapper
+
+# === 版面配置常數 / Layout constants ===
+NODE_RADIUS = 6  # 節點半徑 / node radius
+ROW_HEIGHT = 28  # 每一列高度 / row height
+LANE_WIDTH = 22  # 每一條 lane 的寬度 / lane width
+LEFT_MARGIN = 220  # 左側 SHA/訊息文字區域保留空間 / left margin for SHA/message text
+
+# === 顏色設定 / Colors ===
+EDGE_COLOR = QColor("#888")  # 邊線顏色 / edge color
+TEXT_COLOR = QColor("#222")  # 文字顏色 / text color
+BG_COLOR = QColor("#ffffff")  # 背景顏色 / background color
+
+# Lane 顏色調色盤 / Lane color palette
+LANE_COLORS = [
+    QColor("#4C78A8"),
+    QColor("#F58518"),
+    QColor("#E45756"),
+    QColor("#72B7B2"),
+    QColor("#54A24B"),
+    QColor("#EECA3B"),
+    QColor("#B279A2"),
+    QColor("#FF9DA6"),
+    QColor("#9D755D"),
+    QColor("#BAB0AC"),
+]
+
+
+def lane_color(lane: int) -> QColor:
+    """
+    根據 lane index 回傳顏色
+    Return color based on lane index
+    """
+    if lane < 0:
+        return QColor("#999")
+    return LANE_COLORS[lane % len(LANE_COLORS)]
+
+
+class CommitGraphView(QGraphicsView):
+    """
+    Git Commit Graph 視覺化檢視器
+    Git commit graph visualization view
+    """
+
+    def __init__(self, parent=None):
+        super().__init__(parent)
+        self.language_wrapper_get = language_wrapper.language_word_dict.get
+
+        # 設定背景與抗鋸齒 / set background and antialiasing
+        self.setBackgroundBrush(QBrush(BG_COLOR))
+        self.setRenderHints(
+            self.renderHints()
+            | QPainter.RenderHint.Antialiasing
+            | QPainter.RenderHint.TextAntialiasing
+        )
+
+        # 啟用拖曳與縮放模式 / enable drag and zoom mode
+        self.setDragMode(QGraphicsView.DragMode.ScrollHandDrag)
+        self.setTransformationAnchor(QGraphicsView.ViewportAnchor.AnchorUnderMouse)
+
+        # 建立場景 / create scene
+        self._scene = QGraphicsScene(self)
+        self.setScene(self._scene)
+
+        self.graph: Optional[CommitGraph] = None
+        self._padding = 40
+        self._zoom = 1.0
+
+    def set_graph(self, graph: CommitGraph):
+        """
+        設定 commit graph 並重新繪製
+        Set commit graph and redraw
+        """
+        self.graph = graph
+        self._redraw()
+
+    def _lane_x(self, lane: int) -> float:
+        # 計算 lane 的 X 座標 / calculate X position for lane
+        return lane * LANE_WIDTH + NODE_RADIUS * 2
+
+    def _row_y(self, row: int) -> float:
+        # 計算 row 的 Y 座標 / calculate Y position for row
+        return row * ROW_HEIGHT + NODE_RADIUS * 2
+
+    def _redraw(self):
+        """
+        重新繪製整個 commit graph
+        Redraw the entire commit graph
+        """
+        self._scene.clear()
+        if not self.graph or not self.graph.nodes:
+            self._scene.setSceneRect(QRectF(0, 0, 800, 400))
+            return
+
+        # === 繪製邊線 (父子關係) / Draw edges (parent-child relationships) ===
+        edge_pen = QPen(EDGE_COLOR, 2)
+        for row, node in enumerate(self.graph.nodes):
+            if not node.parent_shas:
+                continue
+            x0 = self._lane_x(node.lane_index)
+            y0 = self._row_y(row)
+            for p in node.parent_shas:
+                if p not in self.graph.index:
+                    continue
+                prow = self.graph.index[p]
+                parent_node = self.graph.nodes[prow]
+                x1 = self._lane_x(parent_node.lane_index)
+                y1 = self._row_y(prow)
+
+                # 使用三次貝茲曲線繪製邊線 / cubic Bezier curve for edge
+                path = QPainterPath(QPointF(x0, y0))
+                ctrl_y = (y0 + y1) / 2.0
+                path.cubicTo(QPointF(x0, ctrl_y), QPointF(x1, ctrl_y), QPointF(x1, y1))
+                edge_item = QGraphicsPathItem(path)
+                edge_item.setPen(edge_pen)
+                self._scene.addItem(edge_item)
+
+        # === 繪製節點 (commit 圓點) / Draw nodes (commit circles) ===
+        for row, node in enumerate(self.graph.nodes):
+            cx = self._lane_x(node.lane_index)
+            cy = self._row_y(row)
+
+            circle = QGraphicsEllipseItem(
+                QRectF(cx - NODE_RADIUS, cy - NODE_RADIUS, NODE_RADIUS * 2, NODE_RADIUS * 2)
+            )
+            circle.setBrush(QBrush(lane_color(node.lane_index)))
+            circle.setPen(QPen(Qt.PenStyle.NoPen))
+            # 設定 tooltip 顯示 commit 資訊 / tooltip with commit info
+            circle.setToolTip(self.language_wrapper_get(
+                "git_graph_tooltip_commit"
+            ).format(short=node.commit_sha[:7],
+                     author=node.author_name,
+                     date=node.commit_date,
+                     msg=node.commit_message))
+            self._scene.addItem(circle)
+
+            # 在左側顯示行號 / show row number on the left
+            label_item = QGraphicsSimpleTextItem(str(row + 1))
+            label_item.setBrush(QBrush(TEXT_COLOR))
+            label_item.setPos(-30, cy - NODE_RADIUS * 2)
+            self._scene.addItem(label_item)
+
+        # 設定場景大小 / set scene rect
+        self._scene.setSceneRect(
+            QRectF(-40, 0,
+                   self._lane_x(max(n.lane_index for n in self.graph.nodes) + 1) + self._padding,
+                   self._row_y(len(self.graph.nodes)) + self._padding)
+        )
+
+    def _apply_initial_view(self):
+        """
+        初始縮放設定
+        Apply initial zoom setting
+        """
+        self._zoom = 0.9
+        self._apply_zoom_transform()
+
+    # === 滑鼠滾輪縮放 / Mouse wheel zoom ===
+    def wheelEvent(self, event):
+        if not self._scene.items():
+            return
+
+        # Ctrl+滾輪縮放，否則正常滾動
+        # Ctrl+Wheel to zoom, else scroll normally
+        if event.modifiers() & Qt.KeyboardModifier.ControlModifier:
+            angle = event.angleDelta().y()
+            factor = 1.0 + (0.1 if angle > 0 else -0.1)
+            self._zoom = max(0.1, min(3.0, self._zoom * factor))
+            self._apply_zoom_transform()
+            event.accept()
+        else:
+            super().wheelEvent(event)
+
+    def _apply_zoom_transform(self):
+        """
+        套用縮放轉換
+        Apply zoom transform
+        """
+        t = QTransform()
+        t.scale(self._zoom, self._zoom)
+        self.setTransform(t)
+
+    def resizeEvent(self, event):
+        """
+        視窗大小改變時的事件
+        Handle resize event (keep scene rect, no auto-fit)
+        """
+        super().resizeEvent(event)
+        # 保持場景大小，不自動縮放
+        # Keep scene rect; do not auto-fit
+
+    # === 外部控制輔助方法 / Helper for external controllers ===
+    def focus_row(self, row: int):
+        """
+        將檢視器聚焦到指定的 row
+        Center the view around a specific row
+        """
+        if not self.graph or row < 0 or row >= len(self.graph.nodes):
+            return
+        y = self._row_y(row)
+        rect = QRectF(0, y - ROW_HEIGHT * 2, self._scene.width(), ROW_HEIGHT * 4)
+        self.fitInView(rect, Qt.AspectRatioMode.KeepAspectRatio)
+        # 聚焦後恢復使用者縮放比例 / restore user zoom after focusing
+        self._apply_zoom_transform()

je_editor/pyside_ui/main_ui/ai_widget/ai_config.py

@@ -0,0 +1,34 @@
+from queue import Queue
+
+
+class AIConfig(object):
+    """
+    AIConfig 類別：用來管理 AI 模型的設定與訊息佇列
+    AIConfig class: manages AI model configuration and message queue
+    """
+
+    def __init__(self):
+        # 當前 AI 模型的系統提示詞 (system prompt)
+        # Current AI model system prompt
+        self.current_ai_model_system_prompt: str = ""
+
+        # 可選擇的 AI 模型設定字典
+        # Dictionary of choosable AI model configurations
+        # 結構: { "AI_model": { "ai_base_url": ..., "ai_api_key": ..., "chat_model": ..., "prompt_template": ... } }
+        self.choosable_ai: dict[str, dict[str, str]] = {
+            "AI_model": {
+                "ai_base_url": "",  # AI 服務的基礎 URL / Base URL of AI service
+                "ai_api_key": "",  # API 金鑰 / API key
+                "chat_model": "",  # 模型名稱 / Model name
+                "prompt_template": "",  # 提示詞模板 / Prompt template
+            }
+        }
+
+        # 訊息佇列，用來暫存與 AI 的互動訊息
+        # Message queue for storing AI interaction messages
+        self.message_queue = Queue()
+
+
+# 建立全域唯一的 AIConfig 實例
+# Create a global singleton instance of AIConfig
+ai_config = AIConfig()

je_editor/pyside_ui/main_ui/ai_widget/ask_thread.py

@@ -0,0 +1,36 @@
+from threading import Thread
+
+from je_editor.pyside_ui.main_ui.ai_widget.ai_config import ai_config
+from je_editor.pyside_ui.main_ui.ai_widget.langchain_interface import LangChainInterface
+
+
+class AskThread(Thread):
+    """
+    AskThread 類別：用來在背景執行緒中呼叫 AI 模型，避免阻塞主執行緒
+    AskThread class: runs AI model calls in a background thread to avoid blocking the main thread
+    """
+
+    def __init__(self, lang_chain_interface: LangChainInterface, prompt):
+        """
+        初始化 AskThread
+        Initialize AskThread
+        :param lang_chain_interface: LangChainInterface 實例，用來呼叫 AI 模型
+                                     LangChainInterface instance for calling AI model
+        :param prompt: 傳給 AI 模型的提示詞
+                       Prompt to send to the AI model
+        """
+        super().__init__()
+        self.lang_chain_interface = lang_chain_interface
+        self.prompt = prompt
+
+    def run(self):
+        """
+        執行緒的主要邏輯：
+        1. 呼叫 AI 模型並取得回應
+        2. 將回應放入 ai_config 的 message_queue 中，供主程式使用
+        Thread main logic:
+        1. Call AI model and get response
+        2. Put response into ai_config.message_queue for main program to consume
+        """
+        ai_response = self.lang_chain_interface.call_ai_model(prompt=self.prompt)
+        ai_config.message_queue.put(ai_response)

je_editor/pyside_ui/main_ui/ai_widget/chat_ui.py

@@ -0,0 +1,147 @@
+from __future__ import annotations
+
+from pathlib import Path
+from typing import TYPE_CHECKING, Union
+
+from PySide6.QtCore import Qt, QTimer
+from PySide6.QtGui import QFontDatabase
+from PySide6.QtWidgets import QWidget, QPlainTextEdit, QScrollArea, QLabel, QComboBox, QGridLayout, QPushButton, \
+    QMessageBox, QSizePolicy, QLineEdit
+
+from je_editor.pyside_ui.dialog.ai_dialog.set_ai_dialog import SetAIDialog
+from je_editor.pyside_ui.main_ui.ai_widget.ai_config import AIConfig, ai_config
+from je_editor.pyside_ui.main_ui.ai_widget.ask_thread import AskThread
+from je_editor.pyside_ui.main_ui.ai_widget.langchain_interface import LangChainInterface
+from je_editor.utils.json.json_file import read_json
+from je_editor.utils.multi_language.multi_language_wrapper import language_wrapper
+
+if TYPE_CHECKING:
+    from je_editor.pyside_ui.main_ui.main_editor import EditorMain
+
+
+class ChatUI(QWidget):
+
+    def __init__(self, main_window: EditorMain):
+        super().__init__()
+        self.setAttribute(Qt.WidgetAttribute.WA_DeleteOnClose)  # 關閉視窗時自動釋放資源 / Auto delete on close
+        self.main_window = main_window
+
+        # ---------------- Chat Panel 聊天面板 ----------------
+        self.chat_panel = QPlainTextEdit()  # 顯示聊天訊息的文字框 / Text area for chat messages
+        self.chat_panel.setLineWrapMode(self.chat_panel.LineWrapMode.NoWrap)  # 不自動換行 / Disable line wrap
+        self.chat_panel.setReadOnly(True)  # 設為唯讀，避免使用者直接輸入 / Read-only
+        self.chat_panel_scroll_area = QScrollArea()  # 加入滾動區域 / Scroll area for chat panel
+        self.chat_panel_scroll_area.setWidgetResizable(True)
+        self.chat_panel_scroll_area.setViewportMargins(0, 0, 0, 0)
+        self.chat_panel_scroll_area.setWidget(self.chat_panel)
+        self.chat_panel.setFont(QFontDatabase.font(self.font().family(), "", 16))  # 設定字體大小 / Set font size
+
+        # ---------------- Prompt Input 輸入框 ----------------
+        self.prompt_input = QLineEdit()  # 使用者輸入提示詞 / Input field for prompts
+        self.prompt_input.setSizePolicy(QSizePolicy.Policy.Expanding, QSizePolicy.Policy.Minimum)
+        self.prompt_input.returnPressed.connect(self.call_ai_model)  # 按 Enter 時呼叫 AI / Call AI on Enter
+
+        # ---------------- Font Size Combobox 字體大小選單 ----------------
+        self.font_size_label = QLabel(language_wrapper.language_word_dict.get("font_size"))  # 標籤 / Label
+        self.font_size_combobox = QComboBox()  # 下拉選單 / Dropdown for font size
+        for font_size in range(2, 101, 2):  # 提供 2~100 的字體大小選項 / Font size options
+            self.font_size_combobox.addItem(str(font_size))
+        self.font_size_combobox.setCurrentText("16")  # 預設字體大小 / Default font size
+        self.font_size_combobox.currentTextChanged.connect(self.update_panel_text_size)
+
+        # ---------------- Buttons 按鈕 ----------------
+        self.set_ai_config_button = QPushButton(language_wrapper.language_word_dict.get("chat_ui_set_ai_button"))
+        self.set_ai_config_button.clicked.connect(self.set_ai_config)  # 開啟 AI 設定視窗 / Open AI config dialog
+
+        self.load_ai_config_button = QPushButton(language_wrapper.language_word_dict.get("chat_ui_load_ai_button"))
+        self.load_ai_config_button.clicked.connect(
+            lambda: self.load_ai_config(show_load_complete=True))  # 載入設定 / Load config
+
+        self.call_ai_model_button = QPushButton(language_wrapper.language_word_dict.get("chat_ui_call_ai_model_button"))
+        self.call_ai_model_button.clicked.connect(self.call_ai_model)  # 呼叫 AI / Call AI
+
+        # ---------------- Layout 版面配置 ----------------
+        self.grid_layout = QGridLayout()
+        self.grid_layout.addWidget(self.chat_panel_scroll_area, 0, 0, 1, 4)  # 聊天面板 / Chat panel
+        self.grid_layout.addWidget(self.call_ai_model_button, 1, 0)  # 呼叫 AI 按鈕 / Call AI button
+        self.grid_layout.addWidget(self.font_size_combobox, 1, 1)  # 字體大小選單 / Font size combobox
+        self.grid_layout.addWidget(self.set_ai_config_button, 1, 2)  # 設定 AI 按鈕 / Set AI config button
+        self.grid_layout.addWidget(self.load_ai_config_button, 1, 3)  # 載入設定按鈕 / Load AI config button
+        self.grid_layout.addWidget(self.prompt_input, 2, 0, 1, 4)  # 輸入框 / Prompt input
+
+        # ---------------- Variables 變數 ----------------
+        self.ai_config: AIConfig = ai_config  # AI 設定物件 / AI config object
+        self.lang_chain_interface: Union[LangChainInterface, None] = None  # LangChain 介面 / LangChain interface
+        self.set_ai_config_dialog = None  # 設定對話框 / Config dialog
+
+        # ---------------- Timer 計時器 ----------------
+        self.pull_message_timer = QTimer(self)  # 定時檢查訊息佇列 / Timer to pull messages
+        self.pull_message_timer.setInterval(1000)  # 每秒檢查一次 / Check every 1 second
+        self.pull_message_timer.timeout.connect(self.pull_message)
+        self.pull_message_timer.start()
+
+        # ---------------- Set Layout 設定版面 ----------------
+        self.setLayout(self.grid_layout)
+
+        # ---------------- Load AI Config 載入 AI 設定 ----------------
+        self.load_ai_config()
+
+    # 更新聊天面板字體大小 / Update chat panel font size
+    def update_panel_text_size(self):
+        self.chat_panel.setFont(
+            QFontDatabase.font(self.font().family(), "", int(self.font_size_combobox.currentText())))
+
+    # 載入 AI 設定檔 / Load AI configuration file
+    def load_ai_config(self, show_load_complete: bool = False):
+        ai_config_file = Path(str(Path.cwd()) + "/" + ".jeditor/ai_config.json")
+        if ai_config_file.exists():
+            with open(ai_config_file, "r", encoding="utf-8"):
+                json_data: dict = read_json(str(ai_config_file))
+            if json_data:
+                # 確認 AI_model 設定存在且格式正確 / Ensure AI_model config exists and valid
+                if json_data.get("AI_model") and len(json_data.get("AI_model")) == 4:
+                    ai_info: dict = json_data.get("AI_model")
+                    if ai_info.get("ai_base_url") and ai_info.get("chat_model"):
+                        ai_config.choosable_ai.update(json_data)  # 更新全域設定 / Update global config
+                    # 建立 LangChain 介面 / Initialize LangChain interface
+                    self.lang_chain_interface = LangChainInterface(
+                        main_window=self,
+                        api_key=ai_info.get("ai_api_key"),
+                        base_url=ai_info.get("ai_base_url"),
+                        chat_model=ai_info.get("chat_model"),
+                        prompt_template=ai_info.get("prompt_template"),
+                    )
+            if show_load_complete:
+                load_complete = QMessageBox(self)
+                load_complete.setWindowTitle(language_wrapper.language_word_dict.get("load_ai_messagebox_title"))
+                load_complete.setText(language_wrapper.language_word_dict.get("load_ai_messagebox_text"))
+                load_complete.exec()
+
+    # 呼叫 AI 模型 / Call AI model
+    def call_ai_model(self):
+        if isinstance(self.lang_chain_interface, LangChainInterface):
+            # 建立新執行緒處理 AI 請求 / Start a new thread for AI request
+            thread = AskThread(lang_chain_interface=self.lang_chain_interface, prompt=self.prompt_input.text())
+            thread.start()
+        else:
+            # 若未正確設定 AI，顯示錯誤訊息 / Show error if AI not configured
+            ai_info = ai_config.choosable_ai.get('AI_model')
+            QMessageBox.warning(self,
+                                language_wrapper.language_word_dict.get("call_ai_model_error_title"),
+                                language_wrapper.language_word_dict.get(
+                                    f"ai_api_key: {ai_info.get('ai_api_key')}, \n"
+                                    f"ai_base_url: {ai_info.get('ai_base_url')}, \n"
+                                    f"chat_model: {ai_info.get('chat_model')}, \n"
+                                    f"prompt_template: {ai_info.get('prompt_template')}"))
+
+    # 從訊息佇列中取出 AI 回覆並顯示 / Pull AI response from queue
+    def pull_message(self):
+        if not ai_config.message_queue.empty():
+            ai_response = ai_config.message_queue.get_nowait()
+            self.chat_panel.appendPlainText(ai_response)  # 顯示回覆 / Display response
+            self.chat_panel.appendPlainText("\n")
+
+    # 開啟 AI 設定對話框 / Open AI config dialog
+    def set_ai_config(self):
+        self.set_ai_config_dialog = SetAIDialog()
+        self.set_ai_config_dialog.show()

je_editor/pyside_ui/main_ui/ai_widget/langchain_interface.py

@@ -0,0 +1,84 @@
+from __future__ import annotations
+
+import os
+import re
+from typing import Union, TYPE_CHECKING
+
+from PySide6.QtWidgets import QMessageBox
+from langchain_core.prompts.chat import SystemMessagePromptTemplate
+from langchain_openai import ChatOpenAI
+from pydantic import SecretStr
+
+from je_editor.utils.multi_language.multi_language_wrapper import language_wrapper
+
+if TYPE_CHECKING:
+    from je_editor.pyside_ui.main_ui.ai_widget.chat_ui import ChatUI
+
+
+class LangChainInterface(object):
+    """
+    LangChainInterface 負責與 LangChain + OpenAI 模型互動
+    LangChainInterface is responsible for interacting with LangChain + OpenAI model
+    """
+
+    def __init__(self, main_window: ChatUI, prompt_template: str, base_url: str,
+                 api_key: Union[SecretStr, str], chat_model: str):
+        """
+        初始化 LangChainInterface
+        Initialize LangChainInterface
+
+        :param main_window: 主視窗，用於顯示錯誤訊息 / Main window, used for showing error messages
+        :param prompt_template: 系統提示詞模板 / System prompt template
+        :param base_url: OpenAI API 的基礎 URL / Base URL for OpenAI API
+        :param api_key: OpenAI API 金鑰 / OpenAI API key
+        :param chat_model: 使用的聊天模型名稱 / Chat model name
+        """
+        # 建立系統提示詞模板 / Create system message prompt template
+        self.system_message_prompt = SystemMessagePromptTemplate.from_template(prompt_template)
+
+        # 儲存基本設定 / Store basic settings
+        self.base_url = base_url
+        self.api_key = api_key
+        self.chat_model = chat_model
+        self.main_window = main_window
+
+        # 將設定寫入環境變數，方便其他套件讀取
+        # Save settings into environment variables for other packages to read
+        os.environ["OPENAI_BASE_URL"] = self.base_url
+        os.environ["OPENAI_API_KEY"] = self.api_key
+        os.environ["CHAT_MODEL"] = self.chat_model
+
+        # 初始化 ChatOpenAI 物件，用於呼叫模型
+        # Initialize ChatOpenAI object for model invocation
+        self.chat_ai = ChatOpenAI(base_url=self.base_url, api_key=self.api_key, model=self.chat_model)
+
+    def call_ai_model(self, prompt: str) -> str | None:
+        """
+        呼叫 AI 模型並回傳結果
+        Call AI model and return response
+
+        :param prompt: 使用者輸入的提示詞 / User input prompt
+        :return: AI 回覆文字或 None / AI response text or None
+        """
+        message = None
+        try:
+            # 呼叫 AI 並取得回覆 / Invoke AI and get response
+            message = self.chat_ai.invoke(prompt).text()
+
+            # 嘗試過濾掉 <think> 標籤前的內容，只保留主要回覆
+            # Try to filter out content before </think>, keep only main response
+            match = re.search(r"</think>\s*(.*)", message, re.DOTALL)
+            if match:
+                message = match.group(1).strip()
+            else:
+                message = message
+
+        except Exception as error:
+            # 發生錯誤時，彈出警告視窗顯示錯誤訊息
+            # Show error message in a warning dialog if exception occurs
+            QMessageBox.warning(
+                self.main_window,
+                language_wrapper.language_word_dict.get("call_ai_model_error_title"),
+                str(error)
+            )
+        return message