ryzokuken/cosmoline
ryzokuken
/
cosmoline
2
2
Fork 1
Code Issues 5 Pull Requests 1 Projects 1 Releases Activity

src: move reading and generation logic into trait

This commit is contained in:
Ujjwal Sharma 2021-03-22 11:36:53 +05:30
parent 2f44ee021d
commit 4d32e8e55f
Signed by: ryzokuken
GPG Key ID: 460B292812C67D9F
1 changed files with 17 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
src/main.rs
View File

@ -12,6 +12,7 @@ use std::io::prelude::*;
trait SSBKeypair {
fn to_json(&self) -> JsonValue;
fn from_json(obj: JsonValue) -> Self;
fn read_or_generate(path: PathBuf) -> Self;
}
impl SSBKeypair for Keypair {
@ -47,6 +48,21 @@ impl SSBKeypair for Keypair {
Keypair { public: pubkey, secret: privkey }
}
fn read_or_generate(path: PathBuf) -> Self {
if path.exists() {
let mut secret_file = File::open(path).unwrap();
let mut secret = String::new();
secret_file.read_to_string(&mut secret).unwrap();
let re = Regex::new(r"\s*#[^\n]*").unwrap();
let secret = re.replace_all(secret.as_str(), "");
SSBKeypair::from_json(json::parse(&secret).unwrap())
} else {
let mut csprng = OsRng {};
Keypair::generate(&mut csprng)
// TODO: write this keypair to a fresh secret file
}
}
}
fn main() {
@ -69,18 +85,6 @@ fn main() {
Some(path) => PathBuf::from(path.as_str().unwrap()),
None => dirs::home_dir().unwrap().join(".cosmoline"),
};
let secret_path = path.join("secret");
let keypair = if secret_path.exists() {
let mut secret_file = File::open(secret_path).unwrap();
let mut secret = String::new();
secret_file.read_to_string(&mut secret).unwrap();
let re = Regex::new(r"\s*#[^\n]*").unwrap();
let secret = re.replace_all(secret.as_str(), "");
SSBKeypair::from_json(json::parse(&secret).unwrap())
} else {
let mut csprng = OsRng {};
Keypair::generate(&mut csprng)
// TODO: write this keypair to a fresh secret file
};
let keypair = Keypair::read_or_generate(path.join("secret"));
println!("{}", keypair.to_json().pretty(2));
}