mirror of
https://github.com/arkorty/rustcm-cli.git
synced 2026-03-18 00:57:17 +00:00
Compare commits
5 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| a0a7f051c5 | |||
| 7a734ac2f4 | |||
| f2a57b1078 | |||
| 1d6ab1291d | |||
| bfbd9d6b1f |
74
.github/workflows/release.yml
vendored
Normal file
74
.github/workflows/release.yml
vendored
Normal file
@@ -0,0 +1,74 @@
|
||||
name: Rust CI/CD
|
||||
|
||||
on:
|
||||
push:
|
||||
branches: [ "master" ]
|
||||
pull_request:
|
||||
branches: [ "master" ]
|
||||
|
||||
env:
|
||||
CARGO_TERM_COLOR: always
|
||||
|
||||
jobs:
|
||||
build:
|
||||
runs-on: ubuntu-latest
|
||||
steps:
|
||||
- uses: actions/checkout@v4
|
||||
- name: Build
|
||||
run: cargo build --verbose
|
||||
- name: Run tests
|
||||
run: cargo test --verbose
|
||||
|
||||
release:
|
||||
needs: build
|
||||
if: github.event_name == 'push' && github.ref == 'refs/heads/master'
|
||||
runs-on: ubuntu-latest
|
||||
permissions:
|
||||
contents: write
|
||||
steps:
|
||||
- uses: actions/checkout@v4
|
||||
- name: Install Rust
|
||||
uses: actions-rs/toolchain@v1
|
||||
with:
|
||||
toolchain: stable
|
||||
override: true
|
||||
- name: Install Windows GNU toolchain
|
||||
run: |
|
||||
sudo apt-get update
|
||||
sudo apt-get install -y gcc-mingw-w64-x86-64
|
||||
- name: Get version
|
||||
id: get_version
|
||||
run: echo "VERSION=$(grep '^version =' Cargo.toml | cut -d '"' -f2)" >> $GITHUB_OUTPUT
|
||||
- name: Build Release for Linux
|
||||
run: cargo build --release --target x86_64-unknown-linux-gnu
|
||||
- name: Build Release for Windows
|
||||
run: |
|
||||
rustup target add x86_64-pc-windows-gnu
|
||||
cargo build --release --target x86_64-pc-windows-gnu
|
||||
- name: Verify binaries and create release
|
||||
env:
|
||||
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
|
||||
run: |
|
||||
linux_binary="./target/x86_64-unknown-linux-gnu/release/rustcm-cli"
|
||||
windows_binary="./target/x86_64-pc-windows-gnu/release/rustcm-cli.exe"
|
||||
|
||||
if [ ! -f "$linux_binary" ]; then
|
||||
echo "Error: Linux binary not found at $linux_binary"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
if [ ! -f "$windows_binary" ]; then
|
||||
echo "Error: Windows binary not found at $windows_binary"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
version="${{ steps.get_version.outputs.VERSION }}"
|
||||
tag_name="v$version"
|
||||
release_name="Release v$version"
|
||||
|
||||
gh release create "$tag_name" \
|
||||
--title "$release_name" \
|
||||
--notes "Automated release for version $version" \
|
||||
--prerelease \
|
||||
"$linux_binary#rustcm-cli-linux-x86_64" \
|
||||
"$windows_binary#rustcm-cli-windows-x86_64.exe"
|
||||
@@ -3,13 +3,7 @@ name = "rustcm-cli"
|
||||
version = "0.1.3-alpha"
|
||||
edition = "2021"
|
||||
|
||||
[lib]
|
||||
name = "rustcm_cli"
|
||||
path = "src/lib.rs"
|
||||
|
||||
[[bin]]
|
||||
name = "rustcm-cli"
|
||||
path = "src/main.rs"
|
||||
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
|
||||
|
||||
[dependencies]
|
||||
orion = "0.17.5"
|
||||
|
||||
@@ -1,4 +1,4 @@
|
||||

|
||||

|
||||
|
||||
# Rust Simple Text Cipher Machine / ru·s·t·c·m /
|
||||
|
||||
|
||||
|
Before Width: | Height: | Size: 38 KiB After Width: | Height: | Size: 38 KiB |
@@ -1,72 +0,0 @@
|
||||
use colored::Colorize;
|
||||
use orion::{aead, kdf};
|
||||
use std::process::exit;
|
||||
|
||||
use crate::utils::into_match;
|
||||
|
||||
pub fn encrypt(plaintext: String, secret_key: orion::kdf::SecretKey) -> Vec<u8> {
|
||||
let plaintext = plaintext.into_bytes();
|
||||
match aead::seal(&secret_key, &plaintext) {
|
||||
Ok(temp) => {
|
||||
println!("{} Data was encrypted", "Success:".bright_green());
|
||||
temp
|
||||
}
|
||||
Err(_) => {
|
||||
eprintln!("{} Could not encrypt the data", "Error:".bright_red());
|
||||
exit(1);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
pub fn decrypt(ciphertext: Vec<u8>, secret_key: orion::kdf::SecretKey) -> String {
|
||||
let plaintext = match aead::open(&secret_key, &ciphertext) {
|
||||
Ok(temp) => {
|
||||
println!("{} Data was decrypted", "Success:".bright_green());
|
||||
temp
|
||||
}
|
||||
Err(_) => {
|
||||
eprintln!(
|
||||
"{} Failed to decrypt the file, please check the password",
|
||||
"Error:".bright_red()
|
||||
);
|
||||
exit(1);
|
||||
}
|
||||
};
|
||||
match String::from_utf8(plaintext) {
|
||||
Ok(temp) => temp,
|
||||
Err(_) => {
|
||||
eprintln!("{} Could not convert to String", "Error:".bright_red());
|
||||
exit(1);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
pub fn get_secret_key(salt_bytes: [u8; 32], password: String) -> orion::kdf::SecretKey {
|
||||
let password = into_match(
|
||||
kdf::Password::from_slice(password.as_bytes()),
|
||||
"Error: Could not use the password",
|
||||
);
|
||||
let salt = into_match(
|
||||
kdf::Salt::from_slice(&salt_bytes),
|
||||
"Error: Could not create the salt",
|
||||
);
|
||||
into_match(
|
||||
kdf::derive_key(&password, &salt, 3, 8, 32),
|
||||
"Error: Could not generate the secret key",
|
||||
)
|
||||
}
|
||||
|
||||
pub fn get_salt_bytes() -> [u8; 32] {
|
||||
let mut salt_bytes = [0u8; 32];
|
||||
match orion::util::secure_rand_bytes(&mut salt_bytes) {
|
||||
Ok(_) => (),
|
||||
Err(_) => {
|
||||
eprintln!(
|
||||
"{} Could not generate the random bytes for the salt",
|
||||
"Error:".bright_red()
|
||||
);
|
||||
exit(1);
|
||||
}
|
||||
};
|
||||
salt_bytes
|
||||
}
|
||||
73
src/io.rs
73
src/io.rs
@@ -1,73 +0,0 @@
|
||||
use colored::Colorize;
|
||||
use std::fs::{read_to_string, File};
|
||||
use std::io::{Read, Write};
|
||||
use std::process::exit;
|
||||
|
||||
use crate::utils::into_match;
|
||||
|
||||
pub fn write_plain(path: String, plaintext: String) {
|
||||
let mut file = into_match(
|
||||
File::create(&path),
|
||||
&format!("Error: Could not create {}", path),
|
||||
);
|
||||
into_match(
|
||||
file.write(&plaintext.into_bytes()),
|
||||
&format!("Error: Could not write the data to {}", path),
|
||||
);
|
||||
into_match(
|
||||
file.flush(),
|
||||
&format!("Error: Could not flush data to {}", path),
|
||||
);
|
||||
}
|
||||
|
||||
pub fn write_cipher(path: String, salt_bytes: [u8; 32], ciphertext: Vec<u8>) {
|
||||
let mut file = into_match(
|
||||
File::create(&path),
|
||||
&format!("Error: Could not create {}", path),
|
||||
);
|
||||
into_match(
|
||||
file.write(&salt_bytes),
|
||||
&format!("Error: Could write the salt to {}", path),
|
||||
);
|
||||
into_match(
|
||||
file.write(&ciphertext),
|
||||
&format!("Error: Could not write the ciphertext to {}", path),
|
||||
);
|
||||
into_match(
|
||||
file.flush(),
|
||||
&format!("Error: Could not flush data to {}", path),
|
||||
);
|
||||
}
|
||||
|
||||
pub fn read_plain(path: String) -> String {
|
||||
into_match(
|
||||
read_to_string(&path),
|
||||
&format!("Error: Could not read {}", path),
|
||||
)
|
||||
}
|
||||
|
||||
pub fn read_cipher(path: String) -> (Vec<u8>, Vec<u8>) {
|
||||
let mut file = into_match(
|
||||
File::open(&path),
|
||||
&format!("Error: Could not open {}", path),
|
||||
);
|
||||
let metadata = into_match(
|
||||
File::metadata(&file),
|
||||
&format!("Error: Could not read the metadata off of {}", path),
|
||||
);
|
||||
let mut data: Vec<u8> = vec![0u8; metadata.len() as usize];
|
||||
match file.read(&mut data) {
|
||||
Ok(_) => (),
|
||||
Err(_) => {
|
||||
eprintln!("{} Could not read {}", "Error:".bright_red(), path);
|
||||
exit(1);
|
||||
}
|
||||
};
|
||||
|
||||
let mut salt_bytes: Vec<u8> = vec![0u8; 32];
|
||||
salt_bytes.clone_from_slice(&data[..32]);
|
||||
let mut cypher: Vec<u8> = vec![0u8; (metadata.len() - 32) as usize];
|
||||
cypher.clone_from_slice(&data[32..]);
|
||||
|
||||
(salt_bytes, cypher)
|
||||
}
|
||||
@@ -1,3 +0,0 @@
|
||||
pub mod crypto;
|
||||
pub mod io;
|
||||
pub mod utils;
|
||||
338
src/main.rs
338
src/main.rs
@@ -1,44 +1,233 @@
|
||||
use colored::Colorize;
|
||||
use std::env::args;
|
||||
use std::process::exit;
|
||||
// Program: rustcm-cli (0.1.3-alpha)
|
||||
// License: GNU GPL version 3
|
||||
// Author: Arkaprabha Chakraborty
|
||||
//
|
||||
// Copyright (C) 2023 Arkaprabha Chakraborty
|
||||
|
||||
use rustcm_cli::{
|
||||
crypto::{decrypt, encrypt, get_salt_bytes, get_secret_key},
|
||||
io::{read_cipher, read_plain, write_cipher, write_plain},
|
||||
utils::{get_password, into_match, to_array},
|
||||
};
|
||||
use colored::Colorize;
|
||||
use orion::{aead, kdf};
|
||||
use passterm::prompt_password_tty;
|
||||
use std::env::args;
|
||||
use std::fs::{read_to_string, File};
|
||||
use std::io::{stdout, Read, Write};
|
||||
use std::process::exit;
|
||||
|
||||
const PROGRAM_NAME: &str = "rustcm-cli";
|
||||
const PROGRAM_VERSION: &str = "0.1.3-alpha";
|
||||
|
||||
fn main() {
|
||||
let mut args = args();
|
||||
args.next();
|
||||
fn get_password(prompt: &str) -> String {
|
||||
into_match(
|
||||
stdout().flush(),
|
||||
format!("{} Could not flush date to stdout", "Error".bright_red()).as_str(),
|
||||
);
|
||||
let password = into_match(
|
||||
prompt_password_tty(Some(prompt)),
|
||||
format!("{} Could not read the password", "Error".bright_red()).as_str(),
|
||||
);
|
||||
//println!();
|
||||
|
||||
let arg_str: String = match args.next() {
|
||||
Some(temp) => temp,
|
||||
None => {
|
||||
password
|
||||
}
|
||||
|
||||
fn encrypt(plaintext: String, secret_key: orion::kdf::SecretKey) -> Vec<u8> {
|
||||
let plaintext = plaintext.into_bytes();
|
||||
match aead::seal(&secret_key, &plaintext) {
|
||||
Ok(temp) => {
|
||||
println!("{} Data was encrypted", "Success:".bright_green());
|
||||
temp
|
||||
}
|
||||
Err(_) => {
|
||||
eprintln!("{} Could not encrypt the data", "Error:".bright_red());
|
||||
exit(1);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn get_secret_key(salt_bytes: [u8; 32], password: String) -> orion::kdf::SecretKey {
|
||||
let password = into_match(
|
||||
kdf::Password::from_slice(password.as_bytes()),
|
||||
format!("{} Could not use the password", "Error:".bright_red()).as_str(),
|
||||
);
|
||||
let salt = into_match(
|
||||
kdf::Salt::from_slice(&salt_bytes),
|
||||
format!("{} Could not create the salt", "Error:".bright_red()).as_str(),
|
||||
);
|
||||
into_match(
|
||||
kdf::derive_key(&password, &salt, 3, 8, 32),
|
||||
format!(
|
||||
"{} Could not generate the secret key",
|
||||
"Error:".bright_red()
|
||||
)
|
||||
.as_str(),
|
||||
)
|
||||
}
|
||||
|
||||
fn get_salt_bytes() -> [u8; 32] {
|
||||
let mut salt_bytes = [0u8; 32];
|
||||
match orion::util::secure_rand_bytes(&mut salt_bytes) {
|
||||
Ok(_) => (),
|
||||
Err(_) => {
|
||||
eprintln!(
|
||||
"{} No arguments provided. Use --help for usage information.",
|
||||
"{} Could not generate the random bytes for the salt",
|
||||
"Error:".bright_red()
|
||||
);
|
||||
exit(1);
|
||||
}
|
||||
};
|
||||
|
||||
match arg_str.as_str() {
|
||||
"--help" | "-h" => handle_help(),
|
||||
"--version" | "-v" => handle_version(),
|
||||
"--encrypt" | "-e" => handle_encrypt(&mut args),
|
||||
"--decrypt" | "-d" => handle_decrypt(&mut args),
|
||||
_ => {
|
||||
eprintln!("{} Unrecognized argument", "Error:".bright_red());
|
||||
salt_bytes
|
||||
}
|
||||
|
||||
fn into_match<T, E>(res: Result<T, E>, estr: &str) -> T {
|
||||
let ok_val: T = match res {
|
||||
Ok(temp) => temp,
|
||||
Err(_) => {
|
||||
eprintln!("{estr}");
|
||||
exit(1)
|
||||
}
|
||||
};
|
||||
|
||||
ok_val
|
||||
}
|
||||
|
||||
fn write_plain(path: String, plaintext: String) {
|
||||
let mut file = into_match(
|
||||
File::create(&path),
|
||||
format!("{} Could not create {path}", "Error".bright_red()).as_str(),
|
||||
);
|
||||
into_match(
|
||||
file.write(&plaintext.into_bytes()),
|
||||
format!(
|
||||
"{} Could not write the data to {path}",
|
||||
"Error".bright_red()
|
||||
)
|
||||
.as_str(),
|
||||
);
|
||||
into_match(
|
||||
file.flush(),
|
||||
format!("{} Could not flush data to {path}", "Error".bright_red()).as_str(),
|
||||
);
|
||||
}
|
||||
|
||||
fn write_cipher(path: String, salt_bytes: [u8; 32], ciphertext: Vec<u8>) {
|
||||
let mut file = into_match(
|
||||
File::create(&path),
|
||||
format!("{} Could not create {path}", "Error".bright_red()).as_str(),
|
||||
);
|
||||
into_match(
|
||||
file.write(&salt_bytes),
|
||||
format!("{} Could write the salt to {path}", "Error".bright_red()).as_str(),
|
||||
);
|
||||
into_match(
|
||||
file.write(&ciphertext),
|
||||
format!(
|
||||
"{} Could not write the ciphertext to {path}",
|
||||
"Error:".bright_red()
|
||||
)
|
||||
.as_str(),
|
||||
);
|
||||
into_match(
|
||||
file.flush(),
|
||||
format!("{} Could not flush data to {path}", "Error:".bright_red()).as_str(),
|
||||
);
|
||||
}
|
||||
|
||||
fn read_plain(path: String) -> String {
|
||||
into_match(
|
||||
read_to_string(&path),
|
||||
format!("{} Could not read {path}", "Error:".bright_red()).as_str(),
|
||||
)
|
||||
}
|
||||
|
||||
fn read_cipher(path: String) -> (Vec<u8>, Vec<u8>) {
|
||||
let mut file = into_match(
|
||||
File::open(&path),
|
||||
format!("{} Could not open {path}", "Error:".bright_red()).as_str(),
|
||||
);
|
||||
let metadata = into_match(
|
||||
File::metadata(&file),
|
||||
format!(
|
||||
"{} Could not read the metadata off of {path}",
|
||||
"Error:".bright_red()
|
||||
)
|
||||
.as_str(),
|
||||
);
|
||||
let mut data: Vec<u8> = vec![0u8; metadata.len() as usize];
|
||||
match file.read(&mut data) {
|
||||
Ok(_) => (),
|
||||
Err(_) => {
|
||||
eprintln!("{} Could not read {path}", "Error:".bright_red());
|
||||
exit(1);
|
||||
}
|
||||
};
|
||||
|
||||
let mut salt_bytes: Vec<u8> = vec![0u8; 32];
|
||||
salt_bytes.clone_from_slice(&data[..32]);
|
||||
let mut cypher: Vec<u8> = vec![0u8; (metadata.len() - 32) as usize];
|
||||
cypher.clone_from_slice(&data[32..]);
|
||||
|
||||
(salt_bytes, cypher)
|
||||
}
|
||||
|
||||
fn handle_help() {
|
||||
fn decrypt(ciphertext: Vec<u8>, secret_key: orion::kdf::SecretKey) -> String {
|
||||
let plaintext = match aead::open(&secret_key, &ciphertext) {
|
||||
Ok(temp) => {
|
||||
println!("{} Data was decrypted", "Success:".bright_green());
|
||||
temp
|
||||
}
|
||||
Err(_) => {
|
||||
eprintln!(
|
||||
"{} Failed to decrypt the file, please check the password",
|
||||
"Error:".bright_red()
|
||||
);
|
||||
exit(1);
|
||||
}
|
||||
};
|
||||
match String::from_utf8(plaintext) {
|
||||
Ok(temp) => temp,
|
||||
Err(_) => {
|
||||
eprintln!("{} Could not convert to String", "Error:".bright_red());
|
||||
exit(1);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn to_array(v: Vec<u8>) -> [u8; 32] {
|
||||
let slice = v.as_slice();
|
||||
let array: [u8; 32] = match slice.try_into() {
|
||||
Ok(bytes) => bytes,
|
||||
Err(_) => {
|
||||
eprintln!(
|
||||
"{} Expected a vector of length {} but it was {}",
|
||||
"Error:".bright_red(),
|
||||
32,
|
||||
v.len()
|
||||
);
|
||||
exit(1);
|
||||
}
|
||||
};
|
||||
|
||||
array
|
||||
}
|
||||
|
||||
fn main() {
|
||||
let mut args = args();
|
||||
while args.next() != None {
|
||||
let arg_str: String = match args.next() {
|
||||
Some(temp) => temp,
|
||||
None => exit(0),
|
||||
};
|
||||
|
||||
match arg_str.as_str() {
|
||||
"--help" | "-h" => {
|
||||
match args.next() {
|
||||
Some(_) => {
|
||||
eprintln!("{} Too many arguments", "Error:".bright_red());
|
||||
exit(1);
|
||||
}
|
||||
None => (),
|
||||
};
|
||||
|
||||
println!(
|
||||
"{} {}
|
||||
Rust Simple Text Cipher Machine.
|
||||
@@ -77,7 +266,15 @@ COMMAND:
|
||||
exit(0);
|
||||
}
|
||||
|
||||
fn handle_version() {
|
||||
"--version" | "-v" => {
|
||||
match args.next() {
|
||||
Some(_) => {
|
||||
eprintln!("{} Too many arguments", "Error:".bright_red());
|
||||
exit(1);
|
||||
}
|
||||
None => (),
|
||||
};
|
||||
|
||||
println!(
|
||||
"{} ({})
|
||||
Copyright (C) 2023 Arkaprabha Chakraborty
|
||||
@@ -92,49 +289,92 @@ Written by Arkaprabha Chakraborty",
|
||||
);
|
||||
exit(0);
|
||||
}
|
||||
"--encrypt" | "-e" => {
|
||||
let path: String = match args.next() {
|
||||
Some(temp) => temp,
|
||||
None => {
|
||||
eprintln!(
|
||||
"{} Was expecting two arguments but received none",
|
||||
"Error:".bright_red()
|
||||
);
|
||||
exit(1);
|
||||
}
|
||||
};
|
||||
|
||||
let output: String = match args.next() {
|
||||
Some(temp) => temp,
|
||||
None => {
|
||||
eprintln!(
|
||||
"{} Was expecting two arguments but received one",
|
||||
"Error:".bright_red()
|
||||
);
|
||||
exit(1);
|
||||
}
|
||||
};
|
||||
|
||||
match args.next() {
|
||||
Some(_) => {
|
||||
eprintln!("{} Too many arguments", "Error:".bright_red());
|
||||
exit(1);
|
||||
}
|
||||
None => (),
|
||||
};
|
||||
|
||||
fn handle_encrypt(args: &mut std::env::Args) {
|
||||
let (path, output) = get_args(args);
|
||||
let plaintext: String = read_plain(path);
|
||||
let salt_bytes = get_salt_bytes();
|
||||
let password: String = get_password("Password: ");
|
||||
let secret_key = get_secret_key(salt_bytes, password);
|
||||
let ciphertext = encrypt(plaintext, secret_key);
|
||||
|
||||
write_cipher(output, salt_bytes, ciphertext);
|
||||
println!(
|
||||
"{} Encryption completed successfully",
|
||||
"Success:".bright_green()
|
||||
);
|
||||
|
||||
exit(0);
|
||||
}
|
||||
"--decrypt" | "-d" => {
|
||||
let path: String = match args.next() {
|
||||
Some(temp) => temp,
|
||||
None => {
|
||||
eprintln!(
|
||||
"{} Was expecting two arguments but received none",
|
||||
"Error:".bright_red()
|
||||
);
|
||||
exit(1);
|
||||
}
|
||||
};
|
||||
|
||||
let output: String = match args.next() {
|
||||
Some(temp) => temp,
|
||||
None => {
|
||||
eprintln!(
|
||||
"{} Was expecting two arguments but received one",
|
||||
"Error:".bright_red()
|
||||
);
|
||||
exit(1);
|
||||
}
|
||||
};
|
||||
|
||||
match args.next() {
|
||||
Some(_) => {
|
||||
eprintln!("{} Too many arguments", "Error:".bright_red());
|
||||
exit(1);
|
||||
}
|
||||
None => (),
|
||||
};
|
||||
|
||||
fn handle_decrypt(args: &mut std::env::Args) {
|
||||
let (path, output) = get_args(args);
|
||||
let (salt_bytes, ciphertext) = read_cipher(path);
|
||||
let salt_bytes: [u8; 32] = to_array(salt_bytes);
|
||||
let password: String = get_password("Password: ");
|
||||
let secret_key = get_secret_key(salt_bytes, password);
|
||||
let plaintext = decrypt(ciphertext, secret_key);
|
||||
|
||||
write_plain(output, plaintext);
|
||||
println!(
|
||||
"{} Decryption completed successfully",
|
||||
"Success:".bright_green()
|
||||
);
|
||||
|
||||
exit(0);
|
||||
}
|
||||
|
||||
fn get_args(args: &mut std::env::Args) -> (String, String) {
|
||||
let path: String = into_match(
|
||||
Err(args.next()),
|
||||
"Error: Was expecting two arguments but received none",
|
||||
);
|
||||
let output: String = into_match(
|
||||
Err(args.next()),
|
||||
"Error: Was expecting two arguments but received one",
|
||||
);
|
||||
if args.next().is_some() {
|
||||
eprintln!("{} Too many arguments", "Error:".bright_red());
|
||||
_ => {
|
||||
eprintln!("{} Unrecognized argument", "Error:".bright_red());
|
||||
exit(1);
|
||||
}
|
||||
(path, output)
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
40
src/utils.rs
40
src/utils.rs
@@ -1,40 +0,0 @@
|
||||
use colored::Colorize;
|
||||
use passterm::prompt_password_tty;
|
||||
use std::io::stdout;
|
||||
use std::io::Write;
|
||||
use std::process::exit;
|
||||
|
||||
pub fn get_password(prompt: &str) -> String {
|
||||
into_match(stdout().flush(), "Error: Could not flush date to stdout");
|
||||
let password = into_match(
|
||||
prompt_password_tty(Some(prompt)),
|
||||
"Error: Could not read the password",
|
||||
);
|
||||
password
|
||||
}
|
||||
|
||||
pub fn into_match<T, E>(res: Result<T, E>, estr: &str) -> T {
|
||||
match res {
|
||||
Ok(temp) => temp,
|
||||
Err(_) => {
|
||||
eprintln!("{}", estr.bright_red());
|
||||
exit(1)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
pub fn to_array(v: Vec<u8>) -> [u8; 32] {
|
||||
let slice = v.as_slice();
|
||||
match slice.try_into() {
|
||||
Ok(bytes) => bytes,
|
||||
Err(_) => {
|
||||
eprintln!(
|
||||
"{} Expected a vector of length {} but it was {}",
|
||||
"Error:".bright_red(),
|
||||
32,
|
||||
v.len()
|
||||
);
|
||||
exit(1);
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user