Make it possible to launch main binary without database and/or library #88

Merged
wojtek merged 3 commits from 87---make-it-possible-to-launch-main-binary-without-database-and/or-library into main 2024-01-11 23:27:01 +01:00
3 changed files with 61 additions and 68 deletions
Showing only changes of commit 8dc9ff6978 - Show all commits

View File

@ -16,7 +16,7 @@ type MH = MusicHoard<NoLibrary, JsonDatabase<JsonDatabaseFileBackend>>;
struct Opt {
#[structopt(
long = "database",
name = "database file path",
help = "Database file path",
default_value = "database.json"
)]
database_file_path: PathBuf,

View File

@ -788,46 +788,35 @@ impl<LIB: ILibrary, DB: IDatabase> MusicHoard<LIB, DB> {
}
pub fn rescan_library(&mut self) -> Result<(), Error> {
match self.library {
Some(ref mut library) => {
if let Some(ref mut library) = self.library {
let items = library.list(&Query::new())?;
let mut library_collection = Self::items_to_artists(items);
Self::sort(&mut library_collection);
let collection = mem::take(&mut self.collection);
self.collection = Self::merge(library_collection, collection);
}
Ok(())
}
None => Err(Error::LibraryError(String::from("library not provided"))),
}
}
pub fn load_from_database(&mut self) -> Result<(), Error> {
match self.database {
Some(ref mut database) => {
if let Some(ref mut database) = self.database {
let mut database_collection: Collection = vec![];
database.load(&mut database_collection)?;
Self::sort(&mut database_collection);
let collection = mem::take(&mut self.collection);
self.collection = Self::merge(collection, database_collection);
}
Ok(())
}
None => Err(Error::DatabaseError(String::from("database not provided"))),
}
}
pub fn save_to_database(&mut self) -> Result<(), Error> {
match self.database {
Some(ref mut database) => {
if let Some(ref mut database) = self.database {
database.save(&self.collection)?;
}
Ok(())
}
None => Err(Error::DatabaseError(String::from("database not provided"))),
}
}
pub fn add_artist<ID: Into<ArtistId>>(&mut self, artist_id: ID) {
let artist_id: ArtistId = artist_id.into();
@ -1963,10 +1952,7 @@ mod tests {
let database = Some(MockIDatabase::new());
let mut music_hoard = MusicHoard::new(library, database);
let actual_err = music_hoard.rescan_library().unwrap_err();
let expected_err = Error::LibraryError(String::from("library not provided"));
assert_eq!(actual_err, expected_err);
assert!(music_hoard.rescan_library().is_ok());
}
#[test]
@ -1997,15 +1983,8 @@ mod tests {
let database: Option<NoDatabase> = None;
let mut music_hoard = MusicHoard::new(library, database);
let expected_err = Error::DatabaseError(String::from("database not provided"));
let actual_err = music_hoard.load_from_database().unwrap_err();
assert_eq!(actual_err, expected_err);
assert_eq!(actual_err.to_string(), expected_err.to_string());
let actual_err = music_hoard.save_to_database().unwrap_err();
assert_eq!(actual_err, expected_err);
assert_eq!(actual_err.to_string(), expected_err.to_string());
assert!(music_hoard.load_from_database().is_ok());
assert!(music_hoard.save_to_database().is_ok());
}
#[test]

View File

@ -3,6 +3,7 @@ use std::path::PathBuf;
use std::{ffi::OsString, io};
use musichoard::Collection;
use musichoard::library::NoLibrary;
use ratatui::{backend::CrosstermBackend, Terminal};
use structopt::StructOpt;
@ -27,22 +28,28 @@ use tui::{event::EventChannel, handler::EventHandler, listener::EventListener, T
#[derive(StructOpt)]
struct Opt {
#[structopt(long = "ssh", name = "beets SSH URI")]
#[structopt(long = "ssh", help = "Beets SSH URI")]
beets_ssh_uri: Option<OsString>,
#[structopt(long = "beets", name = "beets config file path")]
#[structopt(long = "beets", help = "Beets config file path")]
beets_config_file_path: Option<OsString>,
#[structopt(long = "no-library", help = "Do not connect to the library")]
no_library: bool,
#[structopt(
long = "database",
name = "database file path",
help = "Database file path",
default_value = "database.json"
)]
database_file_path: PathBuf,
#[structopt(long = "no-database", help = "Do not read from/write to the database")]
no_database: bool,
}
fn with<LIB: ILibrary, DB: IDatabase>(lib: LIB, db: DB) {
let music_hoard = MusicHoard::new(Some(lib), Some(db));
fn with<LIB: ILibrary, DB: IDatabase>(lib: Option<LIB>, db: Option<DB>) {
let music_hoard = MusicHoard::new(lib, db);
// Initialize the terminal user interface.
let backend = CrosstermBackend::new(io::stdout());
@ -61,6 +68,9 @@ fn with<LIB: ILibrary, DB: IDatabase>(lib: LIB, db: DB) {
fn main() {
let opt = Opt::from_args();
let db = if opt.no_database {
None
} else {
// Create an empty database file if it does not exist.
match OpenOptions::new()
.write(true)
@ -79,9 +89,13 @@ fn main() {
},
}
// Create the application.
let db_exec = JsonDatabaseFileBackend::new(&opt.database_file_path);
if let Some(uri) = opt.beets_ssh_uri {
Some(JsonDatabase::new(db_exec))
};
if opt.no_library {
with(None::<NoLibrary>, db);
} else if let Some(uri) = opt.beets_ssh_uri {
let uri = uri.into_string().expect("invalid SSH URI");
let beets_config_file_path = opt
.beets_config_file_path
@ -91,10 +105,10 @@ fn main() {
let lib_exec = BeetsLibrarySshExecutor::new(uri)
.expect("failed to initialise beets")
.config(beets_config_file_path);
with(BeetsLibrary::new(lib_exec), JsonDatabase::new(db_exec));
with(Some(BeetsLibrary::new(lib_exec)), db);
} else {
let lib_exec = BeetsLibraryProcessExecutor::default().config(opt.beets_config_file_path);
with(BeetsLibrary::new(lib_exec), JsonDatabase::new(db_exec));
with(Some(BeetsLibrary::new(lib_exec)), db);
}
}