## Summary What changed and why? ## Change Type - [ ] Cowork feature - [ ] Bug fix - [ ] Core AI contribution - [ ] Test / hardening - [ ] Performance - [ ] Documentation ## Related Work Cowork Task: Core Repo: http://34.143.229.138/gitea-admin/fsg-ai-core-assets Core AI Issue: Core Task: Related PR: ## Scope What is intentionally included? What is intentionally NOT included? ## Validation - [ ] Unit tests - [ ] Integration tests - [ ] Manual verification - [ ] Regression check Commands / evidence: ## Security Impact Permission / credential / network / customer data impact: ## Compatibility - [ ] No breaking change - [ ] Breaking change documented ## Reviewer Notes Anything Cowork reviewers should pay attention to. --------- Co-authored-by: thanhnv <thanhnv.ip@gmail.com> Co-authored-by: Vu Dam Tuan <vudt15@fpt.com> Reviewed-on: #9
This commit was merged in pull request #9.
This commit is contained in:
thanhnv
@@ -0,0 +1,212 @@
|
||||
"""DF-007 — browse OneDrive/SharePoint via Microsoft Graph and pick a folder
|
||||
to use as (a local mirror of) a project's working directory. See
|
||||
``core/cloud_workspace_sync.py`` for the mirror/sync side and
|
||||
``ui/ms365_signin_dialog.py`` for the sign-in gate this calls first.
|
||||
"""
|
||||
from __future__ import annotations
|
||||
|
||||
from typing import Optional
|
||||
|
||||
from PySide6.QtCore import Qt
|
||||
from PySide6.QtWidgets import (
|
||||
QDialog, QHBoxLayout, QLabel, QLineEdit, QMessageBox, QPushButton,
|
||||
QRadioButton, QTreeWidget, QTreeWidgetItem, QVBoxLayout,
|
||||
)
|
||||
|
||||
from ..core import ms365_auth
|
||||
from ..core import ms365_graph as graph
|
||||
from ..i18n import tr
|
||||
from .icons import icon
|
||||
from .ms365_signin_dialog import ensure_signed_in
|
||||
|
||||
_ITEM_KIND = Qt.UserRole
|
||||
_ITEM_NAME = Qt.UserRole + 1
|
||||
|
||||
|
||||
class CloudFolderPickerDialog(QDialog):
|
||||
"""Chọn "OneDrive của tôi" hoặc tìm 1 site SharePoint, rồi duyệt thư mục
|
||||
con của nó; "Chọn thư mục này" trả về thư mục ĐANG HIỂN THỊ (không phải
|
||||
dòng đang bôi đen — giống hành vi ``QFileDialog`` khi đang ở trong 1
|
||||
thư mục)."""
|
||||
|
||||
def __init__(self, config, parent=None):
|
||||
super().__init__(parent)
|
||||
self._config = config
|
||||
ms365 = (config.ms365 if config is not None else {}) or {}
|
||||
self._token = ms365_auth.get_access_token(
|
||||
ms365.get("tenant_id", ""), ms365.get("client_id", ""))
|
||||
self._site_id = ""
|
||||
self._site_name = ""
|
||||
self._path_parts: list[str] = [] # relative path segments from the drive root
|
||||
|
||||
self.setWindowTitle(tr("cloud_picker.title"))
|
||||
self.setModal(True)
|
||||
self.resize(520, 480)
|
||||
layout = QVBoxLayout(self)
|
||||
|
||||
source_row = QHBoxLayout()
|
||||
self._onedrive_radio = QRadioButton(tr("cloud_picker.source_onedrive"))
|
||||
self._onedrive_radio.setChecked(True)
|
||||
self._onedrive_radio.toggled.connect(self._on_source_toggled)
|
||||
self._sharepoint_radio = QRadioButton(tr("cloud_picker.source_sharepoint"))
|
||||
source_row.addWidget(self._onedrive_radio)
|
||||
source_row.addWidget(self._sharepoint_radio)
|
||||
source_row.addStretch(1)
|
||||
layout.addLayout(source_row)
|
||||
|
||||
search_row = QHBoxLayout()
|
||||
self._site_search = QLineEdit()
|
||||
self._site_search.setPlaceholderText(tr("cloud_picker.search_sites_placeholder"))
|
||||
self._site_search.setEnabled(False)
|
||||
self._site_search.returnPressed.connect(self._search_sites)
|
||||
self._search_btn = QPushButton(tr("cloud_picker.search_btn"))
|
||||
self._search_btn.setEnabled(False)
|
||||
self._search_btn.clicked.connect(self._search_sites)
|
||||
search_row.addWidget(self._site_search, 1)
|
||||
search_row.addWidget(self._search_btn)
|
||||
layout.addLayout(search_row)
|
||||
|
||||
self._path_lbl = QLabel()
|
||||
self._path_lbl.setWordWrap(True)
|
||||
layout.addWidget(self._path_lbl)
|
||||
|
||||
self._tree = QTreeWidget()
|
||||
self._tree.setHeaderHidden(True)
|
||||
self._tree.itemDoubleClicked.connect(self._on_item_activated)
|
||||
layout.addWidget(self._tree, 1)
|
||||
|
||||
btn_row = QHBoxLayout()
|
||||
self._choose_btn = QPushButton(tr("cloud_picker.choose_here"))
|
||||
self._choose_btn.setObjectName("primary")
|
||||
self._choose_btn.setIcon(icon("cloud"))
|
||||
self._choose_btn.clicked.connect(self.accept)
|
||||
self._cancel_btn = QPushButton(tr("cloud_picker.cancel"))
|
||||
self._cancel_btn.clicked.connect(self.reject)
|
||||
btn_row.addStretch(1)
|
||||
btn_row.addWidget(self._cancel_btn)
|
||||
btn_row.addWidget(self._choose_btn)
|
||||
layout.addLayout(btn_row)
|
||||
|
||||
self._refresh_path_label()
|
||||
self._reload()
|
||||
|
||||
# ---- source switching -------------------------------------------------
|
||||
def _on_source_toggled(self, _checked: bool) -> None:
|
||||
is_sharepoint = self._sharepoint_radio.isChecked()
|
||||
self._site_search.setEnabled(is_sharepoint)
|
||||
self._search_btn.setEnabled(is_sharepoint)
|
||||
self._choose_btn.setEnabled(not is_sharepoint or bool(self._site_id))
|
||||
if not is_sharepoint:
|
||||
self._site_id = ""
|
||||
self._site_name = ""
|
||||
self._path_parts = []
|
||||
self._refresh_path_label()
|
||||
self._reload()
|
||||
|
||||
def _search_sites(self) -> None:
|
||||
query = self._site_search.text().strip()
|
||||
if not query:
|
||||
return
|
||||
self._tree.clear()
|
||||
try:
|
||||
sites = graph.list_sharepoint_sites(self._token, query)
|
||||
except graph.Ms365GraphError as exc:
|
||||
QMessageBox.warning(self, tr("cloud_picker.title"),
|
||||
tr("cloud_picker.load_failed", err=str(exc)))
|
||||
return
|
||||
if not sites:
|
||||
item = QTreeWidgetItem([tr("cloud_picker.no_sites")])
|
||||
item.setData(0, _ITEM_KIND, "empty")
|
||||
self._tree.addTopLevelItem(item)
|
||||
return
|
||||
for site in sites:
|
||||
item = QTreeWidgetItem([site.get("displayName") or site.get("name") or site.get("id", "")])
|
||||
item.setIcon(0, icon("globe"))
|
||||
item.setData(0, _ITEM_KIND, "site")
|
||||
item.setData(0, _ITEM_NAME, site.get("id", ""))
|
||||
item.setData(0, Qt.UserRole + 2, site.get("displayName") or site.get("name") or "")
|
||||
self._tree.addTopLevelItem(item)
|
||||
self._choose_btn.setEnabled(False) # must pick a site before choosing a folder
|
||||
|
||||
def _on_item_activated(self, item: QTreeWidgetItem, _col: int) -> None:
|
||||
kind = item.data(0, _ITEM_KIND)
|
||||
if kind == "site":
|
||||
self._site_id = item.data(0, _ITEM_NAME)
|
||||
self._site_name = item.data(0, Qt.UserRole + 2)
|
||||
self._path_parts = []
|
||||
self._choose_btn.setEnabled(True)
|
||||
self._refresh_path_label()
|
||||
self._reload()
|
||||
elif kind == "up":
|
||||
self._path_parts.pop()
|
||||
self._refresh_path_label()
|
||||
self._reload()
|
||||
elif kind == "folder":
|
||||
self._path_parts.append(item.data(0, _ITEM_NAME))
|
||||
self._refresh_path_label()
|
||||
self._reload()
|
||||
# kind == "file": not navigable, double-click does nothing
|
||||
|
||||
# ---- listing ------------------------------------------------------------
|
||||
def _current_remote_path(self) -> str:
|
||||
return "/".join(self._path_parts)
|
||||
|
||||
def _refresh_path_label(self) -> None:
|
||||
if self._sharepoint_radio.isChecked():
|
||||
root = self._site_name or tr("cloud_picker.source_sharepoint")
|
||||
else:
|
||||
root = tr("cloud_picker.source_onedrive")
|
||||
parts = "/".join([root] + self._path_parts)
|
||||
self._path_lbl.setText(parts)
|
||||
|
||||
def _reload(self) -> None:
|
||||
self._tree.clear()
|
||||
if self._sharepoint_radio.isChecked() and not self._site_id:
|
||||
return # waiting on a site search/selection
|
||||
if self._path_parts:
|
||||
up = QTreeWidgetItem([tr("cloud_picker.up")])
|
||||
up.setData(0, _ITEM_KIND, "up")
|
||||
self._tree.addTopLevelItem(up)
|
||||
try:
|
||||
if self._sharepoint_radio.isChecked():
|
||||
children = graph.list_sharepoint_files(
|
||||
self._token, self._site_id, self._current_remote_path())
|
||||
else:
|
||||
children = graph.list_onedrive_files(self._token, self._current_remote_path())
|
||||
except graph.Ms365GraphError as exc:
|
||||
QMessageBox.warning(self, tr("cloud_picker.title"),
|
||||
tr("cloud_picker.load_failed", err=str(exc)))
|
||||
return
|
||||
for entry in children:
|
||||
name = entry.get("name", "")
|
||||
if not name:
|
||||
continue
|
||||
is_folder = "folder" in entry
|
||||
item = QTreeWidgetItem([name])
|
||||
item.setIcon(0, icon("folder" if is_folder else "file"))
|
||||
item.setData(0, _ITEM_KIND, "folder" if is_folder else "file")
|
||||
item.setData(0, _ITEM_NAME, name)
|
||||
self._tree.addTopLevelItem(item)
|
||||
|
||||
# ---- result ---------------------------------------------------------
|
||||
def cloud_source(self) -> dict:
|
||||
"""Chỉ gọi sau khi ``exec()`` trả về ``QDialog.Accepted``."""
|
||||
if self._sharepoint_radio.isChecked():
|
||||
return {
|
||||
"provider": "sharepoint", "site_id": self._site_id,
|
||||
"site_name": self._site_name, "remote_path": self._current_remote_path(),
|
||||
}
|
||||
return {"provider": "onedrive", "site_id": "", "site_name": "",
|
||||
"remote_path": self._current_remote_path()}
|
||||
|
||||
|
||||
def pick_cloud_folder(parent, config) -> Optional[dict]:
|
||||
"""Đảm bảo đã đăng nhập MS365 rồi mở dialog duyệt; trả về ``cloud_source``
|
||||
dict nếu người dùng xác nhận một thư mục, ``None`` nếu hủy hoặc chưa đăng
|
||||
nhập."""
|
||||
if not ensure_signed_in(parent, config):
|
||||
return None
|
||||
dialog = CloudFolderPickerDialog(config, parent)
|
||||
if dialog.exec() == QDialog.Accepted:
|
||||
return dialog.cloud_source()
|
||||
return None
|
||||
Reference in New Issue
Block a user