Files
ncspot/src/main.rs
2020-02-05 22:48:35 +01:00

286 lines
7.8 KiB
Rust

extern crate clap;
extern crate crossbeam_channel;
#[macro_use]
extern crate cursive;
#[cfg(feature = "share_clipboard")]
extern crate clipboard;
extern crate directories;
extern crate failure;
extern crate futures;
#[macro_use]
extern crate lazy_static;
extern crate librespot_core;
extern crate librespot_playback;
extern crate librespot_protocol;
extern crate rspotify;
extern crate tokio;
extern crate tokio_core;
extern crate tokio_timer;
extern crate unicode_width;
extern crate webbrowser;
#[cfg(feature = "mpris")]
extern crate dbus;
#[macro_use]
extern crate serde;
extern crate serde_json;
extern crate toml;
#[macro_use]
extern crate log;
extern crate chrono;
extern crate fern;
extern crate rand;
extern crate url;
extern crate strum;
extern crate strum_macros;
use std::fs;
use std::path::PathBuf;
use std::process;
use std::str::FromStr;
use std::sync::Arc;
use clap::{App, Arg};
use cursive::traits::Identifiable;
use cursive::Cursive;
use librespot_core::authentication::Credentials;
mod album;
mod artist;
mod authentication;
mod command;
mod commands;
mod config;
mod events;
mod library;
mod playlist;
mod queue;
mod spotify;
mod theme;
mod track;
mod traits;
mod ui;
#[cfg(feature = "mpris")]
mod mpris;
use commands::CommandManager;
use events::{Event, EventManager};
use library::Library;
use spotify::PlayerEvent;
fn setup_logging(filename: &str) -> Result<(), fern::InitError> {
fern::Dispatch::new()
// Perform allocation-free log formatting
.format(|out, message, record| {
out.finish(format_args!(
"{} [{}] [{}] {}",
chrono::Local::now().format("[%Y-%m-%d][%H:%M:%S]"),
record.target(),
record.level(),
message
))
})
// Add blanket level filter -
.level(log::LevelFilter::Trace)
// - and per-module overrides
.level_for("librespot", log::LevelFilter::Debug)
// Output to stdout, files, and other Dispatch configurations
.chain(fern::log_file(filename)?)
// Apply globally
.apply()?;
Ok(())
}
fn get_credentials(reset: bool) -> Credentials {
let path = config::config_path("credentials.toml");
if reset && fs::remove_file(&path).is_err() {
error!("could not delete credential file");
}
let creds = ::config::load_or_generate_default(&path, authentication::create_credentials, true)
.unwrap_or_else(|e| {
eprintln!("{}", e);
process::exit(1);
});
#[cfg(target_family = "unix")]
std::fs::set_permissions(path, std::os::unix::fs::PermissionsExt::from_mode(0o600))
.unwrap_or_else(|e| {
eprintln!("{}", e);
process::exit(1);
});
creds
}
fn main() {
let matches = App::new("ncspot")
.version(env!("CARGO_PKG_VERSION"))
.author("Henrik Friedrichsen <henrik@affekt.org> and contributors")
.about("cross-platform ncurses Spotify client")
.arg(
Arg::with_name("debug")
.short("d")
.long("debug")
.value_name("FILE")
.help("Enable debug logging to the specified file")
.takes_value(true),
)
.arg(
Arg::with_name("basepath")
.short("b")
.long("basepath")
.value_name("PATH")
.help("custom basepath to config/cache files")
.takes_value(true),
)
.get_matches();
if let Some(filename) = matches.value_of("debug") {
setup_logging(filename).expect("can't setup logging");
}
if let Some(basepath) = matches.value_of("basepath") {
let path = PathBuf::from_str(basepath).expect("invalid path");
if !path.exists() {
fs::create_dir_all(&path).expect("could not create basepath directory");
}
*config::BASE_PATH.write().unwrap() = Some(path);
}
// Things here may cause the process to abort; we must do them before creating curses windows
// otherwise the error message will not be seen by a user
let cfg: ::config::Config = {
let path = config::config_path("config.toml");
::config::load_or_generate_default(path, |_| Ok(::config::Config::default()), false)
.unwrap_or_else(|e| {
eprintln!("{}", e);
process::exit(1);
})
};
let mut credentials = get_credentials(false);
while !spotify::Spotify::test_credentials(credentials.clone()) {
credentials = get_credentials(true);
}
let theme = theme::load(&cfg);
let mut cursive = Cursive::default();
cursive.set_theme(theme.clone());
let event_manager = EventManager::new(cursive.cb_sink().clone());
let spotify = Arc::new(spotify::Spotify::new(event_manager.clone(), credentials));
let queue = Arc::new(queue::Queue::new(spotify.clone()));
#[cfg(feature = "mpris")]
let mpris_manager = Arc::new(mpris::MprisManager::new(spotify.clone(), queue.clone()));
let library = Arc::new(Library::new(
&event_manager,
spotify.clone(),
cfg.use_nerdfont.unwrap_or(false),
));
let mut cmd_manager = CommandManager::new(
spotify.clone(),
queue.clone(),
library.clone(),
cfg.keybindings.clone(),
);
cmd_manager.register_all();
let cmd_manager = Arc::new(cmd_manager);
CommandManager::register_keybindings(cmd_manager.clone(), &mut cursive);
let search = ui::search::SearchView::new(
event_manager.clone(),
spotify.clone(),
queue.clone(),
library.clone(),
);
let libraryview = ui::library::LibraryView::new(queue.clone(), library.clone());
let queueview = ui::queue::QueueView::new(queue.clone(), library.clone());
let helpview = ui::help::HelpView::new(cmd_manager.keybindings().clone());
let status = ui::statusbar::StatusBar::new(
queue.clone(),
library.clone(),
cfg.use_nerdfont.unwrap_or(false),
);
let mut layout = ui::layout::Layout::new(status, &event_manager, theme)
.view("search", search.with_name("search"), "Search")
.view("library", libraryview.with_name("library"), "Library")
.view("queue", queueview, "Queue")
.view("help", helpview, "Help");
// initial view is library
layout.set_view("library");
cursive.add_global_callback(':', move |s| {
s.call_on_name("main", |v: &mut ui::layout::Layout| {
v.enable_cmdline();
});
});
layout.cmdline.set_on_edit(move |s, cmd, _| {
s.call_on_name("main", |v: &mut ui::layout::Layout| {
if cmd.is_empty() {
v.clear_cmdline();
}
});
});
{
let ev = event_manager.clone();
let cmd_manager = cmd_manager.clone();
layout.cmdline.set_on_submit(move |s, cmd| {
{
let mut main = s.find_name::<ui::layout::Layout>("main").unwrap();
main.clear_cmdline();
}
let c = &cmd[1..];
let parsed = command::parse(c);
if let Some(parsed) = parsed {
cmd_manager.handle(s, parsed);
}
ev.trigger();
});
}
cursive.add_fullscreen_layer(layout.with_name("main"));
// cursive event loop
while cursive.is_running() {
cursive.step();
for event in event_manager.msg_iter() {
match event {
Event::Player(state) => {
trace!("event received: {:?}", state);
spotify.update_status(state.clone());
#[cfg(feature = "mpris")]
mpris_manager.update();
if state == PlayerEvent::FinishedTrack {
queue.next(false);
}
}
}
}
}
}