feat: cmd info works

This commit is contained in:
2023-02-12 15:17:33 +08:00
parent cfa66dabaf
commit ff89bf6711
4 changed files with 55 additions and 14 deletions

34
src/cmd_info.rs Normal file
View File

@@ -0,0 +1,34 @@
use std::fs::File;
use std::path::PathBuf;
use rust_util::{information, opt_result, simple_error, XResult};
use crate::file;
pub fn info(path: PathBuf) -> XResult<()> {
let path_display = format!("{}", path.display());
let mut file_in = opt_result!(File::open(path), "Open file: {} failed: {}", &path_display);
let meta = opt_result!(file::read_tiny_encrypt_meta(&mut file_in), "Read file: {}, failed: {}", &path_display);
// println!("{}", serde_json::to_string_pretty(&meta).expect("SHOULD NOT HAPPEN"));
let mut infos = vec![];
infos.push("Tiny Encrypt File Info".to_string());
infos.push(format!("File..............: {}", path_display));
infos.push(format!("File size.........: {} bytes", meta.file_length));
infos.push(format!("Enc file summary..: Version: {}, Agent: {}", meta.version, meta.user_agent));
infos.push(format!("Last modified.....: {}", meta.file_last_modified));
infos.push(format!("Enc file created..: {}", meta.created));
infos.push(format!("Envelops..........: KMS: {}, PGP: {}",
to_yes_or_no(&meta.envelop),
to_yes_or_no(&meta.pgp_envelop)
));
meta.pgp_fingerprint.map(|fingerprint| {
infos.push(format!("PGP fingerprint...: {}", fingerprint));
});
infos.push(format!("Encrypted comment.: {}", to_yes_or_no(&meta.encrypted_comment)));
information!("{}\n", infos.join("\n"));
Ok(())
}
fn to_yes_or_no(opt: &Option<String>) -> String {
opt.as_ref().map(|_| "YES".to_string()).unwrap_or_else(|| "NO".to_string())
}

View File

@@ -5,7 +5,7 @@ use rust_util::{opt_result, simple_error, XResult};
use crate::spec::TinyEncryptMeta; use crate::spec::TinyEncryptMeta;
fn write_tiny_encrypt_meta<W: Write>(w: &mut W, meta: &TinyEncryptMeta) -> XResult<usize> { pub fn write_tiny_encrypt_meta<W: Write>(w: &mut W, meta: &TinyEncryptMeta) -> XResult<usize> {
let meta_json = opt_result!( serde_json::to_string(meta), "Meta to JSON failed: {}"); let meta_json = opt_result!( serde_json::to_string(meta), "Meta to JSON failed: {}");
let meta_json_bytes = meta_json.as_bytes(); let meta_json_bytes = meta_json.as_bytes();
let meta_json_bytes_len = meta_json_bytes.len(); let meta_json_bytes_len = meta_json_bytes.len();
@@ -17,7 +17,7 @@ fn write_tiny_encrypt_meta<W: Write>(w: &mut W, meta: &TinyEncryptMeta) -> XResu
Ok(meta_json_bytes_len + 2 + 4) Ok(meta_json_bytes_len + 2 + 4)
} }
fn read_tiny_encrypt_meta<R: Read>(r: &mut R) -> XResult<TinyEncryptMeta> { pub fn read_tiny_encrypt_meta<R: Read>(r: &mut R) -> XResult<TinyEncryptMeta> {
let mut tag_buff = [0_u8; 2]; let mut tag_buff = [0_u8; 2];
opt_result!(r.read_exact(&mut tag_buff), "Read tag failed: {}"); opt_result!(r.read_exact(&mut tag_buff), "Read tag failed: {}");
let tag = u16::from_be_bytes(tag_buff); let tag = u16::from_be_bytes(tag_buff);
@@ -32,8 +32,8 @@ fn read_tiny_encrypt_meta<R: Read>(r: &mut R) -> XResult<TinyEncryptMeta> {
return simple_error!("Meta too large: {}", length); return simple_error!("Meta too large: {}", length);
} }
let mut meta_buff = Vec::with_capacity(length as usize); let mut meta_buff = vec![0; length as usize];
opt_result!( r.read_exact(meta_buff.as_mut_slice()), "Read meta failed: {}"); opt_result!(r.read_exact(meta_buff.as_mut_slice()), "Read meta failed: {}");
Ok(opt_result!(serde_json::from_slice(&meta_buff), "Parse meta failed: {}")) Ok(opt_result!(serde_json::from_slice(&meta_buff), "Parse meta failed: {}"))
} }

View File

@@ -1,13 +1,15 @@
extern crate core; extern crate core;
use std::fs::File;
use std::path::PathBuf; use std::path::PathBuf;
use clap::{Parser, Subcommand}; use clap::{Parser, Subcommand};
use rust_util::information; use rust_util::{information, XResult};
mod spec; mod spec;
mod crypto; mod crypto;
mod file; mod file;
mod cmd_info;
#[derive(Debug, Parser)] #[derive(Debug, Parser)]
#[command(name = "tiny-encrypt-rs")] #[command(name = "tiny-encrypt-rs")]
@@ -23,28 +25,33 @@ enum Commands {
#[command(arg_required_else_help = true, short_flag = 'e')] #[command(arg_required_else_help = true, short_flag = 'e')]
Encrypt { Encrypt {
/// Files need to be encrypted /// Files need to be encrypted
files: Vec<PathBuf>, paths: Vec<PathBuf>,
}, },
/// Decrypt file(s) /// Decrypt file(s)
#[command(arg_required_else_help = true, short_flag = 'd')] #[command(arg_required_else_help = true, short_flag = 'd')]
Decrypt { Decrypt {
/// Files need to be decrypted /// Files need to be decrypted
files: Vec<PathBuf>, paths: Vec<PathBuf>,
}, },
/// Show file info /// Show file info
#[command(arg_required_else_help = true, short_flag = 'I')] #[command(arg_required_else_help = true, short_flag = 'I')]
Info { Info {
file: PathBuf, path: PathBuf,
}, },
} }
fn main() { fn main() -> XResult<()> {
let args = Cli::parse(); let args = Cli::parse();
match args.command { match args.command {
Commands::Encrypt { files } => { Commands::Encrypt { paths } => {
files.iter().for_each(|f| information!("{:?}", f)); paths.iter().for_each(|f| information!("{:?}", f));
Ok(())
}
Commands::Decrypt { .. } => {
Ok(())
}
Commands::Info { path } => {
cmd_info::info(path)
} }
Commands::Decrypt { .. } => todo!(),
Commands::Info { .. } => todo!()
} }
} }

View File

@@ -16,7 +16,7 @@ pub struct TinyEncryptMeta {
pub nonce: String, pub nonce: String,
pub file_length: u64, pub file_length: u64,
pub file_last_modified: u64, pub file_last_modified: u64,
pub compress: Option<bool>, pub compress: bool,
} }
pub fn get_user_agent() -> String { pub fn get_user_agent() -> String {