mirror of
https://github.com/moghtech/komodo.git
synced 2026-04-29 04:10:01 -05:00
move some libraries out
This commit is contained in:
17
lib/git/Cargo.toml
Normal file
17
lib/git/Cargo.toml
Normal file
@@ -0,0 +1,17 @@
|
||||
[package]
|
||||
name = "git"
|
||||
version.workspace = true
|
||||
edition.workspace = true
|
||||
authors.workspace = true
|
||||
license.workspace = true
|
||||
repository.workspace = true
|
||||
homepage.workspace = true
|
||||
|
||||
[dependencies]
|
||||
monitor_client.workspace = true
|
||||
command.workspace = true
|
||||
#
|
||||
run_command.workspace = true
|
||||
#
|
||||
tracing.workspace = true
|
||||
anyhow.workspace = true
|
||||
261
lib/git/src/lib.rs
Normal file
261
lib/git/src/lib.rs
Normal file
@@ -0,0 +1,261 @@
|
||||
use std::path::Path;
|
||||
|
||||
use anyhow::Context;
|
||||
use command::run_monitor_command;
|
||||
use monitor_client::entities::{
|
||||
all_logs_success, monitor_timestamp, to_monitor_name, update::Log,
|
||||
CloneArgs, LatestCommit, SystemCommand,
|
||||
};
|
||||
use run_command::async_run_command;
|
||||
use tracing::instrument;
|
||||
|
||||
#[tracing::instrument]
|
||||
pub async fn pull(
|
||||
path: &Path,
|
||||
branch: &Option<String>,
|
||||
commit: &Option<String>,
|
||||
on_pull: &Option<SystemCommand>,
|
||||
) -> Vec<Log> {
|
||||
let branch = match branch {
|
||||
Some(branch) => branch.to_owned(),
|
||||
None => "main".to_string(),
|
||||
};
|
||||
|
||||
let command =
|
||||
format!("cd {} && git pull origin {branch}", path.display());
|
||||
|
||||
let pull_log = run_monitor_command("git pull", command).await;
|
||||
|
||||
let mut logs = vec![pull_log];
|
||||
|
||||
if !logs[0].success {
|
||||
return logs;
|
||||
}
|
||||
|
||||
if let Some(commit) = commit {
|
||||
let reset_log = run_monitor_command(
|
||||
"set commit",
|
||||
format!("cd {} && git reset --hard {commit}", path.display()),
|
||||
)
|
||||
.await;
|
||||
logs.push(reset_log);
|
||||
}
|
||||
|
||||
let commit_hash_log =
|
||||
get_commit_hash_log(path).await.unwrap_or(Log::simple(
|
||||
"latest commit",
|
||||
String::from("failed to get latest commit"),
|
||||
));
|
||||
logs.push(commit_hash_log);
|
||||
|
||||
if let Some(on_pull) = on_pull {
|
||||
if !on_pull.path.is_empty() && !on_pull.command.is_empty() {
|
||||
let path = path.join(&on_pull.path);
|
||||
let on_pull_log = run_monitor_command(
|
||||
"on pull",
|
||||
format!("cd {} && {}", path.display(), on_pull.command),
|
||||
)
|
||||
.await;
|
||||
logs.push(on_pull_log);
|
||||
}
|
||||
}
|
||||
|
||||
logs
|
||||
}
|
||||
|
||||
#[tracing::instrument(skip(repo_dir, github_token))]
|
||||
pub async fn clone<T>(
|
||||
clone_args: T,
|
||||
repo_dir: &Path,
|
||||
github_token: Option<String>,
|
||||
) -> anyhow::Result<Vec<Log>>
|
||||
where
|
||||
T: Into<CloneArgs> + std::fmt::Debug,
|
||||
{
|
||||
let CloneArgs {
|
||||
name,
|
||||
repo,
|
||||
branch,
|
||||
commit,
|
||||
on_clone,
|
||||
on_pull,
|
||||
..
|
||||
} = clone_args.into();
|
||||
|
||||
// let access_token =
|
||||
// match (github_token, get_github_token(&github_account)) {
|
||||
// (Some(token), _) => Some(token),
|
||||
// (None, Ok(token)) => token,
|
||||
// (None, Err(e)) => return Err(e),
|
||||
// };
|
||||
|
||||
let repo = repo.as_ref().context("build has no repo attached")?;
|
||||
let name = to_monitor_name(&name);
|
||||
|
||||
let repo_dir = repo_dir.join(name);
|
||||
|
||||
let mut logs =
|
||||
clone_inner(repo, &repo_dir, &branch, &commit, github_token)
|
||||
.await;
|
||||
|
||||
if !all_logs_success(&logs) {
|
||||
tracing::warn!("repo at {repo_dir:?} failed to clone");
|
||||
return Ok(logs);
|
||||
}
|
||||
|
||||
tracing::info!("repo at {repo_dir:?} cloned with clone_inner");
|
||||
|
||||
let commit_hash_log = get_commit_hash_log(&repo_dir).await?;
|
||||
logs.push(commit_hash_log);
|
||||
|
||||
if let Some(command) = on_clone {
|
||||
if !command.path.is_empty() && !command.command.is_empty() {
|
||||
let on_clone_path = repo_dir.join(&command.path);
|
||||
let on_clone_log = run_monitor_command(
|
||||
"on clone",
|
||||
format!(
|
||||
"cd {} && {}",
|
||||
on_clone_path.display(),
|
||||
command.command
|
||||
),
|
||||
)
|
||||
.await;
|
||||
tracing::info!(
|
||||
"run repo on_clone command | command: {} | cwd: {:?}",
|
||||
command.command,
|
||||
on_clone_path
|
||||
);
|
||||
logs.push(on_clone_log);
|
||||
}
|
||||
}
|
||||
if let Some(command) = on_pull {
|
||||
if !command.path.is_empty() && !command.command.is_empty() {
|
||||
let on_pull_path = repo_dir.join(&command.path);
|
||||
let on_pull_log = run_monitor_command(
|
||||
"on pull",
|
||||
format!(
|
||||
"cd {} && {}",
|
||||
on_pull_path.display(),
|
||||
command.command
|
||||
),
|
||||
)
|
||||
.await;
|
||||
tracing::info!(
|
||||
"run repo on_pull command | command: {} | cwd: {:?}",
|
||||
command.command,
|
||||
on_pull_path
|
||||
);
|
||||
logs.push(on_pull_log);
|
||||
}
|
||||
}
|
||||
Ok(logs)
|
||||
}
|
||||
|
||||
#[tracing::instrument(skip(destination, access_token))]
|
||||
async fn clone_inner(
|
||||
repo: &str,
|
||||
destination: &Path,
|
||||
branch: &Option<String>,
|
||||
commit: &Option<String>,
|
||||
access_token: Option<String>,
|
||||
) -> Vec<Log> {
|
||||
let _ = std::fs::remove_dir_all(destination);
|
||||
let access_token_at = match &access_token {
|
||||
Some(token) => format!("{token}@"),
|
||||
None => String::new(),
|
||||
};
|
||||
let branch = match branch {
|
||||
Some(branch) => format!(" -b {branch}"),
|
||||
None => String::new(),
|
||||
};
|
||||
let repo_url =
|
||||
format!("https://{access_token_at}github.com/{repo}.git");
|
||||
let command =
|
||||
format!("git clone {repo_url} {}{branch}", destination.display());
|
||||
let start_ts = monitor_timestamp();
|
||||
let output = async_run_command(&command).await;
|
||||
let success = output.success();
|
||||
let (command, stderr) = if !access_token_at.is_empty() {
|
||||
// know that access token can't be none if access token non-empty
|
||||
let access_token = access_token.unwrap();
|
||||
(
|
||||
command.replace(&access_token, "<TOKEN>"),
|
||||
output.stderr.replace(&access_token, "<TOKEN>"),
|
||||
)
|
||||
} else {
|
||||
(command, output.stderr)
|
||||
};
|
||||
let mut logs = vec![Log {
|
||||
stage: "clone repo".to_string(),
|
||||
command,
|
||||
success,
|
||||
stdout: output.stdout,
|
||||
stderr,
|
||||
start_ts,
|
||||
end_ts: monitor_timestamp(),
|
||||
}];
|
||||
|
||||
if !logs[0].success {
|
||||
return logs;
|
||||
}
|
||||
|
||||
if let Some(commit) = commit {
|
||||
let reset_log = run_monitor_command(
|
||||
"set commit",
|
||||
format!(
|
||||
"cd {} && git reset --hard {commit}",
|
||||
destination.display()
|
||||
),
|
||||
)
|
||||
.await;
|
||||
logs.push(reset_log);
|
||||
}
|
||||
|
||||
logs
|
||||
}
|
||||
|
||||
#[instrument(level = "debug")]
|
||||
pub async fn get_commit_hash_info(
|
||||
repo_dir: &Path,
|
||||
) -> anyhow::Result<LatestCommit> {
|
||||
let command = format!("cd {} && git rev-parse --short HEAD && git rev-parse HEAD && git log -1 --pretty=%B", repo_dir.display());
|
||||
let output = async_run_command(&command).await;
|
||||
let mut split = output.stdout.split('\n');
|
||||
let (hash, _, message) = (
|
||||
split
|
||||
.next()
|
||||
.context("failed to get short commit hash")?
|
||||
.to_string(),
|
||||
split.next().context("failed to get long commit hash")?,
|
||||
split
|
||||
.next()
|
||||
.context("failed to get commit message")?
|
||||
.to_string(),
|
||||
);
|
||||
Ok(LatestCommit { hash, message })
|
||||
}
|
||||
|
||||
#[instrument(level = "debug")]
|
||||
pub async fn get_commit_hash_log(
|
||||
repo_dir: &Path,
|
||||
) -> anyhow::Result<Log> {
|
||||
let start_ts = monitor_timestamp();
|
||||
let command = format!("cd {} && git rev-parse --short HEAD && git rev-parse HEAD && git log -1 --pretty=%B", repo_dir.display());
|
||||
let output = async_run_command(&command).await;
|
||||
let mut split = output.stdout.split('\n');
|
||||
let (short, _, msg) = (
|
||||
split.next().context("failed to get short commit hash")?,
|
||||
split.next().context("failed to get long commit hash")?,
|
||||
split.next().context("failed to get commit message")?,
|
||||
);
|
||||
let log = Log {
|
||||
stage: "latest commit".into(),
|
||||
command,
|
||||
stdout: format!("hash: {short}\nmessage: {msg}"),
|
||||
stderr: String::new(),
|
||||
success: true,
|
||||
start_ts,
|
||||
end_ts: monitor_timestamp(),
|
||||
};
|
||||
Ok(log)
|
||||
}
|
||||
Reference in New Issue
Block a user