Artifact
9de705aadb5f9d11de42c51711e3ce92f3e20ac9c4f4f4b209ec11ab7b233d05:
0000: 75 73 65 20 63 72 61 74 65 3a 3a 75 74 69 6c 73 use crate::utils
0010: 3a 3a 7b 0a 09 41 74 74 61 63 68 6d 65 6e 74 2c ::{..Attachment,
0020: 0a 09 76 61 6c 69 64 61 74 65 2c 0a 7d 3b 0a 0a ..validate,.};..
0030: 75 73 65 20 73 74 64 3a 3a 7b 0a 09 63 6f 6c 6c use std::{..coll
0040: 65 63 74 69 6f 6e 73 3a 3a 48 61 73 68 4d 61 70 ections::HashMap
0050: 2c 0a 09 66 6d 74 3a 3a 44 65 62 75 67 2c 0a 7d ,..fmt::Debug,.}
0060: 3b 0a 0a 75 73 65 20 73 74 61 63 6b 65 64 5f 65 ;..use stacked_e
0070: 72 72 6f 72 73 3a 3a 7b 0a 09 52 65 73 75 6c 74 rrors::{..Result
0080: 2c 0a 09 53 74 61 63 6b 61 62 6c 65 45 72 72 2c ,..StackableErr,
0090: 0a 7d 3b 0a 75 73 65 20 74 67 62 6f 74 3a 3a 7b .};.use tgbot::{
00a0: 0a 09 61 70 69 3a 3a 43 6c 69 65 6e 74 2c 0a 09 ..api::Client,..
00b0: 74 79 70 65 73 3a 3a 7b 0a 09 09 43 68 61 74 50 types::{...ChatP
00c0: 65 65 72 49 64 2c 0a 09 09 49 6e 70 75 74 46 69 eerId,...InputFi
00d0: 6c 65 2c 0a 09 09 49 6e 70 75 74 46 69 6c 65 52 le,...InputFileR
00e0: 65 61 64 65 72 2c 0a 09 09 49 6e 70 75 74 4d 65 eader,...InputMe
00f0: 64 69 61 44 6f 63 75 6d 65 6e 74 2c 0a 09 09 4d diaDocument,...M
0100: 65 64 69 61 47 72 6f 75 70 2c 0a 09 09 4d 65 64 ediaGroup,...Med
0110: 69 61 47 72 6f 75 70 49 74 65 6d 2c 0a 09 09 4d iaGroupItem,...M
0120: 65 73 73 61 67 65 2c 0a 09 09 50 61 72 73 65 4d essage,...ParseM
0130: 6f 64 65 3a 3a 48 74 6d 6c 2c 0a 09 09 53 65 6e ode::Html,...Sen
0140: 64 4d 65 64 69 61 47 72 6f 75 70 2c 0a 09 09 53 dMediaGroup,...S
0150: 65 6e 64 4d 65 73 73 61 67 65 2c 0a 09 09 53 65 endMessage,...Se
0160: 6e 64 44 6f 63 75 6d 65 6e 74 2c 0a 09 7d 2c 0a ndDocument,..},.
0170: 7d 3b 0a 0a 23 5b 64 65 72 69 76 65 28 44 65 62 };..#[derive(Deb
0180: 75 67 29 5d 0a 70 75 62 20 73 74 72 75 63 74 20 ug)].pub struct
0190: 54 65 6c 65 67 72 61 6d 54 72 61 6e 73 70 6f 72 TelegramTranspor
01a0: 74 20 7b 0a 09 74 67 3a 20 43 6c 69 65 6e 74 2c t {..tg: Client,
01b0: 0a 09 72 65 63 69 70 69 65 6e 74 73 3a 20 48 61 ..recipients: Ha
01c0: 73 68 4d 61 70 3c 53 74 72 69 6e 67 2c 20 43 68 shMap<String, Ch
01d0: 61 74 50 65 65 72 49 64 3e 2c 0a 09 70 75 62 20 atPeerId>,..pub
01e0: 64 65 66 61 75 6c 74 3a 20 43 68 61 74 50 65 65 default: ChatPee
01f0: 72 49 64 2c 0a 7d 0a 0a 69 6d 70 6c 20 54 65 6c rId,.}..impl Tel
0200: 65 67 72 61 6d 54 72 61 6e 73 70 6f 72 74 20 7b egramTransport {
0210: 0a 09 2f 2f 2f 20 43 72 65 61 74 65 73 20 6e 65 ../// Creates ne
0220: 77 20 54 65 6c 65 67 72 61 6d 54 72 61 6e 73 70 w TelegramTransp
0230: 6f 72 74 20 6f 62 6a 65 63 74 2e 0a 09 70 75 62 ort object...pub
0240: 20 66 6e 20 6e 65 77 20 28 61 70 69 5f 6b 65 79 fn new (api_key
0250: 3a 20 53 74 72 69 6e 67 2c 20 72 65 63 69 70 69 : String, recipi
0260: 65 6e 74 73 3a 20 48 61 73 68 4d 61 70 3c 53 74 ents: HashMap<St
0270: 72 69 6e 67 2c 20 69 36 34 3e 2c 20 73 65 74 74 ring, i64>, sett
0280: 69 6e 67 73 3a 20 26 63 6f 6e 66 69 67 3a 3a 43 ings: &config::C
0290: 6f 6e 66 69 67 29 20 2d 3e 20 52 65 73 75 6c 74 onfig) -> Result
02a0: 3c 54 65 6c 65 67 72 61 6d 54 72 61 6e 73 70 6f <TelegramTranspo
02b0: 72 74 3e 20 7b 0a 09 09 6c 65 74 20 64 65 66 61 rt> {...let defa
02c0: 75 6c 74 20 3d 20 73 65 74 74 69 6e 67 73 2e 67 ult = settings.g
02d0: 65 74 5f 69 6e 74 28 22 64 65 66 61 75 6c 74 22 et_int("default"
02e0: 29 0a 09 09 09 2e 63 6f 6e 74 65 78 74 28 22 5b ).....context("[
02f0: 73 6d 74 70 32 74 67 2e 74 6f 6d 6c 5d 20 6d 69 smtp2tg.toml] mi
0300: 73 73 69 6e 67 20 5c 22 64 65 66 61 75 6c 74 5c ssing \"default\
0310: 22 20 72 65 63 69 70 69 65 6e 74 2e 5c 6e 22 29 " recipient.\n")
0320: 3f 3b 0a 09 09 6c 65 74 20 61 70 69 5f 67 61 74 ?;...let api_gat
0330: 65 77 61 79 20 3d 20 73 65 74 74 69 6e 67 73 2e eway = settings.
0340: 67 65 74 5f 73 74 72 69 6e 67 28 22 61 70 69 5f get_string("api_
0350: 67 61 74 65 77 61 79 22 29 0a 09 09 09 2e 63 6f gateway").....co
0360: 6e 74 65 78 74 28 22 5b 73 6d 74 70 32 74 67 2e ntext("[smtp2tg.
0370: 74 6f 6d 6c 5d 20 6d 69 73 73 69 6e 67 20 5c 22 toml] missing \"
0380: 61 70 69 5f 67 61 74 65 77 61 79 5c 22 20 64 65 api_gateway\" de
0390: 73 74 69 6e 61 74 69 6f 6e 2e 5c 6e 22 29 3f 3b stination.\n")?;
03a0: 0a 0a 09 09 6c 65 74 20 74 67 20 3d 20 43 6c 69 ....let tg = Cli
03b0: 65 6e 74 3a 3a 6e 65 77 28 61 70 69 5f 6b 65 79 ent::new(api_key
03c0: 29 0a 09 09 09 2e 63 6f 6e 74 65 78 74 28 22 46 ).....context("F
03d0: 61 69 6c 65 64 20 74 6f 20 63 72 65 61 74 65 20 ailed to create
03e0: 41 50 49 2e 5c 6e 22 29 3f 0a 09 09 09 2e 77 69 API.\n")?.....wi
03f0: 74 68 5f 68 6f 73 74 28 61 70 69 5f 67 61 74 65 th_host(api_gate
0400: 77 61 79 29 3b 0a 09 09 6c 65 74 20 72 65 63 69 way);...let reci
0410: 70 69 65 6e 74 73 20 3d 20 72 65 63 69 70 69 65 pients = recipie
0420: 6e 74 73 2e 69 6e 74 6f 5f 69 74 65 72 28 29 0a nts.into_iter().
0430: 09 09 09 2e 6d 61 70 28 7c 28 61 2c 20 62 29 7c ....map(|(a, b)|
0440: 20 28 61 2c 20 43 68 61 74 50 65 65 72 49 64 3a (a, ChatPeerId:
0450: 3a 66 72 6f 6d 28 62 29 29 29 2e 63 6f 6c 6c 65 :from(b))).colle
0460: 63 74 28 29 3b 0a 09 09 6c 65 74 20 64 65 66 61 ct();...let defa
0470: 75 6c 74 20 3d 20 43 68 61 74 50 65 65 72 49 64 ult = ChatPeerId
0480: 3a 3a 66 72 6f 6d 28 64 65 66 61 75 6c 74 29 3b ::from(default);
0490: 0a 0a 09 09 4f 6b 28 54 65 6c 65 67 72 61 6d 54 ....Ok(TelegramT
04a0: 72 61 6e 73 70 6f 72 74 20 7b 0a 09 09 09 74 67 ransport {....tg
04b0: 2c 0a 09 09 09 72 65 63 69 70 69 65 6e 74 73 2c ,....recipients,
04c0: 0a 09 09 09 64 65 66 61 75 6c 74 2c 0a 09 09 7d ....default,...}
04d0: 29 0a 09 7d 0a 0a 09 2f 2f 2f 20 53 65 6e 64 20 )..}.../// Send
04e0: 6d 65 73 73 61 67 65 20 74 6f 20 64 65 66 61 75 message to defau
04f0: 6c 74 20 75 73 65 72 2c 20 75 73 65 64 20 66 6f lt user, used fo
0500: 72 20 64 65 62 75 67 2f 6c 6f 67 2f 69 6e 66 6f r debug/log/info
0510: 20 70 75 72 70 6f 73 65 73 0a 09 70 75 62 20 61 purposes..pub a
0520: 73 79 6e 63 20 66 6e 20 64 65 62 75 67 20 28 26 sync fn debug (&
0530: 73 65 6c 66 2c 20 6d 73 67 3a 20 26 73 74 72 29 self, msg: &str)
0540: 20 2d 3e 20 52 65 73 75 6c 74 3c 4d 65 73 73 61 -> Result<Messa
0550: 67 65 3e 20 7b 0a 09 09 73 65 6c 66 2e 73 65 6e ge> {...self.sen
0560: 64 28 26 73 65 6c 66 2e 64 65 66 61 75 6c 74 2c d(&self.default,
0570: 20 66 6f 72 6d 61 74 21 28 22 3c 70 72 65 3e 7b format!("<pre>{
0580: 7d 3c 2f 70 72 65 3e 22 2c 20 76 61 6c 69 64 61 }</pre>", valida
0590: 74 65 28 6d 73 67 29 2e 73 74 61 63 6b 28 29 3f te(msg).stack()?
05a0: 29 29 2e 61 77 61 69 74 0a 09 7d 0a 0a 09 2f 2f )).await..}...//
05b0: 2f 20 47 65 74 20 72 65 63 69 70 69 65 6e 74 20 / Get recipient
05c0: 62 79 20 61 64 64 72 65 73 73 0a 09 70 75 62 20 by address..pub
05d0: 66 6e 20 67 65 74 20 28 26 73 65 6c 66 2c 20 6e fn get (&self, n
05e0: 61 6d 65 3a 20 26 73 74 72 29 20 2d 3e 20 52 65 ame: &str) -> Re
05f0: 73 75 6c 74 3c 26 43 68 61 74 50 65 65 72 49 64 sult<&ChatPeerId
0600: 3e 20 7b 0a 09 09 73 65 6c 66 2e 72 65 63 69 70 > {...self.recip
0610: 69 65 6e 74 73 2e 67 65 74 28 6e 61 6d 65 29 0a ients.get(name).
0620: 09 09 09 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 ....with_context
0630: 28 7c 7c 20 66 6f 72 6d 61 74 21 28 22 52 65 63 (|| format!("Rec
0640: 69 70 69 65 6e 74 20 5c 22 7b 6e 61 6d 65 7d 5c ipient \"{name}\
0650: 22 20 6e 6f 74 20 66 6f 75 6e 64 20 69 6e 20 63 " not found in c
0660: 6f 6e 66 69 67 75 72 61 74 69 6f 6e 22 29 29 0a onfiguration")).
0670: 09 7d 0a 0a 09 2f 2f 2f 20 53 65 6e 64 20 6d 65 .}.../// Send me
0680: 73 73 61 67 65 20 74 6f 20 73 70 65 63 69 66 69 ssage to specifi
0690: 65 64 20 75 73 65 72 0a 09 70 75 62 20 61 73 79 ed user..pub asy
06a0: 6e 63 20 66 6e 20 73 65 6e 64 20 3c 53 3e 20 28 nc fn send <S> (
06b0: 26 73 65 6c 66 2c 20 74 6f 3a 20 26 43 68 61 74 &self, to: &Chat
06c0: 50 65 65 72 49 64 2c 20 6d 73 67 3a 20 53 29 20 PeerId, msg: S)
06d0: 2d 3e 20 52 65 73 75 6c 74 3c 4d 65 73 73 61 67 -> Result<Messag
06e0: 65 3e 0a 09 77 68 65 72 65 20 53 3a 20 49 6e 74 e>..where S: Int
06f0: 6f 3c 53 74 72 69 6e 67 3e 20 2b 20 44 65 62 75 o<String> + Debu
0700: 67 7b 0a 09 09 73 65 6c 66 2e 74 67 2e 65 78 65 g{...self.tg.exe
0710: 63 75 74 65 28 0a 09 09 09 53 65 6e 64 4d 65 73 cute(....SendMes
0720: 73 61 67 65 3a 3a 6e 65 77 28 2a 74 6f 2c 20 6d sage::new(*to, m
0730: 73 67 29 0a 09 09 09 2e 77 69 74 68 5f 70 61 72 sg).....with_par
0740: 73 65 5f 6d 6f 64 65 28 48 74 6d 6c 29 0a 09 09 se_mode(Html)...
0750: 29 2e 61 77 61 69 74 2e 73 74 61 63 6b 28 29 0a ).await.stack().
0760: 09 7d 0a 0a 09 2f 2f 2f 20 53 65 6e 64 20 6d 65 .}.../// Send me
0770: 64 69 61 20 74 6f 20 73 70 65 63 69 66 69 65 64 dia to specified
0780: 20 75 73 65 72 0a 09 70 75 62 20 61 73 79 6e 63 user..pub async
0790: 20 66 6e 20 73 65 6e 64 67 72 6f 75 70 20 28 26 fn sendgroup (&
07a0: 73 65 6c 66 2c 20 74 6f 3a 20 26 43 68 61 74 50 self, to: &ChatP
07b0: 65 65 72 49 64 2c 20 6d 65 64 69 61 3a 20 56 65 eerId, media: Ve
07c0: 63 3c 41 74 74 61 63 68 6d 65 6e 74 3e 2c 20 6d c<Attachment>, m
07d0: 73 67 3a 20 26 73 74 72 29 20 2d 3e 20 52 65 73 sg: &str) -> Res
07e0: 75 6c 74 3c 28 29 3e 20 7b 0a 09 09 69 66 20 6d ult<()> {...if m
07f0: 65 64 69 61 2e 6c 65 6e 28 29 20 3e 20 31 20 7b edia.len() > 1 {
0800: 0a 09 09 09 6c 65 74 20 6d 75 74 20 61 74 74 61 ....let mut atta
0810: 63 68 20 3d 20 76 65 63 21 5b 5d 3b 0a 09 09 09 ch = vec![];....
0820: 6c 65 74 20 6d 75 74 20 70 6f 73 20 3d 20 6d 65 let mut pos = me
0830: 64 69 61 2e 6c 65 6e 28 29 3b 0a 09 09 09 66 6f dia.len();....fo
0840: 72 20 66 69 6c 65 20 69 6e 20 6d 65 64 69 61 20 r file in media
0850: 7b 0a 09 09 09 09 6c 65 74 20 6d 75 74 20 63 61 {.....let mut ca
0860: 70 74 69 6f 6e 20 3d 20 49 6e 70 75 74 4d 65 64 ption = InputMed
0870: 69 61 44 6f 63 75 6d 65 6e 74 3a 3a 64 65 66 61 iaDocument::defa
0880: 75 6c 74 28 29 3b 0a 09 09 09 09 69 66 20 70 6f ult();.....if po
0890: 73 20 3d 3d 20 31 20 7b 0a 09 09 09 09 09 63 61 s == 1 {......ca
08a0: 70 74 69 6f 6e 20 3d 20 63 61 70 74 69 6f 6e 2e ption = caption.
08b0: 77 69 74 68 5f 63 61 70 74 69 6f 6e 28 6d 73 67 with_caption(msg
08c0: 29 0a 09 09 09 09 09 09 2e 77 69 74 68 5f 63 61 )........with_ca
08d0: 70 74 69 6f 6e 5f 70 61 72 73 65 5f 6d 6f 64 65 ption_parse_mode
08e0: 28 48 74 6d 6c 29 3b 0a 09 09 09 09 7d 0a 09 09 (Html);.....}...
08f0: 09 09 70 6f 73 20 2d 3d 20 31 3b 0a 09 09 09 09 ..pos -= 1;.....
0900: 61 74 74 61 63 68 2e 70 75 73 68 28 0a 09 09 09 attach.push(....
0910: 09 09 4d 65 64 69 61 47 72 6f 75 70 49 74 65 6d ..MediaGroupItem
0920: 3a 3a 66 6f 72 5f 64 6f 63 75 6d 65 6e 74 28 0a ::for_document(.
0930: 09 09 09 09 09 09 49 6e 70 75 74 46 69 6c 65 3a ......InputFile:
0940: 3a 66 72 6f 6d 28 0a 09 09 09 09 09 09 09 49 6e :from(........In
0950: 70 75 74 46 69 6c 65 52 65 61 64 65 72 3a 3a 66 putFileReader::f
0960: 72 6f 6d 28 66 69 6c 65 2e 64 61 74 61 29 0a 09 rom(file.data)..
0970: 09 09 09 09 09 09 09 2e 77 69 74 68 5f 66 69 6c ........with_fil
0980: 65 5f 6e 61 6d 65 28 66 69 6c 65 2e 6e 61 6d 65 e_name(file.name
0990: 29 0a 09 09 09 09 09 09 29 2c 0a 09 09 09 09 09 ).......),......
09a0: 09 63 61 70 74 69 6f 6e 0a 09 09 09 09 09 29 0a .caption......).
09b0: 09 09 09 09 29 3b 0a 09 09 09 7d 0a 09 09 09 73 ....);....}....s
09c0: 65 6c 66 2e 74 67 2e 65 78 65 63 75 74 65 28 53 elf.tg.execute(S
09d0: 65 6e 64 4d 65 64 69 61 47 72 6f 75 70 3a 3a 6e endMediaGroup::n
09e0: 65 77 28 2a 74 6f 2c 20 4d 65 64 69 61 47 72 6f ew(*to, MediaGro
09f0: 75 70 3a 3a 6e 65 77 28 61 74 74 61 63 68 29 2e up::new(attach).
0a00: 73 74 61 63 6b 28 29 3f 29 29 2e 61 77 61 69 74 stack()?)).await
0a10: 2e 73 74 61 63 6b 28 29 3f 3b 0a 09 09 7d 20 65 .stack()?;...} e
0a20: 6c 73 65 20 7b 0a 09 09 09 73 65 6c 66 2e 74 67 lse {....self.tg
0a30: 2e 65 78 65 63 75 74 65 28 0a 09 09 09 09 53 65 .execute(.....Se
0a40: 6e 64 44 6f 63 75 6d 65 6e 74 3a 3a 6e 65 77 28 ndDocument::new(
0a50: 0a 09 09 09 09 09 2a 74 6f 2c 0a 09 09 09 09 09 ......*to,......
0a60: 49 6e 70 75 74 46 69 6c 65 52 65 61 64 65 72 3a InputFileReader:
0a70: 3a 66 72 6f 6d 28 6d 65 64 69 61 5b 30 5d 2e 64 :from(media[0].d
0a80: 61 74 61 2e 63 6c 6f 6e 65 28 29 29 0a 09 09 09 ata.clone())....
0a90: 09 09 2e 77 69 74 68 5f 66 69 6c 65 5f 6e 61 6d ...with_file_nam
0aa0: 65 28 6d 65 64 69 61 5b 30 5d 2e 6e 61 6d 65 2e e(media[0].name.
0ab0: 63 6c 6f 6e 65 28 29 29 0a 09 09 09 09 29 2e 77 clone()).....).w
0ac0: 69 74 68 5f 63 61 70 74 69 6f 6e 28 6d 73 67 29 ith_caption(msg)
0ad0: 0a 09 09 09 09 2e 77 69 74 68 5f 63 61 70 74 69 ......with_capti
0ae0: 6f 6e 5f 70 61 72 73 65 5f 6d 6f 64 65 28 48 74 on_parse_mode(Ht
0af0: 6d 6c 29 0a 09 09 09 29 2e 61 77 61 69 74 2e 73 ml)....).await.s
0b00: 74 61 63 6b 28 29 3f 3b 0a 09 09 7d 0a 09 09 4f tack()?;...}...O
0b10: 6b 28 28 29 29 0a 09 7d 0a 7d 0a k(())..}.}.