Add support for winget and chocolatey package counts on Windows

Cargo.toml

@@ -58,7 +58,7 @@ windows = { version = "0.39.0", features = [
 [target.'cfg(not(target_os = "windows"))'.dependencies]
 if-addrs = "0.10.2"
 
-[target.'cfg(any(target_os="freebsd", target_os = "linux"))'.dependencies]
+[target.'cfg(any(target_os="freebsd", target_os = "linux", target_os = "windows"))'.dependencies]
 sqlite = "0.36.0"
 
 [target.'cfg(any(target_os="freebsd", target_os = "netbsd"))'.dependencies]

src/traits.rs

@@ -668,6 +668,8 @@ pub enum PackageManager {
     Pkg,
     Scoop,
     Nix,
+    Winget,
+    Chocolatey,
 }
 
 impl std::fmt::Display for PackageManager {
@@ -691,6 +693,8 @@ impl std::fmt::Display for PackageManager {
             PackageManager::Pkg => write!(f, "pkg"),
             PackageManager::Scoop => write!(f, "Scoop"),
             PackageManager::Nix => write!(f, "nix"),
+            PackageManager::Winget => write!(f, "winget"),
+            PackageManager::Chocolatey => write!(f, "choco"),
         }
     }
 }

src/windows/mod.rs

@@ -1,5 +1,7 @@
 use crate::traits::*;
 use std::collections::HashMap;
+use std::env;
+use std::fs::read_dir;
 use std::path::{Path, PathBuf};
 use winreg::enums::*;
 use winreg::RegKey;
@@ -421,6 +423,12 @@ impl PackageReadout for WindowsPackageReadout {
         if let Some(c) = WindowsPackageReadout::count_scoop() {
             packages.push((PackageManager::Scoop, c));
         }
+        if let Some(c) = WindowsPackageReadout::count_winget() {
+            packages.push((PackageManager::Winget, c));
+        }
+        if let Some(c) = WindowsPackageReadout::count_chocolatey() {
+            packages.push((PackageManager::Chocolatey, c));
+        }
         packages
     }
 }
@@ -440,6 +448,38 @@ impl WindowsPackageReadout {
             _ => None,
         }
     }
+
+    fn count_winget() -> Option<usize> {
+        if let Ok(username) = env::var("USERNAME") {
+            let db = format!("C:\\Users\\{username}\\AppData\\Local\\Packages\\Microsoft.DesktopAppInstaller_8wekyb3d8bbwe\\LocalState\\Microsoft.Winget.Source_8wekyb3d8bbwe\\installed.db");
+            if !Path::new(&db).is_file() {
+                return None;
+            }
+            let connection = sqlite::open(db);
+            if let Ok(con) = connection {
+                let statement = con.prepare("SELECT COUNT(*) FROM ids");
+                if let Ok(mut s) = statement {
+                    if s.next().is_ok() {
+                        return match s.read::<Option<i64>, _>(0) {
+                            Ok(Some(count)) => Some(count as usize),
+                            _ => None,
+                        };
+                    }
+                }
+            }
+        }
+        None
+    }
+
+    fn count_chocolatey() -> Option<usize> {
+        let chocolatey_dir = Path::new("C:\\ProgramData\\chocolatey\\lib");
+        if chocolatey_dir.is_dir() {
+            if let Ok(read_dir) = read_dir(chocolatey_dir) {
+                return Some(read_dir.count());
+            }
+        }
+        None
+    }
 }
 
 pub struct WindowsNetworkReadout;