kubie 0.15.3

A more powerful alternative to kubectx and kubens.
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
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116

use std::env;
use std::fs;
use std::fs::Permissions;
use std::os::unix::fs::PermissionsExt;
use std::path::Path;

use anyhow::{Context, Result};
use cfg_if::cfg_if;
use serde::Deserialize;

const VERSION: &str = env!("CARGO_PKG_VERSION");
const LATEST_RELEASE_URL: &str = "https://api.github.com/repos/sbstp/kubie/releases/latest";

#[derive(Debug, Deserialize)]
pub struct Release {
    tag_name: String,
    prerelease: bool,
    assets: Vec<Asset>,
}

impl Release {
    pub fn get_latest() -> Result<Release> {
        let latest_release = attohttpc::get(&LATEST_RELEASE_URL).send()?.json()?;
        Ok(latest_release)
    }

    // Get the right binary name based on which OS and architecture kubie was built-on.
    // Names match the GitHub releases: https://github.com/sbstp/kubie/releases
    fn get_binary_name() -> Option<&'static str> {
        cfg_if! {
            if #[cfg(all(target_os = "linux", target_arch = "x86_64"))] {
                Some("kubie-linux-amd64")
            } else if #[cfg(all(target_os = "linux", target_arch = "arm"))] {
                Some("kubie-linux-arm32")
            } else if #[cfg(all(target_os = "linux", target_arch = "aarch64"))] {
                Some("kubie-linux-arm64")
            } else if #[cfg(all(target_os = "macos", target_arch = "x86_64"))] {
                Some("kubie-darwin-amd64")
            } else if #[cfg(all(target_os = "macos", target_arch = "aarch64"))] {
                Some("kubie-darwin-arm64")
            } else {
                None
            }
        }
    }

    pub fn get_binary_url(&self) -> Option<&str> {
        let binary_name = Self::get_binary_name()?;

        for asset in self.assets.iter() {
            if asset.browser_download_url.contains(binary_name) {
                return Some(&asset.browser_download_url);
            }
        }
        None
    }
}

#[derive(Debug, Deserialize)]
struct Asset {
    name: String,
    browser_download_url: String,
    state: String,
}

pub fn update() -> Result<()> {
    let latest_release = Release::get_latest()?;
    if latest_release.tag_name == format!("v{}", VERSION) {
        println!("Kubie is up-to-date : v{}", VERSION);
    } else {
        println!(
            "A new version of Kubie is available ({}), the new version will be installed by replacing this binary.",
            latest_release.tag_name
        );

        let download_url = latest_release.get_binary_url().context("Sorry, this release has no build for your OS, please create an issue : https://github.com/sbstp/kubie/issues")?;
        println!("Download url is: {}", download_url);

        let resp = attohttpc::get(download_url).send()?;
        if resp.is_success() {
            let temp_file = tempfile::Builder::new().prefix("kubie").tempfile()?;
            resp.write_to(&temp_file)?;

            let old_file = env::current_exe().expect("Could not get own binary path");
            replace_file(&old_file, temp_file.path()).context("Update failed. Consider using sudo?")?;

            println!("Kubie has been updated successfully: {}", Path::display(&old_file));
        }
    }
    Ok(())
}

pub fn replace_file(old_file: &Path, new_file: &Path) -> std::io::Result<()> {
    fs::set_permissions(new_file, Permissions::from_mode(0o755))?;
    fs::remove_file(old_file)?;
    fs::copy(&new_file, old_file)?;
    Ok(())
}

#[test]
#[cfg(all(target_os = "linux", target_arch = "x86_64"))]
fn test_binary_name() {
    assert_eq!(Release::get_binary_name(), Some("kubie-linux-amd64"))
}

#[test]
#[cfg(all(target_os = "macos", target_arch = "x86_64"))]
fn test_binary_name() {
    assert_eq!(Release::get_binary_name(), Some("kubie-darwin-amd64"))
}

#[test]
#[cfg(all(target_os = "macos", target_arch = "aarch64"))]
fn test_binary_name() {
    assert_eq!(Release::get_binary_name(), Some("kubie-darwin-arm64"))
}