1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82

use crate::CACHEDIR;
use anyhow::{Context, Result};
use ijson::IString;
use serde::{Deserialize, Serialize};
use std::{
    collections::HashMap,
    fs::{self, File},
    io::Write,
    path::Path,
    process::Command,
};

use super::nixos::{self, getnixospkgs};

#[derive(Debug, Clone, Serialize, Deserialize)]
struct FlakePkg {
    pname: IString,
    version: IString,
}

/// Gets a list of all packages in the NixOS system with their name and version.
/// Can be used to find what versions of system packages are currently installed.
/// Will only work on NixOS systems.
pub fn flakespkgs() -> Result<String> {
    let versionout = Command::new("nixos-version").arg("--json").output()?;
    let version: HashMap<String, String> = serde_json::from_slice(&versionout.stdout)?;

    let nixosversion = version.get("nixosVersion").context("No NixOS version found")?;

    // If cache directory doesn't exist, create it
    if !std::path::Path::new(&*CACHEDIR).exists() {
        std::fs::create_dir_all(&*CACHEDIR)?;
    }

    // Check if latest version is already downloaded
    if let Ok(prevver) = fs::read_to_string(&format!("{}/flakespkgs.ver", &*CACHEDIR)) {
        if prevver.eq(nixosversion) && Path::new(&format!("{}/flakespkgs.json", &*CACHEDIR)).exists() {
            println!("No new version of NixOS flakes found");
            return Ok(format!("{}/flakespkgs.json", &*CACHEDIR));
        }
    }

    // Get list of packages from flake
    let pkgsout = if let Some(rev) = version.get("nixpkgsRevision") {
        Command::new("nix")
            .arg("search")
            .arg("--json")
            .arg(&format!("nixpkgs/{}", rev))
            .output()?
    } else {
        Command::new("nix")
            .arg("search")
            .arg("--json")
            // .arg("--inputs-from")
            // .arg(&flakepath)
            .arg("nixpkgs")
            .output()?
    };

    let mut pkgsjson: HashMap<IString, FlakePkg> =
        serde_json::from_str(&String::from_utf8(pkgsout.stdout)?)?;
    pkgsjson = pkgsjson
        .iter()
        .map(|(k, v)| (IString::from(k.split('.').collect::<Vec<_>>()[2..].join(".")), v.clone()))
        .collect::<HashMap<_, _>>();
    let outjson = serde_json::to_string(&pkgsjson)?;

    // Write json to cache file
    let mut cachefile = File::create(&format!("{}/flakespkgs.json", &*CACHEDIR))?;
    cachefile.write_all(outjson.as_bytes())?;

    // Write version downloaded to file
    File::create(format!("{}/flakespkgs.ver", &*CACHEDIR))?.write_all(nixosversion.as_bytes())?;

    Ok(format!("{}/flakespkgs.json", &*CACHEDIR))
}

/// Returns a list of all installed system packages with their attribute and version
/// The input `paths` should be the paths to the `configuration.nix` files containing `environment.systemPackages`
pub fn getflakepkgs(paths: &[&str]) -> Result<HashMap<String, String>> {
    getnixospkgs(paths, nixos::NixosType::Flake)
}