Added cli interface

Author: comradez

.gitignore

@@ -1 +1,3 @@
 /target
+id_rsa
+id_rsa.pub

Cargo.lock

@@ -8,7 +8,5 @@ edition = "2021"
 [dependencies]
 base64 = "0.13.0"
 num = { version = "0.4.0", features = ["rand"] }
-# num-traits = "0.2.14"
-# num-primes  = "0.3.0"
-# num-bigint-dig = { version = "0.8.0", features = ["prime"] }
-rand = "0.8.4"
+rand = "0.8.4"
+clap = { version = "3", features = ["derive"] }

Cargo.toml

@@ -1,16 +1,68 @@
-use prime_check::miller_rabin_single;
+#![allow(dead_code)]
+extern crate clap;
 
+use std::io::Read;
+use clap::{Parser, Subcommand};
+use num::BigUint;
+use prime_check::{oct_to_base64, base64_to_oct};
 use crate::prime_check::{PrimeUtils, encrypt, decrypt};
 
 mod prime_check;
-fn main() {
+
+#[derive(Parser)]
+#[clap(author, version, about, long_about = None)]
+struct Cli {
+    #[clap(subcommand)]
+    command: Commands
+}
+
+#[derive(Subcommand)]
+enum Commands {
+    Gen,
+    Encrypt { message: Option<String> },
+    Decrypt { secret: Option<String> },
+}
+
+fn key_to_base64(key: &(BigUint, BigUint)) -> String {
+    oct_to_base64(&key.0) + "-" + &oct_to_base64(&key.1)
+}
+
+fn base64_to_key(base64: &str) -> (BigUint, BigUint) {
+    let parts: Vec<&str> = base64.split("-").collect();
+    (base64_to_oct(parts[0]), base64_to_oct(parts[1]))
+}
+
+fn main() -> std::io::Result<()> {
+    let end_char = if cfg!(target_os = "windows") { 'Z' } else { 'D' };
     let mut checker = PrimeUtils::new(1024);
-    let (pub_key, pri_key) = checker.gen_key();
-    let message = "moderncryptography";
-    let secret = encrypt(&pri_key, message);
-    println!("{}", secret);
-    let message = decrypt(&pub_key, &secret);
-    // checker.is_prime(testee)
-    println!("{}", message);
-    // println!("{}", miller_rabin_single(&341_u32, 20_u32));
+    let cli = Cli::parse();
+    match cli.command {
+        Commands::Gen => {
+            let (pub_key, pri_key) = checker.gen_key();
+            std::fs::write("id_rsa.pub", key_to_base64(&pub_key).as_bytes())?;
+            std::fs::write("id_rsa", key_to_base64(&pri_key).as_bytes())?;
+            println!("id_rsa.pub & id_rsa have been generated.");
+        },
+        Commands::Encrypt { message } => {
+            let message = message.unwrap_or_else(|| {
+                println!("Please input the message. Ctrl + {} to end.", end_char);
+                let mut content = String::new();
+                std::io::stdin().read_to_string(&mut content).unwrap();
+                content
+            });
+            let public_key = base64_to_key(&String::from_utf8(std::fs::read("id_rsa.pub").unwrap()).unwrap());
+            println!("{}", encrypt(&public_key, &message))
+        },
+        Commands::Decrypt { secret} => {
+            let secret = secret.unwrap_or_else(|| {
+                println!("Please input the secret. Ctrl + {} to end.", end_char);
+                let mut content = String::new();
+                std::io::stdin().read_to_string(&mut content).unwrap();
+                content
+            });
+            let private_key = base64_to_key(&String::from_utf8(std::fs::read("id_rsa").unwrap()).unwrap());
+            print!("\n{}", decrypt(&private_key, &secret))
+        }
+    }
+    Ok(())
 }