opensmtpd-filter-dkimout/src/main.rs

mod entry;
mod handshake;
mod message;
mod stdin_reader;

use entry::Entry;
use message::Message;
use std::collections::HashMap;
use stdin_reader::StdinReader;

const DEFAULT_BUFF_SIZE: usize = 1024;

fn main() {
	let mut reader = StdinReader::new();
	let mut messages: HashMap<String, Message> = HashMap::new();
	handshake::read_config(&mut reader);
	handshake::register_filter();
	loop {
		match Entry::from_bytes(&reader.read_line()) {
			Ok(entry) => {
				let msg_id = entry.get_msg_id();
				match messages.get_mut(&msg_id) {
					Some(msg) => {
						if !entry.is_end_of_message() {
							msg.append_line(entry.get_data());
						} else {
							msg.sign_and_return();
							messages.remove(&msg_id);
						}
					}
					None => {
						let msg = Message::from_entry(&entry);
						if !entry.is_end_of_message() {
							messages.insert(msg_id, msg);
						} else {
							msg.sign_and_return();
						}
					}
				}
			}
			Err(err) => {
				eprintln!("{err}");
			}
		}
	}
}