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