Fix bugs

Author: zxelzy

app/utility/update.py

@@ -15,37 +15,45 @@ def run_update():
         pass
 
 
-    # 1. Determine Root with Precision
+    # Determine Root with Precision
     base_dir = os.path.dirname(os.path.realpath(__file__))
     project_root = os.path.abspath(os.path.join(base_dir, "..", ".."))
     os.chdir(project_root)
 
     print(f"{C.SUCCESS}[*] Attempting to update Storm Framework.{C.RESET}")
 
-    # 2. Check Connection & Fetch
-    # Use rebase to be cleaner and not create messy merge commits.
-    print(f"[*] Checking for new versions at github.com/storm-os/Cyber-Pentest")
-    
-    # We use subprocess to catch errors more elegantly.
+    # 1. Get the latest info without changing the locale first
     subprocess.run(["git", "fetch", "--all"], stdout=subprocess.DEVNULL)
+    
+    # 2. CHECK CHANGES: Compare local (HEAD) with server (origin/main)
+    # Check if there are any different .go or .c files
+    check_diff = subprocess.run(
+        ["git", "diff", "--name-only", "HEAD", "origin/main"],
+        capture_output=True, text=True
+    )
+    
+    # Filter: are there any files ending in .go or .c?
+    changed_files = check_diff.stdout.splitlines()
+    needs_recompile = any(f.endswith(('.go', '.c')) for f in changed_files)
+
+    # 3. Reset Execution (Update file to the latest version)
     process = subprocess.run(["git", "reset", "--hard", "origin/main"], 
-                             stderr=subprocess.PIPE,
-                             text=True)
+                             stderr=subprocess.PIPE, text=True)
 
     if process.returncode == 0:
-        print(f"{C.SUCCESS}\n[+] Git synchronization complete.{C.RESET}")
-            
-        # 3. Trigger Compiler (The Hook)
-        compiler_path = os.path.join(project_root, "compiler")
-        if os.path.exists(compiler_path):
-            print(f"{C.SUCCESS}\n[*] Triggering framework recompilation.{C.RESET}")
-            os.system(f'bash -c "source {compiler_path} && compile_modules"')
-            print(f"{C.SUCCESS}\n[✓] Framework updated to v{latest_version}{C.RESET}")
+        print(f"{C.SUCCESS}\n[+] System updated to latest version.{C.RESET}")
+        
+        # 4. Trigger Compiler ONLY IF needed
+        if needs_recompile:
+            compiler_path = os.path.join(project_root, "compiler")
+            if os.path.exists(compiler_path):
+                print(f"{C.SUCCESS}\n[*] Source code changes detected. Recompiling...{C.RESET}")
+                os.system(f'bash -c "source {compiler_path} && compile_modules"')
+                print(f"{C.SUCCESS}\n[✓] Framework recompiled successfully.{C.RESET}")
         else:
-            print(f"{C.ERROR}\n[x] ERROR: Compiler hook not found at {compiler_path}{C.RESET}")
-    else:
-        print(f"{C.ERROR}\n[x] Update failed!{C.RESET}")
-        print(f"{C.SUCCESS}[!] Logic: {process.stderr}{C.RESET}")
-        print(f"[*] Suggestion: Run 'git stash' if you have local changes.")
+            print(f"{C.SUCCESS}[*] No source code changes. Skipping compilation. 😴{C.RESET}")
+            
+        print(f"{C.SUCCESS}\n[✓] Storm is now v{latest_version}{C.RESET}")
+
 
     sys.exit()