From 6a07360dc1f9514a0d8b94e728a820cf68277be5 Mon Sep 17 00:00:00 2001
From: Hatter Jiang <jht5945@gmail.com>
Date: Sat, 16 Nov 2024 21:59:36 +0800
Subject: [PATCH] feat: pendig simple encrypt/decrypt

---
 src/cmd_simple_encrypt_decrypt.rs | 78 +++++++++++++++++++++++++++++++
 src/lib.rs                        |  3 ++
 src/main.rs                       |  6 ++-
 3 files changed, 86 insertions(+), 1 deletion(-)
 create mode 100644 src/cmd_simple_encrypt_decrypt.rs

diff --git a/src/cmd_simple_encrypt_decrypt.rs b/src/cmd_simple_encrypt_decrypt.rs
new file mode 100644
index 0000000..0378bba
--- /dev/null
+++ b/src/cmd_simple_encrypt_decrypt.rs
@@ -0,0 +1,78 @@
+use crate::config::TinyEncryptConfig;
+use crate::consts::TINY_ENC_CONFIG_FILE;
+use clap::Args;
+use rust_util::{debugging, failure, iff, information, opt_result, simple_error, success, util_size, warning, XResult};
+use serde::Serialize;
+use std::process::exit;
+
+#[derive(Debug, Args)]
+pub struct CmdSimpleEncrypt {
+    /// Encryption profile (use default when --key-filter is assigned)
+    #[arg(long, short = 'p')]
+    pub profile: Option<String>,
+
+    /// Encryption key filter (key_id or type:TYPE(e.g. ecdh, pgp, ecdh-p384, pgp-ed25519), multiple joined by ',', ALL for all)
+    #[arg(long, short = 'k')]
+    pub key_filter: Option<String>,
+}
+
+#[derive(Serialize)]
+pub struct CmdResult {
+    pub code: i32,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub message: Option<String>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub result: Option<String>,
+}
+
+impl CmdResult {
+    pub fn fail(code: i32, message: &str) -> Self {
+        Self {
+            code,
+            message: Some(message.to_string()),
+            result: None,
+        }
+    }
+
+    pub fn success(result: &str) -> Self {
+        Self {
+            code: 0,
+            message: None,
+            result: Some(result.to_string()),
+        }
+    }
+
+    pub fn print_exit(&self) {
+        let result = serde_json::to_string_pretty(self).unwrap();
+        println!("{}", result);
+        exit(self.code);
+    }
+}
+
+pub fn simple_encrypt(cmd_encrypt: CmdSimpleEncrypt) -> XResult<()> {
+    if let Err(inner_result_error) = inner_simple_encrypt(cmd_encrypt) {
+        CmdResult::fail(-1, &format!("{}", inner_result_error)).print_exit();
+    }
+    Ok(())
+}
+
+pub fn inner_simple_encrypt(cmd_encrypt: CmdSimpleEncrypt) -> XResult<()> {
+    let config = TinyEncryptConfig::load(TINY_ENC_CONFIG_FILE)?;
+    debugging!("Found tiny encrypt config: {:?}", config);
+    let envelops = config.find_envelops(&cmd_encrypt.profile, &cmd_encrypt.key_filter)?;
+    if envelops.is_empty() { return simple_error!("Cannot find any valid envelops"); }
+    debugging!("Found envelops: {:?}", envelops);
+    let envelop_tkids: Vec<_> = envelops.iter()
+        .map(|e| format!("{}:{}", e.r#type.get_name(), e.kid))
+        .collect();
+    debugging!("Matched {} envelop(s): \n- {}", envelops.len(), envelop_tkids.join("\n- "));
+
+    if envelop_tkids.is_empty() {
+        return simple_error!("no matched envelops found");
+    }
+    // TODO ...
+
+
+    Ok(())
+}
+
diff --git a/src/lib.rs b/src/lib.rs
index 18d13ab..84ca37e 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -9,9 +9,11 @@ pub use cmd_decrypt::decrypt_single;
 pub use cmd_directdecrypt::CmdDirectDecrypt;
 pub use cmd_directdecrypt::direct_decrypt;
 pub use cmd_encrypt::CmdEncrypt;
+pub use cmd_simple_encrypt_decrypt::CmdSimpleEncrypt;
 pub use cmd_encrypt::encrypt;
 pub use cmd_encrypt::encrypt_single;
 pub use cmd_encrypt::encrypt_single_file_out;
+pub use cmd_simple_encrypt_decrypt::simple_encrypt;
 #[cfg(feature = "decrypt")]
 pub use cmd_execenv::CmdExecEnv;
 #[cfg(feature = "decrypt")]
@@ -57,6 +59,7 @@ mod cmd_info;
 #[cfg(feature = "decrypt")]
 mod cmd_decrypt;
 mod cmd_encrypt;
+mod cmd_simple_encrypt_decrypt;
 mod cmd_directdecrypt;
 #[cfg(feature = "macos")]
 mod cmd_initkeychain;
diff --git a/src/main.rs b/src/main.rs
index 4120f28..d5edee1 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -3,7 +3,6 @@ extern crate core;
 use clap::{Parser, Subcommand};
 use rust_util::XResult;
 
-use tiny_encrypt::{CmdConfig, CmdDirectDecrypt, CmdEncrypt, CmdInfo, CmdVersion};
 #[cfg(feature = "decrypt")]
 use tiny_encrypt::CmdDecrypt;
 #[cfg(feature = "decrypt")]
@@ -12,6 +11,7 @@ use tiny_encrypt::CmdExecEnv;
 use tiny_encrypt::CmdInitKeychain;
 #[cfg(feature = "smartcard")]
 use tiny_encrypt::CmdInitPiv;
+use tiny_encrypt::{CmdConfig, CmdDirectDecrypt, CmdEncrypt, CmdInfo, CmdSimpleEncrypt, CmdVersion};
 
 #[derive(Debug, Parser)]
 #[command(name = "tiny-encrypt-rs")]
@@ -26,6 +26,9 @@ enum Commands {
     /// Encrypt file(s)
     #[command(arg_required_else_help = true, short_flag = 'e')]
     Encrypt(CmdEncrypt),
+    /// Simple encrypt file(s)
+    #[command(arg_required_else_help = true)]
+    SimpleEncrypt(CmdSimpleEncrypt),
     #[cfg(feature = "decrypt")]
     /// Decrypt file(s)
     #[command(arg_required_else_help = true, short_flag = 'd')]
@@ -60,6 +63,7 @@ fn main() -> XResult<()> {
     let args = Cli::parse();
     match args.command {
         Commands::Encrypt(cmd_encrypt) => tiny_encrypt::encrypt(cmd_encrypt),
+        Commands::SimpleEncrypt(cmd_simple_encrypt) => tiny_encrypt::simple_encrypt(cmd_simple_encrypt),
         #[cfg(feature = "decrypt")]
         Commands::Decrypt(cmd_decrypt) => tiny_encrypt::decrypt(cmd_decrypt),
         Commands::DirectDecrypt(cmd_direct_decrypt) => tiny_encrypt::direct_decrypt(cmd_direct_decrypt),