feat: initialize project structure with main UI, asset editors, and utility modules

Author: dertwist

src/editors/asset_exporter/dependency_resolver.py

@@ -1,16 +1,6 @@
 import os
-from src.common import Kv3ToJson
 
 class DependencyResolver:
-    ASSET_DEPENDENCY_KEYS = [
-        # vmdl / vsmart
-        'm_refMeshes', 'm_refPhysicsData', 'model',
-        # vmat
-        'TextureColor', 'TextureNormal', 'TextureRoughness',
-        'TextureMetalness', 'g_tColor', 'g_tNormal',
-        # generic
-        'material', 'mesh', 'texture', 'm_strPsd'
-    ]
 
     def __init__(self, addon_content_path: str):
         self.addon_content_path = addon_content_path
@@ -33,28 +23,18 @@ def _walk(self, path: str):
             self._parse_kv3_deps(path)
 
     def _parse_kv3_deps(self, path: str):
+        import re
         try:
             with open(path, 'r', encoding='utf-8', errors='ignore') as f:
                 content = f.read()
-            # If it's pure binary or random data, kv3 parsing will fail, which is fine
-            # We catch exceptions
-            data = Kv3ToJson(content)
-            if data:
-                self._find_deps_in_dict(data)
+            
+            pattern = r'"([^"]+\.(?:vmdl|vsmart|vmat|vpcf|vsndevts|vtex|vsnd|txt|kv3|vmap|vpost)(?:_c)?)"'
+            matches = re.findall(pattern, content, re.IGNORECASE)
+            for m in matches:
+                self._add_dep(m)
         except Exception as e:
             print(f"DependencyResolver: Error parsing {path}: {e}")
 
-    def _find_deps_in_dict(self, data):
-        if isinstance(data, dict):
-            for k, v in data.items():
-                if k in self.ASSET_DEPENDENCY_KEYS and isinstance(v, str):
-                    self._add_dep(v)
-                else:
-                    self._find_deps_in_dict(v)
-        elif isinstance(data, list):
-            for item in data:
-                self._find_deps_in_dict(item)
-
     def _add_dep(self, rel_path: str):
         if not rel_path:
             return

src/editors/asset_exporter/exporter.py

@@ -7,31 +7,78 @@ class ExportWorker(QThread):
     file_copied = Signal(str)
     finished_export = Signal(str)
 
-    def __init__(self, files, addon_content_path, dest_root, layout, addon_name, asset_stem):
+    def __init__(self, files, addon_content_path, dest_root, layout, addon_name, asset_stem, export_to_zip=False, zip_name="Export"):
         super().__init__()
         self.files = files
         self.addon_content_path = addon_content_path
         self.dest_root = dest_root
         self.layout = layout
         self.addon_name = addon_name
         self.asset_stem = asset_stem
+        self.export_to_zip = export_to_zip
+        self.zip_name = zip_name
 
     def run(self):
+        import tempfile
+        import zipfile
+        SCANNABLE_EXTS = {'.vmdl', '.vsmart', '.vmat', '.vpcf', '.vsndevts', '.vsnd'}
+        
+        path_mapping = {}
+        for src in self.files:
+            old_rel = os.path.relpath(src, self.addon_content_path).replace('\\', '/')
+            new_rel = self._compute_dest_rel(src)
+            path_mapping[old_rel] = new_rel
+
+        if self.export_to_zip:
+            temp_dir = tempfile.mkdtemp()
+            base_dir = temp_dir
+        else:
+            base_dir = self.dest_root
+
         for i, src in enumerate(self.files):
-            dest = self._compute_dest(src)
+            dest = self._compute_dest(src, base_dir)
             os.makedirs(os.path.dirname(dest), exist_ok=True)
             try:
-                shutil.copy2(src, dest)
+                ext = os.path.splitext(src)[1].lower()
+                if ext in SCANNABLE_EXTS:
+                    with open(src, 'r', encoding='utf-8', errors='ignore') as f:
+                        content = f.read()
+                    
+                    for old_rel in sorted(path_mapping.keys(), key=len, reverse=True):
+                        new_rel = path_mapping[old_rel]
+                        if old_rel != new_rel and old_rel in content:
+                            content = content.replace(old_rel, new_rel)
+
+                    with open(dest, 'w', encoding='utf-8') as f:
+                        f.write(content)
+                else:
+                    shutil.copy2(src, dest)
                 self.file_copied.emit(src)
             except Exception as e:
                 print(f"Error copying {src} to {dest}: {e}")
             self.progress.emit(int((i + 1) / len(self.files) * 100))
+
+        if self.export_to_zip:
+            zip_path = os.path.join(self.dest_root, self.zip_name)
+            shutil.make_archive(zip_path, 'zip', temp_dir)
+            shutil.rmtree(temp_dir)
+            
         self.finished_export.emit(self.dest_root)
 
-    def _compute_dest(self, src):
-        rel = os.path.relpath(src, self.addon_content_path)
+    def _compute_dest_rel(self, src):
+        rel = os.path.relpath(src, self.addon_content_path).replace('\\', '/')
         if self.layout == 'thirdparty':
-            return os.path.join(self.dest_root,
-                'folder_thirdparty', self.addon_name,
-                self.asset_stem, rel)
-        return os.path.join(self.dest_root, rel)
+            parts = rel.split("/", 1)
+            if len(parts) == 2:
+                root_dir, rest = parts[0], parts[1]
+                path_suffix = "thirdparty"
+                if self.addon_name:
+                    path_suffix = f"{path_suffix}/{self.addon_name}"
+                if self.asset_stem:
+                    path_suffix = f"{path_suffix}/{self.asset_stem}"
+                return f"{root_dir}/{path_suffix}/{rest}"
+        return rel
+
+    def _compute_dest(self, src, base_dir):
+        rel = self._compute_dest_rel(src)
+        return os.path.normpath(os.path.join(base_dir, rel))

src/editors/asset_exporter/main.py

@@ -6,7 +6,6 @@
 from .dependency_resolver import DependencyResolver
 from .exporter import ExportWorker
 from src.settings.main import get_addon_name, get_cs2_path
-from PySide6.QtWidgets import QLineEdit
 from src.common import enable_dark_title_bar
 from src.styles.common import apply_stylesheets, qt_stylesheet_widgetlist2
 
@@ -19,28 +18,6 @@ def __init__(self, parent=None):
         enable_dark_title_bar(self)
         apply_stylesheets(self)
 
-        qt_stylesheet_lineedit = """
-        QLineEdit {
-            font: 580 10pt "Segoe UI";
-            border: 2px solid black;
-            border-radius: 2px;
-            border-color: rgba(80, 80, 80, 255);
-            height:22px;
-            padding-top: 2px;
-            padding-bottom:2px;
-            padding-left: 4px;
-            padding-right: 4px;
-            color: #E3E3E3;
-            background-color: #1C1C1C;
-        }
-        QLineEdit:hover {
-            background-color: #414956;
-            color: white;
-        }
-        """
-        for line_edit in self.findChildren(QLineEdit):
-            line_edit.setStyleSheet(qt_stylesheet_lineedit)
-            
         self.ui.deps_list.setStyleSheet(qt_stylesheet_widgetlist2)
 
         self.cs2_path = get_cs2_path()
@@ -76,6 +53,11 @@ def __init__(self, parent=None):
 
         self.toggle_thirdparty_fields()
 
+        from PySide6.QtWidgets import QCheckBox
+        self.ui.checkbox_zip = QCheckBox("Export to ZIP Archive (name based on selected asset)")
+        self.ui.checkbox_zip.setChecked(True)
+        self.ui.verticalLayout.insertWidget(6, self.ui.checkbox_zip)
+        
         self.deps_model = QStandardItemModel()
         self.ui.deps_list.setModel(self.deps_model)
 
@@ -155,13 +137,21 @@ def start_export(self):
 
         layout = 'thirdparty' if is_thirdparty else 'preserve'
 
+        export_to_zip = self.ui.checkbox_zip.isChecked()
+        zip_name = "Export"
+        if self.sources_to_export:
+            first_src = os.path.basename(self.sources_to_export[0])
+            zip_name = os.path.splitext(first_src)[0]
+
         self.worker = ExportWorker(
             files_to_export,
             self.addon_content_path,
             output_dir,
             layout,
             addon_name,
-            asset_stem
+            asset_stem,
+            export_to_zip,
+            zip_name
         )
         self.worker.progress.connect(self.ui.progress_bar.setValue)
         self.worker.finished_export.connect(self.on_export_finished)

src/editors/asset_exporter/ui_main.py

@@ -111,7 +111,7 @@ def retranslateUi(self, AssetExporterWidget):
         self.btn_resolve.setText(QCoreApplication.translate("AssetExporterWidget", u"Resolve Dependencies", None))
         self.groupBox.setTitle(QCoreApplication.translate("AssetExporterWidget", u"Export Options", None))
         self.radio_preserve.setText(QCoreApplication.translate("AssetExporterWidget", u"Preserve Addon Structure", None))
-        self.radio_thirdparty.setText(QCoreApplication.translate("AssetExporterWidget", u"Third-Party Package Layout (folder_thirdparty/...)", None))
+        self.radio_thirdparty.setText(QCoreApplication.translate("AssetExporterWidget", u"Third-Party Package Layout (models/thirdparty/...)", None))
         self.edit_addon_name.setPlaceholderText(QCoreApplication.translate("AssetExporterWidget", u"Addon Name", None))
         self.edit_asset_stem.setPlaceholderText(QCoreApplication.translate("AssetExporterWidget", u"Asset Stem", None))
         self.edit_output_dir.setPlaceholderText(QCoreApplication.translate("AssetExporterWidget", u"Output Directory", None))

src/editors/asset_manager/main.py

@@ -5,7 +5,6 @@
 from .ui_main import Ui_AssetManagerWidget
 from .move_worker import MoveWorker
 from src.settings.main import get_addon_name, get_cs2_path, app_dir
-from PySide6.QtWidgets import QLineEdit
 from src.common import enable_dark_title_bar
 from src.styles.common import apply_stylesheets
 
@@ -18,28 +17,6 @@ def __init__(self, parent=None):
         enable_dark_title_bar(self)
         apply_stylesheets(self)
 
-        qt_stylesheet_lineedit = """
-        QLineEdit {
-            font: 580 10pt "Segoe UI";
-            border: 2px solid black;
-            border-radius: 2px;
-            border-color: rgba(80, 80, 80, 255);
-            height:22px;
-            padding-top: 2px;
-            padding-bottom:2px;
-            padding-left: 4px;
-            padding-right: 4px;
-            color: #E3E3E3;
-            background-color: #1C1C1C;
-        }
-        QLineEdit:hover {
-            background-color: #414956;
-            color: white;
-        }
-        """
-        for line_edit in self.findChildren(QLineEdit):
-            line_edit.setStyleSheet(qt_stylesheet_lineedit)
-
         self.cs2_path = get_cs2_path()
         if not self.cs2_path:
             return
@@ -48,7 +25,7 @@ def __init__(self, parent=None):
         self.addon_content_path = os.path.join(self.cs2_path, 'content', 'csgo_addons', self.addon_name)
 
         self.setWindowFlags(Qt.Window)
-        self.setWindowTitle("Asset Manager")
+        self.setWindowTitle("Move Assets")
         self.ui.source_tree.hide()
         self.sources_to_move = []