rpk 0.2.2

A lightweight, cross-platform cli package manager.
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
91
92
93
94

use std::{collections::BTreeMap, fs, path::PathBuf, str};

use anyhow::{Context as _, Result};
use serde::{Deserialize, Serialize};
use url::Url;

use crate::{
    context::Context,
    util::{load_toml, not_found_err},
};

use super::Source;

#[derive(Debug, Serialize, Deserialize)]
#[serde(rename_all = "lowercase")]
pub struct LockedConfig {
    #[serde(flatten)]
    pub ctx: Context,

    #[serde(default)]
    pub pkgs: BTreeMap<String, LockedPackage>,
}

#[derive(Debug, Serialize, Deserialize, Clone, PartialEq, Eq)]
#[serde(tag = "source")]
#[serde(rename_all = "lowercase")]
pub struct LockedPackage {
    #[serde(skip)]
    pub name:         String,
    #[serde(default)]
    pub bins:         Vec<String>,
    pub version:      String,
    #[serde(flatten)]
    pub source:       Source,
    pub desc:         Option<String>,
    pub filename:     String,
    pub download_url: Url,
}

impl LockedConfig {
    pub fn new(ctx: Context, pkgs: BTreeMap<String, LockedPackage>) -> Self {
        Self { ctx, pkgs }
    }

    pub fn load(ctx: &Context) -> Result<Self> {
        let mut lcfg = match load_toml(&ctx.lock_file) {
            Err(e) if not_found_err(e.root_cause()) =>
                LockedConfig::new(ctx.clone(), Default::default()),
            lcfg => lcfg.with_context(|| format!("failed to load {}", ctx.lock_file.display()))?,
        };

        lcfg.ctx = ctx.clone();

        // Set the package names for convenience.
        for (name, lpkg) in lcfg.pkgs.iter_mut() {
            lpkg.name = name.clone();
            if lpkg.bins.is_empty() {
                lpkg.bins.push(name.clone());
            }
        }
        Ok(lcfg)
    }

    /// Write this `LockedConfig` to the given path.
    pub fn save(&self) -> Result<()> {
        let buf = toml::to_string_pretty(self).context("failed to serialize `LockedConfig`")?;
        fs::write(&self.ctx.lock_file, buf)
            .with_context(|| format!("failed to save {}", self.ctx.lock_file.display()))
    }

    /// Update a package in the configuration. If the package does not exist, add it.
    pub fn upsert(&mut self, lpkg: LockedPackage) {
        assert!(
            !lpkg.name.is_empty(),
            "locked package must have at least one binary"
        );
        self.pkgs.insert(lpkg.name.clone(), lpkg);
    }
}

impl LockedPackage {
    /// Get the asset path for this package.
    pub fn asset_path(&self, ctx: &Context) -> PathBuf {
        ctx.cache_dir
            .join(&self.name)
            .join(&self.version)
            .join(&self.filename)
    }

    /// Get the install directory for this package.
    pub fn install_dir(&self, ctx: &Context) -> PathBuf {
        ctx.data_dir.join(&self.name).join(&self.version)
    }
}