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