"""Cột trái: thư viện workflow, agent, skill — R08-T09. Bốn mục gập được (WORKFLOWS / AGENTS / SKILLS / FLOW STATUS). Trạng thái gập của từng mục là thứ người dùng đặt rồi mong nó giữ nguyên, nên nó nằm trong cấu hình chứ không phải trong widget. """ from __future__ import annotations import re from typing import List from PySide6.QtCore import QSize, Qt from PySide6.QtWidgets import QHBoxLayout, QListWidget, QListWidgetItem, QPushButton, QSplitter, QVBoxLayout, QWidget from ...core import co4e, skills as skills_mod from ...i18n import tr from ...ui.icons import icon from ...presentation.co4e.agent_list_panel import AgentListPanel from ...presentation.co4e.co4e_chat_view import _skill_names from ...presentation.co4e.palette_list import _PaletteList from ...presentation.co4e.skills_list_panel import SkillsListPanel class Co4ESidebarMixin: def _build_sidebar(self) -> QWidget: # ONE COLUMN, four named sections — no icon tabs. Every list is on screen # at once, so "what can I drag onto the canvas" is answered by looking # rather than by clicking through three unlabeled tabs. # A vertical splitter, not a fixed stack: on a short window four stacked # lists otherwise squeeze down to one visible row each. The splitter # hands out the available height by weight and lets the user re-balance # it by dragging; each list keeps a small minimum so none disappears. self._sections: dict = {} self.sidebar = QWidget() outer_col = QVBoxLayout(self.sidebar) outer_col.setContentsMargins(6, 6, 6, 6) outer_col.setSpacing(0) self.side_split = QSplitter(Qt.Vertical) self.side_split.setChildrenCollapsible(False) self.side_split.setHandleWidth(8) outer_col.addWidget(self.side_split, 1) class _Col: """Adapter so the section builders below read the same as before.""" def __init__(self, split): self._split = split def addWidget(self, w, stretch=1): self._split.addWidget(w) self._split.setStretchFactor(self._split.count() - 1, stretch) col = _Col(self.side_split) # --- WORKFLOWS --------------------------------------------------- self.wf_new_btn = QPushButton(tr("co4e.new")) self.wf_new_btn.setIcon(icon("plus")) self.wf_new_btn.setToolTip(tr("co4e.tt_new_wf")) self.wf_new_btn.setObjectName("co4eSectionAction") self.wf_new_btn.setFlat(True) self.wf_new_btn.setCursor(Qt.PointingHandCursor) self.wf_new_btn.clicked.connect(self._new_workflow) wf_body = QWidget(); wl = QVBoxLayout(wf_body) wl.setContentsMargins(0, 0, 0, 0); wl.setSpacing(4) # Draggable: drag a flow onto the canvas to merge it in (Nova-style); # double-click loads it onto the canvas. self.wf_list = _PaletteList(payload_role=Qt.UserRole + 2) self.wf_list.setToolTip(tr("co4e.drag_hint")) self.wf_list.itemDoubleClicked.connect(self._load_selected_workflow) self.wf_list.setContextMenuPolicy(Qt.CustomContextMenu) self.wf_list.customContextMenuRequested.connect(self._wf_context_menu) wl.addWidget(self.wf_list, 1) wf_btns = QHBoxLayout(); wf_btns.setSpacing(4) self.wf_edit_btn = self._icon_btn("edit", "co4e.tt_edit_wf", self._edit_selected_workflow) self.wf_dup_btn = self._icon_btn("branch", "co4e.tt_dup_wf", self._duplicate_selected_workflow) self.wf_del_btn = self._icon_btn("trash", "co4e.tt_del_wf", self._delete_selected_workflow) for b in (self.wf_edit_btn, self.wf_dup_btn, self.wf_del_btn): wf_btns.addWidget(b) wf_btns.addStretch(1) wl.addLayout(wf_btns) # Its own row: sharing one line with the three icon buttons cut "Chạy # nền" down to "Chạ" as soon as the sidebar hit its narrow width. self.wf_runbg_btn = QPushButton(tr("co4e.run_bg")); self.wf_runbg_btn.setIcon(icon("play")) self.wf_runbg_btn.setToolTip(tr("co4e.tt_run_bg")) self.wf_runbg_btn.clicked.connect(self._run_selected_in_background) wl.addWidget(self.wf_runbg_btn) col.addWidget(self._section("co4e.tab_workflows", wf_body, self.wf_new_btn), 3) # --- AGENTS ------------------------------------------------------ # Widget cua khu vuc nay da doi sang AgentListPanel (xem # presentation/co4e/agent_list_panel.py); o day chi con giu # ag_new_btn/agent_list/ag_edit_btn/ag_del_btn nhu 4 ten thuoc tinh cu # va tu noi signal - dung nguyen tac panel khong tu wire, Co4ETab moi # biet _new_agent/_edit_agent/_delete_agent. self._agent_panel = AgentListPanel() self.ag_new_btn = self._agent_panel.new_btn self.ag_new_btn.clicked.connect(self._new_agent) self.agent_list = self._agent_panel.list_widget self.ag_edit_btn = self._agent_panel.edit_btn self.ag_edit_btn.clicked.connect(self._edit_agent) self.ag_del_btn = self._agent_panel.del_btn self.ag_del_btn.clicked.connect(self._delete_agent) col.addWidget(self._section("co4e.tab_agents", self._agent_panel, self.ag_new_btn), 3) # --- SKILLS ------------------------------------------------------ # Widget cua khu vuc nay da doi sang SkillsListPanel (xem # presentation/co4e/skills_list_panel.py); o day chi con giu # sk_manage_btn/skill_list nhu 2 ten thuoc tinh cu va tu noi signal - # dung nguyen tac panel khong tu wire, Co4ETab moi biet _manage_skills. self._skills_panel = SkillsListPanel() self.sk_manage_btn = self._skills_panel.manage_btn self.sk_manage_btn.clicked.connect(self._manage_skills) self.skill_list = self._skills_panel.list_widget col.addWidget(self._section("co4e.tab_skills", self._skills_panel, self.sk_manage_btn), 2) # --- RUNS -------------------------------------------------------- # A short, always-visible view of the same runs the Flow Status page # tables in full. Clicking one opens that page with the run selected. # Icon only: the heading beside it already reads FLOW STATUS, and the # label was long enough to be cut in half in a narrow sidebar. self.runs_more_btn = QPushButton() self.runs_more_btn.setIcon(icon("chevron-right")) self.runs_more_btn.setFixedWidth(30) self.runs_more_btn.setFlat(True) self.runs_more_btn.setToolTip(tr("co4e.tt_runs_tab")) self.runs_more_btn.clicked.connect(lambda: self._show_runs(True)) runs_body = QWidget(); rl = QVBoxLayout(runs_body) rl.setContentsMargins(0, 0, 0, 0); rl.setSpacing(4) self.runs_side_list = QListWidget() self.runs_side_list.setToolTip(tr("co4e.tt_runs_tab")) self.runs_side_list.itemClicked.connect(self._on_side_run_clicked) rl.addWidget(self.runs_side_list, 1) col.addWidget(self._section("co4e.runs_tab", runs_body, self.runs_more_btn), 2) # Small enough that all four still fit on a laptop screen, large enough # that each shows more than a single row. for lst in (self.wf_list, self.agent_list, self.skill_list, self.runs_side_list): lst.setMinimumHeight(56) return self.sidebar def _refresh_side_runs(self) -> None: """Mirror the newest runs into the sidebar's short list.""" lst = getattr(self, "runs_side_list", None) if lst is None: return dots = {"running": "▶", "done": "✓", "error": "✕", "stopped": "■"} lst.clear() for h in list(reversed(self.manager.runs()))[:self._SIDE_RUNS]: it = QListWidgetItem(f"{dots.get(h.status, '•')} {h.name}" f" {h.progress_text()}") it.setData(Qt.UserRole, h.id) it.setToolTip(f"{h.name} · {tr('co4e.status.' + h.status)} · {h.created_at or '-'}") lst.addItem(it) def _on_side_run_clicked(self, item) -> None: """Open the full Flow Status page with this run selected.""" run_id = item.data(Qt.UserRole) self._show_runs(True) for r in range(self.runs_table.rowCount()): cell = self.runs_table.item(r, 0) if cell is not None and cell.data(Qt.UserRole) == run_id: self.runs_table.setCurrentCell(r, 0) break def _section(self, key: str, body: QWidget, action: QPushButton | None = None, stretch: int = 1) -> QWidget: """One named, foldable section of the sidebar column. Replaces the three icon-only tabs: all the lists are visible at once (WORKFLOWS · AGENTS · SKILLS · runs), each under its own heading with the action that belongs to it. Clicking the heading folds the section, so a narrow window can still get to everything. """ box = QWidget() v = QVBoxLayout(box) v.setContentsMargins(0, 0, 0, 0) v.setSpacing(2) row = QHBoxLayout() row.setContentsMargins(0, 0, 0, 0) row.setSpacing(4) head = QPushButton() head.setObjectName("co4eSectionHdr") head.setCheckable(True) head.setChecked(True) head.setCursor(Qt.PointingHandCursor) head.setFlat(True) head.toggled.connect(lambda on, w=body, b=box: self._fold_section(key, w, b, on)) row.addWidget(head, 1) if action is not None: row.addWidget(action, 0) v.addLayout(row) v.addWidget(body, 1) self._sections[key] = (head, body, stretch) self._sync_section_arrow(key) return box def _fold_section(self, key: str, body: QWidget, box: QWidget, on: bool) -> None: """Fold/unfold a section AND give its height back to the others. Inside a splitter, hiding the body is not enough — the pane keeps its share of the height, so folding would free nothing. Clamping the whole section to its header height makes the splitter re-deal the space. """ body.setVisible(on) if on: box.setMaximumHeight(16777215) else: box.setMaximumHeight(box.layout().itemAt(0).sizeHint().height() + 4) self._sync_section_arrow(key) def _sync_section_arrow(self, key: str) -> None: head, _body, _s = self._sections[key] head.setText(("▾ " if head.isChecked() else "▸ ") + tr(key).upper()) def _icon_btn(self, icon_name: str, tip_key: str, slot) -> QPushButton: b = QPushButton(); b.setIcon(icon(icon_name)); b.setToolTip(tr(tip_key)) b.setFixedWidth(34) b.clicked.connect(slot) return b def _reload_sidebar(self) -> None: self.wf_list.clear() for wf in co4e.list_workflows(): tag = tr("co4e.template") if wf.is_template else tr("co4e.saved") it = QListWidgetItem(icon("workspaces"), f"{wf.name} · {tag}") it.setData(Qt.UserRole, ("saved", wf.id)) it.setData(Qt.UserRole + 2, {"kind": "workflow", "workflow": co4e.workflow_to_dict(wf)}) self.wf_list.addItem(it) # Agents: only the Parallel fan-out node + the user's own custom agents # (create your own with "+ New agent"; drag onto the canvas). The blank # "New Step" palette entry was removed — use the toolbar "+ Add" instead. self.agent_list.clear() self.agent_list.addItem(self._palette_item( tr("co4e.parallel_node"), "server", {"variant": "parallel", "label": "Parallel", "role": "PARALLEL", "icon": "server", "sub_agents": []})) for ca in co4e.list_custom_agents(): step = co4e.Step(label=ca.name, agent_slug=co4e.slugify(ca.name), role=ca.role or "AGENT", icon=ca.icon, instructions=ca.instructions, context=getattr(ca, "context", ""), model=ca.model, permission_preset=ca.permission_preset, skills=list(ca.skills), attachments=list(getattr(ca, "attachments", []) or [])) it = self._palette_item(f"{ca.name} · {ca.role} · {tr('co4e.custom')}", ca.icon or "robot", co4e._step_dict(step)) it.setData(Qt.UserRole + 1, ca.id) self.agent_list.addItem(it) # Skills self.skill_list.clear() for name in _skill_names(): content = skills_mod.skill_prefix_for(name) payload = co4e._step_dict(co4e.Step( label=name, agent_slug=co4e.slugify(name), role="SKILL", icon="sparkle", instructions=content, skills=[name])) self.skill_list.addItem(self._palette_item(name, "sparkle", payload)) @staticmethod def _palette_item(text: str, icon_name: str, payload: dict) -> QListWidgetItem: it = QListWidgetItem(icon(icon_name), text) it.setData(Qt.UserRole, payload) return it