More

utils.py

@@ -0,0 +1,83 @@
+import os
+import shutil
+import sys
+
+
+RED = "\033[1;31m"
+GREEN = "\033[1;32m"
+YELLOW = "\033[1;33m"
+CYAN = "\033[1;36m"
+RESET = "\033[0m"
+
+
+def log_info(msg):    print(f"{CYAN}[*]{RESET} {msg}")
+def log_success(msg): print(f"{GREEN}[✓]{RESET} {msg}")
+def log_warn(msg):    print(f"{YELLOW}[!]{RESET} {msg}")
+def log_error(msg):   print(f"{RED}[✗]{RESET} {msg}")
+
+
+def ask_override(path):
+    resp = input(f"{YELLOW}'{path}' already exists. Override? (Y/N): {RESET}").strip().lower()
+    return resp == 'y'
+
+
+def choice_files(src_root, dst_root, choices_map, override_all=False):
+    for final_name, options in choices_map.items():
+        existing_options = [
+            f for f in options
+            if os.path.exists(os.path.join(src_root, f.lstrip('/')))
+        ]
+        if not existing_options:
+            log_warn(f"No source file found among options for {final_name}. Skipping.")
+            continue
+        
+        if len(existing_options) == 1:
+            chosen = existing_options[0]
+        else:
+            log_info(f"Choose which file to copy as '{final_name}':")
+            for idx, opt in enumerate(existing_options, 1):
+                print(f"{idx}: {opt}")
+            while True:
+                try:
+                    choice_idx = int(input(f"{CYAN}Enter number (1-{len(existing_options)}): {RESET}"))
+                    if 1 <= choice_idx <= len(existing_options):
+                        chosen = existing_options[choice_idx - 1]
+                        break
+                except ValueError:
+                    pass
+                log_warn("Invalid choice, try again.")
+        
+        src_file = os.path.join(src_root, chosen.lstrip('/'))
+        dst_file = os.path.join(dst_root, final_name.lstrip('/'))
+
+        if os.path.exists(dst_file):
+            if not override_all and not ask_override(dst_file):
+                log_warn(f"Skipped overriding {dst_file}")
+                continue
+
+        shutil.copy2(src_file, dst_file)
+        log_success(f"Copied {chosen} -> {final_name}")
+
+def copy_with_structure(src_root, dst_root, override_all=False):
+    for root, dirs, files in os.walk(src_root):
+        rel_path = os.path.relpath(root, src_root)
+        target_dir = os.path.join(dst_root, rel_path) if rel_path != '.' else dst_root
+
+        os.makedirs(target_dir, exist_ok=True)
+
+        for file in files:
+            src_file = os.path.join(root, file)
+            dst_file = os.path.join(target_dir, file)
+
+            if os.path.exists(dst_file):
+                if not override_all and not ask_override(dst_file):
+                    log_warn(f"Skipped overriding {dst_file}")
+                    continue
+
+            shutil.copy2(src_file, dst_file)
+            log_success(f"Copied: {src_file} -> {dst_file}")
+
+def require_root():
+    if os.geteuid() != 0:
+        log_error("This script must be run as root (sudo). Exiting.")
+        sys.exit(1)