musichoard/src/main.rs

use std::fs::OpenOptions;
use std::path::PathBuf;
use std::{ffi::OsString, io};

use musichoard::Collection;
use ratatui::{backend::CrosstermBackend, Terminal};
use structopt::StructOpt;

use musichoard::{
    database::{
        json::{backend::JsonDatabaseFileBackend, JsonDatabase},
        IDatabase,
    },
    library::{
        beets::{
            executor::{ssh::BeetsLibrarySshExecutor, BeetsLibraryProcessExecutor},
            BeetsLibrary,
        },
        ILibrary,
    },
    MusicHoard,
};

mod tui;
use tui::ui::Ui;
use tui::{event::EventChannel, handler::EventHandler, listener::EventListener, Tui};

#[derive(StructOpt)]
struct Opt {
    #[structopt(long = "ssh", name = "beets SSH URI")]
    beets_ssh_uri: Option<OsString>,

    #[structopt(long = "beets", name = "beets config file path")]
    beets_config_file_path: Option<OsString>,

    #[structopt(
        long = "database",
        name = "database file path",
        default_value = "database.json"
    )]
    database_file_path: PathBuf,
}

fn with<LIB: ILibrary, DB: IDatabase>(lib: LIB, db: DB) {
    let music_hoard = MusicHoard::new(lib, db);

    // 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 main() {
    let opt = Opt::from_args();

    // Create an empty database file if it does not exist.
    match OpenOptions::new()
        .write(true)
        .create_new(true)
        .open(&opt.database_file_path)
    {
        Ok(f) => {
            drop(f);
            JsonDatabase::new(JsonDatabaseFileBackend::new(&opt.database_file_path))
                .save::<Collection>(&vec![])
                .expect("failed to create empty database");
        }
        Err(e) => match e.kind() {
            io::ErrorKind::AlreadyExists => {}
            _ => panic!("failed to access database file"),
        },
    }

    // Create the application.
    let db_exec = JsonDatabaseFileBackend::new(&opt.database_file_path);
    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
            .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(BeetsLibrary::new(lib_exec), JsonDatabase::new(db_exec));
    } else {
        let lib_exec = BeetsLibraryProcessExecutor::default().config(opt.beets_config_file_path);
        with(BeetsLibrary::new(lib_exec), JsonDatabase::new(db_exec));
    }
}

#[cfg(test)]
#[macro_use]
mod testlib;

#[cfg(test)]
mod tests {
    use once_cell::sync::Lazy;

    use musichoard::*;

    pub static COLLECTION: Lazy<Vec<Artist>> = Lazy::new(|| collection!());
}