Try remove cmd

mpsd-software.py

@@ -15,6 +15,7 @@ import time
 from pathlib import Path
 from typing import List, Tuple, Union
 import re
+import shutil
 
 # If 'rich' is available ("pip install rich" or "apt-get install python3-rich"),
 # then use coloured output, otherwise proceed as before
@@ -979,13 +980,40 @@ def install_environment(
             )
 
 
-def remove_environment(release, package_sets, target_dir):
+def remove_environment(mpsd_release, root_dir, package_sets="NONE", force_remove=False):
     """Remove release from installation."""
     msg = (
-        f"Removing release {release} with package_sets {package_sets} from {target_dir}"
+        f"Removing release {mpsd_release}"
+        f" with package_sets {package_sets} from {root_dir}"
     )
-    logging.info(msg)
-    raise NotImplementedError(msg)
+    logging.warning(msg)
+    if package_sets == "NONE":
+        logging.warning(
+            "Please specify package_sets to remove, or 'ALL' to remove all toolchains"
+        )
+        sys.exit(1)
+    if "ALL" in package_sets:
+        # we need to remove the entire release folder
+        logging.warning(
+            f"Removing release {mpsd_release} from {root_dir}"
+            "do you want to continue? [y/n]"
+        )
+        if force_remove or input().lower() == "y":
+            folders_to_remove = os.listdir(root_dir / mpsd_release)
+            # skip logs folder
+            if "logs" in folders_to_remove:
+                folders_to_remove.remove("logs")
+            for folder in folders_to_remove:
+                shutil.rmtree(root_dir / mpsd_release / folder)
+                sys.exit(0)
+    for package_set in package_sets:
+        # we load the spack environment and remove the package_set
+        spack_env = ""
+        commands_to_execute = [
+            f"source {spack_env}",
+            f"spack env remove -y {package_set}",
+        ]
+        run(" && ".join(commands_to_execute), shell=True, check=True)
 
 
 def start_new_environment(release, from_release, target_dir):
@@ -1179,7 +1207,7 @@ def main():
 
     # Check the command and run related function
     if args.action == "remove":
-        remove_environment(args.release, args.package_set, root_dir)
+        remove_environment(args.release, root_dir, args.package_set)
     elif args.action == "start-new":
         start_new_environment(args.from_release, args.to_release, root_dir)
     elif args.action == "install":