-
-
Notifications
You must be signed in to change notification settings - Fork 892
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
2 changed files
with
137 additions
and
125 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,134 @@ | ||
| use super::MARKDOWN_PARSER; | ||
| use crate::settings::SETTINGS; | ||
| use markdown_it::plugins::cmark::inline::image::Image; | ||
| use url::Url; | ||
| use urlencoding::encode; | ||
|
|
||
| /// Rewrites all links to remote domains in markdown, so they go through `/api/v3/image_proxy`. | ||
| pub fn markdown_rewrite_image_links(mut src: String) -> (String, Vec<Url>) { | ||
| let ast = MARKDOWN_PARSER.parse(&src); | ||
| let mut links_offsets = vec![]; | ||
|
|
||
| // Walk the syntax tree to find positions of image links | ||
| ast.walk(|node, _depth| { | ||
| if let Some(image) = node.cast::<Image>() { | ||
| // srcmap is always present for image | ||
| // https://github.com/markdown-it-rust/markdown-it/issues/36#issuecomment-1777844387 | ||
| let node_offsets = node.srcmap.expect("srcmap is none").get_byte_offsets(); | ||
| // necessary for custom emojis which look like `` | ||
| let start_offset = node_offsets.1 | ||
| - image.url.len() | ||
| - 1 | ||
| - image | ||
| .title | ||
| .as_ref() | ||
| .map(|t| t.len() + 3) | ||
| .unwrap_or_default(); | ||
| let end_offset = node_offsets.1 - 1; | ||
|
|
||
| links_offsets.push((start_offset, end_offset)); | ||
| } | ||
| }); | ||
|
|
||
| let mut links = vec![]; | ||
| // Go through the collected links in reverse order | ||
| while let Some((start, end)) = links_offsets.pop() { | ||
| let content = src.get(start..end).unwrap_or_default(); | ||
| // necessary for custom emojis which look like `` | ||
| let (url, extra) = if content.contains(' ') { | ||
| let split = content.split_once(' ').expect("split is valid"); | ||
| (split.0, Some(split.1)) | ||
| } else { | ||
| (content, None) | ||
| }; | ||
| match Url::parse(url) { | ||
| Ok(parsed) => { | ||
| links.push(parsed.clone()); | ||
| // If link points to remote domain, replace with proxied link | ||
| if parsed.domain() != Some(&SETTINGS.hostname) { | ||
| let mut proxied = format!( | ||
| "{}/api/v3/image_proxy?url={}", | ||
| SETTINGS.get_protocol_and_hostname(), | ||
| encode(url), | ||
| ); | ||
| // restore custom emoji format | ||
| if let Some(extra) = extra { | ||
| proxied = format!("{proxied} {extra}"); | ||
| } | ||
| src.replace_range(start..end, &proxied); | ||
| } | ||
| } | ||
| Err(_) => { | ||
| // If its not a valid url, replace with empty text | ||
| src.replace_range(start..end, ""); | ||
| } | ||
| } | ||
| } | ||
|
|
||
| (src, links) | ||
| } | ||
|
|
||
| #[cfg(test)] | ||
| #[expect(clippy::unwrap_used)] | ||
| mod tests { | ||
|
|
||
| use super::*; | ||
| use pretty_assertions::assert_eq; | ||
|
|
||
| #[test] | ||
| fn test_markdown_proxy_images() { | ||
| let tests: Vec<_> = | ||
| vec", | ||
| "", | ||
| ), | ||
| ( | ||
| "local image unproxied", | ||
| "", | ||
| "", | ||
| ), | ||
| ( | ||
| "multiple image links", | ||
| " ", | ||
| " ", | ||
| ), | ||
| ( | ||
| "empty link handled", | ||
| "![image]()", | ||
| "![image]()" | ||
| ), | ||
| ( | ||
| "empty label handled", | ||
| "", | ||
| "" | ||
| ), | ||
| ( | ||
| "invalid image link removed", | ||
| "", | ||
| "![image]()" | ||
| ), | ||
| ( | ||
| "label with nested markdown handled", | ||
| "", | ||
| "" | ||
| ), | ||
| ( | ||
| "custom emoji support", | ||
| r#""#, | ||
| r#""# | ||
| ) | ||
| ]; | ||
|
|
||
| tests.iter().for_each(|&(msg, input, expected)| { | ||
| let result = markdown_rewrite_image_links(input.to_string()); | ||
|
|
||
| assert_eq!( | ||
| result.0, expected, | ||
| "Testing {}, with original input '{}'", | ||
| msg, input | ||
| ); | ||
| }); | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters