Annotation For src/main.rs
Logged in as anonymous

Origin for each line in src/main.rs from check-in e66352b9cc:

1db9dbe390 2024-11-28    1: //! Simple SMTP-to-Telegram gateway. Can parse email and send them as telegram
1db9dbe390 2024-11-28    2: //! messages to specified chats, generally you specify which email address is
1db9dbe390 2024-11-28    3: //! available in configuration, everything else is sent to default address.
1db9dbe390 2024-11-28    4: 
e81897ec87 2024-05-24    5: use anyhow::{
e81897ec87 2024-05-24    6: 	anyhow,
1db9dbe390 2024-11-28    7: 	bail,
e81897ec87 2024-05-24    8: 	Result,
e81897ec87 2024-05-24    9: };
1db9dbe390 2024-11-28   10: use async_std::{
1db9dbe390 2024-11-28   11: 	io::Error,
1db9dbe390 2024-11-28   12: 	task,
1db9dbe390 2024-11-28   13: };
e66352b9cc 2025-01-21   14: use just_getopt::{
e66352b9cc 2025-01-21   15: 	OptFlags,
e66352b9cc 2025-01-21   16: 	OptSpecs,
e66352b9cc 2025-01-21   17: 	OptValueType,
e66352b9cc 2025-01-21   18: };
1db9dbe390 2024-11-28   19: use mailin_embedded::{
1db9dbe390 2024-11-28   20: 	Response,
1db9dbe390 2024-11-28   21: 	response::*,
7620f854a7 2024-05-21   22: };
f4cad2a5c0 2024-05-26   23: use teloxide::{
f4cad2a5c0 2024-05-26   24: 	Bot,
f4cad2a5c0 2024-05-26   25: 	prelude::{
f4cad2a5c0 2024-05-26   26: 		Requester,
f4cad2a5c0 2024-05-26   27: 		RequesterExt,
f4cad2a5c0 2024-05-26   28: 	},
f4cad2a5c0 2024-05-26   29: 	types::{
f4cad2a5c0 2024-05-26   30: 		ChatId,
866aad57a4 2024-06-15   31: 		InputMedia,
866aad57a4 2024-06-15   32: 		Message,
f4cad2a5c0 2024-05-26   33: 		ParseMode::MarkdownV2,
f4cad2a5c0 2024-05-26   34: 	},
7620f854a7 2024-05-21   35: };
7620f854a7 2024-05-21   36: 
7620f854a7 2024-05-21   37: use std::{
7620f854a7 2024-05-21   38: 	borrow::Cow,
61238a3618 2024-05-22   39: 	collections::{
61238a3618 2024-05-22   40: 		HashMap,
61238a3618 2024-05-22   41: 		HashSet,
7620f854a7 2024-05-21   42: 	},
e66352b9cc 2025-01-21   43: 	path::Path,
7620f854a7 2024-05-21   44: 	vec::Vec,
7620f854a7 2024-05-21   45: };
7620f854a7 2024-05-21   46: 
1db9dbe390 2024-11-28   47: /// `SomeHeaders` object to store data through SMTP session
1db9dbe390 2024-11-28   48: #[derive(Clone, Debug)]
1db9dbe390 2024-11-28   49: struct SomeHeaders {
1db9dbe390 2024-11-28   50: 	from: String,
1db9dbe390 2024-11-28   51: 	to: Vec<String>,
1db9dbe390 2024-11-28   52: }
1db9dbe390 2024-11-28   53: 
1db9dbe390 2024-11-28   54: /// `TelegramTransport` Central object with TG api and configuration
1db9dbe390 2024-11-28   55: #[derive(Clone)]
1db9dbe390 2024-11-28   56: struct TelegramTransport {
1db9dbe390 2024-11-28   57: 	data: Vec<u8>,
1db9dbe390 2024-11-28   58: 	headers: Option<SomeHeaders>,
1db9dbe390 2024-11-28   59: 	recipients: HashMap<String, ChatId>,
1db9dbe390 2024-11-28   60: 	relay: bool,
1db9dbe390 2024-11-28   61: 	tg: teloxide::adaptors::DefaultParseMode<teloxide::adaptors::Throttle<Bot>>,
28fde40f7a 2024-12-27   62: 	fields: HashSet<String>,
da7fc7983d 2024-05-23   63: }
da7fc7983d 2024-05-23   64: 
da7fc7983d 2024-05-23   65: impl TelegramTransport {
1db9dbe390 2024-11-28   66: 	/// Initialize API and read configuration
1db9dbe390 2024-11-28   67: 	fn new(settings: config::Config) -> TelegramTransport {
f4cad2a5c0 2024-05-26   68: 		let tg = Bot::new(settings.get_string("api_key")
f4cad2a5c0 2024-05-26   69: 			.expect("[smtp2tg.toml] missing \"api_key\" parameter.\n"))
ce79786e06 2024-06-11   70: 			.throttle(teloxide::adaptors::throttle::Limits::default())
f4cad2a5c0 2024-05-26   71: 			.parse_mode(MarkdownV2);
f4cad2a5c0 2024-05-26   72: 		let recipients: HashMap<String, ChatId> = settings.get_table("recipients")
da7fc7983d 2024-05-23   73: 			.expect("[smtp2tg.toml] missing table \"recipients\".\n")
f4cad2a5c0 2024-05-26   74: 			.into_iter().map(|(a, b)| (a, ChatId (b.into_int()
da7fc7983d 2024-05-23   75: 				.expect("[smtp2tg.toml] \"recipient\" table values should be integers.\n")
da7fc7983d 2024-05-23   76: 				))).collect();
da7fc7983d 2024-05-23   77: 		if !recipients.contains_key("_") {
da7fc7983d 2024-05-23   78: 			eprintln!("[smtp2tg.toml] \"recipient\" table misses \"default_recipient\".\n");
da7fc7983d 2024-05-23   79: 			panic!("no default recipient");
da7fc7983d 2024-05-23   80: 		}
28fde40f7a 2024-12-27   81: 		let fields = HashSet::<String>::from_iter(settings.get_array("fields")
28fde40f7a 2024-12-27   82: 			.expect("[smtp2tg.toml] \"fields\" should be an array")
28fde40f7a 2024-12-27   83: 			.iter().map(|x| x.clone().into_string().expect("should be strings")));
1db9dbe390 2024-11-28   84: 		let value = settings.get_string("unknown");
1db9dbe390 2024-11-28   85: 		let relay = match value {
1db9dbe390 2024-11-28   86: 			Ok(value) => {
1db9dbe390 2024-11-28   87: 				match value.as_str() {
1db9dbe390 2024-11-28   88: 					"relay" => true,
1db9dbe390 2024-11-28   89: 					"deny" => false,
1db9dbe390 2024-11-28   90: 					_ => {
1db9dbe390 2024-11-28   91: 						eprintln!("[smtp2tg.toml] \"unknown\" should be either \"relay\" or \"deny\".\n");
1db9dbe390 2024-11-28   92: 						panic!("bad setting");
1db9dbe390 2024-11-28   93: 					},
1db9dbe390 2024-11-28   94: 				}
1db9dbe390 2024-11-28   95: 			},
1db9dbe390 2024-11-28   96: 			Err(err) => {
1db9dbe390 2024-11-28   97: 				eprintln!("[smtp2tg.toml] can't get \"unknown\":\n {}\n", err);
1db9dbe390 2024-11-28   98: 				panic!("bad setting");
1db9dbe390 2024-11-28   99: 			},
1db9dbe390 2024-11-28  100: 		};
f4cad2a5c0 2024-05-26  101: 
f4cad2a5c0 2024-05-26  102: 		TelegramTransport {
1db9dbe390 2024-11-28  103: 			data: vec!(),
1db9dbe390 2024-11-28  104: 			headers: None,
f4cad2a5c0 2024-05-26  105: 			recipients,
1db9dbe390 2024-11-28  106: 			relay,
1db9dbe390 2024-11-28  107: 			tg,
28fde40f7a 2024-12-27  108: 			fields,
f4cad2a5c0 2024-05-26  109: 		}
f4cad2a5c0 2024-05-26  110: 	}
f4cad2a5c0 2024-05-26  111: 
1db9dbe390 2024-11-28  112: 	/// Send message to default user, used for debug/log/info purposes
1db9dbe390 2024-11-28  113: 	async fn debug<'b, S>(&self, msg: S) -> Result<Message>
f4cad2a5c0 2024-05-26  114: 	where S: Into<String> {
f4cad2a5c0 2024-05-26  115: 		Ok(self.tg.send_message(*self.recipients.get("_").unwrap(), msg).await?)
f4cad2a5c0 2024-05-26  116: 	}
f4cad2a5c0 2024-05-26  117: 
1db9dbe390 2024-11-28  118: 	/// Send message to specified user
1db9dbe390 2024-11-28  119: 	async fn send<'b, S>(&self, to: &ChatId, msg: S) -> Result<Message>
f4cad2a5c0 2024-05-26  120: 	where S: Into<String> {
f4cad2a5c0 2024-05-26  121: 		Ok(self.tg.send_message(*to, msg).await?)
f4cad2a5c0 2024-05-26  122: 	}
f4cad2a5c0 2024-05-26  123: 
1db9dbe390 2024-11-28  124: 	/// Attempt to deliver one message
1db9dbe390 2024-11-28  125: 	async fn relay_mail (&self) -> Result<()> {
1db9dbe390 2024-11-28  126: 		if let Some(headers) = &self.headers {
1db9dbe390 2024-11-28  127: 			let mail = mail_parser::MessageParser::new().parse(&self.data)
1db9dbe390 2024-11-28  128: 				.ok_or(anyhow!("Failed to parse mail"))?;
1db9dbe390 2024-11-28  129: 
1db9dbe390 2024-11-28  130: 			// Adding all known addresses to recipient list, for anyone else adding default
1db9dbe390 2024-11-28  131: 			// Also if list is empty also adding default
1db9dbe390 2024-11-28  132: 			let mut rcpt: HashSet<&ChatId> = HashSet::new();
1db9dbe390 2024-11-28  133: 			if headers.to.is_empty() {
1db9dbe390 2024-11-28  134: 				bail!("No recipient addresses.");
1db9dbe390 2024-11-28  135: 			}
1db9dbe390 2024-11-28  136: 			for item in &headers.to {
1db9dbe390 2024-11-28  137: 				match self.recipients.get(item) {
1db9dbe390 2024-11-28  138: 					Some(addr) => rcpt.insert(addr),
1db9dbe390 2024-11-28  139: 					None => {
7f8a9994b3 2024-12-10  140: 						self.debug(format!("Recipient [{}] not found\\.", &item)).await?;
1db9dbe390 2024-11-28  141: 						rcpt.insert(self.recipients.get("_")
7f8a9994b3 2024-12-10  142: 							.ok_or(anyhow!("Missing default address in recipient table\\."))?)
1db9dbe390 2024-11-28  143: 					}
1db9dbe390 2024-11-28  144: 				};
1db9dbe390 2024-11-28  145: 			};
1db9dbe390 2024-11-28  146: 			if rcpt.is_empty() {
7f8a9994b3 2024-12-10  147: 				self.debug("No recipient or envelope address\\.").await?;
1db9dbe390 2024-11-28  148: 				rcpt.insert(self.recipients.get("_")
1db9dbe390 2024-11-28  149: 					.ok_or(anyhow!("Missing default address in recipient table."))?);
1db9dbe390 2024-11-28  150: 			};
1db9dbe390 2024-11-28  151: 
1db9dbe390 2024-11-28  152: 			// prepating message header
1db9dbe390 2024-11-28  153: 			let mut reply: Vec<Cow<'_, str>> = vec![];
28fde40f7a 2024-12-27  154: 			if self.fields.contains("subject") {
28fde40f7a 2024-12-27  155: 				if let Some(subject) = mail.subject() {
3d13e7f81d 2024-12-28  156: 					reply.push(format!("__*Subject:*__ `{}`", subject).into());
28fde40f7a 2024-12-27  157: 				} else if let Some(thread) = mail.thread_name() {
3d13e7f81d 2024-12-28  158: 					reply.push(format!("__*Thread:*__ `{}`", thread).into());
28fde40f7a 2024-12-27  159: 				}
28fde40f7a 2024-12-27  160: 			}
3d13e7f81d 2024-12-28  161: 			let mut short_headers: Vec<Cow<'_, str>> = vec![];
3d13e7f81d 2024-12-28  162: 			// do we need to replace spaces here?
28fde40f7a 2024-12-27  163: 			if self.fields.contains("from") {
3d13e7f81d 2024-12-28  164: 				short_headers.push(format!("__*From:*__ `{}`", headers.from).into());
28fde40f7a 2024-12-27  165: 			}
28fde40f7a 2024-12-27  166: 			if self.fields.contains("date") {
28fde40f7a 2024-12-27  167: 				if let Some(date) = mail.date() {
3d13e7f81d 2024-12-28  168: 					short_headers.push(format!("__*Date:*__ `{}`", date).into());
28fde40f7a 2024-12-27  169: 				}
28fde40f7a 2024-12-27  170: 			}
3d13e7f81d 2024-12-28  171: 			reply.push(short_headers.join(" ").into());
3d13e7f81d 2024-12-28  172: 			let header_size = reply.join(" ").len() + 1;
1db9dbe390 2024-11-28  173: 
1db9dbe390 2024-11-28  174: 			let html_parts = mail.html_body_count();
1db9dbe390 2024-11-28  175: 			let text_parts = mail.text_body_count();
1db9dbe390 2024-11-28  176: 			let attachments = mail.attachment_count();
1db9dbe390 2024-11-28  177: 			if html_parts != text_parts {
7f8a9994b3 2024-12-10  178: 				self.debug(format!("Hm, we have {} HTML parts and {} text parts\\.", html_parts, text_parts)).await?;
1db9dbe390 2024-11-28  179: 			}
1db9dbe390 2024-11-28  180: 			//let mut html_num = 0;
1db9dbe390 2024-11-28  181: 			let mut text_num = 0;
1db9dbe390 2024-11-28  182: 			let mut file_num = 0;
1db9dbe390 2024-11-28  183: 			// let's display first html or text part as body
1db9dbe390 2024-11-28  184: 			let mut body = "".into();
1db9dbe390 2024-11-28  185: 			/*
1db9dbe390 2024-11-28  186: 			 * actually I don't wanna parse that html stuff
1db9dbe390 2024-11-28  187: 			if html_parts > 0 {
1db9dbe390 2024-11-28  188: 				let text = mail.body_html(0).unwrap();
1db9dbe390 2024-11-28  189: 				if text.len() < 4096 - header_size {
1db9dbe390 2024-11-28  190: 					body = text;
1db9dbe390 2024-11-28  191: 					html_num = 1;
1db9dbe390 2024-11-28  192: 				}
1db9dbe390 2024-11-28  193: 			};
1db9dbe390 2024-11-28  194: 			*/
1db9dbe390 2024-11-28  195: 			if body == "" && text_parts > 0 {
1db9dbe390 2024-11-28  196: 				let text = mail.body_text(0)
1db9dbe390 2024-11-28  197: 					.ok_or(anyhow!("Failed to extract text from message."))?;
1db9dbe390 2024-11-28  198: 				if text.len() < 4096 - header_size {
1db9dbe390 2024-11-28  199: 					body = text;
1db9dbe390 2024-11-28  200: 					text_num = 1;
1db9dbe390 2024-11-28  201: 				}
1db9dbe390 2024-11-28  202: 			};
1db9dbe390 2024-11-28  203: 			reply.push("```".into());
1db9dbe390 2024-11-28  204: 			reply.extend(body.lines().map(|x| x.into()));
1db9dbe390 2024-11-28  205: 			reply.push("```".into());
1db9dbe390 2024-11-28  206: 
1db9dbe390 2024-11-28  207: 			// and let's collect all other attachment parts
1db9dbe390 2024-11-28  208: 			let mut files_to_send = vec![];
1db9dbe390 2024-11-28  209: 			/*
1db9dbe390 2024-11-28  210: 			 * let's just skip html parts for now, they just duplicate text?
1db9dbe390 2024-11-28  211: 			while html_num < html_parts {
1db9dbe390 2024-11-28  212: 				files_to_send.push(mail.html_part(html_num).unwrap());
1db9dbe390 2024-11-28  213: 				html_num += 1;
1db9dbe390 2024-11-28  214: 			}
1db9dbe390 2024-11-28  215: 			*/
1db9dbe390 2024-11-28  216: 			while text_num < text_parts {
1db9dbe390 2024-11-28  217: 				files_to_send.push(mail.text_part(text_num)
1db9dbe390 2024-11-28  218: 					.ok_or(anyhow!("Failed to get text part from message"))?);
1db9dbe390 2024-11-28  219: 				text_num += 1;
1db9dbe390 2024-11-28  220: 			}
1db9dbe390 2024-11-28  221: 			while file_num < attachments {
1db9dbe390 2024-11-28  222: 				files_to_send.push(mail.attachment(file_num)
1db9dbe390 2024-11-28  223: 					.ok_or(anyhow!("Failed to get file part from message"))?);
1db9dbe390 2024-11-28  224: 				file_num += 1;
1db9dbe390 2024-11-28  225: 			}
1db9dbe390 2024-11-28  226: 
1db9dbe390 2024-11-28  227: 			let msg = reply.join("\n");
1db9dbe390 2024-11-28  228: 			for chat in rcpt {
1db9dbe390 2024-11-28  229: 				if !files_to_send.is_empty() {
1db9dbe390 2024-11-28  230: 					let mut files = vec![];
1db9dbe390 2024-11-28  231: 					let mut first_one = true;
1db9dbe390 2024-11-28  232: 					for chunk in &files_to_send {
1db9dbe390 2024-11-28  233: 						let data = chunk.contents();
1db9dbe390 2024-11-28  234: 						let mut filename: Option<String> = None;
1db9dbe390 2024-11-28  235: 						for header in chunk.headers() {
1db9dbe390 2024-11-28  236: 							if header.name() == "Content-Type" {
1db9dbe390 2024-11-28  237: 								match header.value() {
1db9dbe390 2024-11-28  238: 									mail_parser::HeaderValue::ContentType(contenttype) => {
1db9dbe390 2024-11-28  239: 										if let Some(fname) = contenttype.attribute("name") {
1db9dbe390 2024-11-28  240: 											filename = Some(fname.to_owned());
1db9dbe390 2024-11-28  241: 										}
1db9dbe390 2024-11-28  242: 									},
1db9dbe390 2024-11-28  243: 									_ => {
562951709a 2024-12-11  244: 										self.debug("Attachment has bad ContentType header\\.").await?;
1db9dbe390 2024-11-28  245: 									},
1db9dbe390 2024-11-28  246: 								};
1db9dbe390 2024-11-28  247: 							};
1db9dbe390 2024-11-28  248: 						};
1db9dbe390 2024-11-28  249: 						let filename = if let Some(fname) = filename {
1db9dbe390 2024-11-28  250: 							fname
1db9dbe390 2024-11-28  251: 						} else {
1db9dbe390 2024-11-28  252: 							"Attachment.txt".into()
1db9dbe390 2024-11-28  253: 						};
1db9dbe390 2024-11-28  254: 						let item = teloxide::types::InputMediaDocument::new(
1db9dbe390 2024-11-28  255: 							teloxide::types::InputFile::memory(data.to_vec())
1db9dbe390 2024-11-28  256: 							.file_name(filename));
1db9dbe390 2024-11-28  257: 						let item = if first_one {
1db9dbe390 2024-11-28  258: 							first_one = false;
1db9dbe390 2024-11-28  259: 							item.caption(&msg).parse_mode(MarkdownV2)
1db9dbe390 2024-11-28  260: 						} else {
1db9dbe390 2024-11-28  261: 							item
1db9dbe390 2024-11-28  262: 						};
1db9dbe390 2024-11-28  263: 						files.push(InputMedia::Document(item));
1db9dbe390 2024-11-28  264: 					}
1db9dbe390 2024-11-28  265: 					self.sendgroup(chat, files).await?;
1db9dbe390 2024-11-28  266: 				} else {
1db9dbe390 2024-11-28  267: 					self.send(chat, &msg).await?;
1db9dbe390 2024-11-28  268: 				}
1db9dbe390 2024-11-28  269: 			}
1db9dbe390 2024-11-28  270: 		} else {
1db9dbe390 2024-11-28  271: 			bail!("No headers.");
1db9dbe390 2024-11-28  272: 		}
1db9dbe390 2024-11-28  273: 		Ok(())
1db9dbe390 2024-11-28  274: 	}
1db9dbe390 2024-11-28  275: 
1db9dbe390 2024-11-28  276: 	/// Send media to specified user
866aad57a4 2024-06-15  277: 	pub async fn sendgroup<M>(&self, to: &ChatId, media: M) -> Result<Vec<Message>>
866aad57a4 2024-06-15  278: 	where M: IntoIterator<Item = InputMedia> {
f4cad2a5c0 2024-05-26  279: 		Ok(self.tg.send_media_group(*to, media).await?)
f4cad2a5c0 2024-05-26  280: 	}
f4cad2a5c0 2024-05-26  281: }
f4cad2a5c0 2024-05-26  282: 
1db9dbe390 2024-11-28  283: impl mailin_embedded::Handler for TelegramTransport {
1db9dbe390 2024-11-28  284: 	/// Just deny login auth
1db9dbe390 2024-11-28  285: 	fn auth_login (&mut self, _username: &str, _password: &str) -> Response {
1db9dbe390 2024-11-28  286: 		INVALID_CREDENTIALS
1db9dbe390 2024-11-28  287: 	}
1db9dbe390 2024-11-28  288: 
1db9dbe390 2024-11-28  289: 	/// Just deny plain auth
1db9dbe390 2024-11-28  290: 	fn auth_plain (&mut self, _authorization_id: &str, _authentication_id: &str, _password: &str) -> Response {
1db9dbe390 2024-11-28  291: 		INVALID_CREDENTIALS
1db9dbe390 2024-11-28  292: 	}
1db9dbe390 2024-11-28  293: 
1db9dbe390 2024-11-28  294: 	/// Verify whether address is deliverable
1db9dbe390 2024-11-28  295: 	fn rcpt (&mut self, to: &str) -> Response {
1db9dbe390 2024-11-28  296: 		if self.relay {
1db9dbe390 2024-11-28  297: 			OK
1db9dbe390 2024-11-28  298: 		} else {
1db9dbe390 2024-11-28  299: 			match self.recipients.get(to) {
1db9dbe390 2024-11-28  300: 				Some(_) => OK,
1db9dbe390 2024-11-28  301: 				None => {
1db9dbe390 2024-11-28  302: 					if self.relay {
1db9dbe390 2024-11-28  303: 						OK
1db9dbe390 2024-11-28  304: 					} else {
1db9dbe390 2024-11-28  305: 						NO_MAILBOX
1db9dbe390 2024-11-28  306: 					}
1db9dbe390 2024-11-28  307: 				}
1db9dbe390 2024-11-28  308: 			}
1db9dbe390 2024-11-28  309: 		}
1db9dbe390 2024-11-28  310: 	}
1db9dbe390 2024-11-28  311: 
1db9dbe390 2024-11-28  312: 	/// Save headers we need
1db9dbe390 2024-11-28  313: 	fn data_start (&mut self, _domain: &str, from: &str, _is8bit: bool, to: &[String]) -> Response {
1db9dbe390 2024-11-28  314: 		self.headers = Some(SomeHeaders{
1db9dbe390 2024-11-28  315: 			from: from.to_string(),
1db9dbe390 2024-11-28  316: 			to: to.to_vec(),
1db9dbe390 2024-11-28  317: 		});
1db9dbe390 2024-11-28  318: 		OK
1db9dbe390 2024-11-28  319: 	}
1db9dbe390 2024-11-28  320: 
1db9dbe390 2024-11-28  321: 	/// Save chunk(?) of data
1db9dbe390 2024-11-28  322: 	fn data(&mut self, buf: &[u8]) -> Result<(), Error> {
1db9dbe390 2024-11-28  323: 		self.data.append(buf.to_vec().as_mut());
1db9dbe390 2024-11-28  324: 		Ok(())
1db9dbe390 2024-11-28  325: 	}
1db9dbe390 2024-11-28  326: 
1db9dbe390 2024-11-28  327: 	/// Attempt to send email, return temporary error if that fails
1db9dbe390 2024-11-28  328: 	fn data_end(&mut self) -> Response {
1db9dbe390 2024-11-28  329: 		let mut result = OK;
1db9dbe390 2024-11-28  330: 		task::block_on(async {
1db9dbe390 2024-11-28  331: 			// relay mail
1db9dbe390 2024-11-28  332: 			if let Err(err) = self.relay_mail().await {
1db9dbe390 2024-11-28  333: 				result = INTERNAL_ERROR;
1db9dbe390 2024-11-28  334: 				// in case that fails - inform default recipient
1db9dbe390 2024-11-28  335: 				if let Err(err) = self.debug(format!("Sending emails failed:\n{:?}", err)).await {
37a0139d49 2024-05-26  336: 					// in case that also fails - write some logs and bail
37a0139d49 2024-05-26  337: 					eprintln!("Failed to contact Telegram:\n{:?}", err);
37a0139d49 2024-05-26  338: 				};
1db9dbe390 2024-11-28  339: 			};
1db9dbe390 2024-11-28  340: 		});
1db9dbe390 2024-11-28  341: 		// clear - just in case
1db9dbe390 2024-11-28  342: 		self.data = vec![];
1db9dbe390 2024-11-28  343: 		self.headers = None;
1db9dbe390 2024-11-28  344: 		result
1db9dbe390 2024-11-28  345: 	}
1db9dbe390 2024-11-28  346: }
1db9dbe390 2024-11-28  347: 
1db9dbe390 2024-11-28  348: #[async_std::main]
1db9dbe390 2024-11-28  349: async fn main() -> Result<()> {
e66352b9cc 2025-01-21  350: 	let specs = OptSpecs::new()
e66352b9cc 2025-01-21  351: 		.option("help", "h", OptValueType::None)
e66352b9cc 2025-01-21  352: 		.option("help", "help", OptValueType::None)
e66352b9cc 2025-01-21  353: 		.option("config", "c", OptValueType::Required)
e66352b9cc 2025-01-21  354: 		.option("config", "config", OptValueType::Required)
e66352b9cc 2025-01-21  355: 		.flag(OptFlags::OptionsEverywhere);
e66352b9cc 2025-01-21  356: 	let mut args = std::env::args();
e66352b9cc 2025-01-21  357: 	args.next();
e66352b9cc 2025-01-21  358: 	let parsed = specs.getopt(args);
e66352b9cc 2025-01-21  359: 	for u in &parsed.unknown {
e66352b9cc 2025-01-21  360: 		println!("Unknown option: {}", u);
e66352b9cc 2025-01-21  361: 	}
e66352b9cc 2025-01-21  362: 	if !(parsed.unknown.is_empty()) || parsed.options_first("help").is_some() {
e66352b9cc 2025-01-21  363: 		println!("SMTP2TG v{}, (C) 2024 - 2025\n\n\
e66352b9cc 2025-01-21  364: 			\t-h|--help\tDisplay this help\n\
e66352b9cc 2025-01-21  365: 			\t-c|-config …\tSet configuration file location.",
e66352b9cc 2025-01-21  366: 			env!("CARGO_PKG_VERSION"));
e66352b9cc 2025-01-21  367: 		return Ok(());
e66352b9cc 2025-01-21  368: 	};
e66352b9cc 2025-01-21  369: 	let config_file = Path::new(if let Some(path) = parsed.options_value_last("config") {
e66352b9cc 2025-01-21  370: 		&path[..]
e66352b9cc 2025-01-21  371: 	} else {
e66352b9cc 2025-01-21  372: 		"smtp2tg.toml"
e66352b9cc 2025-01-21  373: 	});
e66352b9cc 2025-01-21  374: 	if !config_file.exists() {
e66352b9cc 2025-01-21  375: 		eprintln!("Error: can't read configuration from {:?}", config_file);
e66352b9cc 2025-01-21  376: 		std::process::exit(1);
e66352b9cc 2025-01-21  377: 	};
1db9dbe390 2024-11-28  378: 	let settings: config::Config = config::Config::builder()
28fde40f7a 2024-12-27  379: 		.set_default("fields", vec!["date", "from", "subject"]).unwrap()
1db9dbe390 2024-11-28  380: 		.set_default("hostname", "smtp.2.tg").unwrap()
28fde40f7a 2024-12-27  381: 		.set_default("listen_on", "0.0.0.0:1025").unwrap()
1db9dbe390 2024-11-28  382: 		.set_default("unknown", "relay").unwrap()
e66352b9cc 2025-01-21  383: 		.add_source(config::File::from(config_file))
1db9dbe390 2024-11-28  384: 		.build()
e66352b9cc 2025-01-21  385: 		.expect(&format!("[{:?}] there was an error reading config\n\
e66352b9cc 2025-01-21  386: 			\tplease consult \"smtp2tg.toml.example\" for details",
e66352b9cc 2025-01-21  387: 			config_file)[..]);
1db9dbe390 2024-11-28  388: 
1db9dbe390 2024-11-28  389: 	let listen_on = settings.get_string("listen_on")?;
1db9dbe390 2024-11-28  390: 	let server_name = settings.get_string("hostname")?;
1db9dbe390 2024-11-28  391: 	let core = TelegramTransport::new(settings);
1db9dbe390 2024-11-28  392: 	let mut server = mailin_embedded::Server::new(core);
1db9dbe390 2024-11-28  393: 
1db9dbe390 2024-11-28  394: 	server.with_name(server_name)
1db9dbe390 2024-11-28  395: 		.with_ssl(mailin_embedded::SslConfig::None).unwrap()
1db9dbe390 2024-11-28  396: 		.with_addr(listen_on).unwrap();
1db9dbe390 2024-11-28  397: 	server.serve().unwrap();
1db9dbe390 2024-11-28  398: 
1db9dbe390 2024-11-28  399: 	Ok(())
7620f854a7 2024-05-21  400: }