From 9bc8e6caaa194e5d91aab944a417f0ea2cb3b4fd Mon Sep 17 00:00:00 2001
From: Bo Maryniuk <bo@maryniuk.net>
Date: Wed, 13 Sep 2023 18:26:09 +0200
Subject: [PATCH] Define paths to keep and to prune

---
 src/procdata.rs | 10 ++++++++++
 src/profile.rs  | 40 ++++++++++++++++++++++++++++++++++++----
 2 files changed, 46 insertions(+), 4 deletions(-)

diff --git a/src/procdata.rs b/src/procdata.rs
index 03c0d53..20ae3fb 100644
--- a/src/procdata.rs
+++ b/src/procdata.rs
@@ -84,6 +84,16 @@ impl TintProcessor {
             paths.extend(databuf);
         }
 
+        // Explicitly keep paths
+        // XXX: Support globbing
+        paths.extend(self.profile.get_keep_paths());
+
+        // Explicitly knock-out paths
+        // XXX: Support globbing
+        for p in self.profile.get_prune_paths() {
+            paths.remove(&p);
+        }
+
         // Scan rootfs
         log::debug!("Scanning existing rootfs");
         let mut p = rootfs::RootFS::new()
diff --git a/src/profile.rs b/src/profile.rs
index 5441598..3ac5fd8 100644
--- a/src/profile.rs
+++ b/src/profile.rs
@@ -1,11 +1,12 @@
 use serde::{Deserialize, Serialize};
-use std::fmt::Debug;
+use std::{fmt::Debug, path::PathBuf};
 use std::{fs, io::Error, path::Path};
 
 #[derive(Serialize, Deserialize, PartialEq, Debug)]
 pub struct PConfig {
     filters: Option<Vec<String>>,
     prune: Option<Vec<String>>,
+    keep: Option<Vec<String>>,
 }
 
 #[derive(Serialize, Deserialize, PartialEq, Debug)]
@@ -24,7 +25,8 @@ pub struct Profile {
     f_man: bool,
     f_dir: bool,
     f_log: bool,
-    f_prune: Vec<String>,
+    f_expl_prune: Vec<PathBuf>,
+    f_expl_keep: Vec<PathBuf>,
 
     packages: Vec<String>,
     targets: Vec<String>,
@@ -43,7 +45,8 @@ impl Profile {
             f_log: true,
             packages: vec![],
             targets: vec![],
-            f_prune: vec![],
+            f_expl_prune: vec![],
+            f_expl_keep: vec![],
         }
     }
 
@@ -77,8 +80,13 @@ impl Profile {
                     }
                 }
             }
+
             if let Some(prn) = cfg.prune {
-                self.f_prune.extend(prn);
+                self.f_expl_prune.extend(prn.iter().map(PathBuf::from).collect::<Vec<PathBuf>>());
+            }
+
+            if let Some(keep) = cfg.keep {
+                self.f_expl_keep.extend(keep.iter().map(PathBuf::from).collect::<Vec<PathBuf>>());
             }
         }
 
@@ -135,11 +143,35 @@ impl Profile {
         self
     }
 
+    /// Add path prune
+    #[allow(dead_code)]
+    pub fn prune_path(&mut self, pth: String) -> &mut Self {
+        self.f_expl_prune.push(PathBuf::from(pth));
+        self
+    }
+
+    /// Add path to be kept
+    #[allow(dead_code)]
+    pub fn keep_path(&mut self, pth: String) -> &mut Self {
+        self.f_expl_keep.push(PathBuf::from(pth));
+        self
+    }
+
     /// Get targets
     pub fn get_targets(&self) -> &Vec<String> {
         &self.targets
     }
 
+    /// Get paths to be explicitly pruned
+    pub fn get_prune_paths(&self) -> Vec<PathBuf> {
+        self.f_expl_prune.clone()
+    }
+
+    /// Get paths to be explicitly kept
+    pub fn get_keep_paths(&self) -> Vec<PathBuf> {
+        self.f_expl_keep.clone()
+    }
+
     /// Returns true if localisation data needs to be removed
     pub fn filter_l10n(&self) -> bool {
         !self.f_l10n