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

// 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 anyhow::Context;
use anyhow::Result;
use argh::FromArgs;
use lazy_static::lazy_static;
use lium::chroot::Chroot;
use lium::repo::get_repo_dir;
use regex::Regex;

lazy_static! {
    static ref RE_GERRIT_CL: Regex = Regex::new(r"^(?P<cl>[0-9]+)/(?P<patchset>[0-9+])$").unwrap();
}

#[derive(FromArgs, PartialEq, Debug)]
/// manage CL (Change List)
#[argh(subcommand, name = "cl")]
pub struct Args {
    #[argh(subcommand)]
    nested: SubCommand,
}
#[derive(FromArgs, PartialEq, Debug)]
#[argh(subcommand)]
enum SubCommand {
    Pick(ArgsPick),
}
#[tracing::instrument(level = "trace")]
pub fn run(args: &Args) -> Result<()> {
    match &args.nested {
        SubCommand::Pick(args) => run_pick(args),
    }
}

#[derive(FromArgs, PartialEq, Debug)]
/// cherry-pick a CL
#[argh(subcommand, name = "pick")]
pub struct ArgsPick {
    /// target cros repo dir
    #[argh(option)]
    repo: Option<String>,

    /// dir to run git commands, relative to cros checkout (e.g.
    /// src/platform/crosvm)
    #[argh(option)]
    dir: String,

    /// CLs to checkout (e.g. "4196467", "4196467/2")
    #[argh(positional)]
    cl: String,
}
fn run_pick(args: &ArgsPick) -> Result<()> {
    let capture = RE_GERRIT_CL
        .captures(&args.cl)
        .context("Invalid CL id. please specify patchset number as well (like '1234/5').")?;
    let cl = &capture["cl"];
    let cl_suffix = &cl[cl.len() - 2..];
    let patchset = &capture["patchset"];
    let dir = &args.dir;
    let chroot = Chroot::new(&get_repo_dir(&args.repo)?)?;
    chroot.run_bash_script_in_chroot(
        "checkout",
        &format!(
            r###"
cd ~/chromiumos
cd {dir}
export PROJ=`repo info . | grep -e 'Project:' | cut -d ' ' -f 2`
echo "PROJ=${{PROJ}}"
git fetch https://chromium.googlesource.com/${{PROJ}} \
  refs/changes/{cl_suffix}/{cl}/{patchset}
git cherry-pick FETCH_HEAD || git cherry-pick --abort
"###,
        ),
        None,
    )?;
    Ok(())
}

#[cfg(test)]
mod tests {
    use super::*;
    #[test]
    fn regex() {
        assert!(&RE_GERRIT_CL.captures("1234").is_none());
        assert_eq!(&RE_GERRIT_CL.captures("1234/5").unwrap()["cl"], "1234");
        assert_eq!(&RE_GERRIT_CL.captures("1234/5").unwrap()["patchset"], "5");
    }
}