Archives
/
cargo-remote
mirror of https://github.com/sgeisler/cargo-remote
2
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

refactor out hack to make structopt work with cargo subcommand

Browse Source
pull/1/head
Sebastian Geisler 6 years ago
parent da0142361c
commit 7d4a9a1ca7
2 changed files with 126 additions and 117 deletions
Show Stats Download Patch File Download Diff File

2
Cargo.toml
Unescape Escape View File

@ -5,7 +5,7 @@ authors = ["Sebastian Geisler <sgeisler@wh2.tu-dresden.de>"]
[dependencies]
structopt = "0.1.6"
structopt-derive = "0.1.6"
structopt-derive = {git = "https://github.com/sgeisler/structopt"}
cargo_metadata = "0.4.0"
log = "0.4.1"
simple_logger = "0.4.0"

241
src/main.rs
Unescape Escape View File

@ -20,37 +20,37 @@ use structopt::StructOpt;
use toml::Value;
#[derive(StructOpt, Debug)]
#[structopt(name = "cargo remote")]
struct Opts {
// workaround for "remote" argument when calling "cargo remote"
_unused: String,
#[structopt(help = "cargo command that will be executed remotely")]
command: String,
#[structopt(short = "r", long = "remote", help = "remote ssh build server")]
remote: Option<String>,
#[structopt(
short = "c",
long = "copy-back",
help = "transfer the target folder back to the local machine"
)]
copy_back: bool,
#[structopt(
long = "manifest-path",
help = "Path to the manifest to execute",
parse(from_os_str)
)]
manifest_path: Option<PathBuf>,
#[structopt(
short = "h",
long = "transfer-hidden",
help = "transfer hidden files and directories to the build server"
)]
hidden: bool
#[structopt(name = "cargo-remote", bin_name = "cargo")]
enum Opts {
#[structopt(name = "remote")]
Remote {
#[structopt(help = "cargo command that will be executed remotely")]
command: String,
#[structopt(short = "r", long = "remote", help = "remote ssh build server")]
remote: Option<String>,
#[structopt(
short = "c",
long = "copy-back",
help = "transfer the target folder back to the local machine"
)]
copy_back: bool,
#[structopt(
long = "manifest-path",
help = "Path to the manifest to execute",
parse(from_os_str)
)]
manifest_path: Option<PathBuf>,
#[structopt(
short = "h",
long = "transfer-hidden",
help = "transfer hidden files and directories to the build server"
)]
hidden: bool,
}
}
fn main() {
@ -58,90 +58,99 @@ fn main() {
let options = Opts::from_args();
let manifest_path = options.manifest_path.as_ref().map(PathBuf::borrow);
let project_metadata = cargo_metadata::metadata(manifest_path)
.unwrap_or_else(|e| {
error!("Could not read cargo metadata: {}", e);
exit(-1);
});
// for now, assume that there is only one project and find it's root directory
let (project_dir, project_name) = project_metadata.packages.first().map_or_else(|| {
error!("No project found.");
exit(-2);
}, |project| {
(
Path::new(&project.manifest_path)
.parent()
.expect("Cargo.toml seems to have no parent directory?"),
&project.name
)
});
// TODO: refactor argument and config parsing and merging
let build_server = options.remote.unwrap_or_else(|| {
let config_path = project_dir.join(".cargo-remote.toml");
File::open(config_path).ok().and_then(|mut file| {
let mut config_file_string = "".to_owned();
// ignore the result for now, the whole config/argument parsing needs to be refactored
let _ = file.read_to_string(&mut config_file_string);
config_file_string.parse::<Value>().ok()
}).and_then(|value| {
value["remote"].as_str().map(str::to_owned)
}).unwrap_or_else(|| {
error!("No remote build server was defined (use config file or --remote flag)");
exit(-3);
})
});
info!("Transferring sources to build server.");
// transfer project to build server
let mut rsync_to = Command::new("rsync");
rsync_to.arg("-a".to_owned())
.arg("--delete")
.arg("--info=progress2")
.arg("--exclude")
.arg("target");
if !options.hidden {
rsync_to.arg("--exclude")
.arg(".*");
}
rsync_to.arg("--rsync-path")
.arg("mkdir -p remote-builds && rsync")
.arg(format!("{}/", project_dir.to_string_lossy()))
.arg(format!("{}:~/remote-builds/{}/", build_server, project_name))
.stdout(Stdio::inherit())
.stderr(Stdio::inherit())
.stdin(Stdio::inherit())
.output()
.expect("failed to transfer project to build server");
let build_command = format!("cd ~/remote-builds/{}/; $HOME/.cargo/bin/cargo {}", project_name, options.command);
info!("Starting build process.");
Command::new("ssh")
.arg(&build_server)
.arg(build_command)
.stdout(Stdio::inherit())
.stderr(Stdio::inherit())
.stdin(Stdio::inherit())
.output()
.expect("failed to build project");
if options.copy_back {
info!("Transferring artifacts back to client.");
Command::new("rsync")
.arg("-a")
.arg("--delete")
.arg("--info=progress2")
.arg(format!("{}:~/remote-builds/{}/target/", build_server, project_name))
.arg(format!("{}/target/", project_dir.to_string_lossy()))
.stdout(Stdio::inherit())
.stderr(Stdio::inherit())
.stdin(Stdio::inherit())
.output()
.expect("failed to transfer built project to client");
match options {
Opts::Remote{command, remote, copy_back, manifest_path, hidden} => {
let manifest_path = manifest_path.as_ref().map(PathBuf::borrow);
let project_metadata = cargo_metadata::metadata(manifest_path)
.unwrap_or_else(|e| {
error!("Could not read cargo metadata: {}", e);
exit(-1);
});
// for now, assume that there is only one project and find it's root directory
let (project_dir, project_name) = project_metadata.packages.first().map_or_else(|| {
error!("No project found.");
exit(-2);
}, |project| {
(
Path::new(&project.manifest_path)
.parent()
.expect("Cargo.toml seems to have no parent directory?"),
&project.name
)
});
// TODO: refactor argument and config parsing and merging
let build_server = remote.unwrap_or_else(|| {
let config_path = project_dir.join(".cargo-remote.toml");
File::open(config_path).ok().and_then(|mut file| {
let mut config_file_string = "".to_owned();
// ignore the result for now, the whole config/argument parsing needs to
// be refactored
let _ = file.read_to_string(&mut config_file_string);
config_file_string.parse::<Value>().ok()
}).and_then(|value| {
value["remote"].as_str().map(str::to_owned)
}).unwrap_or_else(|| {
error!("No remote build server was defined (use config file or --remote flag)");
exit(-3);
})
});
info!("Transferring sources to build server.");
// transfer project to build server
let mut rsync_to = Command::new("rsync");
rsync_to.arg("-a".to_owned())
.arg("--delete")
.arg("--info=progress2")
.arg("--exclude")
.arg("target");
if !hidden {
rsync_to.arg("--exclude")
.arg(".*");
}
rsync_to.arg("--rsync-path")
.arg("mkdir -p remote-builds && rsync")
.arg(format!("{}/", project_dir.to_string_lossy()))
.arg(format!("{}:~/remote-builds/{}/", build_server, project_name))
.stdout(Stdio::inherit())
.stderr(Stdio::inherit())
.stdin(Stdio::inherit())
.output()
.expect("failed to transfer project to build server");
let build_command = format!(
"cd ~/remote-builds/{}/; $HOME/.cargo/bin/cargo {}",
project_name,
command
);
info!("Starting build process.");
Command::new("ssh")
.arg(&build_server)
.arg(build_command)
.stdout(Stdio::inherit())
.stderr(Stdio::inherit())
.stdin(Stdio::inherit())
.output()
.expect("failed to build project");
if copy_back {
info!("Transferring artifacts back to client.");
Command::new("rsync")
.arg("-a")
.arg("--delete")
.arg("--info=progress2")
.arg(format!("{}:~/remote-builds/{}/target/", build_server, project_name))
.arg(format!("{}/target/", project_dir.to_string_lossy()))
.stdout(Stdio::inherit())
.stderr(Stdio::inherit())
.stdin(Stdio::inherit())
.output()
.expect("failed to transfer built project to client");
}
}
}
}

Write Preview
Loading…
Cancel
Save