cowork-local/presentation/folder/folder_tab.py

"""FolderTab shell (R08-T12) — assembles
``workspace_file_tree.py::WorkspaceFileTree``,
``document_preview_manager.py::DocumentPreviewManager`` and
``ai_file_editor_dialog.py::AiFileEditorDialog`` behind the splitter/terminal
layout that used to be inline in ``ui/folder_tab.py::FolderTab.__init__``
(lines 238-384 of the original 1587-line file).

The AI-panel toggle button (``ai_btn``) lives here because it controls
things two different children own: the panel's own visibility AND the
content splitter's sizing — a genuine shell-level concern, not either
child's.
"""
from __future__ import annotations

from PySide6.QtWidgets import QHBoxLayout, QPushButton, QSplitter, QVBoxLayout, QWidget
from PySide6.QtCore import Qt, Signal

from cowork_local.i18n import on_language_changed, tr
from cowork_local.presentation.folder.ai_file_editor_dialog import AiFileEditorDialog
from cowork_local.presentation.folder.document_preview_manager import DocumentPreviewManager
from cowork_local.presentation.folder.workspace_file_tree import WorkspaceFileTree
from cowork_local.state import AppContext
from cowork_local.ui.icons import icon


class FolderTab(QWidget):
    """Two-pane file explorer: directory tree + view/edit pane (+ collapsible
    AI-edit panel, + collapsible terminal)."""

    status_message = Signal(str)

    def __init__(self, ctx: AppContext, cowork=None):
        """Ghép ba phần của tab Thư mục: cây tệp, khung xem và terminal."""
        super().__init__()
        self.ctx = ctx
        self._root = str(ctx.config.cowork_output_dir())
        # Goc ma project dang chon ap xuong, de phan biet "doi project" voi
        # "chi la refresh" — xem set_project_root().
        self._project_root = ""

        root_layout = QVBoxLayout(self)
        split = QSplitter(Qt.Horizontal)

        self.tree = WorkspaceFileTree(self._root)
        split.addWidget(self.tree)

        right = QWidget()
        rl = QVBoxLayout(right)
        rl.setContentsMargins(0, 0, 0, 0)

        self.preview = DocumentPreviewManager(self._root)
        self.ai_panel = AiFileEditorDialog(ctx, self.preview, cowork=cowork)

        self.ai_btn = QPushButton()        # expand/collapse the AI-edit panel
        self.ai_btn.setIcon(icon("sparkle"))
        self.ai_btn.setCheckable(True)
        self.ai_btn.clicked.connect(self._toggle_ai_panel)
        # Same visual position as the original single-class header: between
        # the Preview⇄Edit toggle and Save (file_label=0, mode_btn=1).
        self.preview.header_layout.insertWidget(2, self.ai_btn)
        self.ai_panel.badge_changed.connect(self._on_ai_badge_changed)

        content_split = QSplitter(Qt.Horizontal)
        content_split.addWidget(self.preview)
        content_split.addWidget(self.ai_panel)
        content_split.setStretchFactor(0, 1)
        content_split.setStretchFactor(1, 0)
        content_split.setSizes([700, 320])
        self._content_split = content_split
        self.ai_panel.setVisible(False)   # default collapsed
        rl.addWidget(content_split, 1)

        split.addWidget(right)
        split.setStretchFactor(0, 0)
        split.setStretchFactor(1, 1)
        split.setSizes([300, 800])
        root_layout.addWidget(split, 1)

        # Terminal CLI below the file view — collapsible, default collapsed;
        # opening it points the shell at the current workspace folder.
        from cowork_local.ui.terminal_panel import TerminalPanel

        self.terminal = TerminalPanel()
        self.terminal.set_cwd(self._root)
        self.terminal.expanded.connect(lambda: self.terminal.set_cwd(self._root))
        root_layout.addWidget(self.terminal)

        # ``root_changed`` truoc day KHONG co ai lang nghe: nguoi dung tu chon
        # thu muc trong cay thi chi cai cay doi goc, con khung xem va terminal o
        # lai thu muc cu.
        self.tree.root_changed.connect(self._on_user_picked_root)
        self.tree.file_selected.connect(self.preview.open_file)
        self.preview.status_message.connect(self.status_message.emit)
        self.ai_panel.status_message.connect(self.status_message.emit)

        on_language_changed(self._retranslate)
        self._retranslate()

    # ---- public API ---------------------------------------------------------
    def set_root(self, path: str) -> None:
        """Đổi thư mục gốc của cả ba widget con.

        ``WorkspaceFileTree`` lặng lẽ từ chối đường dẫn không hợp lệ, nên chỉ lan
        sang khung xem và terminal khi cây thật sự đã nhận.
        """
        self.tree.set_root(path)
        # WorkspaceFileTree silently no-ops on an invalid path (same guard
        # the original single-class _root setter had) — mirror that here by
        # only propagating when the tree actually accepted it.
        if self.tree.root == path:
            self._root = path
            self.preview.set_root(path)
            self.terminal.set_cwd(path)

    def _on_user_picked_root(self, path: str) -> None:
        """Người dùng tự chọn thư mục trong cây: lan sang khung xem và terminal."""
        self._root = path
        self.preview.set_root(path)
        self.terminal.set_cwd(path)

    def set_project_root(self, path: str) -> None:
        """Áp thư mục gốc theo project đang chọn.

        Bỏ qua nếu project KHÔNG đổi. ``WorkspaceTab.refresh()`` — và qua đó
        ``_load_current`` — chạy lại mỗi lần người dùng vào lại màn Workspace
        (``_goto`` gọi nó), nên gọi ``set_root`` vô điều kiện sẽ kéo thư mục về
        workspace của project và xoá mất lựa chọn tay: chọn folder khác, chuyển
        tab rồi quay lại là mất.

        Đổi sang project khác thì vẫn re-root — thư mục của màn này thuộc về
        project, chỉ có lần refresh trong CÙNG một project là không được đụng.
        """
        if path == self._project_root:
            return
        self._project_root = path
        self.set_root(path)

    def _toggle_ai_panel(self) -> None:
        """Gập/mở panel AI-Edit; mở ra thì báo cho panel biết để nó nạp model lần đầu."""
        show = self.ai_btn.isChecked()
        self.ai_panel.setVisible(show)
        if show:
            self._content_split.setSizes([700, 320])
            self.ai_panel.on_opened()

    def _on_ai_badge_changed(self, suffix: str) -> None:
        """Cập nhật huy hiệu trên nút AI-Edit (số việc đang chờ)."""
        self.ai_btn.setText(tr("folder.ai_edit") + suffix)

    def _retranslate(self) -> None:
        """Chuyển lệnh dịch lại xuống cho cả ba widget con."""
        self.tree.retranslate()
        self.preview.retranslate()
        self.ai_panel.retranslate()
        self.ai_btn.setText(tr("folder.ai_edit"))
        self.ai_btn.setToolTip(tr("folder.ai_edit_tooltip"))


__all__ = ["FolderTab"]