lium 0.1.2

Abstraction Layer of ChromiumOS development
Documentation 
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

// Copyright 2023 The ChromiumOS Authors
//
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file or at
// https://developers.google.com/open-source/licenses/bsd

use std::process::Command;

use anyhow::bail;
use anyhow::Context;
use anyhow::Result;
use regex_macro::regex;
use tracing::info;

use crate::cache::KvCache;
use crate::google_storage;
use crate::util::shell_helpers::run_bash_command;

static VERSION_TO_MILESTONE_CACHE: KvCache<String> = KvCache::new("version_cache");

// TODO #83 create an enum to represent board that can be converted to string
// (adds some type safety)
pub fn lookup_full_version(input: &str, board: &str) -> Result<String> {
    let input = input.trim();
    let re_cros_version_without_milestone = regex!(r"^\d+\.\d+\.\d+$");
    let re_cros_version = regex!(r"/(R\d+\-\d+\.\d+\.\d+)/");
    let re_full_cros_version = regex!(r"(R\d+\-\d+\.\d+\.\d+)");
    if let Some(captures) = re_full_cros_version.captures(input) {
        let captures = captures.get(1).context("No match found")?;
        Ok(captures.as_str().to_string())
    } else if re_cros_version_without_milestone.is_match(input) {
        VERSION_TO_MILESTONE_CACHE.get_or_else(input, &|key| {
            let output = google_storage::list_gs_files(&format!(
                "gs://chromeos-image-archive/{}-release/R*-{}/chromiumos_test_image.tar.xz",
                board, key
            ))
            .context(
                "gsutil command failed (maybe you need depot_tools and/or `gsutil.py config` with \
                 'chromeos-swarming' project)",
            )?;
            let output = re_cros_version
                .captures(output.trim())
                .context("Invalid gsutil output")?;
            let output = output.get(1).context("No match found")?;
            Ok(output.as_str().to_string())
        })
    } else {
        bail!("Invalid version format: {}", input)
    }
}

pub fn ensure_testing_rsa_is_there() -> Result<()> {
    let cmd = "
if ! [ -f ~/.ssh/testing_rsa ]; then
    curl -s https://chromium.googlesource.com/chromiumos/chromite/+/master/ssh_keys/testing_rsa?format=TEXT | base64 --decode > ~/.ssh/testing_rsa
    chmod 600 ~/.ssh/testing_rsa
fi
";
    let output = run_bash_command(cmd, None)?;
    if output.status.success() {
        Ok(())
    } else {
        bail!("Downloading testing_rsa failed")
    }
}

pub fn setup_cros_repo(repo: &str, version: &str, reference: &Option<String>) -> Result<()> {
    let url = if version == "tot" {
        "https://chrome-internal.googlesource.com/chromeos/manifest-internal"
    } else {
        "https://chrome-internal.googlesource.com/chromeos/manifest-versions"
    };

    let mut cmd = Command::new("repo");
    cmd.current_dir(repo)
        .arg("init")
        .arg("--repo-url")
        .arg("https://chromium.googlesource.com/external/repo.git")
        .arg("-u")
        .arg(url)
        .arg("-b")
        .arg("main");

    if let Some(reference) = reference {
        info!("Using {reference} as a local mirror.");
        cmd.args(["--reference", reference]);
    }

    if version != "tot" {
        let re_cros_version = regex!(r"R(\d+)\-(\d+\.\d+\.\d+)");
        let output = re_cros_version
            .captures(version.trim())
            .context("Invalid cros version")?;
        let milestone = output.get(1).context("No match found")?.as_str();
        let version = output.get(2).context("No match found")?.as_str();
        cmd.arg("-m");
        cmd.arg(format!("buildspecs/{}/{}.xml", milestone, version));
    };

    println!("Running repo init with the given version...");
    let cld = cmd.spawn().context("Failed to execute repo init")?;
    cld.wait_with_output()
        .context("Failed to wait for repo init")?;
    Ok(())
}