Artifact
b825a3dd51773b9e3891e2f01477f51ef24f8a79ebb5746e6be27e838a09b6c2:
0000: 2f 2f 21 20 55 74 69 6c 69 74 79 20 66 75 6e 63 //! Utility func
0010: 74 69 6f 6e 73 20 61 6e 64 20 74 79 70 65 73 20 tions and types
0020: 66 6f 72 20 74 68 65 20 61 70 70 6c 69 63 61 74 for the applicat
0030: 69 6f 6e 2e 0a 0a 75 73 65 20 63 72 61 74 65 3a ion...use crate:
0040: 3a 43 75 72 73 6f 72 3b 0a 0a 75 73 65 20 73 74 :Cursor;..use st
0050: 64 3a 3a 7b 0a 09 62 6f 72 72 6f 77 3a 3a 43 6f d::{..borrow::Co
0060: 77 2c 0a 09 73 79 6e 63 3a 3a 4c 61 7a 79 4c 6f w,..sync::LazyLo
0070: 63 6b 2c 0a 7d 3b 0a 0a 75 73 65 20 68 74 6d 6c ck,.};..use html
0080: 5f 65 73 63 61 70 65 3a 3a 65 6e 63 6f 64 65 5f _escape::encode_
0090: 74 65 78 74 3b 0a 75 73 65 20 72 65 67 65 78 3a text;.use regex:
00a0: 3a 7b 0a 09 52 65 67 65 78 2c 0a 09 52 65 67 65 :{..Regex,..Rege
00b0: 78 42 75 69 6c 64 65 72 2c 0a 7d 3b 0a 75 73 65 xBuilder,.};.use
00c0: 20 73 74 61 63 6b 65 64 5f 65 72 72 6f 72 73 3a stacked_errors:
00d0: 3a 7b 0a 09 62 61 69 6c 2c 0a 09 52 65 73 75 6c :{..bail,..Resul
00e0: 74 2c 0a 7d 3b 0a 0a 70 75 62 20 73 74 61 74 69 t,.};..pub stati
00f0: 63 20 52 45 5f 44 4f 4d 41 49 4e 3a 20 4c 61 7a c RE_DOMAIN: Laz
0100: 79 4c 6f 63 6b 3c 52 65 67 65 78 3e 20 3d 20 4c yLock<Regex> = L
0110: 61 7a 79 4c 6f 63 6b 3a 3a 6e 65 77 28 7c 7c 20 azyLock::new(||
0120: 7b 0a 09 52 65 67 65 78 3a 3a 6e 65 77 28 72 22 {..Regex::new(r"
0130: 5e 5b 61 2d 7a 30 2d 39 5d 28 5b 2d 61 2d 7a 30 ^[a-z0-9]([-a-z0
0140: 2d 39 5d 2a 5b 61 2d 7a 30 2d 39 5d 29 3f 28 5c -9]*[a-z0-9])?(\
0150: 2e 5b 61 2d 7a 30 2d 39 5d 28 5b 2d 61 2d 7a 30 .[a-z0-9]([-a-z0
0160: 2d 39 5d 2a 5b 61 2d 7a 30 2d 39 5d 29 3f 29 2a -9]*[a-z0-9])?)*
0170: 24 22 29 2e 65 78 70 65 63 74 28 22 49 6e 76 61 $").expect("Inva
0180: 6c 69 64 20 64 6f 6d 61 69 6e 20 72 65 67 65 78 lid domain regex
0190: 22 29 0a 7d 29 3b 0a 70 75 62 20 73 74 61 74 69 ").});.pub stati
01a0: 63 20 52 45 5f 43 4c 4f 53 49 4e 47 3a 20 4c 61 c RE_CLOSING: La
01b0: 7a 79 4c 6f 63 6b 3c 52 65 67 65 78 3e 20 3d 20 zyLock<Regex> =
01c0: 4c 61 7a 79 4c 6f 63 6b 3a 3a 6e 65 77 28 7c 7c LazyLock::new(||
01d0: 20 7b 0a 09 52 65 67 65 78 42 75 69 6c 64 65 72 {..RegexBuilder
01e0: 3a 3a 6e 65 77 28 72 22 3c 2f 5b 20 5c 74 5d 2a ::new(r"</[ \t]*
01f0: 28 70 72 65 7c 63 6f 64 65 29 5b 20 5c 74 5d 2a (pre|code)[ \t]*
0200: 3e 22 29 0a 09 09 2e 63 61 73 65 5f 69 6e 73 65 >")....case_inse
0210: 6e 73 69 74 69 76 65 28 74 72 75 65 29 2e 62 75 nsitive(true).bu
0220: 69 6c 64 28 29 2e 65 78 70 65 63 74 28 22 49 6e ild().expect("In
0230: 76 61 6c 69 64 20 63 6c 6f 73 69 6e 67 20 74 61 valid closing ta
0240: 67 20 72 65 67 65 78 22 29 0a 7d 29 3b 0a 0a 2f g regex").});../
0250: 2f 2f 20 53 74 6f 72 65 73 20 62 69 6e 61 72 79 // Stores binary
0260: 20 61 74 74 61 63 68 6d 65 6e 74 20 64 61 74 61 attachment data
0270: 20 61 6e 64 20 6d 65 74 61 64 61 74 61 20 66 6f and metadata fo
0280: 72 20 54 65 6c 65 67 72 61 6d 20 6d 65 73 73 61 r Telegram messa
0290: 67 65 73 2e 0a 2f 2f 2f 20 54 68 65 20 64 61 74 ges../// The dat
02a0: 61 20 69 73 20 77 72 61 70 70 65 64 20 69 6e 20 a is wrapped in
02b0: 61 20 60 43 75 72 73 6f 72 3c 56 65 63 3c 75 38 a `Cursor<Vec<u8
02c0: 3e 3e 60 20 66 6f 72 20 65 66 66 69 63 69 65 6e >>` for efficien
02d0: 74 20 73 74 72 65 61 6d 69 6e 67 2c 0a 2f 2f 2f t streaming,.///
02e0: 20 77 68 69 6c 65 20 60 6e 61 6d 65 60 20 68 6f while `name` ho
02f0: 6c 64 73 20 74 68 65 20 66 69 6c 65 6e 61 6d 65 lds the filename
0300: 20 6f 72 20 64 69 73 70 6c 61 79 20 6e 61 6d 65 or display name
0310: 20 6f 66 20 74 68 65 20 61 74 74 61 63 68 6d 65 of the attachme
0320: 6e 74 2e 0a 23 5b 64 65 72 69 76 65 28 44 65 62 nt..#[derive(Deb
0330: 75 67 29 5d 0a 70 75 62 20 73 74 72 75 63 74 20 ug)].pub struct
0340: 41 74 74 61 63 68 6d 65 6e 74 20 7b 0a 09 70 75 Attachment {..pu
0350: 62 20 64 61 74 61 3a 20 43 75 72 73 6f 72 3c 56 b data: Cursor<V
0360: 65 63 3c 75 38 3e 3e 2c 0a 09 70 75 62 20 6e 61 ec<u8>>,..pub na
0370: 6d 65 3a 20 53 74 72 69 6e 67 2c 0a 7d 0a 0a 2f me: String,.}../
0380: 2f 2f 20 56 61 6c 69 64 61 74 65 73 20 74 65 78 // Validates tex
0390: 74 20 74 6f 20 65 6e 73 75 72 65 20 69 74 20 64 t to ensure it d
03a0: 6f 65 73 6e 27 74 20 62 72 65 61 6b 20 54 65 6c oesn't break Tel
03b0: 65 67 72 61 6d 27 73 20 70 72 65 66 6f 72 6d 61 egram's preforma
03c0: 74 74 65 64 20 62 6c 6f 63 6b 73 2e 0a 2f 2f 2f tted blocks..///
03d0: 0a 2f 2f 2f 20 45 73 63 61 70 65 73 20 48 54 4d ./// Escapes HTM
03e0: 4c 20 73 70 65 63 69 61 6c 20 63 68 61 72 61 63 L special charac
03f0: 74 65 72 73 20 74 6f 20 70 72 65 76 65 6e 74 20 ters to prevent
0400: 69 6e 6a 65 63 74 69 6f 6e 2e 0a 2f 2f 2f 0a 2f injection..///./
0410: 2f 2f 20 23 20 41 72 67 75 6d 65 6e 74 73 0a 2f // # Arguments./
0420: 2f 2f 20 2a 20 60 74 65 78 74 60 20 2d 20 54 65 // * `text` - Te
0430: 78 74 20 74 6f 20 76 61 6c 69 64 61 74 65 20 61 xt to validate a
0440: 6e 64 20 65 73 63 61 70 65 2e 0a 2f 2f 2f 0a 2f nd escape..///./
0450: 2f 2f 20 23 20 52 65 74 75 72 6e 73 0a 2f 2f 2f // # Returns.///
0460: 20 2a 20 60 52 65 73 75 6c 74 3c 43 6f 77 3c 27 * `Result<Cow<'
0470: 61 2c 20 73 74 72 3e 3e 60 20 2d 20 45 73 63 61 a, str>>` - Esca
0480: 70 65 64 20 74 65 78 74 20 6f 72 20 65 72 72 6f ped text or erro
0490: 72 20 69 66 20 69 6e 76 61 6c 69 64 2e 0a 2f 2f r if invalid..//
04a0: 2f 0a 2f 2f 2f 20 23 20 45 72 72 6f 72 73 0a 2f /./// # Errors./
04b0: 2f 2f 20 52 65 74 75 72 6e 73 20 61 6e 20 65 72 // Returns an er
04c0: 72 6f 72 20 69 66 20 74 68 65 20 74 65 78 74 20 ror if the text
04d0: 63 6f 6e 74 61 69 6e 73 20 54 65 6c 65 67 72 61 contains Telegra
04e0: 6d 20 63 6c 6f 73 69 6e 67 20 74 61 67 73 20 28 m closing tags (
04f0: 60 3c 2f 70 72 65 3e 60 2c 20 60 3c 2f 63 6f 64 `</pre>`, `</cod
0500: 65 3e 60 29 2e 0a 70 75 62 20 66 6e 20 76 61 6c e>`)..pub fn val
0510: 69 64 61 74 65 20 3c 27 61 3e 28 74 65 78 74 3a idate <'a>(text:
0520: 20 26 27 61 20 73 74 72 29 20 2d 3e 20 52 65 73 &'a str) -> Res
0530: 75 6c 74 3c 43 6f 77 3c 27 61 2c 20 73 74 72 3e ult<Cow<'a, str>
0540: 3e 20 7b 0a 09 69 66 20 52 45 5f 43 4c 4f 53 49 > {..if RE_CLOSI
0550: 4e 47 2e 69 73 5f 6d 61 74 63 68 28 74 65 78 74 NG.is_match(text
0560: 29 20 7b 0a 09 09 62 61 69 6c 21 28 22 54 65 6c ) {...bail!("Tel
0570: 65 67 72 61 6d 20 63 6c 6f 73 69 6e 67 20 74 61 egram closing ta
0580: 67 20 66 6f 75 6e 64 2e 22 29 3b 0a 09 7d 20 65 g found.");..} e
0590: 6c 73 65 20 7b 0a 09 09 4f 6b 28 65 6e 63 6f 64 lse {...Ok(encod
05a0: 65 5f 74 65 78 74 28 74 65 78 74 29 29 0a 09 7d e_text(text))..}
05b0: 0a 7d 0a .}.