Artifact
dfced9c660a0870702b2db683a63f9fec1579c0ade40979527cefba9928f3614:
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 52 45 5f 53 50 45 43 49 41 4c 2c 0a 7d 3b ..RE_SPECIAL,.};
0030: 0a 0a 75 73 65 20 73 74 64 3a 3a 7b 0a 09 62 6f ..use std::{..bo
0040: 72 72 6f 77 3a 3a 43 6f 77 2c 0a 09 63 6f 6c 6c rrow::Cow,..coll
0050: 65 63 74 69 6f 6e 73 3a 3a 48 61 73 68 4d 61 70 ections::HashMap
0060: 2c 0a 09 66 6d 74 3a 3a 44 65 62 75 67 2c 0a 7d ,..fmt::Debug,.}
0070: 3b 0a 0a 75 73 65 20 73 74 61 63 6b 65 64 5f 65 ;..use stacked_e
0080: 72 72 6f 72 73 3a 3a 7b 0a 09 52 65 73 75 6c 74 rrors::{..Result
0090: 2c 0a 09 53 74 61 63 6b 61 62 6c 65 45 72 72 2c ,..StackableErr,
00a0: 0a 7d 3b 0a 75 73 65 20 74 67 62 6f 74 3a 3a 7b .};.use tgbot::{
00b0: 0a 09 61 70 69 3a 3a 43 6c 69 65 6e 74 2c 0a 09 ..api::Client,..
00c0: 74 79 70 65 73 3a 3a 7b 0a 09 09 43 68 61 74 50 types::{...ChatP
00d0: 65 65 72 49 64 2c 0a 09 09 49 6e 70 75 74 46 69 eerId,...InputFi
00e0: 6c 65 2c 0a 09 09 49 6e 70 75 74 46 69 6c 65 52 le,...InputFileR
00f0: 65 61 64 65 72 2c 0a 09 09 49 6e 70 75 74 4d 65 eader,...InputMe
0100: 64 69 61 44 6f 63 75 6d 65 6e 74 2c 0a 09 09 4d diaDocument,...M
0110: 65 64 69 61 47 72 6f 75 70 2c 0a 09 09 4d 65 64 ediaGroup,...Med
0120: 69 61 47 72 6f 75 70 49 74 65 6d 2c 0a 09 09 4d iaGroupItem,...M
0130: 65 73 73 61 67 65 2c 0a 09 09 50 61 72 73 65 4d essage,...ParseM
0140: 6f 64 65 3a 3a 4d 61 72 6b 64 6f 77 6e 56 32 2c ode::MarkdownV2,
0150: 0a 09 09 53 65 6e 64 4d 65 64 69 61 47 72 6f 75 ...SendMediaGrou
0160: 70 2c 0a 09 09 53 65 6e 64 4d 65 73 73 61 67 65 p,...SendMessage
0170: 2c 0a 09 09 53 65 6e 64 44 6f 63 75 6d 65 6e 74 ,...SendDocument
0180: 2c 0a 09 7d 2c 0a 7d 3b 0a 0a 2f 2f 2f 20 45 6e ,..},.};../// En
0190: 63 6f 64 65 73 20 73 70 65 63 69 61 6c 20 48 54 codes special HT
01a0: 4d 4c 20 65 6e 74 69 74 69 65 73 20 74 6f 20 70 ML entities to p
01b0: 72 65 76 65 6e 74 20 74 68 65 6d 20 69 6e 74 65 revent them inte
01c0: 72 66 65 72 69 6e 67 20 77 69 74 68 20 54 65 6c rfering with Tel
01d0: 65 67 72 61 6d 20 48 54 4d 4c 0a 70 75 62 20 66 egram HTML.pub f
01e0: 6e 20 65 6e 63 6f 64 65 20 28 74 65 78 74 3a 20 n encode (text:
01f0: 26 73 74 72 29 20 2d 3e 20 43 6f 77 3c 27 5f 2c &str) -> Cow<'_,
0200: 20 73 74 72 3e 20 7b 0a 09 52 45 5f 53 50 45 43 str> {..RE_SPEC
0210: 49 41 4c 2e 72 65 70 6c 61 63 65 5f 61 6c 6c 28 IAL.replace_all(
0220: 74 65 78 74 2c 20 22 5c 5c 24 31 22 29 0a 7d 0a text, "\\$1").}.
0230: 0a 23 5b 64 65 72 69 76 65 28 44 65 62 75 67 29 .#[derive(Debug)
0240: 5d 0a 70 75 62 20 73 74 72 75 63 74 20 54 65 6c ].pub struct Tel
0250: 65 67 72 61 6d 54 72 61 6e 73 70 6f 72 74 20 7b egramTransport {
0260: 0a 09 74 67 3a 20 43 6c 69 65 6e 74 2c 0a 09 72 ..tg: Client,..r
0270: 65 63 69 70 69 65 6e 74 73 3a 20 48 61 73 68 4d ecipients: HashM
0280: 61 70 3c 53 74 72 69 6e 67 2c 20 43 68 61 74 50 ap<String, ChatP
0290: 65 65 72 49 64 3e 2c 0a 09 70 75 62 20 64 65 66 eerId>,..pub def
02a0: 61 75 6c 74 3a 20 43 68 61 74 50 65 65 72 49 64 ault: ChatPeerId
02b0: 2c 0a 7d 0a 0a 69 6d 70 6c 20 54 65 6c 65 67 72 ,.}..impl Telegr
02c0: 61 6d 54 72 61 6e 73 70 6f 72 74 20 7b 0a 0a 09 amTransport {...
02d0: 70 75 62 20 66 6e 20 6e 65 77 20 28 61 70 69 5f pub fn new (api_
02e0: 6b 65 79 3a 20 53 74 72 69 6e 67 2c 20 72 65 63 key: String, rec
02f0: 69 70 69 65 6e 74 73 3a 20 48 61 73 68 4d 61 70 ipients: HashMap
0300: 3c 53 74 72 69 6e 67 2c 20 69 36 34 3e 2c 20 73 <String, i64>, s
0310: 65 74 74 69 6e 67 73 3a 20 26 63 6f 6e 66 69 67 ettings: &config
0320: 3a 3a 43 6f 6e 66 69 67 29 20 2d 3e 20 52 65 73 ::Config) -> Res
0330: 75 6c 74 3c 54 65 6c 65 67 72 61 6d 54 72 61 6e ult<TelegramTran
0340: 73 70 6f 72 74 3e 20 7b 0a 09 09 6c 65 74 20 64 sport> {...let d
0350: 65 66 61 75 6c 74 20 3d 20 73 65 74 74 69 6e 67 efault = setting
0360: 73 2e 67 65 74 5f 69 6e 74 28 22 64 65 66 61 75 s.get_int("defau
0370: 6c 74 22 29 0a 09 09 09 2e 63 6f 6e 74 65 78 74 lt").....context
0380: 28 22 5b 73 6d 74 70 32 74 67 2e 74 6f 6d 6c 5d ("[smtp2tg.toml]
0390: 20 6d 69 73 73 69 6e 67 20 5c 22 64 65 66 61 75 missing \"defau
03a0: 6c 74 5c 22 20 72 65 63 69 70 69 65 6e 74 2e 5c lt\" recipient.\
03b0: 6e 22 29 3f 3b 0a 09 09 6c 65 74 20 61 70 69 5f n")?;...let api_
03c0: 67 61 74 65 77 61 79 20 3d 20 73 65 74 74 69 6e gateway = settin
03d0: 67 73 2e 67 65 74 5f 73 74 72 69 6e 67 28 22 61 gs.get_string("a
03e0: 70 69 5f 67 61 74 65 77 61 79 22 29 0a 09 09 09 pi_gateway")....
03f0: 2e 63 6f 6e 74 65 78 74 28 22 5b 73 6d 74 70 32 .context("[smtp2
0400: 74 67 2e 74 6f 6d 6c 5d 20 6d 69 73 73 69 6e 67 tg.toml] missing
0410: 20 5c 22 61 70 69 5f 67 61 74 65 77 61 79 5c 22 \"api_gateway\"
0420: 20 64 65 73 74 69 6e 61 74 69 6f 6e 2e 5c 6e 22 destination.\n"
0430: 29 3f 3b 0a 09 09 6c 65 74 20 74 67 20 3d 20 43 )?;...let tg = C
0440: 6c 69 65 6e 74 3a 3a 6e 65 77 28 61 70 69 5f 6b lient::new(api_k
0450: 65 79 29 0a 09 09 09 2e 63 6f 6e 74 65 78 74 28 ey).....context(
0460: 22 46 61 69 6c 65 64 20 74 6f 20 63 72 65 61 74 "Failed to creat
0470: 65 20 41 50 49 2e 5c 6e 22 29 3f 0a 09 09 09 2e e API.\n")?.....
0480: 77 69 74 68 5f 68 6f 73 74 28 61 70 69 5f 67 61 with_host(api_ga
0490: 74 65 77 61 79 29 3b 0a 09 09 6c 65 74 20 72 65 teway);...let re
04a0: 63 69 70 69 65 6e 74 73 20 3d 20 72 65 63 69 70 cipients = recip
04b0: 69 65 6e 74 73 2e 69 6e 74 6f 5f 69 74 65 72 28 ients.into_iter(
04c0: 29 0a 09 09 09 2e 6d 61 70 28 7c 28 61 2c 20 62 ).....map(|(a, b
04d0: 29 7c 20 28 61 2c 20 43 68 61 74 50 65 65 72 49 )| (a, ChatPeerI
04e0: 64 3a 3a 66 72 6f 6d 28 62 29 29 29 2e 63 6f 6c d::from(b))).col
04f0: 6c 65 63 74 28 29 3b 0a 09 09 6c 65 74 20 64 65 lect();...let de
0500: 66 61 75 6c 74 20 3d 20 43 68 61 74 50 65 65 72 fault = ChatPeer
0510: 49 64 3a 3a 66 72 6f 6d 28 64 65 66 61 75 6c 74 Id::from(default
0520: 29 3b 0a 0a 09 09 4f 6b 28 54 65 6c 65 67 72 61 );....Ok(Telegra
0530: 6d 54 72 61 6e 73 70 6f 72 74 20 7b 0a 09 09 09 mTransport {....
0540: 74 67 2c 0a 09 09 09 72 65 63 69 70 69 65 6e 74 tg,....recipient
0550: 73 2c 0a 09 09 09 64 65 66 61 75 6c 74 2c 0a 09 s,....default,..
0560: 09 7d 29 0a 09 7d 0a 0a 09 2f 2f 2f 20 53 65 6e .})..}.../// Sen
0570: 64 20 6d 65 73 73 61 67 65 20 74 6f 20 64 65 66 d message to def
0580: 61 75 6c 74 20 75 73 65 72 2c 20 75 73 65 64 20 ault user, used
0590: 66 6f 72 20 64 65 62 75 67 2f 6c 6f 67 2f 69 6e for debug/log/in
05a0: 66 6f 20 70 75 72 70 6f 73 65 73 0a 09 70 75 62 fo purposes..pub
05b0: 20 61 73 79 6e 63 20 66 6e 20 64 65 62 75 67 20 async fn debug
05c0: 28 26 73 65 6c 66 2c 20 6d 73 67 3a 20 26 73 74 (&self, msg: &st
05d0: 72 29 20 2d 3e 20 52 65 73 75 6c 74 3c 4d 65 73 r) -> Result<Mes
05e0: 73 61 67 65 3e 20 7b 0a 09 09 73 65 6c 66 2e 73 sage> {...self.s
05f0: 65 6e 64 28 26 73 65 6c 66 2e 64 65 66 61 75 6c end(&self.defaul
0600: 74 2c 20 65 6e 63 6f 64 65 28 6d 73 67 29 29 2e t, encode(msg)).
0610: 61 77 61 69 74 0a 09 7d 0a 0a 09 2f 2f 2f 20 47 await..}.../// G
0620: 65 74 20 72 65 63 69 70 69 65 6e 74 20 62 79 20 et recipient by
0630: 61 64 64 72 65 73 73 0a 09 70 75 62 20 66 6e 20 address..pub fn
0640: 67 65 74 20 28 26 73 65 6c 66 2c 20 6e 61 6d 65 get (&self, name
0650: 3a 20 26 73 74 72 29 20 2d 3e 20 52 65 73 75 6c : &str) -> Resul
0660: 74 3c 26 43 68 61 74 50 65 65 72 49 64 3e 20 7b t<&ChatPeerId> {
0670: 0a 09 09 73 65 6c 66 2e 72 65 63 69 70 69 65 6e ...self.recipien
0680: 74 73 2e 67 65 74 28 6e 61 6d 65 29 0a 09 09 09 ts.get(name)....
0690: 2e 77 69 74 68 5f 63 6f 6e 74 65 78 74 28 7c 7c .with_context(||
06a0: 20 66 6f 72 6d 61 74 21 28 22 52 65 63 69 70 69 format!("Recipi
06b0: 65 6e 74 20 5c 22 7b 6e 61 6d 65 7d 5c 22 20 6e ent \"{name}\" n
06c0: 6f 74 20 66 6f 75 6e 64 20 69 6e 20 63 6f 6e 66 ot found in conf
06d0: 69 67 75 72 61 74 69 6f 6e 22 29 29 0a 09 7d 0a iguration"))..}.
06e0: 0a 09 2f 2f 2f 20 53 65 6e 64 20 6d 65 73 73 61 ../// Send messa
06f0: 67 65 20 74 6f 20 73 70 65 63 69 66 69 65 64 20 ge to specified
0700: 75 73 65 72 0a 09 70 75 62 20 61 73 79 6e 63 20 user..pub async
0710: 66 6e 20 73 65 6e 64 20 3c 53 3e 20 28 26 73 65 fn send <S> (&se
0720: 6c 66 2c 20 74 6f 3a 20 26 43 68 61 74 50 65 65 lf, to: &ChatPee
0730: 72 49 64 2c 20 6d 73 67 3a 20 53 29 20 2d 3e 20 rId, msg: S) ->
0740: 52 65 73 75 6c 74 3c 4d 65 73 73 61 67 65 3e 0a Result<Message>.
0750: 09 77 68 65 72 65 20 53 3a 20 49 6e 74 6f 3c 53 .where S: Into<S
0760: 74 72 69 6e 67 3e 20 2b 20 44 65 62 75 67 7b 0a tring> + Debug{.
0770: 09 09 73 65 6c 66 2e 74 67 2e 65 78 65 63 75 74 ..self.tg.execut
0780: 65 28 0a 09 09 09 53 65 6e 64 4d 65 73 73 61 67 e(....SendMessag
0790: 65 3a 3a 6e 65 77 28 2a 74 6f 2c 20 6d 73 67 29 e::new(*to, msg)
07a0: 0a 09 09 09 2e 77 69 74 68 5f 70 61 72 73 65 5f .....with_parse_
07b0: 6d 6f 64 65 28 4d 61 72 6b 64 6f 77 6e 56 32 29 mode(MarkdownV2)
07c0: 0a 09 09 29 2e 61 77 61 69 74 2e 73 74 61 63 6b ...).await.stack
07d0: 28 29 0a 09 7d 0a 0a 09 2f 2f 2f 20 53 65 6e 64 ()..}.../// Send
07e0: 20 6d 65 64 69 61 20 74 6f 20 73 70 65 63 69 66 media to specif
07f0: 69 65 64 20 75 73 65 72 0a 09 70 75 62 20 61 73 ied user..pub as
0800: 79 6e 63 20 66 6e 20 73 65 6e 64 67 72 6f 75 70 ync fn sendgroup
0810: 20 28 26 73 65 6c 66 2c 20 74 6f 3a 20 26 43 68 (&self, to: &Ch
0820: 61 74 50 65 65 72 49 64 2c 20 6d 65 64 69 61 3a atPeerId, media:
0830: 20 56 65 63 3c 41 74 74 61 63 68 6d 65 6e 74 3e Vec<Attachment>
0840: 2c 20 6d 73 67 3a 20 26 73 74 72 29 20 2d 3e 20 , msg: &str) ->
0850: 52 65 73 75 6c 74 3c 28 29 3e 20 7b 0a 09 09 69 Result<()> {...i
0860: 66 20 6d 65 64 69 61 2e 6c 65 6e 28 29 20 3e 20 f media.len() >
0870: 31 20 7b 0a 09 09 09 6c 65 74 20 6d 75 74 20 61 1 {....let mut a
0880: 74 74 61 63 68 20 3d 20 76 65 63 21 5b 5d 3b 0a ttach = vec![];.
0890: 09 09 09 6c 65 74 20 6d 75 74 20 70 6f 73 20 3d ...let mut pos =
08a0: 20 6d 65 64 69 61 2e 6c 65 6e 28 29 3b 0a 09 09 media.len();...
08b0: 09 66 6f 72 20 66 69 6c 65 20 69 6e 20 6d 65 64 .for file in med
08c0: 69 61 20 7b 0a 09 09 09 09 6c 65 74 20 6d 75 74 ia {.....let mut
08d0: 20 63 61 70 74 69 6f 6e 20 3d 20 49 6e 70 75 74 caption = Input
08e0: 4d 65 64 69 61 44 6f 63 75 6d 65 6e 74 3a 3a 64 MediaDocument::d
08f0: 65 66 61 75 6c 74 28 29 3b 0a 09 09 09 09 69 66 efault();.....if
0900: 20 70 6f 73 20 3d 3d 20 31 20 7b 0a 09 09 09 09 pos == 1 {.....
0910: 09 63 61 70 74 69 6f 6e 20 3d 20 63 61 70 74 69 .caption = capti
0920: 6f 6e 2e 77 69 74 68 5f 63 61 70 74 69 6f 6e 28 on.with_caption(
0930: 6d 73 67 29 0a 09 09 09 09 09 09 2e 77 69 74 68 msg)........with
0940: 5f 63 61 70 74 69 6f 6e 5f 70 61 72 73 65 5f 6d _caption_parse_m
0950: 6f 64 65 28 4d 61 72 6b 64 6f 77 6e 56 32 29 3b ode(MarkdownV2);
0960: 0a 09 09 09 09 7d 0a 09 09 09 09 70 6f 73 20 2d .....}.....pos -
0970: 3d 20 31 3b 0a 09 09 09 09 61 74 74 61 63 68 2e = 1;.....attach.
0980: 70 75 73 68 28 0a 09 09 09 09 09 4d 65 64 69 61 push(......Media
0990: 47 72 6f 75 70 49 74 65 6d 3a 3a 66 6f 72 5f 64 GroupItem::for_d
09a0: 6f 63 75 6d 65 6e 74 28 0a 09 09 09 09 09 09 49 ocument(.......I
09b0: 6e 70 75 74 46 69 6c 65 3a 3a 66 72 6f 6d 28 0a nputFile::from(.
09c0: 09 09 09 09 09 09 09 49 6e 70 75 74 46 69 6c 65 .......InputFile
09d0: 52 65 61 64 65 72 3a 3a 66 72 6f 6d 28 66 69 6c Reader::from(fil
09e0: 65 2e 64 61 74 61 29 0a 09 09 09 09 09 09 09 09 e.data).........
09f0: 2e 77 69 74 68 5f 66 69 6c 65 5f 6e 61 6d 65 28 .with_file_name(
0a00: 66 69 6c 65 2e 6e 61 6d 65 29 0a 09 09 09 09 09 file.name)......
0a10: 09 29 2c 0a 09 09 09 09 09 09 63 61 70 74 69 6f .),.......captio
0a20: 6e 0a 09 09 09 09 09 29 0a 09 09 09 09 29 3b 0a n......).....);.
0a30: 09 09 09 7d 0a 09 09 09 73 65 6c 66 2e 74 67 2e ...}....self.tg.
0a40: 65 78 65 63 75 74 65 28 53 65 6e 64 4d 65 64 69 execute(SendMedi
0a50: 61 47 72 6f 75 70 3a 3a 6e 65 77 28 2a 74 6f 2c aGroup::new(*to,
0a60: 20 4d 65 64 69 61 47 72 6f 75 70 3a 3a 6e 65 77 MediaGroup::new
0a70: 28 61 74 74 61 63 68 29 2e 73 74 61 63 6b 28 29 (attach).stack()
0a80: 3f 29 29 2e 61 77 61 69 74 2e 73 74 61 63 6b 28 ?)).await.stack(
0a90: 29 3f 3b 0a 09 09 7d 20 65 6c 73 65 20 7b 0a 09 )?;...} else {..
0aa0: 09 09 73 65 6c 66 2e 74 67 2e 65 78 65 63 75 74 ..self.tg.execut
0ab0: 65 28 0a 09 09 09 09 53 65 6e 64 44 6f 63 75 6d e(.....SendDocum
0ac0: 65 6e 74 3a 3a 6e 65 77 28 0a 09 09 09 09 09 2a ent::new(......*
0ad0: 74 6f 2c 0a 09 09 09 09 09 49 6e 70 75 74 46 69 to,......InputFi
0ae0: 6c 65 52 65 61 64 65 72 3a 3a 66 72 6f 6d 28 6d leReader::from(m
0af0: 65 64 69 61 5b 30 5d 2e 64 61 74 61 2e 63 6c 6f edia[0].data.clo
0b00: 6e 65 28 29 29 0a 09 09 09 09 09 2e 77 69 74 68 ne()).......with
0b10: 5f 66 69 6c 65 5f 6e 61 6d 65 28 6d 65 64 69 61 _file_name(media
0b20: 5b 30 5d 2e 6e 61 6d 65 2e 63 6c 6f 6e 65 28 29 [0].name.clone()
0b30: 29 0a 09 09 09 09 29 2e 77 69 74 68 5f 63 61 70 ).....).with_cap
0b40: 74 69 6f 6e 28 6d 73 67 29 0a 09 09 09 09 2e 77 tion(msg)......w
0b50: 69 74 68 5f 63 61 70 74 69 6f 6e 5f 70 61 72 73 ith_caption_pars
0b60: 65 5f 6d 6f 64 65 28 4d 61 72 6b 64 6f 77 6e 56 e_mode(MarkdownV
0b70: 32 29 0a 09 09 09 29 2e 61 77 61 69 74 2e 73 74 2)....).await.st
0b80: 61 63 6b 28 29 3f 3b 0a 09 09 7d 0a 09 09 4f 6b ack()?;...}...Ok
0b90: 28 28 29 29 0a 09 7d 0a 7d 0a (())..}.}.