Merge pull request #15 from ChocoScaff/context_menu

Context menu

Author: ChocoScaff

caption.py

@@ -16,14 +16,16 @@ def __init__(self, sourceSDK) -> None:
 
         self.sdk = sourceSDK
 
-    def build_caption(self):
+    def build_caption(self, file=None):
         """
         Compile caption
         """
-
-        filenameTXT = filedialog.askopenfile(title="Select .txt file", filetypes=[("TXT files", "closecaption*.txt")], initialdir=self.sdk.selected_folder + "/resource")
+        if file == None:
+            filenameTXT = filedialog.askopenfile(title="Select .txt file", filetypes=[("TXT files", "closecaption*.txt")], initialdir=self.sdk.selected_folder + "/resource")
+            file = filenameTXT.name
+        
         captioncompiler = (self.sdk.bin_folder + "/captioncompiler.exe")
-        command = ('"' + captioncompiler + '"' + " -game " + '"' + self.sdk.selected_folder + '"' + " " + '"' + filenameTXT.name + '"')
+        command = ('"' + captioncompiler + '"' + " -game " + '"' + self.sdk.selected_folder + '"' + " " + '"' + file + '"')
         print(command)
         result = subprocess.run(command, shell=True, capture_output=True, text=True)
         print(result)

file.py

@@ -4,6 +4,9 @@
 from fileListApp import FileListApp  # Assuming this is your custom module
 from open import Open  # Assuming this is your custom module
 from PIL import Image, ImageTk
+from model import Model
+from texture import Texture
+from map import Map
 
 class File:
     """
@@ -112,6 +115,7 @@ def display_files(self):
 
         # Bind double-click event to open the selected file
         self.tree.bind("<Double-Button-1>", self.open_file)
+        self.tree.bind("<Button-3>", self.show_context_menu)
 
         self.fileList = FileListApp(self.sdk, self.main_root)
 
@@ -189,14 +193,12 @@ def load_thumbnail(self, file_path, parent=""):
                 ".smd": "txt.png",
                 ".cfg": "txt.png",
                 ".sln": "Visual_Studio.png",
-                ".vpk": "fileexplorer.png",
                 ".wav": "audio.png",
                 ".mp3": "audio.png",
                 ".bik": "video.png",
                 ".bat": "terminal.png"
             }
 
-            #ext = os.path.splitext(file_path)[1]
             file_name, file_extension = os.path.splitext(file_path)
 
             if file_extension in file_icons:
@@ -217,3 +219,58 @@ def load_thumbnail(self, file_path, parent=""):
         except Exception as e:
             print("Error loading thumbnail:", e)
         return None
+    
+    def show_context_menu(self, event):
+        """
+        Show the context menu on right-click.
+        """
+        selected_item = self.tree.identify_row(event.y)
+               
+        if selected_item:
+            self.tree.selection_set(selected_item)
+
+            filename = self.tree.item(selected_item, 'text')
+            print(filename)
+
+            parent_item = self.tree.parent(selected_item)
+            
+            file_path_parts = [filename]
+
+            while parent_item:
+                item_text = self.tree.item(parent_item, "text")
+                file_path_parts.append(item_text)
+                parent_item = self.tree.parent(parent_item)
+
+            file_path_parts.reverse()
+            file_path = os.path.join(self.sdk.parent_folder, *file_path_parts)
+
+            print(file_path)
+
+            self.context_menu = tk.Menu(self.tree, tearoff=0)
+
+            file_extension = os.path.splitext(filename)[1]
+
+            if file_extension == ".qc":
+                model = Model(self.sdk)
+                self.context_menu.add_command(label="Compile Model", command=lambda: model.build_model(file_path))
+            elif file_extension == ".tga":
+                texture = Texture(self.sdk)
+                self.context_menu.add_command(label="Compile Texture", command=lambda: texture.build_texture(file_path))
+            elif file_extension == ".vmf":
+                map = Map(self.sdk)
+                self.context_menu.add_command(label="Compile Map", command=lambda: map.build_map(file_path))
+               
+            self.context_menu.add_command(label="Delete", command=lambda: self.delete_file(file_path, selected_item))
+
+            self.context_menu.post(event.x_root, event.y_root)
+
+    def delete_file(self, file_path, tree_item):
+        """
+        Delete the specified file and update the Treeview.
+        """
+        try:
+            os.remove(file_path)
+            self.tree.delete(tree_item)
+            print(f"Deleted: {file_path}")
+        except Exception as e:
+            print(f"Error deleting file: {e}")

fileListApp.py

@@ -3,6 +3,9 @@
 from PIL import Image, ImageTk
 import os
 from open import Open
+from model import Model
+from texture import Texture
+from map import Map
 
 class FileListApp:
     def __init__(self, sourceSDK, root):
@@ -94,13 +97,19 @@ def load_files(self, folder):
 
             if os.path.isdir(file_path):
                 bind_func = lambda e, path=file_path: self.load_files(path)
+                bind_right = lambda e, path=file_path: self.show_context_menu(e, path)
             else:
                 bind_func = lambda e, path=file_path: self.open_file(path)
+                bind_right = lambda e, path=file_path: self.show_context_menu(e, path)
 
             frame.bind("<Double-Button-1>", bind_func)
             label.bind("<Double-Button-1>", bind_func)
+            frame.bind("<Button-3>", bind_right)
+            label.bind("<Button-3>", bind_right)
+
             if thumbnail:
                 thumbnail_label.bind("<Double-Button-1>", bind_func)
+                thumbnail_label.bind("<Button-3>", bind_right)
 
             col += 1
             if col >= columns:
@@ -188,4 +197,39 @@ def resize(self, event):
         if new_width != self.previous_width:
             self.previous_width = new_width
             # Add the code you want to execute when the width changes
-            self.load_files(self.current_folder)
+            self.load_files(self.current_folder)
+    
+    def show_context_menu(self, event, file_path):
+        """
+        Show the context menu on right-click.
+        """
+        file_name, file_extension = os.path.splitext(file_path)
+
+        self.context_menu = tk.Menu(self.root, tearoff=0)
+
+        if file_extension == ".qc":
+            model = Model(self.sdk)
+            self.context_menu.add_command(label="Compile Model", command=lambda: model.build_model(file_path))
+        elif file_extension == ".tga":
+            texture = Texture(self.sdk)
+            self.context_menu.add_command(label="Compile Texture", command=lambda: texture.build_texture(file_path))
+        elif file_extension == ".vmf":
+                map = Map(self.sdk)
+                self.context_menu.add_command(label="Compile Map", command=lambda: map.build_map(file_path))
+
+        self.context_menu.add_command(label="Delete", command=lambda: self.delete_file(file_path))
+
+
+        self.context_menu.post(event.x_root, event.y_root)
+
+    def delete_file(self, file_path):
+        """
+        Delete the specified file and update the Treeview.
+        """
+        try:
+            os.remove(file_path)
+            print(f"Deleted: {file_path}")
+        except Exception as e:
+            print(f"Error deleting file: {e}")
+        
+        self.load_files(self.current_folder)

map.py

@@ -19,19 +19,21 @@ def __init__(self, sourceSDK) -> None:
         """
         self.sdk = sourceSDK
 
-    def build_map(self):
+    def build_map(self, file=None):
         """
         """
 
         mapsrc_directory = os.path.join(self.sdk.selected_folder, "mapsrc")
         map_directory = os.path.join(self.sdk.selected_folder, "maps")
+        
+        if file == None:
+            filenameVMF = filedialog.askopenfile(title="Select .vmf file", filetypes=[("VMF files", "*.vmf")], initialdir=mapsrc_directory)
+            file = filenameVMF.name
 
-        filenameVMF = filedialog.askopenfile(title="Select .vmf file", filetypes=[("VMF files", "*.vmf")], initialdir=mapsrc_directory)
-
-        print("file =", filenameVMF.name)
+        print("file =", file)
         # Execute vbsp.exe
 
-        fileBSP = filenameVMF.name
+        fileBSP = file
         #file_directory = os.path.dirname(fileBSP)
         fileBSP = os.path.splitext(os.path.basename(fileBSP))[0]
 
@@ -42,7 +44,7 @@ def build_map(self):
         print(self.sdk.bin_folder)
 
         vbsp = (self.sdk.bin_folder + "/vbsp.exe")
-        command = ('"' + vbsp + '"' + " -game " + '"' + self.sdk.selected_folder + '"' + " " + '"' + filenameVMF.name + '"')
+        command = ('"' + vbsp + '"' + " -game " + '"' + self.sdk.selected_folder + '"' + " " + '"' + file + '"')
         print(command)
         #Execute the command in cmd
         result = subprocess.run(command, shell=True, capture_output=True, text=True)

model.py

@@ -24,13 +24,16 @@ def open_hlmv(self):
 
         subprocess.Popen([self.sdk.bin_folder + "/hlmv.exe"])
 
-    def build_model(self):
+    def build_model(self, file=None):
         """
         """
-
-        filenameQC = filedialog.askopenfile(title="Select .qc file", filetypes=[("QC files", "*.qc")], initialdir=self.sdk.selected_folder + "/modelsrc")
+        if file == None:
+            filenameQC = filedialog.askopenfile(title="Select .qc file", filetypes=[("QC files", "*.qc")], initialdir=self.sdk.selected_folder + "/modelsrc")
+            file= filenameQC.name
+        
+            
         mdl = (self.sdk.bin_folder + "/studiomdl.exe")
-        command = ('"' + mdl + '"' + " -game " + '"' + self.sdk.selected_folder + '"' + " " + '"' + filenameQC.name + '"')
+        command = ('"' + mdl + '"' + " -game " + '"' + self.sdk.selected_folder + '"' + " " + '"' + file + '"')
         print(command)
         result = subprocess.run(command, shell=True, capture_output=True, text=True)
         print(result)

texture.py

@@ -52,13 +52,14 @@ def build_all_texture(self):
                     result = subprocess.run(command, shell=True, capture_output=True, text=True)
                     print(result)
 
-    def build_texture(self):
+    def build_texture(self,file=None):
         """
         """
-
-        filenameTGA = filedialog.askopenfile(title="Select .tga file", filetypes=[("TGA files", "*.tga")], initialdir=self.sdk.selected_folder + "/materialsrc")
+        if file == None:
+            filenameTGA = filedialog.askopenfile(title="Select .tga file", filetypes=[("TGA files", "*.tga")], initialdir=self.sdk.selected_folder + "/materialsrc")
+            file = filenameTGA.name
         vtex = (self.sdk.bin_folder + "/vtex.exe")
-        command = ('"' + vtex + '"' + " -game " + '"' + self.sdk.selected_folder + '"' + " -nopause "  + '"' + filenameTGA.name + '"' )
+        command = ('"' + vtex + '"' + " -game " + '"' + self.sdk.selected_folder + '"' + " -nopause "  + '"' + file + '"' )
         print(command)
         result = subprocess.run(command, shell=True, capture_output=True, text=True)
         print(result)