7620f854a7 2024-05-21 arcade: use anyhow::Result;
7620f854a7 2024-05-21 arcade: use async_std::task;
7620f854a7 2024-05-21 arcade: //use async_trait::async_trait;
7620f854a7 2024-05-21 arcade: //use futures::io::AsyncRead;
7620f854a7 2024-05-21 arcade: //use mail_parser::Message;
7620f854a7 2024-05-21 arcade: use samotop::{
7620f854a7 2024-05-21 arcade: mail::{
7620f854a7 2024-05-21 arcade: Builder,
7620f854a7 2024-05-21 arcade: DebugService,
7620f854a7 2024-05-21 arcade: MailDir,
7620f854a7 2024-05-21 arcade: Name
7620f854a7 2024-05-21 arcade: },
7620f854a7 2024-05-21 arcade: smtp::Prudence,
7620f854a7 2024-05-21 arcade: };
7620f854a7 2024-05-21 arcade: use telegram_bot::{
7620f854a7 2024-05-21 arcade: Api,
7620f854a7 2024-05-21 arcade: ParseMode,
7620f854a7 2024-05-21 arcade: SendMessage,
7620f854a7 2024-05-21 arcade: UserId,
7620f854a7 2024-05-21 arcade: };
7620f854a7 2024-05-21 arcade:
7620f854a7 2024-05-21 arcade: use std::{
7620f854a7 2024-05-21 arcade: borrow::Cow,
7620f854a7 2024-05-21 arcade: collections::HashMap,
7620f854a7 2024-05-21 arcade: io::Read,
7620f854a7 2024-05-21 arcade: path::{
7620f854a7 2024-05-21 arcade: Path,
7620f854a7 2024-05-21 arcade: PathBuf
7620f854a7 2024-05-21 arcade: },
7620f854a7 2024-05-21 arcade: time::Duration,
7620f854a7 2024-05-21 arcade: vec::Vec,
7620f854a7 2024-05-21 arcade: };
7620f854a7 2024-05-21 arcade:
7620f854a7 2024-05-21 arcade:
7620f854a7 2024-05-21 arcade: fn relay_mails(maildir: &Path, core: &Core) -> Result<()> {
7620f854a7 2024-05-21 arcade: use mail_parser::*;
7620f854a7 2024-05-21 arcade:
7620f854a7 2024-05-21 arcade: let new_dir = maildir.join("new");
7620f854a7 2024-05-21 arcade:
7620f854a7 2024-05-21 arcade: std::fs::create_dir_all(&new_dir)?;
7620f854a7 2024-05-21 arcade:
7620f854a7 2024-05-21 arcade: let files = std::fs::read_dir(new_dir)?;
7620f854a7 2024-05-21 arcade: for file in files {
7620f854a7 2024-05-21 arcade: dbg!(&file);
7620f854a7 2024-05-21 arcade: let file = file?;
7620f854a7 2024-05-21 arcade: let mut buf = Vec::new();
7620f854a7 2024-05-21 arcade: std::fs::File::open(file.path())?.read_to_end(&mut buf)?;
7620f854a7 2024-05-21 arcade:
7620f854a7 2024-05-21 arcade: task::block_on(async move {
7620f854a7 2024-05-21 arcade: match MessageParser::default().parse(&buf[..]) {
7620f854a7 2024-05-21 arcade: Some(mail) => {
7620f854a7 2024-05-21 arcade: /*
7620f854a7 2024-05-21 arcade: dbg!(&mail);
7620f854a7 2024-05-21 arcade: let to = match mail.to() {
7620f854a7 2024-05-21 arcade: Some(mail) => mail.into_list().into_iter().map(|a| a.address.unwrap()).collect(),
7620f854a7 2024-05-21 arcade: None => match mail.header("X-Samotop-To").unwrap() {
7620f854a7 2024-05-21 arcade: mail_parser::HeaderValue::Address(addr) => addr.address.unwrap(),
7620f854a7 2024-05-21 arcade: },
7620f854a7 2024-05-21 arcade: };
7620f854a7 2024-05-21 arcade: dbg!(&to);
7620f854a7 2024-05-21 arcade: */
7620f854a7 2024-05-21 arcade: },
7620f854a7 2024-05-21 arcade: None => { core.debug("None mail.").await.unwrap(); },
7620f854a7 2024-05-21 arcade: //send_to_sendgrid(mail, sendgrid_api_key).await;
7620f854a7 2024-05-21 arcade: };
7620f854a7 2024-05-21 arcade: });
7620f854a7 2024-05-21 arcade:
7620f854a7 2024-05-21 arcade: std::fs::remove_file(file.path())?;
7620f854a7 2024-05-21 arcade: }
7620f854a7 2024-05-21 arcade: Ok(())
7620f854a7 2024-05-21 arcade: }
7620f854a7 2024-05-21 arcade:
7620f854a7 2024-05-21 arcade: fn my_prudence() -> Prudence {
7620f854a7 2024-05-21 arcade: Prudence::default().with_read_timeout(Duration::from_secs(60)).with_banner_delay(Duration::from_secs(1))
7620f854a7 2024-05-21 arcade: }
7620f854a7 2024-05-21 arcade:
7620f854a7 2024-05-21 arcade: pub struct Core {
7620f854a7 2024-05-21 arcade: default: UserId,
7620f854a7 2024-05-21 arcade: tg: Api,
7620f854a7 2024-05-21 arcade: recipients: HashMap<String, UserId>,
7620f854a7 2024-05-21 arcade: }
7620f854a7 2024-05-21 arcade:
7620f854a7 2024-05-21 arcade: impl Core {
7620f854a7 2024-05-21 arcade: pub fn new(settings: &config::Config) -> Result<Core> {
7620f854a7 2024-05-21 arcade: let api_key = settings.get_string("api_key").unwrap();
7620f854a7 2024-05-21 arcade: let tg = Api::new(api_key);
7620f854a7 2024-05-21 arcade: let default_recipient = settings.get_string("default")?;
7620f854a7 2024-05-21 arcade: let recipients: HashMap<String, UserId> = settings.get_table("recipients")?.into_iter().map(|(a, b)| (a, UserId::new(b.into_int().unwrap()))).collect();
7620f854a7 2024-05-21 arcade: let default = recipients[&default_recipient];
7620f854a7 2024-05-21 arcade:
7620f854a7 2024-05-21 arcade: Ok(Core {
7620f854a7 2024-05-21 arcade: default,
7620f854a7 2024-05-21 arcade: tg,
7620f854a7 2024-05-21 arcade: recipients,
7620f854a7 2024-05-21 arcade: })
7620f854a7 2024-05-21 arcade: }
7620f854a7 2024-05-21 arcade:
7620f854a7 2024-05-21 arcade: pub async fn debug<'b, S>(&self, msg: S) -> Result<()>
7620f854a7 2024-05-21 arcade: where S: Into<Cow<'b, str>> {
7620f854a7 2024-05-21 arcade: self.tg.send(SendMessage::new(self.default, msg)
7620f854a7 2024-05-21 arcade: .parse_mode(ParseMode::Markdown)).await?;
7620f854a7 2024-05-21 arcade: Ok(())
7620f854a7 2024-05-21 arcade: }
7620f854a7 2024-05-21 arcade:
7620f854a7 2024-05-21 arcade: pub async fn send<'b, S>(&self, to: String, msg: S) -> Result<()>
7620f854a7 2024-05-21 arcade: where S: Into<Cow<'b, str>> {
7620f854a7 2024-05-21 arcade: self.tg.send(SendMessage::new(self.recipients[&to], msg)
7620f854a7 2024-05-21 arcade: .parse_mode(ParseMode::Markdown)).await?;
7620f854a7 2024-05-21 arcade: Ok(())
7620f854a7 2024-05-21 arcade: }
7620f854a7 2024-05-21 arcade: }
7620f854a7 2024-05-21 arcade:
7620f854a7 2024-05-21 arcade: #[async_std::main]
7620f854a7 2024-05-21 arcade: async fn main() {
7620f854a7 2024-05-21 arcade: let settings: config::Config = config::Config::builder()
7620f854a7 2024-05-21 arcade: .add_source(config::File::with_name("smtp2tg.toml"))
7620f854a7 2024-05-21 arcade: .build().unwrap();
7620f854a7 2024-05-21 arcade:
7620f854a7 2024-05-21 arcade: let core = Core::new(&settings).unwrap();
7620f854a7 2024-05-21 arcade: let maildir: PathBuf = settings.get_string("maildir").unwrap().into();
7620f854a7 2024-05-21 arcade: let addr = "./smtp2tg.sock";
7620f854a7 2024-05-21 arcade: let listen_on = settings.get_string("listen_on").unwrap();
7620f854a7 2024-05-21 arcade: let sink = Builder + Name::new("smtp2tg") + DebugService +
7620f854a7 2024-05-21 arcade: samotop::smtp::Esmtp.with(samotop::smtp::SmtpParser) + my_prudence() +
7620f854a7 2024-05-21 arcade: MailDir::new(maildir.clone()).unwrap();
7620f854a7 2024-05-21 arcade:
7620f854a7 2024-05-21 arcade: task::spawn(async move {
7620f854a7 2024-05-21 arcade: loop {
7620f854a7 2024-05-21 arcade: task::sleep(Duration::from_secs(5)).await;
7620f854a7 2024-05-21 arcade: relay_mails(&maildir, &core).unwrap();
7620f854a7 2024-05-21 arcade: }
7620f854a7 2024-05-21 arcade: });
7620f854a7 2024-05-21 arcade:
7620f854a7 2024-05-21 arcade: match listen_on.as_str() {
7620f854a7 2024-05-21 arcade: "socket" => samotop::server::UnixServer::on("./smtp2tg.sock")
7620f854a7 2024-05-21 arcade: .serve(sink.build()).await.unwrap(),
7620f854a7 2024-05-21 arcade: _ => samotop::server::TcpServer::on(listen_on)
7620f854a7 2024-05-21 arcade: .serve(sink.build()).await.unwrap(),
7620f854a7 2024-05-21 arcade: };
7620f854a7 2024-05-21 arcade: /*
7620f854a7 2024-05-21 arcade: task::block_on(async {
7620f854a7 2024-05-21 arcade: let be = MyBackend;
7620f854a7 2024-05-21 arcade:
7620f854a7 2024-05-21 arcade: //let mut s = Server::new(be);
7620f854a7 2024-05-21 arcade:
7620f854a7 2024-05-21 arcade: s.addr = "127.0.0.1:2525".to_string();
7620f854a7 2024-05-21 arcade: s.domain = "localhost".to_string();
7620f854a7 2024-05-21 arcade: s.read_timeout = std::time::Duration::from_secs(10);
7620f854a7 2024-05-21 arcade: s.write_timeout = std::time::Duration::from_secs(10);
7620f854a7 2024-05-21 arcade: s.max_message_bytes = 10 * 1024 * 1024;
7620f854a7 2024-05-21 arcade: s.max_recipients = 50;
7620f854a7 2024-05-21 arcade: s.max_line_length = 1000;
7620f854a7 2024-05-21 arcade: s.allow_insecure_auth = true;
7620f854a7 2024-05-21 arcade:
7620f854a7 2024-05-21 arcade: println!("Starting server on {}", s.addr);
7620f854a7 2024-05-21 arcade: match s.listen_and_serve().await {
7620f854a7 2024-05-21 arcade: Ok(_) => println!("Server stopped"),
7620f854a7 2024-05-21 arcade: Err(e) => println!("Server error: {}", e),
7620f854a7 2024-05-21 arcade: }
7620f854a7 2024-05-21 arcade: Ok(())
7620f854a7 2024-05-21 arcade: })
7620f854a7 2024-05-21 arcade: */
7620f854a7 2024-05-21 arcade: }