# File attachment dialog for AI Interaction Tool
from PyQt5 import QtWidgets, QtCore
import os
from .file_tree import FileTreeView, FileTreeDelegate
from .styles import get_file_dialog_stylesheet, get_context_menu_stylesheet, ModernTheme
from ..utils.translations import get_translation
from ..constants import DEFAULT_PATH
from ..utils.file_utils import (
validate_workspace_path,
validate_file_path_in_workspace,
create_relative_path_with_workspace,
normalize_path_unicode
)
class FileAttachDialog(QtWidgets.QDialog):
"""
Hộp thoại cho phép duyệt và chọn file/folder để đính kèm với workspace support
"""
def __init__(self, parent=None, language="en", translations=None):
super().__init__(parent)
self.language = language
self.translations = translations or {}
self.setWindowTitle(self._get_translation("file_dialog_title"))
self.setMinimumSize(700, 500)
# Workspace root path
self.workspace_path = ""
# Khởi tạo UI
self.init_ui()
# Danh sách file/folder đã chọn (relative paths)
self.selected_items = []
def _get_translation(self, key):
"""Lấy bản dịch cho key dựa trên ngôn ngữ hiện tại"""
if self.translations:
lang_dict = self.translations.get(self.language, {})
return lang_dict.get(key, key)
else:
return get_translation(self.language, key)
def init_ui(self):
"""Khởi tạo giao diện người dùng"""
layout = QtWidgets.QVBoxLayout(self)
# Workspace selection section
workspace_group = QtWidgets.QGroupBox(self._get_translation("workspace_config"))
workspace_layout = QtWidgets.QVBoxLayout()
# Workspace path display and selection
workspace_path_layout = QtWidgets.QHBoxLayout()
self.workspace_label = QtWidgets.QLabel(self._get_translation("no_workspace_selected"))
self.workspace_label.setStyleSheet("QLabel { color: #f38ba8; font-weight: bold; }")
self.select_workspace_btn = QtWidgets.QPushButton(self._get_translation("workspace_browse"), self)
self.select_workspace_btn.clicked.connect(self.select_workspace)
self.select_workspace_btn.setProperty("button-type", "info")
workspace_label_widget = QtWidgets.QLabel(self._get_translation("workspace_label"))
workspace_label_widget.setStyleSheet(f"QLabel {{ color: {ModernTheme.COLORS['text'].name()}; }}")
workspace_path_layout.addWidget(workspace_label_widget)
workspace_path_layout.addWidget(self.workspace_label, 1)
workspace_path_layout.addWidget(self.select_workspace_btn)
# Workspace path input (for pasting/typing path directly)
workspace_input_layout = QtWidgets.QHBoxLayout()
self.workspace_input = QtWidgets.QLineEdit(self)
self.workspace_input.setPlaceholderText(self._get_translation("paste_path_placeholder"))
self.workspace_input.setToolTip(self._get_translation("paste_path_tooltip"))
self.workspace_input.returnPressed.connect(self.set_workspace_from_input)
self.set_workspace_btn = QtWidgets.QPushButton(self._get_translation("set_workspace"), self)
self.set_workspace_btn.clicked.connect(self.set_workspace_from_input)
self.set_workspace_btn.setToolTip(self._get_translation("set_workspace_tooltip"))
self.set_workspace_btn.setProperty("button-type", "info")
paste_label_widget = QtWidgets.QLabel(self._get_translation("or_paste_path"))
paste_label_widget.setStyleSheet(f"QLabel {{ color: {ModernTheme.COLORS['text'].name()}; }}")
workspace_input_layout.addWidget(paste_label_widget)
workspace_input_layout.addWidget(self.workspace_input, 1)
workspace_input_layout.addWidget(self.set_workspace_btn)
workspace_layout.addLayout(workspace_path_layout)
workspace_layout.addLayout(workspace_input_layout)
workspace_group.setLayout(workspace_layout)
layout.addWidget(workspace_group)
# File/Folder selection options
options_layout = QtWidgets.QHBoxLayout()
options_layout.addStretch()
layout.addLayout(options_layout)
# Ô nhập đường dẫn và nút duyệt
path_layout = QtWidgets.QHBoxLayout()
self.path_input = QtWidgets.QLineEdit(self)
self.path_input.setPlaceholderText(self._get_translation("path_placeholder"))
self.browse_btn = QtWidgets.QPushButton(self._get_translation("browse_btn"), self)
self.browse_btn.clicked.connect(self.browse_folder)
self.browse_btn.setProperty("button-type", "secondary")
self.go_btn = QtWidgets.QPushButton(self._get_translation("go_btn"), self)
self.go_btn.clicked.connect(self.navigate_to_path)
self.go_btn.setProperty("button-type", "info")
path_layout.addWidget(self.path_input)
path_layout.addWidget(self.browse_btn)
path_layout.addWidget(self.go_btn)
layout.addLayout(path_layout)
# Cây thư mục
self.file_tree = FileTreeView(self)
self.file_tree.setItemDelegate(FileTreeDelegate(self))
self.file_tree.itemSelected.connect(self.update_selected_items)
# Thiết lập đường dẫn mặc định
default_path = DEFAULT_PATH
self.file_tree.setRootPath(default_path)
self.path_input.setText(default_path)
layout.addWidget(self.file_tree)
# Danh sách items đã chọn
selected_group = QtWidgets.QGroupBox(self._get_translation("selected_items"))
selected_layout = QtWidgets.QVBoxLayout()
# Clear Selected button
clear_selected_layout = QtWidgets.QHBoxLayout()
self.clear_selected_btn = QtWidgets.QPushButton(self._get_translation("clear_selected"), self)
self.clear_selected_btn.clicked.connect(self.clear_selected_items)
self.clear_selected_btn.setEnabled(False) # Always visible, disabled by default # Ẩn ban đầu
self.clear_selected_btn.setProperty("button-type", "warning")
clear_selected_layout.addWidget(self.clear_selected_btn)
clear_selected_layout.addStretch()
selected_layout.addLayout(clear_selected_layout)
# Paste Path to Auto-Select section
paste_select_layout = QtWidgets.QHBoxLayout()
paste_label = QtWidgets.QLabel("📋 " + self._get_translation("paste_to_select") + ":")
paste_label.setStyleSheet(f"QLabel {{ color: {ModernTheme.COLORS['text'].name()}; font-weight: 500; }}")
self.paste_path_input = QtWidgets.QLineEdit(self)
self.paste_path_input.setPlaceholderText(self._get_translation("paste_path_to_select_placeholder"))
self.paste_path_input.setToolTip(self._get_translation("paste_path_to_select_tooltip"))
self.paste_path_input.returnPressed.connect(self.auto_select_from_path)
self.auto_select_btn = QtWidgets.QPushButton("⚡ " + self._get_translation("auto_select"), self)
self.auto_select_btn.clicked.connect(self.auto_select_from_path)
self.auto_select_btn.setProperty("button-type", "special")
self.auto_select_btn.setToolTip(self._get_translation("auto_select_tooltip"))
paste_select_layout.addWidget(paste_label)
paste_select_layout.addWidget(self.paste_path_input, 2) # Give more space to input
paste_select_layout.addWidget(self.auto_select_btn)
selected_layout.addLayout(paste_select_layout)
self.selected_list = QtWidgets.QListWidget(self)
self.selected_list.setAlternatingRowColors(False) # Disable to maintain dark theme consistency
self.selected_list.setSelectionMode(QtWidgets.QAbstractItemView.ExtendedSelection) # Multi-select
self.selected_list.setToolTip(self._get_translation("selected_list_tooltip"))
self.selected_list.setContextMenuPolicy(QtCore.Qt.CustomContextMenu)
self.selected_list.customContextMenuRequested.connect(self.show_selected_context_menu)
self.selected_list.itemSelectionChanged.connect(self.update_selected_button_state)
selected_layout.addWidget(self.selected_list)
selected_group.setLayout(selected_layout)
layout.addWidget(selected_group)
# Nút ở dưới cùng
buttons_layout = QtWidgets.QHBoxLayout()
self.clear_btn = QtWidgets.QPushButton(self._get_translation("clear_all_btn"), self)
self.clear_btn.clicked.connect(self.clear_selection)
self.clear_btn.setProperty("button-type", "danger")
self.attach_btn = QtWidgets.QPushButton(self._get_translation("attach_selected"), self)
self.attach_btn.clicked.connect(self.accept)
self.attach_btn.setEnabled(False) # Disabled until workspace is selected
self.attach_btn.setProperty("button-type", "success")
self.cancel_btn = QtWidgets.QPushButton(self._get_translation("cancel_btn"), self)
self.cancel_btn.clicked.connect(self.reject)
self.cancel_btn.setProperty("button-type", "secondary")
buttons_layout.addWidget(self.clear_btn)
buttons_layout.addStretch()
buttons_layout.addWidget(self.cancel_btn)
buttons_layout.addWidget(self.attach_btn)
layout.addLayout(buttons_layout)
# Áp dụng stylesheet
self.setStyleSheet(get_file_dialog_stylesheet())
# Force refresh styles cho tất cả buttons để apply semantic colors
self._refresh_button_styles()
def select_workspace(self):
"""Chọn workspace root directory"""
folder = QtWidgets.QFileDialog.getExistingDirectory(
self,
self._get_translation("select_workspace_dir"),
self.workspace_path or DEFAULT_PATH
)
if folder:
validation_result = validate_workspace_path(folder)
if not validation_result["valid"]:
QtWidgets.QMessageBox.critical(
self,
self._get_translation("invalid_workspace"),
self._get_translation("workspace_error").format(error=validation_result['error'])
)
return
self.clear_selection()
self.workspace_path = validation_result["normalized_path"]
workspace_name = os.path.basename(self.workspace_path)
self.workspace_label.setText(workspace_name)
self.workspace_label.setStyleSheet("QLabel { color: #a6e3a1; font-weight: bold; }")
self.workspace_label.setToolTip(f"Full path: {self.workspace_path}")
self.attach_btn.setEnabled(True)
self.file_tree.setRootPath(self.workspace_path)
self.path_input.setText(self.workspace_path)
# Auto-expand workspace root để show immediate subdirectories
self._expand_workspace_root()
# Update workspace input field với current workspace
self.workspace_input.setText(self.workspace_path)
def set_workspace_from_input(self):
"""Set workspace từ đường dẫn đã nhập/paste"""
input_path = self.workspace_input.text().strip()
if not input_path:
QtWidgets.QMessageBox.warning(
self,
self._get_translation("empty_path"),
self._get_translation("enter_path_first")
)
return
# Normalize path
try:
normalized_path = normalize_path_unicode(input_path)
except Exception as e:
QtWidgets.QMessageBox.critical(
self,
self._get_translation("invalid_path"),
self._get_translation("invalid_path_format").format(error=str(e))
)
return
# Validate workspace
validation_result = validate_workspace_path(normalized_path)
if not validation_result["valid"]:
QtWidgets.QMessageBox.critical(
self,
self._get_translation("invalid_workspace"),
self._get_translation("workspace_error").format(error=validation_result['error'])
)
return
# Clear existing selections
self.clear_selection()
# Set workspace
self.workspace_path = validation_result["normalized_path"]
workspace_name = os.path.basename(self.workspace_path)
self.workspace_label.setText(workspace_name)
self.workspace_label.setStyleSheet("QLabel { color: #a6e3a1; font-weight: bold; }")
self.workspace_label.setToolTip(f"Full path: {self.workspace_path}")
self.attach_btn.setEnabled(True)
self.file_tree.setRootPath(self.workspace_path)
self.path_input.setText(self.workspace_path)
# Auto-expand workspace root để show immediate subdirectories
self._expand_workspace_root()
# Update workspace input với final normalized path
self.workspace_input.setText(self.workspace_path)
# Success feedback
QtWidgets.QMessageBox.information(
self,
self._get_translation("workspace_set"),
self._get_translation("workspace_success").format(name=workspace_name, path=self.workspace_path)
)
def browse_folder(self):
"""Mở hộp thoại chọn thư mục"""
start_path = self.workspace_path if self.workspace_path else self.path_input.text()
if start_path and not os.path.exists(start_path):
start_path = DEFAULT_PATH
folder = QtWidgets.QFileDialog.getExistingDirectory(
self,
"Select Folder",
start_path
)
if folder:
normalized_folder = normalize_path_unicode(folder)
if os.path.exists(normalized_folder) and os.path.isdir(normalized_folder):
self.path_input.setText(normalized_folder)
self.file_tree.setRootPath(normalized_folder)
else:
QtWidgets.QMessageBox.warning(
self,
self._get_translation("invalid_folder"),
self._get_translation("invalid_folder_msg").format(folder=folder)
)
def navigate_to_path(self):
"""Di chuyển đến đường dẫn đã nhập"""
path = self.path_input.text().strip()
if not path:
QtWidgets.QMessageBox.warning(
self,
self._get_translation("empty_path"),
self._get_translation("enter_path_first")
)
return
normalized_path = normalize_path_unicode(path)
if not os.path.exists(normalized_path):
QtWidgets.QMessageBox.warning(
self,
self._get_translation("invalid_path"),
self._get_translation("path_not_exist")
)
return
if not os.path.isdir(normalized_path):
QtWidgets.QMessageBox.warning(
self,
self._get_translation("not_directory"),
self._get_translation("not_directory_msg").format(path=normalized_path)
)
return
if not os.access(normalized_path, os.R_OK):
QtWidgets.QMessageBox.warning(
self,
self._get_translation("access_denied"),
self._get_translation("access_denied_msg").format(path=normalized_path)
)
return
try:
self.file_tree.setRootPath(normalized_path)
self.path_input.setText(normalized_path)
except Exception as e:
QtWidgets.QMessageBox.critical(
self,
self._get_translation("navigation_error"),
self._get_translation("navigation_error_msg").format(error=str(e))
)
def update_selected_items(self, item_path, selected):
"""Cập nhật danh sách items đã chọn"""
if not self.workspace_path:
QtWidgets.QMessageBox.warning(
self,
self._get_translation("no_workspace"),
self._get_translation("no_workspace_msg")
)
return
try:
validation_result = validate_file_path_in_workspace(item_path, self.workspace_path)
if not validation_result["valid"]:
QtWidgets.QMessageBox.warning(
self,
self._get_translation("invalid_selection"),
self._get_translation("invalid_selection_msg").format(error=validation_result['error'])
)
return
full_relative_path, error = create_relative_path_with_workspace(
item_path, self.workspace_path
)
if error:
QtWidgets.QMessageBox.warning(
self,
self._get_translation("path_error"),
self._get_translation("path_error_msg").format(error=error)
)
return
if selected:
if full_relative_path not in self.selected_items:
self.selected_items.append(full_relative_path)
item_type = "FOLDER" if validation_result["is_dir"] else "FILE"
if validation_result["is_symlink"]:
item_type += " (SYMLINK)"
basename = validation_result["basename"]
display_name = f"[{item_type}] {basename}"
if len(full_relative_path) > 60:
short_path = "..." + full_relative_path[-57:]
display_name += f" ({short_path})"
else:
display_name += f" ({full_relative_path})"
list_item = QtWidgets.QListWidgetItem(display_name)
list_item.setToolTip(self._get_translation("file_item_tooltip").format(path=full_relative_path))
self.selected_list.addItem(list_item)
else:
if full_relative_path in self.selected_items:
index = self.selected_items.index(full_relative_path)
self.selected_items.pop(index)
item = self.selected_list.takeItem(index)
if item:
del item
except Exception as e:
QtWidgets.QMessageBox.critical(
self,
self._get_translation("unexpected_error"),
self._get_translation("unexpected_error_msg").format(error=str(e))
)
def _is_safe_path(self, path):
"""Kiểm tra xem path có an toàn không (deprecated - sử dụng utils functions)"""
# Chuyển sang sử dụng function từ utils
from ..utils.file_utils import contains_dangerous_patterns
return not contains_dangerous_patterns(path)
def show_selected_context_menu(self, position):
"""Hiển thị menu ngữ cảnh cho danh sách items đã chọn"""
if not self.selected_list.count():
return
menu = QtWidgets.QMenu()
# Apply modern context menu styling
menu.setStyleSheet(get_context_menu_stylesheet())
# Get current item để check if valid for scroll action
current_item = self.selected_list.itemAt(position)
current_row = self.selected_list.row(current_item) if current_item else -1
scroll_action = menu.addAction("🔍 " + self._get_translation("scroll_to_location"))
scroll_action.setEnabled(current_row >= 0 and current_row < len(self.selected_items))
scroll_action.setToolTip("Automatically scroll tree view to this item's location")
menu.addSeparator()
remove_action = menu.addAction("🗑️ " + self._get_translation("remove_item"))
remove_action.setToolTip("Remove this item from selected list")
remove_all_action = menu.addAction("🧹 " + self._get_translation("remove_all_items"))
remove_all_action.setToolTip("Remove all items from selected list")
action = menu.exec_(self.selected_list.mapToGlobal(position))
if action == scroll_action:
# Scroll to location in tree view
if current_row >= 0 and current_row < len(self.selected_items):
self._scroll_to_item_location(current_row)
elif action == remove_action:
current_row = self.selected_list.currentRow()
if current_row >= 0 and current_row < len(self.selected_items):
try:
relative_path = self.selected_items[current_row]
self.selected_items.pop(current_row)
list_item = self.selected_list.takeItem(current_row)
if list_item:
del list_item
workspace_name = os.path.basename(self.workspace_path)
if relative_path.startswith(f"{workspace_name}/"):
path_without_workspace = relative_path[len(workspace_name)+1:]
full_path = os.path.join(self.workspace_path, path_without_workspace.replace('/', os.sep))
normalized_full_path = normalize_path_unicode(full_path)
# Use model.setSelected instead of deselectItem để avoid expand side effects
index = self.file_tree.model.index(normalized_full_path)
if index.isValid():
self.file_tree.model.setSelected(index, False)
# Single refresh instead of potential multiple refreshes
self.file_tree.refreshView()
# Update button state after removal
self.update_selected_button_state()
except Exception as e:
QtWidgets.QMessageBox.warning(
self,
self._get_translation("path_error"),
self._get_translation("path_error_msg").format(error=str(e))
)
elif action == remove_all_action:
self.clear_selection()
def update_selected_button_state(self):
"""Cập nhật trạng thái Clear Selected button"""
selected_items = self.selected_list.selectedItems()
has_selection = len(selected_items) > 0
self.clear_selected_btn.setEnabled(has_selection)
if has_selection:
self.clear_selected_btn.setText(f"{self._get_translation('clear_selected')} ({len(selected_items)})")
self.clear_selected_btn.setToolTip(self._get_translation("clear_selected_enabled_tooltip"))
else:
self.clear_selected_btn.setText(self._get_translation("clear_selected"))
self.clear_selected_btn.setToolTip(self._get_translation("clear_selected_disabled_tooltip"))
def clear_selected_items(self):
"""Xóa các items đã chọn trong list"""
selected_items = self.selected_list.selectedItems()
if not selected_items:
QtWidgets.QMessageBox.information(
self,
self._get_translation("no_selection"),
self._get_translation("no_selection_message")
)
return
# Get rows to remove và corresponding relative paths
rows_to_remove = []
relative_paths_to_remove = []
for item in selected_items:
row = self.selected_list.row(item)
if row < len(self.selected_items):
rows_to_remove.append(row)
relative_paths_to_remove.append(self.selected_items[row])
# Sort descending để xóa từ cuối
rows_to_remove.sort(reverse=True)
# Remove from UI list
for row in rows_to_remove:
self.selected_list.takeItem(row)
if row < len(self.selected_items):
self.selected_items.pop(row)
# Deselect in tree view - batch operation để minimize refreshes
if relative_paths_to_remove:
for relative_path in relative_paths_to_remove:
workspace_name = os.path.basename(self.workspace_path)
if relative_path.startswith(f"{workspace_name}/"):
path_without_workspace = relative_path[len(workspace_name)+1:]
full_path = os.path.join(self.workspace_path, path_without_workspace.replace('/', os.sep))
normalized_full_path = normalize_path_unicode(full_path)
# Tìm index và deselect trực tiếp - NO expansion side effects
index = self.file_tree.model.index(normalized_full_path)
if index.isValid():
self.file_tree.model.setSelected(index, False)
# Single refresh sau khi tất cả deselections complete
self.file_tree.refreshView()
# Update button state
self.update_selected_button_state()
def clear_selection(self):
"""Xóa tất cả các lựa chọn"""
try:
self.selected_items.clear()
self.selected_list.clear()
self.file_tree.clearSelection()
self.update_selected_button_state()
except Exception as e:
QtWidgets.QMessageBox.warning(
self,
self._get_translation("clear_error"),
self._get_translation("clear_error_msg").format(error=str(e))
)
def get_selected_files(self):
"""Trả về danh sách các items đã chọn"""
validated_items = []
for relative_path in self.selected_items:
try:
workspace_name = os.path.basename(self.workspace_path)
if relative_path.startswith(f"{workspace_name}/"):
path_without_workspace = relative_path[len(workspace_name)+1:]
full_path = os.path.join(self.workspace_path, path_without_workspace.replace('/', os.sep))
if os.path.exists(full_path) and os.access(full_path, os.R_OK):
validated_items.append(relative_path)
else:
validated_items.append(relative_path)
except Exception:
continue
return validated_items
def get_workspace_path(self):
"""Trả về tên workspace gốc"""
if not self.workspace_path:
return ""
workspace_name = os.path.basename(self.workspace_path)
return workspace_name
def get_full_workspace_path(self):
"""Trả về full path của workspace"""
return self.workspace_path
def restore_workspace_state(self, workspace_path, current_attached_files):
"""Khôi phục workspace state và highlight các items đã select"""
if not workspace_path or not os.path.exists(workspace_path):
return
# Set workspace
validation_result = validate_workspace_path(workspace_path)
if validation_result["valid"]:
self.workspace_path = validation_result["normalized_path"]
workspace_name = os.path.basename(self.workspace_path)
self.workspace_label.setText(workspace_name)
self.workspace_label.setStyleSheet("QLabel { color: #a6e3a1; font-weight: bold; }")
self.workspace_label.setToolTip(f"Full path: {self.workspace_path}")
self.attach_btn.setEnabled(True)
self.file_tree.setRootPath(self.workspace_path)
self.path_input.setText(self.workspace_path)
# Auto-expand workspace root để show immediate subdirectories
self._expand_workspace_root()
# Khôi phục selected items
for item_info in current_attached_files:
try:
relative_path = item_info["relative_path"]
if relative_path not in self.selected_items:
self.selected_items.append(relative_path)
# Thêm vào UI list
item_type = item_info.get("type", "unknown").upper()
basename = item_info.get("name", "unknown")
display_name = f"[{item_type}] {basename}"
if len(relative_path) > 60:
short_path = "..." + relative_path[-57:]
display_name += f" ({short_path})"
else:
display_name += f" ({relative_path})"
list_item = QtWidgets.QListWidgetItem(display_name)
list_item.setToolTip(self._get_translation("file_item_tooltip").format(path=relative_path))
self.selected_list.addItem(list_item)
# Highlight và auto-expand trong tree nếu tìm thấy
workspace_name_prefix = f"{workspace_name}/"
if relative_path.startswith(workspace_name_prefix):
path_without_workspace = relative_path[len(workspace_name_prefix):]
full_path = os.path.join(self.workspace_path, path_without_workspace.replace('/', os.sep))
# Use delayed method để ensure proper expand timing
self._auto_expand_and_highlight_delayed(full_path)
except Exception:
continue
# Update button state after restore
self.update_selected_button_state()
def _highlight_item_in_tree(self, full_path):
"""Highlight một item trong tree view"""
try:
normalized_path = normalize_path_unicode(full_path)
if os.path.exists(normalized_path):
# Tìm index của item trong model
index = self.file_tree.model.index(normalized_path)
if index.isValid():
self.file_tree.model.setSelected(index, True)
self.file_tree.refreshView()
except Exception:
pass
def _auto_expand_and_highlight(self, full_path):
"""Auto-expand parent directories và highlight item trong tree view"""
try:
normalized_path = normalize_path_unicode(full_path)
if not os.path.exists(normalized_path):
return
# Tìm index của target item
target_index = self.file_tree.model.index(normalized_path)
if not target_index.isValid():
return
# Auto-expand tất cả parent directories
current_path = normalized_path
paths_to_expand = []
# Collect tất cả parent paths từ target đến workspace root
while current_path and current_path != self.workspace_path:
parent_path = os.path.dirname(current_path)
if parent_path and parent_path != current_path:
paths_to_expand.append(parent_path)
current_path = parent_path
else:
break
# Reverse để expand từ root down to target
paths_to_expand.reverse()
# Expand each parent directory
for path_to_expand in paths_to_expand:
try:
index = self.file_tree.model.index(path_to_expand)
if index.isValid():
self.file_tree.expand(index)
except Exception:
continue
# Highlight target item
self.file_tree.model.setSelected(target_index, True)
# Scroll to make target visible
self.file_tree.scrollTo(target_index)
# Refresh view
self.file_tree.refreshView()
except Exception as e:
# Fallback to basic highlighting if auto-expand fails
self._highlight_item_in_tree(full_path)
def _refresh_button_styles(self):
"""Force refresh button styles để apply semantic colors"""
buttons_to_refresh = [
self.select_workspace_btn,
self.set_workspace_btn,
self.browse_btn,
self.go_btn,
self.clear_selected_btn,
self.auto_select_btn,
self.clear_btn,
self.attach_btn,
self.cancel_btn
]
for button in buttons_to_refresh:
# Force style refresh by unpolish then polish
button.style().unpolish(button)
button.style().polish(button)
button.update()
def _expand_workspace_root(self):
"""Auto-expand workspace root để show immediate subdirectories"""
try:
if not self.workspace_path:
return
# Expand workspace root directory
root_index = self.file_tree.model.index(self.workspace_path)
if root_index.isValid():
self.file_tree.expand(root_index)
# Optional: Also expand first level subdirectories cho better visibility
# Uncomment nếu muốn expand deeper
# for i in range(self.file_tree.model.rowCount(root_index)):
# child_index = self.file_tree.model.index(i, 0, root_index)
# if child_index.isValid() and self.file_tree.model.isDir(child_index):
# self.file_tree.expand(child_index)
except Exception as e:
pass # Fail silently if expand không thành công
def _scroll_to_item_location(self, item_row):
"""Scroll tree view đến location của item trong selected list"""
try:
if item_row < 0 or item_row >= len(self.selected_items):
return
relative_path = self.selected_items[item_row]
workspace_name = os.path.basename(self.workspace_path)
if not relative_path.startswith(f"{workspace_name}/"):
QtWidgets.QMessageBox.information(
self,
self._get_translation("scroll_error"),
self._get_translation("scroll_invalid_path")
)
return
# Convert relative path to full path
path_without_workspace = relative_path[len(workspace_name)+1:]
full_path = os.path.join(self.workspace_path, path_without_workspace.replace('/', os.sep))
normalized_full_path = normalize_path_unicode(full_path)
if not os.path.exists(normalized_full_path):
QtWidgets.QMessageBox.warning(
self,
self._get_translation("file_not_found"),
self._get_translation("file_not_found_msg").format(path=relative_path)
)
return
# Use delayed auto-expand method để ensure proper timing
self._auto_expand_and_highlight_delayed(normalized_full_path)
except Exception as e:
QtWidgets.QMessageBox.warning(
self,
self._get_translation("scroll_error"),
self._get_translation("scroll_error_msg").format(error=str(e))
)
def auto_select_from_path(self):
"""Auto-select item từ pasted path"""
if not self.workspace_path:
QtWidgets.QMessageBox.warning(
self,
self._get_translation("no_workspace"),
self._get_translation("no_workspace_msg")
)
return
input_path = self.paste_path_input.text().strip()
if not input_path:
QtWidgets.QMessageBox.information(
self,
self._get_translation("empty_path"),
self._get_translation("paste_path_empty_msg")
)
return
try:
# Normalize input path
normalized_input = normalize_path_unicode(input_path)
# Check if it's an absolute path within workspace
if os.path.isabs(normalized_input):
# Check if absolute path is within workspace
if not normalized_input.startswith(self.workspace_path):
QtWidgets.QMessageBox.warning(
self,
self._get_translation("path_outside_workspace"),
self._get_translation("path_outside_workspace_msg")
)
return
target_path = normalized_input
else:
# Treat as relative path within workspace
target_path = os.path.join(self.workspace_path, normalized_input.replace('/', os.sep))
target_path = normalize_path_unicode(target_path)
# Validate target path exists
if not os.path.exists(target_path):
QtWidgets.QMessageBox.warning(
self,
self._get_translation("file_not_found"),
self._get_translation("paste_path_not_found_msg").format(path=input_path)
)
return
# Validate path is within workspace
validation_result = validate_file_path_in_workspace(target_path, self.workspace_path)
if not validation_result["valid"]:
QtWidgets.QMessageBox.warning(
self,
self._get_translation("invalid_selection"),
self._get_translation("invalid_selection_msg").format(error=validation_result['error'])
)
return
# Create relative path với workspace
full_relative_path, error = create_relative_path_with_workspace(
target_path, self.workspace_path
)
if error:
QtWidgets.QMessageBox.warning(
self,
self._get_translation("path_error"),
self._get_translation("path_error_msg").format(error=error)
)
return
# Check if already selected
if full_relative_path in self.selected_items:
QtWidgets.QMessageBox.information(
self,
self._get_translation("already_selected"),
self._get_translation("already_selected_msg").format(path=full_relative_path)
)
# Still scroll to it for user convenience với delayed method
self._auto_expand_and_highlight_delayed(target_path)
return
# Add to selected items
self.selected_items.append(full_relative_path)
# Determine item type và name
item_type = "folder" if os.path.isdir(target_path) else "file"
basename = os.path.basename(target_path)
# Create display name
display_name = f"[{item_type.upper()}] {basename}"
if len(full_relative_path) > 60:
short_path = "..." + full_relative_path[-57:]
display_name += f" ({short_path})"
else:
display_name += f" ({full_relative_path})"
# Add to UI list
list_item = QtWidgets.QListWidgetItem(display_name)
list_item.setToolTip(self._get_translation("file_item_tooltip").format(path=full_relative_path))
self.selected_list.addItem(list_item)
# Auto-expand và highlight trong tree với delayed scroll
self._auto_expand_and_highlight_delayed(target_path)
# Update button states
self.update_selected_button_state()
# Clear input để prepare for next paste
self.paste_path_input.clear()
# Show success feedback
self.paste_path_input.setPlaceholderText(
f"✅ Added: {basename} - " + self._get_translation("paste_path_to_select_placeholder")
)
# Reset placeholder after 3 seconds
QtCore.QTimer.singleShot(3000, lambda: self.paste_path_input.setPlaceholderText(
self._get_translation("paste_path_to_select_placeholder")
))
except Exception as e:
QtWidgets.QMessageBox.critical(
self,
self._get_translation("auto_select_error"),
self._get_translation("auto_select_error_msg").format(error=str(e))
)
def _auto_expand_and_highlight_delayed(self, full_path):
"""Auto-expand và highlight với delayed scroll để ensure expand hoàn thành"""
try:
normalized_path = normalize_path_unicode(full_path)
if not os.path.exists(normalized_path):
return
# Phase 1: Collect tất cả parent paths cần expand
current_path = normalized_path
paths_to_expand = []
while current_path and current_path != self.workspace_path:
parent_path = os.path.dirname(current_path)
if parent_path and parent_path != current_path:
paths_to_expand.append(parent_path)
current_path = parent_path
else:
break
# Reverse để expand từ root down to target
paths_to_expand.reverse()
# Phase 2: Expand each parent directory với progressive delays
self._expand_paths_progressively(paths_to_expand, normalized_path, 0)
except Exception as e:
# Fallback to basic highlighting if auto-expand fails
self._auto_expand_and_highlight(full_path)
def _expand_paths_progressively(self, paths_to_expand, target_path, current_index):
"""Expand paths progressively với delays để ensure completion"""
if current_index >= len(paths_to_expand):
# All parent paths expanded, now scroll to target với delay
QtCore.QTimer.singleShot(200, lambda: self._final_highlight_and_scroll(target_path))
return
# Expand current path
path_to_expand = paths_to_expand[current_index]
try:
index = self.file_tree.model.index(path_to_expand)
if index.isValid():
self.file_tree.expand(index)
except Exception:
pass
# Schedule next expansion với delay
QtCore.QTimer.singleShot(100, lambda: self._expand_paths_progressively(
paths_to_expand, target_path, current_index + 1
))
def _final_highlight_and_scroll(self, target_path):
"""Final step: highlight target và scroll với focus"""
try:
# Find target index
target_index = self.file_tree.model.index(target_path)
if not target_index.isValid():
return
# Highlight target item
self.file_tree.model.setSelected(target_index, True)
# Scroll to make target visible
self.file_tree.scrollTo(target_index)
# Focus tree view để user thấy highlight
self.file_tree.setFocus()
# Refresh view để ensure visual update
self.file_tree.refreshView()
except Exception:
pass # Fail silently for final step
