upstream-rs 0.5.5

Fetch package updates directly from the source.
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
83
84
85
86
87
88
89
90

use anyhow::{Context, Result};
use std::fs;
use std::path::Path;

pub struct ShellManager<'a> {
    paths_file: &'a Path,
    symlinks_dir: &'a Path,
}

impl<'a> ShellManager<'a> {
    pub fn new(paths_file: &'a Path, symlinks_dir: &'a Path) -> Self {
        Self {
            paths_file,
            symlinks_dir,
        }
    }

    /// Creates or updates the `path.sh` file with the symlinks directory
    pub fn initialize(&self) -> Result<()> {
        const HEADER: &str = "#!/bin/bash\n# Upstream managed PATH additions\n";

        let path_line = format!(
            "export PATH=\"{}:$PATH\"",
            self.symlinks_dir.to_string_lossy()
        );

        if !self.paths_file.exists() {
            let content = format!("{HEADER}{path_line}\n");
            fs::write(self.paths_file, content).context("Failed to create paths file")?;
            return Ok(());
        }

        // Ensure the line exists exactly once
        let existing = fs::read_to_string(self.paths_file).context("Failed to read paths file")?;

        if !existing.contains(&path_line) {
            let updated = format!("{existing}{path_line}\n");
            fs::write(self.paths_file, updated).context("Failed to update paths file")?;
        }

        Ok(())
    }

    /// Adds a package's installation path to PATH by appending an export line
    pub fn add_to_paths(&self, install_path: &Path) -> Result<()> {
        if !install_path.is_dir() {
            anyhow::bail!(
                "Package install directory not found: {}",
                install_path.to_string_lossy()
            );
        }

        let mut content =
            fs::read_to_string(self.paths_file).context("Failed to read paths file")?;

        let escaped = install_path
            .to_string_lossy()
            .replace('$', "\\$")
            .replace('"', "\\\"");

        let export_line = format!("export PATH=\"{escaped}:$PATH\"");

        if !content.contains(&export_line) {
            content.push_str(&format!("{export_line}\n"));
            fs::write(self.paths_file, &content).context("Failed to write paths file")?;
        }

        Ok(())
    }

    /// Removes a package's PATH entry from `path.sh`
    pub fn remove_from_paths(&self, install_path: &Path) -> Result<()> {
        let mut content =
            fs::read_to_string(self.paths_file).context("Failed to read paths file")?;

        let escaped = install_path
            .to_string_lossy()
            .replace('$', "\\$")
            .replace('"', "\\\"");

        let export_line = format!("export PATH=\"{escaped}:$PATH\"");

        content = content.replace(&format!("{export_line}\n"), "");
        content = content.replace(&export_line, "");

        fs::write(self.paths_file, content).context("Failed to write paths file")?;

        Ok(())
    }
}