+36
@@ -0,0 +1,36 @@
|
||||
mod pass;
|
||||
mod sh;
|
||||
|
||||
use std::env;
|
||||
use std::io::{Error, ErrorKind};
|
||||
|
||||
const ENV_VAR_DEFAULT_VAULT: &str = "PASS_VAULT";
|
||||
|
||||
fn main() -> Result<(), Error> {
|
||||
pass::check_pass()?;
|
||||
let vaults = pass::get_vaults()?;
|
||||
if vaults.len() <= 0 {
|
||||
return Err(Error::new(ErrorKind::Other, "No vaults found"));
|
||||
}
|
||||
|
||||
let mut vault = String::new();
|
||||
match env::var(ENV_VAR_DEFAULT_VAULT) {
|
||||
Ok(default_vault) => {
|
||||
if vaults.contains(&default_vault) {
|
||||
vault = default_vault;
|
||||
} else {
|
||||
return Err(Error::new(
|
||||
ErrorKind::Other,
|
||||
format!("Vault {} does not exist", default_vault),
|
||||
));
|
||||
}
|
||||
}
|
||||
Err(_) => {
|
||||
vault = vaults.get(0).unwrap().clone();
|
||||
println!("No default vault found (you can set it with {}), using {}...", ENV_VAR_DEFAULT_VAULT, vault);
|
||||
}
|
||||
};
|
||||
println!("selected: {:?}", vault);
|
||||
|
||||
return Ok(());
|
||||
}
|
||||
+55
@@ -0,0 +1,55 @@
|
||||
use crate::sh;
|
||||
use fancy_regex::Regex;
|
||||
use std::io::{Error, ErrorKind};
|
||||
|
||||
const EXECUTABLE: &str = "pass-cli";
|
||||
const VAULT_NAME_REGEX: &str = r"(?m)(?<=- \[.{88}\]: ).*$";
|
||||
|
||||
pub fn check_pass() -> Result<(), Error> {
|
||||
let which = sh::sh(format!("which {}", EXECUTABLE));
|
||||
if which.is_empty() {
|
||||
return Err(Error::new(
|
||||
ErrorKind::Other,
|
||||
format!("{} is not installed", EXECUTABLE),
|
||||
));
|
||||
}
|
||||
|
||||
let test = sh::sh(format!("{} test", EXECUTABLE));
|
||||
if test != "Connection successful\n" {
|
||||
return Err(Error::new(
|
||||
ErrorKind::Other,
|
||||
format!("{} test failed", EXECUTABLE),
|
||||
));
|
||||
}
|
||||
|
||||
return Ok(());
|
||||
}
|
||||
|
||||
pub fn get_vaults() -> Result<Vec<String>, Error> {
|
||||
let re = Regex::new(VAULT_NAME_REGEX).unwrap();
|
||||
let vaults_raw = sh::sh(format!("{} vault list", EXECUTABLE));
|
||||
|
||||
let captures = re
|
||||
.captures(vaults_raw.as_str())
|
||||
.expect("Error running regex");
|
||||
if let None = captures {
|
||||
return Err(Error::new(ErrorKind::Other, "No vault found"));
|
||||
}
|
||||
|
||||
let mut vaults = Vec::<String>::new();
|
||||
|
||||
let captures = captures.unwrap();
|
||||
for each in captures.iter() {
|
||||
if let None = each {
|
||||
continue;
|
||||
}
|
||||
|
||||
let each = each.unwrap();
|
||||
let vault = each.as_str();
|
||||
if !vault.is_empty() {
|
||||
vaults.push(vault.to_string());
|
||||
}
|
||||
}
|
||||
|
||||
return Ok(vaults);
|
||||
}
|
||||
@@ -0,0 +1,11 @@
|
||||
use std::process::Command;
|
||||
|
||||
pub fn sh(command: impl Into<String>) -> String {
|
||||
let output = Command::new("sh")
|
||||
.arg("-c")
|
||||
.arg(command.into())
|
||||
.output()
|
||||
.expect("Failed to execute command");
|
||||
|
||||
return String::from_utf8(output.stdout).expect("Invalid UTF-8 sequence");
|
||||
}
|
||||
Reference in New Issue
Block a user