From 9e088751790c79306f86125d74f4ff74ee5dd57f Mon Sep 17 00:00:00 2001
From: Ward from fusion-voyager-3 <hotrod.master@hotmail.com>
Date: Thu, 11 Jul 2024 02:53:33 +0300
Subject: [PATCH] apt upgrade

---
 .idea/workspace.xml                    | 10 +------
 src/bin/apt/apt_full_upgrade/main.rs   | 38 +++++++++++++++++++-------
 src/bin/gui/apt_update_page/process.rs |  1 +
 3 files changed, 30 insertions(+), 19 deletions(-)

diff --git a/.idea/workspace.xml b/.idea/workspace.xml
index abd33e3..1651f68 100644
--- a/.idea/workspace.xml
+++ b/.idea/workspace.xml
@@ -8,15 +8,7 @@
   </component>
   <component name="ChangeListManager">
     <list default="true" id="df2ca9e1-e07d-43f4-bc68-0a6113fc1fa2" name="Changes" comment="">
-      <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
       <change beforePath="$PROJECT_DIR$/src/bin/apt/apt_full_upgrade/main.rs" beforeDir="false" afterPath="$PROJECT_DIR$/src/bin/apt/apt_full_upgrade/main.rs" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/src/bin/apt/apt_update/main.rs" beforeDir="false" afterPath="$PROJECT_DIR$/src/bin/apt/apt_update/main.rs" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/src/bin/gui/apt_update_page/mod.rs" beforeDir="false" afterPath="$PROJECT_DIR$/src/bin/gui/apt_update_page/mod.rs" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/src/bin/gui/apt_update_page/process.rs" beforeDir="false" afterPath="$PROJECT_DIR$/src/bin/gui/apt_update_page/process.rs" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/src/lib/apt_install_progress_socket/mod.rs" beforeDir="false" afterPath="$PROJECT_DIR$/src/lib/apt_install_progress_socket/mod.rs" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/src/lib/apt_update_progress_socket/mod.rs" beforeDir="false" afterPath="$PROJECT_DIR$/src/lib/apt_update_progress_socket/mod.rs" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/src/lib/lib.rs" beforeDir="false" afterPath="$PROJECT_DIR$/src/lib/lib.rs" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/src/lib/pika_unixsocket_tools/mod.rs" beforeDir="false" afterPath="$PROJECT_DIR$/src/lib/pika_unixsocket_tools/mod.rs" afterDir="false" />
     </list>
     <option name="SHOW_DIALOG" value="false" />
     <option name="HIGHLIGHT_CONFLICTS" value="true" />
@@ -141,7 +133,7 @@
       <workItem from="1720556059466" duration="12649000" />
       <workItem from="1720594302708" duration="52000" />
       <workItem from="1720597532937" duration="7050000" />
-      <workItem from="1720648284536" duration="6532000" />
+      <workItem from="1720648284536" duration="7092000" />
     </task>
     <servers />
   </component>
diff --git a/src/bin/apt/apt_full_upgrade/main.rs b/src/bin/apt/apt_full_upgrade/main.rs
index 8d06cad..03c479b 100644
--- a/src/bin/apt/apt_full_upgrade/main.rs
+++ b/src/bin/apt/apt_full_upgrade/main.rs
@@ -5,13 +5,14 @@ use rust_apt::new_cache;
 use rust_apt::progress::{AcquireProgress, InstallProgress};
 use std::fs::*;
 use std::process::exit;
+use rust_apt::cache::Upgrade;
 use tokio::runtime::Runtime;
 
 fn main() {
-    let cache = new_cache!().unwrap();
     let percent_socket_path = "/tmp/pika_apt_upgrade_percent.sock";
     let status_socket_path = "/tmp/pika_apt_upgrade_status.sock";
     let json_file_path = "/tmp/pika-apt-exclusions.json";
+    let mut excluded_updates_vec = Vec::new();
 
     let json: serde_json::Value = serde_json::from_str(
         &std::fs::read_to_string(json_file_path).expect("Unable to read file"),
@@ -20,11 +21,32 @@ fn main() {
 
     if let serde_json::Value::Array(exclusions) = &json["exclusions"] {
         for exclusion in exclusions {
-            println!("{}", exclusion["package"])
+            excluded_updates_vec.push(exclusion["package"].as_str().unwrap());
         }
     }
 
-    let pkg = cache.get("neovim").unwrap();
+    let apt_cache = new_cache!().unwrap();
+    let apt_upgrade_cache = new_cache!().unwrap();
+
+    apt_cache.upgrade(Upgrade::FullUpgrade).unwrap();
+
+    for change in apt_cache.get_changes(false) {
+        if !excluded_updates_vec
+            .iter()
+            .any(|e| change.name().contains(e))
+        {
+            let pkg = apt_upgrade_cache.get(change.name()).unwrap();
+            if change.marked_upgrade() || change.marked_install() || change.marked_downgrade() {
+                pkg.mark_install(true, false);
+            } else if change.marked_delete() {
+                pkg.mark_delete(false);
+            }
+            pkg.protect();
+        }
+    }
+
+    apt_upgrade_cache.resolve(true).unwrap();
+
     let mut acquire_progress = AcquireProgress::new(AptUpdateProgressSocket::new(
         percent_socket_path,
         status_socket_path,
@@ -34,13 +56,9 @@ fn main() {
         status_socket_path,
     ));
 
-    pkg.mark_install(true, true);
-    pkg.protect();
-    cache.resolve(true).unwrap();
+    apt_upgrade_cache.resolve(true).unwrap();
 
-    exit(1);
-
-    match cache.get_archives(&mut acquire_progress) {
+    match apt_upgrade_cache.get_archives(&mut acquire_progress) {
         Ok(_) => {
             Runtime::new()
                 .unwrap()
@@ -60,7 +78,7 @@ fn main() {
         }
     };
 
-    match cache.do_install(&mut install_progress) {
+    match apt_upgrade_cache.do_install(&mut install_progress) {
         Ok(_) => {
             Runtime::new()
                 .unwrap()
diff --git a/src/bin/gui/apt_update_page/process.rs b/src/bin/gui/apt_update_page/process.rs
index d0e8c82..62ea3aa 100644
--- a/src/bin/gui/apt_update_page/process.rs
+++ b/src/bin/gui/apt_update_page/process.rs
@@ -123,6 +123,7 @@ fn apt_confirm_window(excluded_updates_vec: &Vec<String>, window: adw::Applicati
             } else if change.marked_delete() {
                 pkg.mark_delete(false);
             }
+            pkg.protect();
         }
     }