From 3fee7c6d5b99e85084a4239de201f99854be44a6 Mon Sep 17 00:00:00 2001
From: eatradish <sakiiily@aosc.io>
Date: Sat, 15 Feb 2025 19:39:05 +0800
Subject: [PATCH] feat(subcmd/clean): add `--keep-downloadable` and
 `--keep-downloadable-and-installed` arguments

---
 src/subcommand/clean.rs | 64 ++++++++++++++++++++++++++++++++++++++---
 1 file changed, 60 insertions(+), 4 deletions(-)

diff --git a/src/subcommand/clean.rs b/src/subcommand/clean.rs
index e6ea4f70..9d7ca09b 100644
--- a/src/subcommand/clean.rs
+++ b/src/subcommand/clean.rs
@@ -1,3 +1,4 @@
+use std::fs::remove_dir_all;
 use std::path::PathBuf;
 
 use crate::subcommand::utils::create_progress_spinner;
@@ -5,7 +6,7 @@ use crate::success;
 use crate::{config::Config, fl};
 use clap::Args;
 use oma_pm::apt::{AptConfig, OmaApt, OmaAptArgs};
-use tracing::info;
+use tracing::{debug, info};
 
 use crate::{error::OutputError, utils::root};
 
@@ -13,6 +14,12 @@ use crate::args::CliExecuter;
 
 #[derive(Debug, Args)]
 pub struct Clean {
+    /// Keep downloadable packages
+    #[arg(long, conflicts_with = "keep_downloadable_and_installed")]
+    keep_downloadable: bool,
+    /// Keep downloadable and installed packages
+    #[arg(long, conflicts_with = "keep_downloadable")]
+    keep_downloadable_and_installed: bool,
     /// Set sysroot target directory
     #[arg(from_global)]
     sysroot: PathBuf,
@@ -30,6 +37,8 @@ impl CliExecuter for Clean {
             sysroot,
             apt_options,
             dry_run,
+            keep_downloadable,
+            keep_downloadable_and_installed,
         } = self;
 
         if dry_run {
@@ -54,9 +63,52 @@ impl CliExecuter for Clean {
 
         let pb = create_progress_spinner(no_progress, fl!("cleaning"));
 
-        for i in dir.flatten() {
-            if i.path().extension().and_then(|x| x.to_str()) == Some("deb") {
-                std::fs::remove_file(i.path()).ok();
+        remove_dir_all(download_dir.join("partial")).ok();
+
+        for i in dir
+            .flatten()
+            .filter(|x| x.path().extension().is_some_and(|name| name == "deb"))
+        {
+            if !keep_downloadable && !keep_downloadable_and_installed {
+                remove_deb(&i);
+            }
+
+            let file_name = i.file_name();
+            let file_name = file_name.to_string_lossy();
+            let mut file_name = file_name.splitn(3, '_');
+
+            let Some((pkg, version)) = Some(()).and_then(|_| {
+                let package = file_name.next()?;
+                let version = file_name.next()?;
+
+                Some((package, version))
+            }) else {
+                debug!(
+                    "Failed to get package name or version: {}, will delete this file",
+                    i.path().display()
+                );
+                remove_deb(&i);
+                continue;
+            };
+
+            let version = version.replace("%3a", ":");
+
+            let Some(version) = apt.cache.get(pkg).and_then(|pkg| pkg.get_version(&version)) else {
+                remove_deb(&i);
+                continue;
+            };
+
+            if !version.is_installed()
+                && !version.is_downloadable()
+                && keep_downloadable_and_installed
+            {
+                remove_deb(&i);
+                continue;
+            }
+
+            if !version.is_downloadable() && keep_downloadable {
+                remove_deb(&i);
+                continue;
             }
         }
 
@@ -69,3 +121,7 @@ impl CliExecuter for Clean {
         Ok(0)
     }
 }
+
+fn remove_deb(i: &std::fs::DirEntry) {
+    std::fs::remove_file(i.path()).ok();
+}