arnekeller/KIT-ILIAS-downloader
1
0
Fork 0
mirror of https://github.com/FliegendeWurst/KIT-ILIAS-downloader.git synced 2024-08-28 04:04:18 +00:00
Code Issues Projects Releases Wiki Activity

Handle external images in forum posts

Browse Source
This commit is contained in:
FliegendeWurst 2021-04-19 18:47:19 +02:00
parent 3ece654b52
commit f291bcbb4b
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/main.rs
View File

@ -640,12 +640,15 @@ async fn process(ilias: Arc<ILIAS>, mut path: PathBuf, obj: Object) -> Result<()
for (id, image) in all_images { for (id, image) in all_images {
let src = URL::from_href(&image)?; let src = URL::from_href(&image)?;
let dl = ilias.download(&src.url).await?; let dl = ilias.download(&src.url).await?;
let m = image_src_regex
.captures(&image)
.context(format!("image src {} unexpected format", image))?;
let (media_id, filename) = (m.get(1).unwrap().as_str(), m.get(2).unwrap().as_str());
let mut path = path.clone(); let mut path = path.clone();
if let Some(m) = image_src_regex.captures(&image) {
// image uploaded to ILIAS
let (media_id, filename) = (m.get(1).unwrap().as_str(), m.get(2).unwrap().as_str());
path.push(file_escape(&format!("{}_{}_{}", id, media_id, filename))); path.push(file_escape(&format!("{}_{}_{}", id, media_id, filename)));
} else {
// external image
path.push(file_escape(&format!("{}_{}", id, image)));
}
spawn!(handle_gracefully(async move { spawn!(handle_gracefully(async move {
let bytes = dl.bytes().await?; let bytes = dl.bytes().await?;
write_file_data(&path, &mut &*bytes) write_file_data(&path, &mut &*bytes)