feat(oma-refresh): support custom Dir::Etc::SourceList

Author: eatradish

oma-refresh/src/db.rs

@@ -216,7 +216,19 @@ impl<'a> OmaRefresh<'a> {
         let paths = if let Some(ref paths) = self.sources_lists_paths {
             paths.to_vec()
         } else {
-            scan_sources_lists_paths_from_sysroot(&self.source)
+            #[cfg(feature = "apt")]
+            let list_file = self.apt_config.get("Dir::Etc::sourcelist").unwrap_or_else(|| "sources.list".to_string());
+
+            #[cfg(feature = "apt")]
+            let list_dir = self.apt_config.get("Dir::Etc::sourceparts").unwrap_or_else(|| "sources.list.d".to_string());
+
+            #[cfg(not(feature = "apt"))]
+            let list_file = "/etc/apt/sources.list";
+
+            #[cfg(not(feature = "apt"))]
+            let list_dir = "/etc/apt/sources.list.d";
+
+            scan_sources_lists_paths_from_sysroot(&self.source, list_file, list_dir)
                 .await
                 .map_err(RefreshError::ScanSourceError)?
         };

oma-refresh/src/sourceslist.rs

@@ -55,17 +55,32 @@ impl Debug for OmaSourceEntry<'_> {
 }
 
 pub(crate) async fn scan_sources_lists_paths_from_sysroot(
-    sysroot: impl AsRef<Path>,
+    dir: impl AsRef<Path>,
+    list_file: impl AsRef<str>,
+    list_dir: impl AsRef<str>,
 ) -> Result<Vec<PathBuf>, SourcesListError> {
     let mut paths = vec![];
-    let default = sysroot.as_ref().join("etc/apt/sources.list");
+    let default = Path::new(list_file.as_ref());
+    let list_dir = Path::new(list_dir.as_ref());
+
+    let default = if !default.is_absolute() {
+        dir.as_ref().join("etc/apt").join(default)
+    } else {
+        dir.as_ref().join(default)
+    };
 
     if default.exists() {
         paths.push(default);
     }
 
-    if sysroot.as_ref().join("etc/apt/sources.list.d/").exists() {
-        let mut dir = tokio::fs::read_dir(sysroot.as_ref().join("etc/apt/sources.list.d/")).await?;
+    let list_dir = if !list_dir.is_absolute() {
+        dir.as_ref().join("etc/apt").join(list_dir)
+    } else {
+        dir.as_ref().join(list_dir)
+    };
+
+    if list_dir.exists() {
+        let mut dir = tokio::fs::read_dir(list_dir).await?;
         while let Some(entry) = dir.next_entry().await? {
             let path = entry.path();
             if !path.is_file() {