use std::error::Error; use std::path::Path; use std::rc::Rc; use std::sync::{Arc, OnceLock}; use cursive::traits::Nameable; use cursive::{Cursive, CursiveRunner}; use log::{error, info, trace}; #[cfg(unix)] use signal_hook::{consts::SIGHUP, consts::SIGTERM, iterator::Signals}; use crate::command::Command; use crate::commands::CommandManager; use crate::config::{Config, PlaybackState}; use crate::events::{Event, EventManager}; use crate::library::Library; use crate::queue::Queue; use crate::spotify::{PlayerEvent, Spotify}; use crate::ui::create_cursive; use crate::{authentication, ui, utils}; use crate::{command, queue, spotify}; #[cfg(feature = "mpris")] use crate::mpris::MprisManager; #[cfg(unix)] use crate::ipc::{self, IpcSocket}; /// Set up the global logger to log to `filename`. pub fn setup_logging(filename: &Path) -> 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) // Set runtime log level for modules .level_for("librespot", log::LevelFilter::Debug) // Output to stdout, files, and other Dispatch configurations .chain(fern::log_file(filename)?) // Apply globally .apply()?; Ok(()) } pub type UserData = Rc; pub struct UserDataInner { pub cmd: CommandManager, } /// The global Tokio runtime for running asynchronous tasks. pub static ASYNC_RUNTIME: OnceLock = OnceLock::new(); /// The representation of an ncspot application. pub struct Application { /// The music queue which controls playback order. queue: Arc, /// Internally shared spotify: Spotify, /// Internally shared event_manager: EventManager, /// An IPC implementation using the D-Bus MPRIS protocol, used to control and inspect ncspot. #[cfg(unix)] ipc: Option, /// The object to render to the terminal. cursive: CursiveRunner, } impl Application { /// Create a new ncspot application. /// /// # Arguments /// /// * `configuration_file_path` - Relative path to the configuration file inside the base path pub fn new(configuration_file_path: Option) -> Result> { // 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 ASYNC_RUNTIME .set( tokio::runtime::Builder::new_multi_thread() .enable_all() .build() .unwrap(), ) .unwrap(); let configuration = Arc::new(Config::new(configuration_file_path)); let credentials = authentication::get_credentials(&configuration)?; let theme = configuration.build_theme(); println!("Connecting to Spotify.."); // DON'T USE STDOUT AFTER THIS CALL! let mut cursive = create_cursive().map_err(|error| error.to_string())?; cursive.set_theme(theme.clone()); #[cfg(all(unix, feature = "pancurses_backend"))] cursive.add_global_callback(cursive::event::Event::CtrlChar('z'), |_s| unsafe { libc::raise(libc::SIGTSTP); }); let event_manager = EventManager::new(cursive.cb_sink().clone()); let mut spotify = spotify::Spotify::new(event_manager.clone(), credentials, configuration.clone())?; let library = Arc::new(Library::new( event_manager.clone(), spotify.clone(), configuration.clone(), )); let queue = Arc::new(queue::Queue::new( spotify.clone(), configuration.clone(), library.clone(), )); #[cfg(feature = "mpris")] let mpris_manager = MprisManager::new( event_manager.clone(), queue.clone(), library.clone(), spotify.clone(), ); #[cfg(feature = "mpris")] spotify.set_mpris(mpris_manager.clone()); // Load the last played track into the player let playback_state = configuration.state().playback_state.clone(); let queue_state = configuration.state().queuestate.clone(); if let Some(playable) = queue.get_current() { spotify.load( &playable, playback_state == PlaybackState::Playing, queue_state.track_progress.as_millis() as u32, ); spotify.update_track(); match playback_state { PlaybackState::Stopped => { spotify.stop(); } PlaybackState::Paused | PlaybackState::Playing | PlaybackState::Default => { spotify.pause(); } } } #[cfg(unix)] let ipc = if let Ok(runtime_directory) = utils::create_runtime_directory() { Some( ipc::IpcSocket::new( ASYNC_RUNTIME.get().unwrap().handle(), runtime_directory.join("ncspot.sock"), event_manager.clone(), ) .map_err(|e| e.to_string())?, ) } else { error!("failed to create IPC socket: no suitable user runtime directory found"); None }; let mut cmd_manager = CommandManager::new( spotify.clone(), queue.clone(), library.clone(), configuration.clone(), event_manager.clone(), ); cmd_manager.register_all(); cmd_manager.register_keybindings(&mut cursive); cursive.set_user_data(Rc::new(UserDataInner { cmd: cmd_manager })); let search = ui::search::SearchView::new(event_manager.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()); #[cfg(feature = "cover")] let coverview = ui::cover::CoverView::new(queue.clone(), library.clone(), &configuration); let status = ui::statusbar::StatusBar::new(queue.clone(), Arc::clone(&library)); let mut layout = ui::layout::Layout::new(status, &event_manager, theme, Arc::clone(&configuration)) .screen("search", search.with_name("search")) .screen("library", libraryview.with_name("library")) .screen("queue", queueview); #[cfg(feature = "cover")] layout.add_screen("cover", coverview.with_name("cover")); // initial screen is library let initial_screen = configuration .values() .initial_screen .clone() .unwrap_or_else(|| "library".to_string()); if layout.has_screen(&initial_screen) { layout.set_screen(initial_screen); } else { error!("Invalid screen name: {}", initial_screen); layout.set_screen("library"); } cursive.add_fullscreen_layer(layout.with_name("main")); Ok(Self { queue, spotify, event_manager, #[cfg(unix)] ipc, cursive, }) } /// Start the application and run the event loop. pub fn run(&mut self) -> Result<(), String> { #[cfg(unix)] let mut signals = Signals::new([SIGTERM, SIGHUP]).expect("could not register signal handler"); // cursive event loop while self.cursive.is_running() { self.cursive.step(); #[cfg(unix)] for signal in signals.pending() { if signal == SIGTERM || signal == SIGHUP { info!("Caught {}, cleaning up and closing", signal); if let Some(data) = self.cursive.user_data::().cloned() { data.cmd.handle(&mut self.cursive, Command::Quit); } } } for event in self.event_manager.msg_iter() { match event { Event::Player(state) => { trace!("event received: {:?}", state); self.spotify.update_status(state.clone()); #[cfg(unix)] if let Some(ref ipc) = self.ipc { ipc.publish(&state, self.queue.get_current()); } if state == PlayerEvent::FinishedTrack { self.queue.next(false); } } Event::Queue(event) => { self.queue.handle_event(event); } Event::SessionDied => { if self.spotify.start_worker(None).is_err() { let data: UserData = self .cursive .user_data() .cloned() .expect("user data should be set"); data.cmd.handle(&mut self.cursive, Command::Quit); }; } Event::IpcInput(input) => match command::parse(&input) { Ok(commands) => { if let Some(data) = self.cursive.user_data::().cloned() { for cmd in commands { info!("Executing command from IPC: {cmd}"); data.cmd.handle(&mut self.cursive, cmd); } } } Err(e) => error!("Parsing error: {e}"), }, } } } Ok(()) } }