Skip to content
41 changes: 13 additions & 28 deletions setup-deploy-keys/src/deploy.rs
Original file line number Diff line number Diff line change
@@ -1,57 +1,42 @@
use std::env;
use std::fs;
use std::io::{Read, Write};
use std::io::Write;
use std::os::unix::net::UnixStream;
use std::os::unix::prelude::*;
use std::process::{Command, Stdio};

fn main() {
let slug = env::var("BUILD_REPOSITORY_ID")
.or_else(|_| env::var("GITHUB_REPOSITORY"))
.unwrap();
let key = env::var("GITHUB_DEPLOY_KEY").unwrap();
let key_b64 = env::var("GITHUB_DEPLOY_KEY").unwrap();

let socket = "/tmp/.github-deploy-socket";
run(Command::new("ssh-agent").arg("-a").arg(socket));
while UnixStream::connect(socket).is_err() {
std::thread::sleep(std::time::Duration::from_millis(5));
}

let mut decode = Command::new("base64")
.arg("-d")
// decode langsung via crate base64
let decoded = base64::decode(&key_b64).expect("invalid base64 key");

// pipe ke ssh-add via stdin
let mut ssh_add = Command::new("ssh-add")
.arg("-")
.env("SSH_AUTH_SOCK", socket)
.stdin(Stdio::piped())
.stdout(Stdio::piped())
.spawn()
.unwrap();
decode
.stdin
.take()
.unwrap()
.write_all(key.as_bytes())
.unwrap();
let mut key = Vec::new();
decode.stdout.take().unwrap().read_to_end(&mut key).unwrap();
decode.wait().unwrap();

let path = "_the_key";
fs::write(path, key).unwrap();
fs::set_permissions(path, fs::Permissions::from_mode(0o600)).unwrap();
run(Command::new("ssh-add")
.arg(path)
.env("SSH_AUTH_SOCK", socket));
fs::remove_file(path).unwrap();
ssh_add.stdin.as_mut().unwrap().write_all(&decoded).unwrap();
ssh_add.wait().unwrap();

let sha = env::var("BUILD_SOURCEVERSION")
.or_else(|_| env::var("GITHUB_SHA"))
.unwrap();
let msg = format!("Deploy {sha} to gh-pages");

drop(fs::remove_dir_all(".git"));
let _ = std::fs::remove_dir_all(".git");
run(Command::new("git").arg("init"));
run(Command::new("git")
.arg("config")
.arg("user.name")
.arg("Deploy from CI"));
run(Command::new("git").arg("config").arg("user.name").arg("Deploy from CI"));
run(Command::new("git").arg("config").arg("user.email").arg(""));
run(Command::new("git").arg("add").arg("."));
run(Command::new("git").arg("commit").arg("-m").arg(&msg));
Expand Down
Loading