Merge pull request #5 from secretlay3r/patch-1

feat: Add tkinter icon and move on_ok() to a dict

Author: neohiro

ExploitProtection.PY

@@ -217,8 +217,6 @@ class XProtApp:
         self.root.resizable(False, False)
 
         # Download and Setup Image
-        self.dlfolder = os.environ["ProgramFiles"]
-        self.img_path = os.path.join(self.dlfolder, "XPROTlogo.png")
         self.download_logo()
 
         # Image Display
@@ -270,30 +268,52 @@ class XProtApp:
         self.lbl_footer.place(x=32, y=490, width=260, height=125)
 
     def download_logo(self):
-        if not os.path.exists(self.img_path):
+        dlfolder = os.environ["ProgramFiles"]
+        
+        # Logo
+        img_path = os.path.join(dlfolder, "XPROTlogo.png")
+        if not os.path.exists(img_path):
             try:
-                url = "https://raw.githubusercontent.com/neohiro/ExploitProtection/M3T4P0D.3XPL01T/media/Exploitprotectionlogo3.png"
-                req = urllib.request.Request(url, headers={'User-Agent': 'Mozilla/5.0'})
-                with urllib.request.urlopen(req) as response, open(self.img_path, 'wb') as out_file:
+                url1 = "https://raw.githubusercontent.com/neohiro/ExploitProtection/M3T4P0D.3XPL01T/media/Exploitprotectionlogo3.png"
+                req = urllib.request.Request(url1, headers={'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64)'})
+                with urllib.request.urlopen(req, timeout=10) as response, open(img_path, 'wb') as out_file:
                     out_file.write(response.read())
-            except Exception as e:
-                print(f"Logo download failed: {e}")
+            except:
+                pass
+        
+        # Icon
+        ico_path = os.path.join(dlfolder, "XPROT.ico")
+        if os.path.exists(ico_path):
+            try:
+                self.root.iconbitmap(ico_path)
+            except:
+                pass
+        else:
+            try:
+                url2 = "https://raw.githubusercontent.com/neohiro/ExploitProtection/refs/heads/M3T4P0D.3XPL01T/media/icons8-windows-defender-70.ico"
+                req = urllib.request.Request(url2, headers={'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64)'})
+                with urllib.request.urlopen(req, timeout=10) as response, open(ico_path, 'wb') as out_file:
+                    out_file.write(response.read())
+                self.root.iconbitmap(ico_path)
+            except:
+                pass
+        self.img_path = img_path
 
     def on_ok(self):
+        choices = {
+            1: update_online_xml,
+            2: backup_current_settings,
+            3: remove_all_protection,
+            4: import_local_xml,
+            5: reboot_device,
+            6: open_support
+        }
+        
         selection = self.selected_option.get()
+        action = choices.get(selection)
 
-        if selection == 1:
-            update_online_xml()
-        elif selection == 2:
-            backup_current_settings()
-        elif selection == 3:
-            remove_all_protection()
-        elif selection == 4:
-            import_local_xml()
-        elif selection == 5:
-            reboot_device()
-        elif selection == 6:
-            open_support()
+        if action:
+            action()
         else:
             messagebox.showerror("Error", "Please choose an option first.")
 
@@ -307,4 +327,5 @@ if __name__ == "__main__":
 
     root = tk.Tk()
     app = XProtApp(root)
-    root.mainloop()
+    root.mainloop()
+