separate ssh platform code into ssh.rs | extend ssh config | improve tui
This commit is contained in:
parent
937a4c9010
commit
3c1dd898d5
@ -1,4 +1,5 @@
|
|||||||
use anyhow::Result;
|
use anyhow::Result;
|
||||||
|
use colored::Colorize;
|
||||||
use crate::{ platform::Platform, input };
|
use crate::{ platform::Platform, input };
|
||||||
use flexbuffers::{ FlexbufferSerializer, Reader };
|
use flexbuffers::{ FlexbufferSerializer, Reader };
|
||||||
use std::collections::BTreeSet;
|
use std::collections::BTreeSet;
|
||||||
@ -19,27 +20,31 @@ pub struct Host {
|
|||||||
|
|
||||||
impl std::fmt::Display for Host {
|
impl std::fmt::Display for Host {
|
||||||
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
||||||
f.write_fmt(std::format_args!("{} via {}@{}", self.platform, self.user, self.host))
|
f.write_fmt(std::format_args!("[{}] {}@{}", self.platform, self.user, self.host))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Host {
|
impl Host {
|
||||||
pub fn edit(&mut self) -> Result<Self> {
|
pub fn edit(&mut self, config: &mut ConfigManager) -> Result<Self> {
|
||||||
let mut data = toml::to_string_pretty(self)?;
|
let data = toml::to_string_pretty(self)?;
|
||||||
|
let t = input::edit_temp_from_string(&data)?;
|
||||||
|
|
||||||
{
|
match toml::from_str::<Self>(&fs::read_to_string(t)?) {
|
||||||
let t = mktemp::Temp::new_file().unwrap();
|
Ok(d) => {
|
||||||
fs::write(t.to_owned(), data)?;
|
println!(" ## {} ## ", "Definition OK".green());
|
||||||
|
|
||||||
let editor = std::env::var("EDITOR").expect("EDITOR is not set");
|
// Replace old config with new config
|
||||||
std::process::Command::new(editor)
|
config.configs.remove(self);
|
||||||
.arg(t.to_owned().to_string_lossy().to_string())
|
config.configs.insert(d.to_owned());
|
||||||
.spawn()?.wait()?;
|
config.save();
|
||||||
|
|
||||||
data = fs::read_to_string(t)?;
|
Ok(d)
|
||||||
|
},
|
||||||
|
Err(e) => {
|
||||||
|
println!(" ## {}: {} ## ", "Rejecting definition".red(), e);
|
||||||
|
Ok(self.to_owned())
|
||||||
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
Ok(toml::from_str(&data)?)
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -53,36 +58,53 @@ pub struct ConfigManager {
|
|||||||
pub configs: BTreeSet<Host>,
|
pub configs: BTreeSet<Host>,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
impl ConfigManager {
|
impl ConfigManager {
|
||||||
pub fn new(given_config_root: Option<String>, search_path: Option<String>) -> Result<Self> {
|
pub fn new(given_config_root: Option<String>, search_path: Option<String>) -> Result<Self> {
|
||||||
|
// Resolve config root
|
||||||
let config_root = match given_config_root {
|
let config_root = match given_config_root {
|
||||||
Some(c) => path::Path::new(&c).to_owned(),
|
Some(c) => path::Path::new(&c).to_owned(),
|
||||||
None => dirs::data_local_dir().unwrap().join("keyman/"),
|
None => dirs::data_local_dir().unwrap().join("keyman/"),
|
||||||
};
|
};
|
||||||
|
|
||||||
|
// Make config root if it doesn't exist (eg: first run)
|
||||||
if !fs::try_exists(config_root.to_owned()).unwrap_or(false) {
|
if !fs::try_exists(config_root.to_owned()).unwrap_or(false) {
|
||||||
fs::create_dir_all(config_root.to_owned())?;
|
fs::create_dir_all(config_root.to_owned())?;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Resolve search path
|
||||||
let search_path = match search_path {
|
let search_path = match search_path {
|
||||||
Some(s) => std::path::Path::new(&s).to_owned(),
|
Some(s) => std::path::Path::new(&s).to_owned(),
|
||||||
None => dirs::home_dir().unwrap().join(".ssh/"),
|
None => dirs::home_dir().unwrap().join(".ssh/"),
|
||||||
};
|
};
|
||||||
|
|
||||||
match fs::read(config_root.join("config")) {
|
// Read config
|
||||||
Ok(c) => {
|
let config = fs::read(config_root.join("config"));
|
||||||
let root = Reader::get_root(c.as_slice())?;
|
if config.is_err() {
|
||||||
match ConfigManager::deserialize(root) {
|
return Ok(Self { config_dir: config_root.to_owned(), search_path, configs: BTreeSet::new() })
|
||||||
Ok(mut mgr) => {
|
|
||||||
mgr.config_dir = config_root;
|
|
||||||
mgr.search_path = search_path;
|
|
||||||
Ok(mgr)
|
|
||||||
},
|
|
||||||
Err(_) => Ok(Self { config_dir: config_root, search_path, configs: BTreeSet::new() }),
|
|
||||||
}
|
}
|
||||||
|
let config = config.unwrap();
|
||||||
|
|
||||||
|
// Deserialize
|
||||||
|
let root = Reader::get_root(config.as_slice())?;
|
||||||
|
let mgr = ConfigManager::deserialize(root);
|
||||||
|
|
||||||
|
if mgr.is_err() {
|
||||||
|
return Ok(Self { config_dir: config_root, search_path, configs: BTreeSet::new() })
|
||||||
}
|
}
|
||||||
Err(_) => Ok(Self { config_dir: config_root, search_path, configs: BTreeSet::new() }),
|
|
||||||
|
let mgr = mgr.unwrap();
|
||||||
|
Ok(Self { config_dir: config_root, search_path, configs: mgr.configs.to_owned() })
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub fn remove(&mut self, host: &Host) {
|
||||||
|
self.configs.remove(host);
|
||||||
|
self.save();
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn insert(&mut self, host: Host) {
|
||||||
|
self.configs.insert(host);
|
||||||
|
self.save();
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn save(&mut self) {
|
pub fn save(&mut self) {
|
||||||
|
31
src/input.rs
31
src/input.rs
@ -32,6 +32,7 @@ pub fn get_host(config: &mut ConfigManager) -> Result<Host> {
|
|||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
pub fn get_platform() -> Result<Platform> {
|
pub fn get_platform() -> Result<Platform> {
|
||||||
let platforms = Platform::all();
|
let platforms = Platform::all();
|
||||||
let c = Select::with_theme(&ColorfulTheme::default())
|
let c = Select::with_theme(&ColorfulTheme::default())
|
||||||
@ -43,6 +44,7 @@ pub fn get_platform() -> Result<Platform> {
|
|||||||
Ok(platforms.get(c).unwrap().to_owned())
|
Ok(platforms.get(c).unwrap().to_owned())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
pub fn get_key(config: &ConfigManager, platform: Platform) -> Result<PlatformKey> {
|
pub fn get_key(config: &ConfigManager, platform: Platform) -> Result<PlatformKey> {
|
||||||
let keys: Vec<PlatformKey> = platform.get_keys(config);
|
let keys: Vec<PlatformKey> = platform.get_keys(config);
|
||||||
let c = FuzzySelect::with_theme(&ColorfulTheme::default())
|
let c = FuzzySelect::with_theme(&ColorfulTheme::default())
|
||||||
@ -58,6 +60,7 @@ pub fn get_key(config: &ConfigManager, platform: Platform) -> Result<PlatformKey
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
pub fn get_custom_key() -> Result<PlatformKey> {
|
pub fn get_custom_key() -> Result<PlatformKey> {
|
||||||
let theme = ColorfulTheme::default();
|
let theme = ColorfulTheme::default();
|
||||||
let mut input = Input::with_theme(&theme);
|
let mut input = Input::with_theme(&theme);
|
||||||
@ -82,3 +85,31 @@ pub fn get_custom_key() -> Result<PlatformKey> {
|
|||||||
}
|
}
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
fn edit_temp(temp: &mktemp::Temp) -> Result<()> {
|
||||||
|
let editor = std::env::var("EDITOR").expect("EDITOR is not set");
|
||||||
|
std::process::Command::new(editor)
|
||||||
|
.arg(&temp.to_str().unwrap())
|
||||||
|
.spawn()?.wait()?;
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
pub fn edit_temp_from_string(data: &String) -> Result<mktemp::Temp> {
|
||||||
|
let t = mktemp::Temp::new_file()?;
|
||||||
|
fs::write(&t, data)?;
|
||||||
|
edit_temp(&t)?;
|
||||||
|
|
||||||
|
Ok(t)
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
pub fn edit_temp_from_file(file: &String) -> Result<mktemp::Temp> {
|
||||||
|
let t = mktemp::Temp::new_file()?;
|
||||||
|
fs::copy(file, &t)?;
|
||||||
|
edit_temp(&t)?;
|
||||||
|
|
||||||
|
Ok(t)
|
||||||
|
}
|
@ -3,12 +3,12 @@
|
|||||||
use anyhow::Result;
|
use anyhow::Result;
|
||||||
use config::ConfigManager;
|
use config::ConfigManager;
|
||||||
use std::io;
|
use std::io;
|
||||||
use termion::screen::IntoAlternateScreen;
|
|
||||||
use tui::{ backend::TermionBackend, Terminal };
|
use tui::{ backend::TermionBackend, Terminal };
|
||||||
|
|
||||||
pub mod config;
|
pub mod config;
|
||||||
pub mod input;
|
pub mod input;
|
||||||
pub mod platform;
|
pub mod platform;
|
||||||
|
pub mod ssh;
|
||||||
|
|
||||||
fn main() -> Result<()> {
|
fn main() -> Result<()> {
|
||||||
let mut search_path: Option<String> = None;
|
let mut search_path: Option<String> = None;
|
||||||
@ -31,9 +31,10 @@ fn main() -> Result<()> {
|
|||||||
|
|
||||||
let mut config: ConfigManager = ConfigManager::new(config_root, search_path)?;
|
let mut config: ConfigManager = ConfigManager::new(config_root, search_path)?;
|
||||||
|
|
||||||
let backend = TermionBackend::new(io::stdout().into_alternate_screen()?);
|
let backend = TermionBackend::new(io::stdout());
|
||||||
// let backend = TermionBackend::new(io::stdout()); // used for debugging
|
// let backend = TermionBackend::new(io::stdout()); // used for debugging
|
||||||
let mut term = Terminal::new(backend)?;
|
let mut term = Terminal::new(backend)?;
|
||||||
|
term.hide_cursor()?;
|
||||||
|
|
||||||
loop {
|
loop {
|
||||||
term.clear()?;
|
term.clear()?;
|
||||||
@ -41,6 +42,6 @@ fn main() -> Result<()> {
|
|||||||
let mut host = input::get_host(&mut config)?;
|
let mut host = input::get_host(&mut config)?;
|
||||||
|
|
||||||
let platform = host.platform.to_owned();
|
let platform = host.platform.to_owned();
|
||||||
platform.run(&mut host, &mut config)?;
|
platform.run(&mut host, &mut config, &mut term)?;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
173
src/platform.rs
173
src/platform.rs
@ -1,11 +1,8 @@
|
|||||||
use anyhow::Result;
|
use anyhow::Result;
|
||||||
use colored::Colorize;
|
use crate::{ ConfigManager, config::Host };
|
||||||
use crate::{ ConfigManager, config::Host, input };
|
|
||||||
use dialoguer::{ Select, Input, Confirm, Editor, theme::ColorfulTheme };
|
|
||||||
use serde::{ Serialize, Deserialize };
|
use serde::{ Serialize, Deserialize };
|
||||||
use ssh2_config::{ SshConfig, ParseRule };
|
use std::io;
|
||||||
use ssh_key::PrivateKey;
|
use tui::{ backend::TermionBackend, Terminal };
|
||||||
use std::fs;
|
|
||||||
|
|
||||||
|
|
||||||
#[derive(Clone, Debug)]
|
#[derive(Clone, Debug)]
|
||||||
@ -53,172 +50,12 @@ impl Platform {
|
|||||||
Platform::SSH => self.ssh_get_keys(config),
|
Platform::SSH => self.ssh_get_keys(config),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
pub fn run(&self, host: &mut Host, config: &mut ConfigManager) -> Result<()> {
|
pub fn run(&self, host: &mut Host, config: &mut ConfigManager, term: &mut Terminal<TermionBackend<io::Stdout>>) -> Result<()> {
|
||||||
match self {
|
match self {
|
||||||
Platform::SSH => self.ssh_run(host, config),
|
Platform::SSH => self.ssh_run(host, config, term),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
// --- --- ---
|
|
||||||
// SSH
|
|
||||||
// --- --- ---
|
|
||||||
|
|
||||||
fn ssh_new_config(&self, host: String, port: i16, keyid: PlatformKey) -> String {
|
|
||||||
format!(r#"# Generated by Keyman
|
|
||||||
|
|
||||||
Host {host}
|
|
||||||
HostName {host}
|
|
||||||
Port {port}
|
|
||||||
IdentityFile {}
|
|
||||||
"#, keyid.file)
|
|
||||||
}
|
|
||||||
|
|
||||||
// New host
|
|
||||||
fn ssh_new_host(&self, config: &mut ConfigManager) -> Result<Host> {
|
|
||||||
println!(" ## Creating new SSH host entry ## ");
|
|
||||||
let keyid = input::get_key(config, self.to_owned())?;
|
|
||||||
|
|
||||||
let userhost = loop {
|
|
||||||
let inp: String = Input::with_theme(&ColorfulTheme::default())
|
|
||||||
.with_prompt("User@Host")
|
|
||||||
.interact_text()?;
|
|
||||||
|
|
||||||
let split = inp.split("@");
|
|
||||||
let split: Vec<String> = split.map(|a| a.to_string()).collect();
|
|
||||||
|
|
||||||
if split.len() < 2 {
|
|
||||||
println!("{} Example: paul@example.com:34 - <user>@<host>[:port]", "Incorrect user@host given.".red());
|
|
||||||
} else {
|
|
||||||
break split
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
let user: String = userhost.get(0).unwrap().to_string();
|
|
||||||
let mut host: String = userhost.get(1).unwrap().to_string();
|
|
||||||
let mut port: i16 = 22;
|
|
||||||
|
|
||||||
let _host = host.to_owned();
|
|
||||||
let h: Vec<&str> = _host.split(":").collect();
|
|
||||||
|
|
||||||
if h.len() > 1 {
|
|
||||||
let portstr = h.get(1).unwrap();
|
|
||||||
port = i16::from_str_radix(portstr, 10)?;
|
|
||||||
host = h.get(0).unwrap().to_string();
|
|
||||||
}
|
|
||||||
|
|
||||||
let config_dir = config.config_dir.join("hosts/").join(host.to_owned());
|
|
||||||
fs::create_dir_all(config_dir.to_owned()).expect(&format!("Couldnt create_dir_all on {:?}", config_dir.to_owned()));
|
|
||||||
let config_file = config_dir.join("config");
|
|
||||||
|
|
||||||
fs::write(config_file.to_owned(), self.ssh_new_config(host.to_owned(), port, keyid.to_owned())).expect(&format!("Couldn't write config at {:?}", config_file.to_owned()));
|
|
||||||
|
|
||||||
Ok(Host {
|
|
||||||
platform: Platform::SSH,
|
|
||||||
id: keyid.file,
|
|
||||||
user,
|
|
||||||
host,
|
|
||||||
port,
|
|
||||||
config: config_file.to_owned().to_string_lossy().to_string()
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
// Get keys
|
|
||||||
fn ssh_get_keys(&self, config: &ConfigManager) -> Vec<PlatformKey> {
|
|
||||||
let mut keys = vec![];
|
|
||||||
let Ok(dir) = std::fs::read_dir(config.search_path.to_owned()) else { panic!("Couldn't read {:?}", config.search_path.to_owned()) };
|
|
||||||
|
|
||||||
for file in dir {
|
|
||||||
let Ok(f) = file else { continue };
|
|
||||||
|
|
||||||
match PrivateKey::read_openssh_file(&f.path()) {
|
|
||||||
Ok(p) => keys.push(PlatformKey {
|
|
||||||
file: format!("{}", f.path().to_string_lossy()),
|
|
||||||
display: format!("{}: {}", p.fingerprint(ssh_key::HashAlg::Sha256), f.file_name().to_string_lossy())
|
|
||||||
}),
|
|
||||||
// Not a private key
|
|
||||||
Err(_) => continue
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
keys
|
|
||||||
}
|
|
||||||
|
|
||||||
// Run
|
|
||||||
fn ssh_run(&self, host: &mut Host, config: &mut ConfigManager) -> Result<()> {
|
|
||||||
Ok(loop {
|
|
||||||
let action = Select::with_theme(&ColorfulTheme::default())
|
|
||||||
.with_prompt(format!("{}@{}", host.user.to_owned(), host.host.to_owned()))
|
|
||||||
.item("Connect")
|
|
||||||
.item("Edit config")
|
|
||||||
.item("Edit definition")
|
|
||||||
.item("Delete")
|
|
||||||
.item("Back")
|
|
||||||
.default(0)
|
|
||||||
.interact()?;
|
|
||||||
|
|
||||||
match action {
|
|
||||||
0 => { // Connect
|
|
||||||
// TODO: clear screen before connect
|
|
||||||
std::process::Command::new("ssh")
|
|
||||||
.arg("-F")
|
|
||||||
.arg(host.config.to_owned())
|
|
||||||
.arg("-l")
|
|
||||||
.arg(host.user.to_owned())
|
|
||||||
.arg(host.host.to_owned()).spawn()?.wait()?;
|
|
||||||
}
|
|
||||||
1 => { // Edit config
|
|
||||||
// TODO: replace this with configuer::editor
|
|
||||||
// if let Some(cnf) = Editor::new().edit(fs::read_to_string(config.config_dir.join("config"))?.as_str())? {
|
|
||||||
// fs::write(config.config_dir.join("config"), cnf)?;
|
|
||||||
// }
|
|
||||||
let t = mktemp::Temp::new_file()?;
|
|
||||||
fs::copy(&host.config, &t)?;
|
|
||||||
|
|
||||||
let editor = std::env::var("EDITOR").expect("EDITOR is not set");
|
|
||||||
std::process::Command::new(editor)
|
|
||||||
.arg(&t.to_str().unwrap())
|
|
||||||
.spawn()?.wait()?;
|
|
||||||
|
|
||||||
let mut file = std::io::BufReader::new(fs::File::open(&t).expect(&format!("Couldn't open config ({})", &host.config)));
|
|
||||||
match SshConfig::default().parse(&mut file, ParseRule::STRICT) {
|
|
||||||
Ok(_) => {
|
|
||||||
println!(" ## {} ## ", "Config OK".green());
|
|
||||||
fs::copy(&t, &host.config)?;
|
|
||||||
},
|
|
||||||
Err(e) => println!(" ## {}: {} ## ", "Rejecting config".red(), e),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
2 => { // Edit definition
|
|
||||||
let edited_host = host.edit()?;
|
|
||||||
config.configs.remove(host);
|
|
||||||
config.configs.insert(edited_host.to_owned());
|
|
||||||
host.clone_from(&edited_host);
|
|
||||||
config.save();
|
|
||||||
}
|
|
||||||
3 => { // Delete
|
|
||||||
let confirm = Confirm::with_theme(&ColorfulTheme::default())
|
|
||||||
.with_prompt("Are you sure you want to delete this definition?")
|
|
||||||
.default(false)
|
|
||||||
.interact()?;
|
|
||||||
|
|
||||||
if confirm {
|
|
||||||
let mut conf = std::path::PathBuf::from(host.config.to_owned());
|
|
||||||
conf.pop();
|
|
||||||
|
|
||||||
config.configs.remove(host);
|
|
||||||
config.save();
|
|
||||||
fs::remove_dir_all(conf)?;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
} // Back
|
|
||||||
4 => break,
|
|
||||||
_ => ()
|
|
||||||
}
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
// --- --- ---
|
// --- --- ---
|
||||||
// GPG
|
// GPG
|
||||||
// --- --- ---
|
// --- --- ---
|
||||||
|
166
src/ssh.rs
Normal file
166
src/ssh.rs
Normal file
@ -0,0 +1,166 @@
|
|||||||
|
use anyhow::Result;
|
||||||
|
use colored::Colorize;
|
||||||
|
use crate::{ ConfigManager, config::Host, input, platform::{ Platform, PlatformKey } };
|
||||||
|
use dialoguer::{ Select, Input, Confirm, theme::ColorfulTheme };
|
||||||
|
use ssh2_config::{ SshConfig, ParseRule };
|
||||||
|
use ssh_key::PrivateKey;
|
||||||
|
use std::fs;
|
||||||
|
use std::io;
|
||||||
|
use tui::{ backend::TermionBackend, Terminal };
|
||||||
|
|
||||||
|
|
||||||
|
fn new_config(host: String, user: String, port: i16, keyid: PlatformKey) -> String {
|
||||||
|
format!(r#"# Generated by Keyman
|
||||||
|
|
||||||
|
Host {host}
|
||||||
|
HostName {host}
|
||||||
|
User {user}
|
||||||
|
Port {port}
|
||||||
|
IdentityFile {}
|
||||||
|
"#, keyid.file)
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Platform {
|
||||||
|
// New host
|
||||||
|
pub fn ssh_new_host(&self, config: &mut ConfigManager) -> Result<Host> {
|
||||||
|
println!(" ## Creating new SSH host entry ## ");
|
||||||
|
let keyid = input::get_key(config, self.to_owned())?;
|
||||||
|
|
||||||
|
let userhost = loop {
|
||||||
|
let inp: String = Input::with_theme(&ColorfulTheme::default())
|
||||||
|
.with_prompt("User@Host")
|
||||||
|
.interact_text()?;
|
||||||
|
|
||||||
|
let split = inp.split("@");
|
||||||
|
let split: Vec<String> = split.map(|a| a.to_string()).collect();
|
||||||
|
|
||||||
|
if split.len() < 2 {
|
||||||
|
println!("{} Example: paul@example.com:34 - <user>@<host>[:port]", "Incorrect user@host given.".red());
|
||||||
|
} else {
|
||||||
|
break split
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let user: String = userhost.get(0).unwrap().to_string();
|
||||||
|
let mut host: String = userhost.get(1).unwrap().to_string();
|
||||||
|
let mut port: i16 = 22;
|
||||||
|
|
||||||
|
let _host = host.to_owned();
|
||||||
|
let h: Vec<&str> = _host.split(":").collect();
|
||||||
|
|
||||||
|
if h.len() > 1 {
|
||||||
|
let portstr = h.get(1).unwrap();
|
||||||
|
port = i16::from_str_radix(portstr, 10)?;
|
||||||
|
host = h.get(0).unwrap().to_string();
|
||||||
|
}
|
||||||
|
|
||||||
|
let config_dir = config.config_dir.join("hosts/").join(host.to_owned());
|
||||||
|
fs::create_dir_all(config_dir.to_owned()).expect(&format!("Couldnt create_dir_all on {:?}", config_dir.to_owned()));
|
||||||
|
let config_file = config_dir.join("config");
|
||||||
|
|
||||||
|
fs::write(config_file.to_owned(), new_config(host.to_owned(), user.to_owned(), port, keyid.to_owned())).expect(&format!("Couldn't write config at {:?}", config_file.to_owned()));
|
||||||
|
|
||||||
|
Ok(Host {
|
||||||
|
platform: Platform::SSH,
|
||||||
|
id: keyid.file,
|
||||||
|
user,
|
||||||
|
host,
|
||||||
|
port,
|
||||||
|
config: config_file.to_owned().to_string_lossy().to_string()
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// Get keys
|
||||||
|
pub fn ssh_get_keys(&self, config: &ConfigManager) -> Vec<PlatformKey> {
|
||||||
|
let mut keys = vec![];
|
||||||
|
let Ok(dir) = std::fs::read_dir(config.search_path.to_owned()) else { panic!("Couldn't read {:?}", config.search_path.to_owned()) };
|
||||||
|
|
||||||
|
for file in dir {
|
||||||
|
let Ok(f) = file else { continue };
|
||||||
|
|
||||||
|
match PrivateKey::read_openssh_file(&f.path()) {
|
||||||
|
Ok(p) => keys.push(PlatformKey {
|
||||||
|
file: format!("{}", f.path().to_string_lossy()),
|
||||||
|
display: format!("{}: {}", p.fingerprint(ssh_key::HashAlg::Sha256), f.file_name().to_string_lossy())
|
||||||
|
}),
|
||||||
|
// Not a private key
|
||||||
|
Err(_) => continue
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
keys
|
||||||
|
}
|
||||||
|
|
||||||
|
// Run
|
||||||
|
pub fn ssh_run(&self, host: &mut Host, config: &mut ConfigManager, term: &mut Terminal<TermionBackend<io::Stdout>>) -> Result<()> {
|
||||||
|
Ok(loop {
|
||||||
|
let action = Select::with_theme(&ColorfulTheme::default())
|
||||||
|
.with_prompt(format!("{}@{}", host.user.to_owned(), host.host.to_owned()))
|
||||||
|
.item("Connect")
|
||||||
|
.item("Edit config")
|
||||||
|
.item("Edit definition")
|
||||||
|
.item("Delete")
|
||||||
|
.item("Back")
|
||||||
|
.default(0)
|
||||||
|
.interact()?;
|
||||||
|
|
||||||
|
term.clear()?;
|
||||||
|
match action {
|
||||||
|
|
||||||
|
// Connect
|
||||||
|
0 => {
|
||||||
|
// TODO: clear screen before connect
|
||||||
|
std::process::Command::new("ssh")
|
||||||
|
.arg("-F")
|
||||||
|
.arg(host.config.to_owned())
|
||||||
|
.arg(host.host.to_owned()).spawn()?.wait()?;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
// Edit config
|
||||||
|
1 => {
|
||||||
|
let t = input::edit_temp_from_file(&host.config)?;
|
||||||
|
|
||||||
|
// Validate SSH config
|
||||||
|
let mut file = std::io::BufReader::new(fs::File::open(&t)?);
|
||||||
|
if let Err(e) = SshConfig::default().parse(&mut file, ParseRule::STRICT) {
|
||||||
|
println!(" ## {}: {} ## ", "Rejecting config".red(), e)
|
||||||
|
} else {
|
||||||
|
println!(" ## {} ## ", "Config OK".green());
|
||||||
|
fs::copy(&t, &host.config)?;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
// Edit definition
|
||||||
|
2 => {
|
||||||
|
let edited_host = host.edit(config)?;
|
||||||
|
host.clone_from(&edited_host);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
// Delete
|
||||||
|
3 => {
|
||||||
|
let confirm = Confirm::with_theme(&ColorfulTheme::default())
|
||||||
|
.with_prompt("Are you sure you want to delete this definition?")
|
||||||
|
.default(false)
|
||||||
|
.interact()?;
|
||||||
|
|
||||||
|
if confirm {
|
||||||
|
let mut conf = std::path::PathBuf::from(host.config.to_owned());
|
||||||
|
conf.pop();
|
||||||
|
|
||||||
|
config.remove(host);
|
||||||
|
fs::remove_dir_all(conf)?;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
// Back
|
||||||
|
_ => break
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user