shep 0.24.2

use std::path::{Path, PathBuf};

use super::*;

fn archive_probe() -> Probe {
    Probe::default()
}

fn channel_of(exe: &str) -> Channel {
    Channel::of(Path::new(exe), &archive_probe())
}

#[test]
fn a_cellar_path_is_homebrew_on_both_prefixes() {
    for exe in [
        "/opt/homebrew/Cellar/shep/0.24.0/bin/shep",
        "/usr/local/Cellar/shep/0.24.0/bin/shep",
        "/home/linuxbrew/.linuxbrew/Cellar/shep/0.24.0/bin/shep",
    ] {
        assert_eq!(channel_of(exe), Channel::Homebrew, "{exe}");
    }
}

#[test]
fn a_binary_in_cargo_homes_bin_is_cargo_and_cargo_home_moves_it() {
    let default = Probe {
        cargo_home: Some(PathBuf::from("/home/ann/.cargo")),
        ..Probe::default()
    };
    let moved = Probe {
        cargo_home: Some(PathBuf::from("/opt/cargo")),
        ..Probe::default()
    };

    let in_default = Path::new("/home/ann/.cargo/bin/shep");
    let in_moved = Path::new("/opt/cargo/bin/shep");
    assert_eq!(Channel::of(in_default, &default), Channel::Cargo);
    assert_eq!(Channel::of(in_moved, &moved), Channel::Cargo);
    assert_eq!(Channel::of(in_default, &moved), Channel::Archive);
    assert_eq!(Channel::of(in_moved, &default), Channel::Archive);
    assert_eq!(
        Channel::of(Path::new("/home/ann/.cargo/registry/shep"), &default),
        Channel::Archive,
        "only the bin directory itself counts"
    );
}

#[test]
fn a_scoop_apps_path_is_scoop_whatever_the_case_or_separator() {
    for exe in [
        r"C:\Users\Ann\scoop\apps\shep\current\shep.exe",
        r"c:\users\ann\SCOOP\APPS\SHEP\0.24.0\shep.exe",
        "C:/Users/Ann/scoop/apps/shep/current/shep.exe",
    ] {
        assert_eq!(channel_of(exe), Channel::Scoop, "{exe}");
    }
}

#[test]
fn a_scoop_root_from_the_environment_is_scoop() {
    let probe = Probe {
        scoop_roots: vec![PathBuf::from(r"D:\Tools\Scoop"), PathBuf::from("E:/global")],
        ..Probe::default()
    };

    for exe in [
        r"D:\Tools\Scoop\apps\shep\current\shep.exe",
        r"E:\Global\apps\shep\0.24.0\shep.exe",
    ] {
        assert_eq!(Channel::of(Path::new(exe), &probe), Channel::Scoop, "{exe}");
    }
    assert_eq!(
        Channel::of(Path::new(r"F:\Tools\apps\shep\shep.exe"), &probe),
        Channel::Archive,
        "apps/shep counts only under a named root"
    );
}

#[test]
fn a_winget_packages_path_is_winget() {
    for exe in [
        r"C:\Users\Ann\AppData\Local\Microsoft\WinGet\Packages\shep-pm.shep_Microsoft.Winget.Source_8wekyb3d8bbwe\shep.exe",
        r"c:\program files\winget\packages\shep-pm.shep_x\shep.exe",
    ] {
        assert_eq!(channel_of(exe), Channel::Winget, "{exe}");
    }
    assert_eq!(
        channel_of(r"C:\Users\Ann\Packages\WinGet\shep.exe"),
        Channel::Archive,
        "Packages has to come after WinGet"
    );
}

#[test]
fn a_chocolatey_lib_path_is_chocolatey() {
    let probe = Probe {
        chocolatey_root: Some(PathBuf::from(r"D:\Choco")),
        ..Probe::default()
    };

    assert_eq!(
        channel_of(r"C:\ProgramData\chocolatey\lib\shep\tools\shep.exe"),
        Channel::Chocolatey
    );
    assert_eq!(
        Channel::of(Path::new(r"D:\choco\lib\shep\tools\shep.exe"), &probe),
        Channel::Chocolatey
    );
    assert_eq!(
        Channel::of(Path::new(r"C:\tools\lib\shep\shep.exe"), &probe),
        Channel::Archive
    );
}

fn dpkg_info_with(list: &str, body: &str) -> (tempfile::TempDir, Probe) {
    let dir = tempfile::tempdir().unwrap();
    std::fs::write(dir.path().join(list), body).unwrap();
    let probe = Probe {
        dpkg_info: dir.path().to_path_buf(),
        ..Probe::default()
    };
    (dir, probe)
}

#[test]
fn usr_bin_shep_named_by_dpkg_is_deb_and_unnamed_is_archive() {
    let exe = Path::new("/usr/bin/shep");
    let (_named, named) = dpkg_info_with("shep.list", "/.\n/usr\n/usr/bin\n/usr/bin/shep\n");
    let (_other, other) = dpkg_info_with("shep.list", "/usr/bin/shepherd\n");
    let (_elsewhere, elsewhere) = dpkg_info_with("coreutils.list", "/usr/bin/shep\n");

    assert_eq!(Channel::of(exe, &named), Channel::Deb);
    assert_eq!(Channel::of(exe, &other), Channel::Archive);
    assert_eq!(
        Channel::of(exe, &elsewhere),
        Channel::Archive,
        "only a list of the shep package counts"
    );
    assert_eq!(
        Channel::of(exe, &Probe::default()),
        Channel::Archive,
        "no dpkg directory is no package"
    );
    assert_eq!(
        Channel::of(Path::new("/usr/local/bin/shep"), &named),
        Channel::Archive,
        "dpkg names /usr/bin/shep, not this binary"
    );
}

#[test]
fn dpkg_lists_for_shep_and_for_shep_with_an_arch_are_both_read() {
    // Names are checked as text: Windows cannot create a file called
    // `shep:amd64.list`, so only the unix half below touches the disk.
    for name in ["shep.list", "shep:amd64.list", "shep:arm64.list"] {
        assert!(is_shep_list(name), "{name}");
    }
    for name in [
        "shepherd.list",
        "shepherd:amd64.list",
        "shep:.list",
        "shep.md5sums",
        "xshep.list",
    ] {
        assert!(!is_shep_list(name), "{name}");
    }

    #[cfg(unix)]
    {
        let exe = Path::new("/usr/bin/shep");
        let (_arch, arch) = dpkg_info_with("shep:amd64.list", "/usr/bin/shep\n");
        let (_other, other) = dpkg_info_with("shepherd:amd64.list", "/usr/bin/shep\n");

        assert_eq!(Channel::of(exe, &arch), Channel::Deb);
        assert_eq!(Channel::of(exe, &other), Channel::Archive);
    }
}

#[test]
fn a_verbatim_windows_prefix_does_not_hide_the_channel() {
    assert_eq!(
        channel_of(r"\\?\C:\Users\Ann\scoop\apps\shep\current\shep.exe"),
        Channel::Scoop
    );
    assert_eq!(
        channel_of(r"\\?\C:\ProgramData\chocolatey\lib\shep\tools\shep.exe"),
        Channel::Chocolatey
    );
    let probe = Probe {
        cargo_home: Some(PathBuf::from(r"C:\Users\Ann\.cargo")),
        ..Probe::default()
    };
    assert_eq!(
        Channel::of(Path::new(r"\\?\C:\Users\Ann\.cargo\bin\shep.exe"), &probe),
        Channel::Cargo
    );
}

#[test]
fn a_near_miss_is_not_a_match() {
    for exe in [
        "/opt/homebrew/Cellar/shep-other/1.0/bin/shep",
        "/opt/homebrew/Cellar/shepherd/1.0/bin/shep",
        r"C:\Users\Ann\scoop\apps\shepherd\current\shep.exe",
        r"C:\Users\Ann\scoop\apps\shep-other\current\shep.exe",
        r"C:\ProgramData\chocolatey\lib\shepherd\tools\shep.exe",
        r"C:\Users\Ann\AppData\Local\Microsoft\WinGet\Links\shep.exe",
        r"C:\Users\Ann\AppData\Local\Microsoft\WinGetPackages\shep.exe",
    ] {
        assert_eq!(channel_of(exe), Channel::Archive, "{exe}");
    }
}

#[test]
fn each_manager_names_its_command_and_deb_and_archive_name_none() {
    assert_eq!(
        Channel::Homebrew.command().as_deref(),
        Some("brew upgrade shep")
    );
    assert_eq!(
        Channel::Cargo.command().as_deref(),
        Some("cargo install --locked shep")
    );
    assert_eq!(
        Channel::Scoop.command().as_deref(),
        Some("scoop update shep")
    );
    assert_eq!(
        Channel::Winget.command().as_deref(),
        Some("winget upgrade shep-pm.shep")
    );
    assert_eq!(
        Channel::Chocolatey.command().as_deref(),
        Some("choco upgrade shep")
    );
    assert_eq!(Channel::Deb.command().as_deref(), None);
    assert_eq!(Channel::Archive.command().as_deref(), None);
}

/// `running()` resolves the binary's path, so the roots it is compared with
/// have to be resolved too: `~/.cargo` is often a symlink.
#[cfg(unix)]
#[test]
fn a_symlinked_root_matches_the_binary_resolved_through_it() {
    let dir = tempfile::tempdir().unwrap();
    let real = dir.path().join("data");
    std::fs::create_dir_all(real.join("cargo/bin")).unwrap();
    std::fs::create_dir_all(real.join("scoop/apps/shep")).unwrap();
    std::fs::create_dir_all(real.join("choco/lib/shep")).unwrap();
    std::os::unix::fs::symlink(real.join("cargo"), dir.path().join("dot-cargo")).unwrap();
    std::os::unix::fs::symlink(real.join("scoop"), dir.path().join("scoop-link")).unwrap();
    std::os::unix::fs::symlink(real.join("choco"), dir.path().join("choco-link")).unwrap();
    let probe = Probe {
        cargo_home: Some(dir.path().join("dot-cargo")),
        scoop_roots: vec![dir.path().join("scoop-link")],
        chocolatey_root: Some(dir.path().join("choco-link")),
        ..Probe::default()
    }
    .resolved();

    for (exe, channel) in [
        ("cargo/bin/shep", Channel::Cargo),
        ("scoop/apps/shep/shep", Channel::Scoop),
        ("choco/lib/shep/shep", Channel::Chocolatey),
    ] {
        let exe = real.join(exe);
        std::fs::write(&exe, "").unwrap();
        let exe = exe.canonicalize().unwrap();
        assert_eq!(Channel::of(&exe, &probe), channel, "{}", exe.display());
    }
}

#[test]
fn the_manual_steps_name_the_release_page_the_checksum_and_the_install_command() {
    let deb = Channel::Deb
        .manual_steps("x86_64-unknown-linux-musl")
        .join("\n");
    assert!(
        deb.contains("https://github.com/shep-pm/shep/releases/latest"),
        "{deb}"
    );
    assert!(deb.contains("SHA256SUMS"), "{deb}");
    assert!(deb.contains("sudo dpkg -i"), "{deb}");

    let archive = Channel::Archive
        .manual_steps("aarch64-apple-darwin")
        .join("\n");
    assert!(
        archive.contains(
            "https://github.com/shep-pm/shep/releases/latest/download/shep-aarch64-apple-darwin.tar.gz"
        ),
        "{archive}"
    );
    assert!(
        archive.contains("https://github.com/shep-pm/shep/releases/latest/download/SHA256SUMS"),
        "{archive}"
    );
    assert!(!archive.contains("dpkg"), "{archive}");

    let windows = Channel::Archive
        .manual_steps("x86_64-pc-windows-msvc")
        .join("\n");
    assert!(
        windows.contains("shep-x86_64-pc-windows-msvc.zip"),
        "{windows}"
    );

    assert!(
        Channel::Homebrew
            .manual_steps("x86_64-apple-darwin")
            .is_empty()
    );
}