135 lines
4.1 KiB
Rust
135 lines
4.1 KiB
Rust
use std::{ffi::OsString, fs::OpenOptions, io, path::PathBuf};
|
|
|
|
use ratatui::{backend::CrosstermBackend, Terminal};
|
|
use structopt::StructOpt;
|
|
|
|
use musichoard::{
|
|
database::{
|
|
json::{backend::JsonDatabaseFileBackend, JsonDatabase},
|
|
IDatabase, NullDatabase,
|
|
},
|
|
library::{
|
|
beets::{
|
|
executor::{ssh::BeetsLibrarySshExecutor, BeetsLibraryProcessExecutor},
|
|
BeetsLibrary,
|
|
},
|
|
ILibrary, NullLibrary,
|
|
},
|
|
MusicHoardBuilder, NoDatabase, NoLibrary,
|
|
};
|
|
|
|
mod tui;
|
|
use tui::{event::EventChannel, handler::EventHandler, listener::EventListener, ui::Ui, Tui};
|
|
|
|
#[derive(StructOpt)]
|
|
struct Opt {
|
|
#[structopt(flatten)]
|
|
lib_opt: LibOpt,
|
|
|
|
#[structopt(flatten)]
|
|
db_opt: DbOpt,
|
|
}
|
|
|
|
#[derive(StructOpt)]
|
|
struct LibOpt {
|
|
#[structopt(long = "ssh", help = "Beets SSH URI")]
|
|
beets_ssh_uri: Option<OsString>,
|
|
|
|
#[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,
|
|
}
|
|
|
|
#[derive(StructOpt)]
|
|
struct DbOpt {
|
|
#[structopt(
|
|
long = "database",
|
|
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>(builder: MusicHoardBuilder<LIB, DB>) {
|
|
let music_hoard = builder.build();
|
|
|
|
// Initialize the terminal user interface.
|
|
let backend = CrosstermBackend::new(io::stdout());
|
|
let terminal = Terminal::new(backend).expect("failed to initialise terminal");
|
|
|
|
let channel = EventChannel::new();
|
|
let listener = EventListener::new(channel.sender());
|
|
let handler = EventHandler::new(channel.receiver());
|
|
|
|
let ui = Ui::new(music_hoard).expect("failed to initialise ui");
|
|
|
|
// Run the TUI application.
|
|
Tui::run(terminal, ui, handler, listener).expect("failed to run tui");
|
|
}
|
|
|
|
fn with_database<LIB: ILibrary>(db_opt: DbOpt, builder: MusicHoardBuilder<LIB, NoDatabase>) {
|
|
if db_opt.no_database {
|
|
with(builder.set_database(NullDatabase));
|
|
} else {
|
|
// Create an empty database file if it does not exist.
|
|
match OpenOptions::new()
|
|
.write(true)
|
|
.create_new(true)
|
|
.open(&db_opt.database_file_path)
|
|
{
|
|
Ok(f) => {
|
|
drop(f);
|
|
JsonDatabase::new(JsonDatabaseFileBackend::new(&db_opt.database_file_path))
|
|
.save(&vec![])
|
|
.expect("failed to create empty database");
|
|
}
|
|
Err(e) => match e.kind() {
|
|
io::ErrorKind::AlreadyExists => {}
|
|
_ => panic!("failed to access database file"),
|
|
},
|
|
}
|
|
|
|
let db_exec = JsonDatabaseFileBackend::new(&db_opt.database_file_path);
|
|
with(builder.set_database(JsonDatabase::new(db_exec)));
|
|
};
|
|
}
|
|
|
|
fn with_library(lib_opt: LibOpt, db_opt: DbOpt, builder: MusicHoardBuilder<NoLibrary, NoDatabase>) {
|
|
if lib_opt.no_library {
|
|
with_database(db_opt, builder.set_library(NullLibrary));
|
|
} else if let Some(uri) = lib_opt.beets_ssh_uri {
|
|
let uri = uri.into_string().expect("invalid SSH URI");
|
|
let beets_config_file_path = lib_opt
|
|
.beets_config_file_path
|
|
.map(|s| s.into_string())
|
|
.transpose()
|
|
.expect("failed to extract beets config file path");
|
|
let lib_exec = BeetsLibrarySshExecutor::new(uri)
|
|
.expect("failed to initialise beets")
|
|
.config(beets_config_file_path);
|
|
with_database(db_opt, builder.set_library(BeetsLibrary::new(lib_exec)));
|
|
} else {
|
|
let lib_exec =
|
|
BeetsLibraryProcessExecutor::default().config(lib_opt.beets_config_file_path);
|
|
with_database(db_opt, builder.set_library(BeetsLibrary::new(lib_exec)));
|
|
}
|
|
}
|
|
|
|
fn main() {
|
|
let opt = Opt::from_args();
|
|
let builder = MusicHoardBuilder::default();
|
|
with_library(opt.lib_opt, opt.db_opt, builder);
|
|
}
|
|
|
|
#[cfg(test)]
|
|
#[macro_use]
|
|
mod testmacros;
|
|
|
|
#[cfg(test)]
|
|
mod testbin;
|