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