use crate::commands::{self, Verbosity};
use crate::error::Error;
use crate::instance::{InstanceDao, InstanceStore};
use crate::ssh_cmd::{get_ssh_private_key_names, Ssh};
use crate::view::SpinnerView;
use clap::Parser;
use std::env;
use std::thread;
use std::time::{Duration, Instant};
fn get_instance_name(target: &str) -> Result<String, Error> {
if target.contains('@') {
target
.split('@')
.nth(1)
.map(|instance| instance.to_string())
.ok_or(Error::InvalidSshTarget(target.to_string()))
} else {
Ok(target.to_string())
}
}
fn get_user_name(target: &str) -> Result<Option<String>, Error> {
if target.contains('@') {
target
.split('@')
.next()
.map(|instance| Some(instance.to_string()))
.ok_or(Error::InvalidSshTarget(target.to_string()))
} else {
Ok(None)
}
}
#[derive(Parser)]
pub struct InstanceSshCommand {
pub instance: String,
#[clap(short = 'X', default_value_t = false)]
pub xforward: bool,
#[clap(short, long, default_value_t = false)]
pub verbose: bool,
#[clap(short, long, default_value_t = false)]
pub quiet: bool,
#[clap(long)]
pub ssh_args: Option<String>,
pub cmd: Option<String>,
}
impl InstanceSshCommand {
pub fn run(&self, instance_dao: &InstanceDao) -> Result<(), Error> {
let name = get_instance_name(&self.instance)?;
let instance = instance_dao.load(&name)?;
let user = get_user_name(&self.instance)?.unwrap_or(instance.user.to_string());
let ssh_port = instance.ssh_port;
let verbosity = Verbosity::new(self.verbose, self.quiet);
commands::InstanceStartCommand {
qemu_args: None,
verbose: self.verbose,
quiet: self.quiet,
wait: true,
instances: vec![name.to_string()],
}
.run(instance_dao)?;
let mut ssh = None;
let mut start_time = Instant::now();
if !verbosity.is_quiet() {
println!("Default login user: cubic / password: cubic");
}
loop {
if ssh.is_none() {
ssh = Some(
Ssh::new()
.set_known_hosts_file(
env::var("HOME")
.map(|dir| format!("{dir}/.ssh/known_hosts"))
.ok(),
)
.set_private_keys(get_ssh_private_key_names()?)
.set_port(Some(ssh_port))
.set_xforward(self.xforward)
.set_args(self.ssh_args.clone().unwrap_or_default())
.set_user(user.clone())
.set_cmd(self.cmd.clone())
.set_verbose(verbosity.is_verbose())
.connect()
.spawn()?,
);
start_time = Instant::now();
}
if let Ok(Some(exit)) = ssh.as_mut().unwrap().try_wait() {
if exit.success() || self.cmd.is_some() || start_time.elapsed().as_secs() > 5 {
break;
}
let spinner = (!verbosity.is_quiet()).then(|| SpinnerView::new("Try to connect"));
thread::sleep(Duration::from_secs(5));
if let Some(mut s) = spinner {
s.stop()
}
ssh = None;
} else {
thread::sleep(Duration::from_secs(1));
}
}
Ok(())
}
}