use std::env;
use std::io;
extern crate sequoia_openpgp as openpgp;
use openpgp::armor;
use openpgp::constants::DataFormat;
use openpgp::parse::Parse;
use openpgp::serialize::stream::{
Message, LiteralWriter, Encryptor, EncryptionMode,
};
fn main() {
let args: Vec<String> = env::args().collect();
if args.len() < 3 {
panic!("A simple encryption filter.\n\n\
Usage: {} [at-rest|for-transport] <keyfile> [<keyfile>...] \
<input >output\n", args[0]);
}
let mode = match args[1].as_ref() {
"at-rest" => EncryptionMode::AtRest,
"for-transport" => EncryptionMode::ForTransport,
x => panic!("invalid mode: {:?}, \
must be either 'at-rest' or 'for-transport'",
x),
};
let tpks: Vec<openpgp::TPK> = args[2..].iter().map(|f| {
openpgp::TPK::from_file(f)
.expect("Failed to read key")
}).collect();
let recipients: Vec<&openpgp::TPK> = tpks.iter().collect();
let sink = armor::Writer::new(io::stdout(), armor::Kind::Message, &[])
.expect("Failed to create an armored writer");
let message = Message::new(sink);
let encryptor = Encryptor::new(message,
&[], &recipients,
mode)
.expect("Failed to create encryptor");
let mut literal_writer = LiteralWriter::new(encryptor, DataFormat::Binary,
None, None)
.expect("Failed to create literal writer");
io::copy(&mut io::stdin(), &mut literal_writer)
.expect("Failed to encrypt");
literal_writer.finalize().unwrap();
}