multimr 0.1.0

Create identical MR/PRs on multiple repositories.
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
117
118
119
120
121
122

//! Provides functionality to create merge requests using the `glab` CLI.
use std::{env, io, process};

use color_eyre::Result;

use super::utils;
use crate::config;

/// Represents a merge request to be created.
#[derive(Debug)]
pub struct MergeRequest {
    pub(crate) title: String,
    pub(crate) description: String,
    pub(crate) reviewers: Vec<String>,
    pub(crate) labels: Vec<String>,
    pub(crate) assignee: Option<String>,
}

impl MergeRequest {
    /// Construct a command to create a merge request for the cwd repo using the `glab` CLI.
    /// If the current branch is main or master, create a new branch
    pub(crate) fn create(&self) -> process::Command {
        let mut cmd = process::Command::new("glab");
        cmd.arg("mr").arg("create");

        if let Some(assignee) = &self.assignee {
            cmd.arg("--assignee").arg(assignee);
        }

        if !self.reviewers.is_empty() {
            for reviewer in &self.reviewers {
                cmd.arg("--reviewer").arg(reviewer);
            }
        }

        if !self.labels.is_empty() {
            for label in &self.labels {
                cmd.arg("--label").arg(label);
            }
        }

        let current_branch = utils::get_current_branch();

        cmd.arg("--title").arg(&self.title);
        cmd.arg("--description").arg(&self.description);

        if config::DEFAULT_BRANCHES.contains(&current_branch.as_str()) {
            // If the current branch is main or master, create a new branch

            println!();

            process::Command::new("git")
                .arg("switch")
                .arg("-c")
                .arg(self.title.replace(' ', "-"))
                .status()
                .expect("Failed to create new branch");

            println!();

            process::Command::new("git")
                .arg("add")
                .arg(".")
                .status()
                .expect("Failed to add changes");

            process::Command::new("git")
                .arg("commit")
                .arg("-am")
                .arg(&self.title)
                .status()
                .or_else(|_e| -> Result<process::ExitStatus, io::Error> {
                    // Retry once if adding and committing fails, this might happen if the pre-commit hook formats the code
                    // TODO: test this.
                    process::Command::new("git")
                        .arg("add")
                        .arg(".")
                        .status()
                        .expect("Failed to add changes Second attempt");

                    println!();

                    let status = process::Command::new("git")
                        .arg("commit")
                        .arg("-am")
                        .arg(&self.title)
                        .status()
                        .expect("Failed to commit changes second attempt");

                    Ok(status)
                })
                .expect("Failed to commit changes twice.");

            cmd.arg("--push");
        } else {
            // If not, just use the current branch
            cmd.arg("--yes");
        }

        cmd
    }

    /// Run the command to create the merge request.
    pub(crate) fn run(&self, mut cmd: process::Command) {
        let status = cmd.status().expect("Failed to execute command");
        if !status.success() {
            eprintln!("Failed to create merge request: {:?}", status);
        } else {
            println!("Merge request created successfully.");
        }
    }

    /// Print the command that would be run, useful for dry runs.
    pub(crate) fn dry_run(&self, cmd: process::Command) {
        println!(
            "Current directory: {}",
            env::current_dir().unwrap().display()
        );

        println!("Dry run command: {:?}", cmd);
    }
}