mirror of
https://github.com/flibusta-apps/books_downloader.git
synced 2025-12-06 15:05:37 +01:00
Add rust implementation
This commit is contained in:
77
src/services/filename_getter.rs
Normal file
77
src/services/filename_getter.rs
Normal file
@@ -0,0 +1,77 @@
|
||||
use translit::{gost779b_ru, CharsMapping, Transliterator};
|
||||
|
||||
use super::book_library::types::{BookAuthor, Book};
|
||||
|
||||
pub fn get_author_short_name(author: BookAuthor) -> String {
|
||||
let mut parts: Vec<String> = vec![];
|
||||
|
||||
if author.last_name.len() != 0 {
|
||||
parts.push(author.last_name);
|
||||
}
|
||||
|
||||
if author.first_name.len() != 0 {
|
||||
let first_char = author.first_name.chars().next().unwrap();
|
||||
parts.push(first_char.to_string());
|
||||
}
|
||||
|
||||
if author.middle_name.len() != 0 {
|
||||
let first_char = author.middle_name.chars().next().unwrap();
|
||||
parts.push(first_char.to_string());
|
||||
}
|
||||
|
||||
parts.join(" ")
|
||||
}
|
||||
|
||||
pub fn get_filename_by_book(book: &Book, file_type: &str, force_zip: bool) -> String {
|
||||
let book_id = book.remote_id;
|
||||
let mut filename_parts: Vec<String> = vec![];
|
||||
|
||||
let file_type_: String = if let "fb2zip" = file_type {
|
||||
"fb2.zip".to_string()
|
||||
} else if force_zip {
|
||||
format!("{file_type}.zip")
|
||||
} else {
|
||||
file_type.to_string()
|
||||
};
|
||||
|
||||
filename_parts.push(
|
||||
book.authors
|
||||
.clone()
|
||||
.into_iter()
|
||||
.map(|author| get_author_short_name(author))
|
||||
.collect::<Vec<String>>()
|
||||
.join("_-_"),
|
||||
);
|
||||
filename_parts.push(book.title.trim().to_string());
|
||||
|
||||
let transliterator = Transliterator::new(gost779b_ru());
|
||||
let mut filename_without_type = transliterator.convert(&filename_parts.join(""), false);
|
||||
|
||||
for char in "(),….’!\"?»«':".get(..) {
|
||||
filename_without_type = filename_without_type.replace(char, "");
|
||||
}
|
||||
|
||||
let replace_char_map: CharsMapping = [
|
||||
("—", "-"),
|
||||
("/", "_"),
|
||||
("№", "N"),
|
||||
(" ", "_"),
|
||||
("–", "-"),
|
||||
("á", "a"),
|
||||
(" ", "_"),
|
||||
("'", ""),
|
||||
("`", ""),
|
||||
]
|
||||
.iter()
|
||||
.cloned()
|
||||
.collect();
|
||||
|
||||
let replace_transliterator = Transliterator::new(replace_char_map);
|
||||
let normal_filename = replace_transliterator.convert(&filename_without_type, false);
|
||||
|
||||
let right_part = format!(".{book_id}.{file_type_}");
|
||||
let normal_filename_slice = std::cmp::min(64 - right_part.len() - 1, normal_filename.len());
|
||||
let left_part = normal_filename.get(..normal_filename_slice).unwrap();
|
||||
|
||||
format!("{left_part}{right_part}")
|
||||
}
|
||||
Reference in New Issue
Block a user