From 35c09a86889357ce1b4d0306978f71b05df2dd67 Mon Sep 17 00:00:00 2001
From: Ubiratan Soares <ubiratan.f.soares@gmail.com>
Date: Thu, 18 Apr 2024 11:47:51 +0200
Subject: [PATCH] Refactors cleaning logic (#27)

---
 src/disk.rs            |  3 ++-
 src/disk/locations.rs  | 30 ++++++++++++++++++------------
 src/disk/operations.rs |  9 ++-------
 src/wiper.rs           |  8 +++++++-
 4 files changed, 29 insertions(+), 21 deletions(-)

diff --git a/src/disk.rs b/src/disk.rs
index bf05f92..6d0478a 100644
--- a/src/disk.rs
+++ b/src/disk.rs
@@ -6,9 +6,10 @@ mod operations;
 mod resources;
 
 pub use locations::find_all_gradle_projects;
+pub use locations::find_associated_filepaths;
 pub use locations::find_gradle_home;
 pub use locations::find_maven_local_repository;
-pub use operations::cleanup;
+pub use operations::cleanup_resources;
 pub use resources::resources_used_by_gradle_home;
 pub use resources::resources_used_by_gradle_projects;
 pub use resources::resources_used_by_maven_local_repository;
diff --git a/src/disk/locations.rs b/src/disk/locations.rs
index ce865e7..7a78202 100644
--- a/src/disk/locations.rs
+++ b/src/disk/locations.rs
@@ -2,7 +2,6 @@
 // SPDX-License-Identifier: MIT
 
 use crate::models::DiskCached;
-use anyhow::anyhow;
 use directories::BaseDirs;
 use std::path::{Path, PathBuf};
 use walkdir::{DirEntry, WalkDir};
@@ -14,20 +13,20 @@ pub fn find_gradle_home() -> anyhow::Result<PathBuf> {
         return Ok(PathBuf::from(custom_gradle_home));
     }
 
-    let base_dir = BaseDirs::new().ok_or(anyhow!("Cannot access base directories"))?;
-    let home_dir = base_dir.home_dir();
+    let base_dirs = BaseDirs::new().expect("Cannot access base directories");
+    let home_dir = base_dirs.home_dir();
     Ok(home_dir.join(".gradle"))
 }
 
 pub fn find_maven_local_repository() -> anyhow::Result<PathBuf> {
-    let base_dir = BaseDirs::new().ok_or(anyhow!("Cannot access base directories"))?;
-    let home_dir = base_dir.home_dir();
+    let base_dirs = BaseDirs::new().expect("Cannot access base directories");
+    let home_dir = base_dirs.home_dir();
     Ok(home_dir.join(".m2"))
 }
 
 pub fn find_all_gradle_projects() -> anyhow::Result<Vec<PathBuf>> {
-    let base_dir = BaseDirs::new().ok_or(anyhow!("Cannot access base directories"))?;
-    let home_dir = base_dir.home_dir();
+    let base_dirs = BaseDirs::new().expect("Cannot access base directories");
+    let home_dir = base_dirs.home_dir();
 
     let android_studio_projects_folder = home_dir.join("AndroidStudioProjects");
     let android_projects = find_gradle_projects(android_studio_projects_folder.as_path())?;
@@ -42,10 +41,7 @@ pub fn find_all_gradle_projects() -> anyhow::Result<Vec<PathBuf>> {
     Ok(all_projects)
 }
 
-pub(crate) fn find_paths(cached: &DiskCached) -> Vec<PathBuf> {
-    let base_dir = BaseDirs::new().expect("Cannot access base directories");
-    let home_dir = base_dir.home_dir();
-
+pub fn find_associated_filepaths(cached: DiskCached) -> Vec<PathBuf> {
     match cached {
         DiskCached::Standalone(_project_level) => {
             let gradle_projects = find_all_gradle_projects().unwrap_or_default();
@@ -71,7 +67,17 @@ pub(crate) fn find_paths(cached: &DiskCached) -> Vec<PathBuf> {
         },
         DiskCached::Shared(user_level) => match user_level.path_relative_to_user_home() {
             None => vec![],
-            Some(path) => vec![home_dir.join(path)],
+            Some(path) => {
+                if let Ok(custom_gradle_home) = std::env::var("GRADLE_USER_HOME") {
+                    if path.to_string_lossy().contains(".gradle") {
+                        return vec![PathBuf::from(custom_gradle_home).join(path)];
+                    }
+                }
+
+                let base_dirs = BaseDirs::new().expect("Cannot access base directories");
+                let home_dir = base_dirs.home_dir();
+                vec![home_dir.join(path)]
+            },
         },
     }
 }
diff --git a/src/disk/operations.rs b/src/disk/operations.rs
index 10d0ca8..ab865ab 100644
--- a/src/disk/operations.rs
+++ b/src/disk/operations.rs
@@ -1,15 +1,10 @@
 // Copyright 2024 Dotanuki Labs
 // SPDX-License-Identifier: MIT
 
-use crate::models::DiskCached;
 use std::fs;
+use std::path::PathBuf;
 
-pub fn cleanup(resources: &[DiskCached]) {
-    let paths_to_remove = resources
-        .iter()
-        .flat_map(crate::disk::locations::find_paths)
-        .collect::<Vec<_>>();
-
+pub fn cleanup_resources(paths_to_remove: &[PathBuf]) {
     let errors = paths_to_remove
         .iter()
         .map(fs::remove_dir_all)
diff --git a/src/wiper.rs b/src/wiper.rs
index 017d239..d6f4e87 100644
--- a/src/wiper.rs
+++ b/src/wiper.rs
@@ -6,6 +6,7 @@ use crate::models::{
     AllocatedResource, DiskCached, EvaluationOutcome, ExecutionOutcome, MachineResource, UserLevelDiskCache,
     WipeAction, WipingOutcome,
 };
+use std::path::PathBuf;
 use ubyte::{ByteUnit, ToByteUnit};
 use MachineResource::{DiskSpace, RamMemory};
 use WipeAction::{DeepWipe, Evaluate, ShallowWipe};
@@ -73,7 +74,12 @@ fn shallow_wipe_disk() -> anyhow::Result<ExecutionOutcome> {
         DiskCached::Shared(UserLevelDiskCache::MavenLocalRepository),
     ];
 
-    disk::cleanup(&caches_to_remove);
+    let paths_to_remove = caches_to_remove
+        .into_iter()
+        .flat_map(disk::find_associated_filepaths)
+        .collect::<Vec<PathBuf>>();
+
+    disk::cleanup_resources(&paths_to_remove);
 
     let after_cleaning = match evaluate_disk_space()? {
         ExecutionOutcome::Evaluation(outcome) => outcome.total_size,