feat: add asset exporter, reference updater, and smartprop comment property editor

Author: dertwist

src/editors/asset_exporter/dependency_resolver.py

@@ -19,7 +19,7 @@ def _walk(self, path: str):
             return
         self._visited.add(path)
         ext = os.path.splitext(path)[1].lower()
-        if ext in ('.vmdl', '.vsmart', '.vmat', '.vpcf', '.vsndevts', '.vtex', '.vsnd'):
+        if ext in ('.vmdl', '.vsmart', '.vmat', '.vpcf', '.vsndevts', '.vtex', '.vsnd', '.vmap', '.vpost', '.vanim', '.vseq', '.vphys'):
             self._parse_kv3_deps(path)
 
     def _parse_kv3_deps(self, path: str):
@@ -28,7 +28,8 @@ def _parse_kv3_deps(self, path: str):
             with open(path, 'r', encoding='utf-8', errors='ignore') as f:
                 content = f.read()
 
-            pattern = r'"([^"]+\.(?:vmdl|vsmart|vmat|vpcf|vsndevts|vtex|vsnd|txt|kv3|vmap|vpost)(?:_c)?)"'
+            # Broaden the regex to capture more file extensions used in Source 2, including raw source files like .tga
+            pattern = r'"(?:resource:)?([^"]+\.(?:vmdl|vsmart|vmat|vpcf|vsndevts|vtex|vsnd|txt|kv3|vmap|vpost|tga|png|jpg|jpeg|psd|wav|mp3|fbx|obj|vfx|vcs|vjs|vcss|vanim|vseq|vphys)(?:_c)?)"'
             matches = re.findall(pattern, content, re.IGNORECASE)
             for m in matches:
                 self._add_dep(m)
@@ -39,6 +40,10 @@ def _add_dep(self, rel_path: str):
         if not rel_path:
             return
 
+        # Strip resource prefixes common in KV3 files
+        for prefix in ["resource:", "panorama:", "file:"]:
+            if rel_path.lower().startswith(prefix):
+                rel_path = rel_path[len(prefix):]
         # Handle compiled asset path mapping
         if rel_path.endswith('_c'):
             rel_path = rel_path[:-2]

src/editors/asset_exporter/exporter.py

@@ -21,7 +21,7 @@ def __init__(self, files, addon_content_path, dest_root, layout, addon_name, ass
     def run(self):
         import tempfile
         import zipfile
-        SCANNABLE_EXTS = {'.vmdl', '.vsmart', '.vmat', '.vpcf', '.vsndevts', '.vsnd'}
+        SCANNABLE_EXTS = {'.vmdl', '.vsmart', '.vmat', '.vpcf', '.vsndevts', '.vsnd', '.vtex', '.vmap', '.vpost', '.vanim', '.vseq', '.vphys'}
 
         path_mapping = {}
         for src in self.files:

src/editors/asset_exporter/main.py

@@ -52,6 +52,9 @@ def __init__(self, parent=None):
         self.ui.radio_preserve.toggled.connect(self.toggle_thirdparty_fields)
 
         self.toggle_thirdparty_fields()
+
+        desktop_dir = os.path.join(os.path.expanduser("~"), "Desktop")
+        self.ui.edit_output_dir.setText(os.path.normpath(desktop_dir))
 
         from PySide6.QtWidgets import QCheckBox
         self.ui.checkbox_zip = QCheckBox("Export to ZIP Archive (name based on selected asset)")
@@ -107,7 +110,8 @@ def resolve_dependencies(self):
             )
 
     def browse_output_dir(self):
-        d = QFileDialog.getExistingDirectory(self, "Select Output Directory")
+        start_dir = self.ui.edit_output_dir.text() or os.path.join(os.path.expanduser("~"), "Desktop")
+        d = QFileDialog.getExistingDirectory(self, "Select Output Directory", start_dir)
         if d:
             self.ui.edit_output_dir.setText(os.path.normpath(d))
 

src/editors/asset_manager/reference_updater.py

@@ -1,7 +1,7 @@
 import os
 
 class ReferenceUpdater:
-    SCANNABLE_EXTS = {'.vmdl', '.vsmart', '.vmat', '.vpcf', '.vsndevts', '.vsnd', '.vmap'}
+    SCANNABLE_EXTS = {'.vmdl', '.vsmart', '.vmat', '.vpcf', '.vsndevts', '.vsnd', '.vmap', '.vpost', '.vanim', '.vseq', '.vphys'}
 
     def __init__(self, addon_content_path: str):
         self.addon_content_path = addon_content_path

src/editors/smartprop_editor/commands.py

@@ -521,3 +521,26 @@ def undo(self):
             self.document._restore_choices(self.old_state)
         except Exception as e:
             print(f"[SPE][Choices] undo: ERROR — {e}")
+
+
+class HierarchyItemRenameCommand(QUndoCommand):
+    """Undo/redo support for renaming hierarchy tree items (m_sLabel)."""
+    def __init__(self, item, old_label, new_label):
+        super().__init__("Rename")
+        self.item = item
+        self.old_label = old_label
+        self.new_label = new_label
+
+    def redo(self):
+        if self.item is not None:
+            self.item.setText(0, self.new_label)
+            # Update the underlying data if it exists
+            if hasattr(self.item, 'data') and isinstance(self.item.data, dict):
+                self.item.data['m_sLabel'] = self.new_label
+
+    def undo(self):
+        if self.item is not None:
+            self.item.setText(0, self.old_label)
+            # Update the underlying data if it exists
+            if hasattr(self.item, 'data') and isinstance(self.item.data, dict):
+                self.item.data['m_sLabel'] = self.old_label

src/editors/smartprop_editor/document.py

@@ -148,6 +148,10 @@ def __init__(self, parent=None, update_title=None):
         # Choices rename undo state: captured on itemDoubleClicked, consumed by itemChanged.
         self._choices_rename_old_state = None
 
+        # Hierarchy item rename undo state: captured on itemDoubleClicked, consumed by itemChanged.
+        self._hierarchy_rename_old_label = None
+        self._hierarchy_rename_item = None
+
         # Choices widget-edit debounce (ComboboxTreeChild, VariableWidget, etc.)
         self._choices_widget_old_state = None
         self._choices_widget_debounce_desc = "Edit Choices"
@@ -187,6 +191,8 @@ def __init__(self, parent=None, update_title=None):
         self.ui.tree_hierarchy_widget.setAcceptDrops(True)
         self.ui.tree_hierarchy_widget.setDropIndicatorShown(True)
         self.ui.tree_hierarchy_widget.setDragDropMode(QTreeWidget.InternalMove)
+        self.ui.tree_hierarchy_widget.itemDoubleClicked.connect(self._on_hierarchy_item_about_to_edit)
+        self.ui.tree_hierarchy_widget.itemChanged.connect(self._on_hierarchy_item_changed)
 
         # Content version
         self.content_version_spinbox = QSpinBox()
@@ -2321,6 +2327,28 @@ def _flush_choices_widget_if_pending(self):
             self._choices_widget_debounce.stop()
             self._push_choices_widget_edit()
 
+    def _on_hierarchy_item_about_to_edit(self, item, column):
+        """Capture the 'before' label when the user starts an inline rename in the hierarchy tree."""
+        if column == 0:
+            self._hierarchy_rename_old_label = item.text(0)
+            self._hierarchy_rename_item = item
+
+    def _on_hierarchy_item_changed(self, item, column):
+        """Push rename undo command once the inline edit is committed in the hierarchy tree."""
+        if (
+            column == 0
+            and self._hierarchy_rename_item is item
+            and self._hierarchy_rename_old_label is not None
+        ):
+            new_label = item.text(0)
+            if new_label != self._hierarchy_rename_old_label:
+                from src.editors.smartprop_editor.commands import HierarchyItemRenameCommand
+                self.undo_stack.push(
+                    HierarchyItemRenameCommand(item, self._hierarchy_rename_old_label, new_label)
+                )
+            self._hierarchy_rename_old_label = None
+            self._hierarchy_rename_item = None
+
     def _setup_history_dock(self):
         self._history_dock = QDockWidget("History", self)
         self._history_dock.setObjectName("SPE_history_dock")

src/editors/smartprop_editor/property/comment.py

@@ -1,6 +1,6 @@
 from src.editors.smartprop_editor.property.ui_comment import Ui_Widget
 from PySide6.QtWidgets import QWidget
-from PySide6.QtCore import Signal
+from PySide6.QtCore import Signal, QTimer, Qt
 
 
 class PropertyComment(QWidget):
@@ -17,12 +17,43 @@ def __init__(self, value_class, value):
         self.ui.text_field.setPlainText(value)
         self.ui.text_field.textChanged.connect(self.on_changed)
 
+        # Timer for debouncing resize operations
+        self._resize_timer = QTimer()
+        self._resize_timer.setSingleShot(True)
+        self._resize_timer.timeout.connect(self._do_auto_resize)
+        self._resize_timer.setInterval(100)
 
         self.change_value()
+        # Initial resize after content is set
+        self._do_auto_resize()
 
     def on_changed(self):
         self.change_value()
         self.edited.emit()
+        # Debounce the resize to avoid excessive updates during rapid typing
+        self._resize_timer.start()
+
+    def _do_auto_resize(self):
+        doc = self.ui.text_field.document()
+        doc_height = doc.size().height()
+
+        padding = 16
+        new_height = max(128, int(doc_height) + padding)
+
+        self.setFixedHeight(new_height)
+        self.updateGeometry()
+
+        from src.editors.smartprop_editor.property_frame import PropertyFrame
+        parent = self.parentWidget()
+        while parent is not None:
+            parent.updateGeometry()
+            parent_layout = QWidget.layout(parent)
+            if parent_layout is not None:
+                parent_layout.activate()
+            parent.adjustSize()
+            if isinstance(parent, PropertyFrame):
+                break
+            parent = parent.parentWidget()
 
     def change_value(self):
         value = self.ui.text_field.toPlainText()

src/editors/smartprop_editor/property/comment.ui

@@ -10,18 +10,6 @@
     <height>157</height>
    </rect>
   </property>
-  <property name="minimumSize">
-   <size>
-    <width>0</width>
-    <height>128</height>
-   </size>
-  </property>
-  <property name="maximumSize">
-   <size>
-    <width>16777215</width>
-    <height>256</height>
-   </size>
-  </property>
   <property name="windowTitle">
    <string>Form</string>
   </property>

src/editors/smartprop_editor/property/ui_comment.py

@@ -23,8 +23,6 @@ def setupUi(self, Widget):
         if not Widget.objectName():
             Widget.setObjectName(u"Widget")
         Widget.resize(830, 157)
-        Widget.setMinimumSize(QSize(0, 128))
-        Widget.setMaximumSize(QSize(16777215, 256))
         Widget.setStyleSheet(u".QWidget {\n"
 "    font: 580 10pt \"Segoe UI\";\n"
 "    border: 2px solid black;\n"