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

use std::{path::Path, process::ExitStatus, sync::mpsc::Sender};

use crate::{exec, pkgx::Pkgx, Extension};
use anyhow::Error;
use fluentci_types::Output;

#[derive(Default)]
pub struct Git {}

impl Git {
    pub fn validate_url(&self, url: &str) -> Result<(), Error> {
        if url.is_empty() {
            return Err(Error::msg("URL is empty"));
        }
        if !regex::Regex::new(
            r"^(?:https:\/\/([^\/]+)\/([^\/]+)\/([^\/]+)|git@([^:]+):([^\/]+)\/([^\/]+))$",
        )
        .unwrap()
        .is_match(url)
        {
            return Err(Error::msg("Invalid URL"));
        }
        Ok(())
    }
}

impl Extension for Git {
    fn exec(
        &self,
        url: &str,
        tx: Sender<String>,
        out: Output,
        last_cmd: bool,
        work_dir: &str,
    ) -> Result<ExitStatus, Error> {
        self.setup()?;

        if self.validate_url(url).is_err() {
            return Err(Error::msg("Invalid URL"));
        }

        let repo = url.split('/').last().unwrap().replace(".git", "");
        let git_dir = format!("{}/{}/.git", work_dir, repo);
        if Path::new(&git_dir).exists() {
            let cmd = format!("git pull {}", url);
            let work_dir = format!("{}/{}", work_dir, repo);
            return exec(&cmd, tx, out, last_cmd, &work_dir);
        }

        let cmd = format!("git clone {}", url);
        exec(&cmd, tx, out, last_cmd, work_dir)
    }

    fn setup(&self) -> Result<(), Error> {
        Pkgx::default().install(vec!["git"])?;
        Ok(())
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_validate_url() {
        let git = Git::default();
        assert!(git.validate_url("https://github.com/tsirysndr/me").is_ok());
        assert!(git.validate_url("git@github.com:tsirysndr/me").is_ok());
        assert!(git.validate_url("github.com:tsirysndr/me").is_err());
    }
}