66 lines
2.0 KiB
Rust
66 lines
2.0 KiB
Rust
use std::{collections::HashMap, path::Path};
|
|
|
|
use calibre_db::data::{
|
|
author::Author as DbAuthor, book::Book as DbBook, series::Series as DbSeries,
|
|
};
|
|
use serde::Serialize;
|
|
|
|
use crate::app_state::AppState;
|
|
|
|
#[derive(Debug, Serialize)]
|
|
pub struct Book {
|
|
pub id: u64,
|
|
pub title: String,
|
|
pub sort: String,
|
|
pub path: String,
|
|
pub author: DbAuthor,
|
|
pub series: Option<(DbSeries, f64)>,
|
|
pub formats: HashMap<String, String>,
|
|
}
|
|
|
|
impl Book {
|
|
pub fn from_db_book(
|
|
db_book: &DbBook,
|
|
db_series: Option<(DbSeries, f64)>,
|
|
author: DbAuthor,
|
|
formats: HashMap<String, String>,
|
|
) -> Self {
|
|
Self {
|
|
id: db_book.id,
|
|
title: db_book.title.clone(),
|
|
sort: db_book.sort.clone(),
|
|
path: db_book.path.clone(),
|
|
author: author.clone(),
|
|
series: db_series.map(|x| (x.0, x.1)),
|
|
formats,
|
|
}
|
|
}
|
|
|
|
fn formats(book: &DbBook, library_path: &Path) -> HashMap<String, String> {
|
|
let book_path = library_path.join(&book.path);
|
|
let mut formats = HashMap::new();
|
|
|
|
for entry in book_path.read_dir().unwrap().flatten() {
|
|
if let Some(extension) = entry.path().extension() {
|
|
let format = match extension.to_string_lossy().to_string().as_str() {
|
|
"pdf" => Some("pdf".to_string()),
|
|
"epub" => Some("epub".to_string()),
|
|
_ => None,
|
|
};
|
|
if let Some(format) = format {
|
|
formats.insert(format, entry.file_name().to_string_lossy().to_string());
|
|
}
|
|
}
|
|
}
|
|
|
|
formats
|
|
}
|
|
|
|
pub fn full_book(book: &DbBook, state: &AppState) -> Option<Book> {
|
|
let formats = Book::formats(book, &state.config.library_path);
|
|
let author = state.calibre.book_author(book.id).ok()?;
|
|
let series = state.calibre.book_series(book.id).ok()?;
|
|
Some(Book::from_db_book(book, series, author, formats))
|
|
}
|
|
}
|