phoenixandphoenix
e3ca2c5781
Reviewed-on: phoenix/icarus_models#67 Co-authored-by: phoenix <kundeng00@pm.me> Co-committed-by: phoenix <kundeng00@pm.me>
72 lines
1.7 KiB
Rust
72 lines
1.7 KiB
Rust
use std::io::Write;
|
|
|
|
use serde::{Deserialize, Serialize};
|
|
|
|
#[derive(Clone, Debug, Default, Deserialize, Serialize, utoipa::ToSchema)]
|
|
pub struct CoverArt {
|
|
pub id: uuid::Uuid,
|
|
pub title: String,
|
|
#[serde(skip)]
|
|
pub path: String,
|
|
#[serde(skip)]
|
|
pub data: Vec<u8>,
|
|
pub song_id: uuid::Uuid,
|
|
}
|
|
|
|
pub mod init {
|
|
use crate::coverart::CoverArt;
|
|
|
|
pub fn init_coverart_only_path(path: String) -> CoverArt {
|
|
CoverArt {
|
|
id: uuid::Uuid::nil(),
|
|
title: String::new(),
|
|
path: path.clone(),
|
|
data: Vec::new(),
|
|
song_id: uuid::Uuid::nil(),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl CoverArt {
|
|
/// Saves the coverart to the filesystem
|
|
pub fn save_to_filesystem(&self) -> Result<(), std::io::Error> {
|
|
match std::fs::File::create(&self.path) {
|
|
Ok(mut file) => match file.write_all(&self.data) {
|
|
Ok(_) => Ok(()),
|
|
Err(err) => Err(err),
|
|
},
|
|
Err(err) => Err(err),
|
|
}
|
|
}
|
|
|
|
// TODO: Add method to remove from filesystem
|
|
}
|
|
|
|
pub mod io {
|
|
use std::io::Read;
|
|
|
|
/// Gets the raw data of the cover art
|
|
pub fn to_data(coverart: &super::CoverArt) -> Result<Vec<u8>, std::io::Error> {
|
|
let path: &String = &coverart.path;
|
|
let mut file = std::fs::File::open(path)?;
|
|
let mut buffer = Vec::new();
|
|
match file.read_to_end(&mut buffer) {
|
|
Ok(_) => Ok(buffer),
|
|
Err(err) => Err(err),
|
|
}
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use crate::coverart;
|
|
|
|
#[test]
|
|
fn test_cover_art_image() {
|
|
let path: String = String::from("somepath");
|
|
let coverart = coverart::init::init_coverart_only_path(path.clone());
|
|
|
|
assert_eq!(path, coverart.path);
|
|
}
|
|
}
|