Something went wrong. Try again.
[READ-ONLY] Mirror of https://github.com/kasper9n/ferrum. Music library app for Mac, Linux and Windows ferrum.kasper.space
Something went wrong. Try again.
5.5 kB · 196 lines
Rust
at sqlite
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197#[cfg(feature = "napi")]use crate::data::Data;#[cfg(feature = "napi")]use crate::filter::SortKey;use crate::library_types::{ItemId, Library, SpecialTrackListName, TrackList, VersionedLibrary};use crate::migrate::migrate_to_sqlite;use anyhow::{Context, Result};use linked_hash_map::LinkedHashMap;use rusqlite::{Connection, OpenFlags};use serde_json::{Value, json};use std::fs::File;use std::io::{ErrorKind, Read, Seek, SeekFrom};use std::path::PathBuf;use std::time::Instant;use std::{fs::create_dir_all, path::Path};
#[derive(Clone)]#[cfg_attr(feature = "napi", napi(object))]pub struct Paths { pub path_separator: String, pub library_dir: String, pub tracks_dir: String, pub library_sqlite: String, pub library_json: String, pub cache_dir: String, pub cache_db: String, pub local_data_dir: String, pub view_options_file: String, pub queue_file: String, pub logs_dir: String,}impl Paths { fn ensure_dirs_exists(&self) -> Result<()> { create_dir_all(&self.library_dir)?; create_dir_all(&self.tracks_dir)?; create_dir_all(&self.cache_dir)?; create_dir_all(&self.local_data_dir)?; // We do not create logs_dir, we create it lazily when a crash occurs return Ok(()); } pub fn get_track_file_path(&self, file: &str) -> PathBuf { PathBuf::from(&self.tracks_dir).join(file) }}
// todo: deletepub fn load_library_json(library_json: &PathBuf) -> Result<Option<Library>> { let mut library_file = match File::open(&library_json) { Ok(file) => file, Err(err) => match err.kind() { ErrorKind::NotFound => return Ok(None), _ => return Err(err).context("Error opening library file"), }, };
let mut json_bytes = Vec::new(); library_file .read_to_end(&mut json_bytes) .context("Error reading library file")?;
let versioned_library: VersionedLibrary = match simd_json::from_slice(&mut json_bytes) { Ok(lib) => lib, Err(_) => { library_file .seek(SeekFrom::Start(0)) .context("Error seeking to start of library file")?; let versioned_library = parse_old_versionless_library_json(&mut library_file)?; versioned_library } };
let library = versioned_library.upgrade().init_libary(); Ok(Some(library))}
// todo: deletefn parse_old_versionless_library_json(library_file: &mut File) -> Result<VersionedLibrary<'_>> { let mut json_str = String::new(); library_file .read_to_string(&mut json_str) .context("Error reading library file")?;
let mut value: Value = serde_json::from_str(&mut json_str).context("Error parsing library file")?; // Migrate version number to string if let Some(obj) = value.as_object_mut() { if let Some(version_field) = obj.get_mut("version") { if let Some(version) = version_field.as_number() { if version.as_u64() == Some(1) { *version_field = json!("1"); } else if version.as_u64() == Some(2) { *version_field = json!("2"); } } } }
let versioned_library: VersionedLibrary = serde_json::from_value(value).context("Error parsing library file")?; Ok(versioned_library)}
pub mod embedded_sql { use refinery::embed_migrations; embed_migrations!("./src-native/migrations");}
pub async fn open_library(paths: &Paths) -> Result<Connection> { let now = Instant::now();
paths .ensure_dirs_exists() .context("Error ensuring folder exists")?; println!("Loading library at path: {}", paths.library_dir);
let library_sqlite = &paths.library_sqlite;
let exists = Path::new(&library_sqlite).exists(); if !exists { migrate_to_sqlite(paths).await?; } let mut db = Connection::open_with_flags(&paths.library_sqlite, OpenFlags::SQLITE_OPEN_READ_WRITE) .context("Error connecting to library database")?; db.execute_batch("PRAGMA foreign_keys = ON;") .context("Error enabling foreign keys")?;
embedded_sql::migrations::runner() .run(&mut db) .context("Could not run database migrations")?;
println!("Open library: {}ms", now.elapsed().as_millis()); Ok(db)}
pub enum TrackField { String, F64, I64, U32,}
#[cfg(feature = "napi")]#[napi(js_name = "get_default_sort_desc")]#[allow(dead_code)]pub fn get_default_sort_desc(field: String) -> Result<bool> { if field == "index" { return Ok(true); } let field = SortKey::from_col_view_key(&field); let desc = match field.field_type() { TrackField::String => false, _ => true, }; Ok(desc)}
#[cfg(feature = "napi")]#[napi(js_name = "get_genres")]#[allow(dead_code)]pub fn get_genres() -> Vec<String> { let mut data = Data::get_blocking(); let genres = data.library.get_genres(); genres.clone()}
#[cfg(feature = "napi")]#[napi(js_name = "get_artists")]#[allow(dead_code)]pub fn get_artists() -> Vec<String> { let mut data = Data::get_blocking(); let genres = data.library.get_artists(); genres.clone()}
pub fn get_tracklist_item_ids(library: &Library, playlist_id: &str) -> Result<Vec<ItemId>> { match library.get_tracklist(playlist_id)? { TrackList::Playlist(playlist) => Ok(playlist.tracks.clone()), TrackList::Folder(folder) => { let mut ids: LinkedHashMap<ItemId, ()> = LinkedHashMap::new(); for child in &folder.children { let child_ids = get_tracklist_item_ids(library, &child)?; for child_id in child_ids { ids.insert(child_id, ()); } } Ok(ids.into_iter().map(|(id, _)| id).collect()) } TrackList::Special(special) => match special.name { SpecialTrackListName::Root => { let item_ids = library.get_track_item_ids().values().cloned().collect(); Ok(item_ids) } }, }}