use futures::sync::oneshot; use futures::{future, Future}; use std; use std::borrow::Cow; use std::io::{Read, Seek, SeekFrom, Result}; use std::mem; use std::process::Command; use std::sync::mpsc::{RecvError, TryRecvError, RecvTimeoutError}; use std::thread; use std::time::Duration; use config::{Bitrate, PlayerConfig}; use core::session::Session; use core::util::SpotifyId; use audio_backend::Sink; use audio::{AudioFile, AudioDecrypt}; use audio::{VorbisDecoder, VorbisPacket}; use metadata::{FileFormat, Track, Metadata}; use mixer::AudioFilter; pub struct Player { commands: Option>, thread_handle: Option>, } struct PlayerInternal { session: Session, config: PlayerConfig, commands: std::sync::mpsc::Receiver, state: PlayerState, sink: Box, sink_running: bool, audio_filter: Option>, } enum PlayerCommand { Load(SpotifyId, bool, u32, oneshot::Sender<()>), Play, Pause, Stop, Seek(u32), } impl Player { pub fn new(config: PlayerConfig, session: Session, audio_filter: Option>, sink_builder: F) -> Player where F: FnOnce() -> Box + Send + 'static { let (cmd_tx, cmd_rx) = std::sync::mpsc::channel(); let handle = thread::spawn(move || { debug!("new Player[{}]", session.session_id()); let internal = PlayerInternal { session: session, config: config, commands: cmd_rx, state: PlayerState::Stopped, sink: sink_builder(), sink_running: false, audio_filter: audio_filter, }; internal.run(); }); Player { commands: Some(cmd_tx), thread_handle: Some(handle), } } fn command(&self, cmd: PlayerCommand) { self.commands.as_ref().unwrap().send(cmd).unwrap(); } pub fn load(&self, track: SpotifyId, start_playing: bool, position_ms: u32) -> oneshot::Receiver<()> { let (tx, rx) = oneshot::channel(); self.command(PlayerCommand::Load(track, start_playing, position_ms, tx)); rx } pub fn play(&self) { self.command(PlayerCommand::Play) } pub fn pause(&self) { self.command(PlayerCommand::Pause) } pub fn stop(&self) { self.command(PlayerCommand::Stop) } pub fn seek(&self, position_ms: u32) { self.command(PlayerCommand::Seek(position_ms)); } } impl Drop for Player { fn drop(&mut self) { debug!("Shutting down player thread ..."); self.commands = None; if let Some(handle) = self.thread_handle.take() { match handle.join() { Ok(_) => (), Err(_) => error!("Player thread panicked!") } } } } type Decoder = VorbisDecoder>>; enum PlayerState { Stopped, Paused { decoder: Decoder, end_of_track: oneshot::Sender<()>, }, Playing { decoder: Decoder, end_of_track: oneshot::Sender<()>, }, Invalid, } impl PlayerState { fn is_playing(&self) -> bool { use self::PlayerState::*; match *self { Stopped | Paused { .. } => false, Playing { .. } => true, Invalid => panic!("invalid state"), } } fn decoder(&mut self) -> Option<&mut Decoder> { use self::PlayerState::*; match *self { Stopped => None, Paused { ref mut decoder, .. } | Playing { ref mut decoder, .. } => Some(decoder), Invalid => panic!("invalid state"), } } fn signal_end_of_track(self) { use self::PlayerState::*; match self { Paused { end_of_track, .. } | Playing { end_of_track, .. } => { let _ = end_of_track.send(()); } Stopped => warn!("signal_end_of_track from stopped state"), Invalid => panic!("invalid state"), } } fn paused_to_playing(&mut self) { use self::PlayerState::*; match ::std::mem::replace(self, Invalid) { Paused { decoder, end_of_track } => { *self = Playing { decoder: decoder, end_of_track: end_of_track, }; } _ => panic!("invalid state"), } } fn playing_to_paused(&mut self) { use self::PlayerState::*; match ::std::mem::replace(self, Invalid) { Playing { decoder, end_of_track } => { *self = Paused { decoder: decoder, end_of_track: end_of_track, }; } _ => panic!("invalid state"), } } } impl PlayerInternal { fn run(mut self) { loop { let cmd = if self.state.is_playing() { if self.sink_running { match self.commands.try_recv() { Ok(cmd) => Some(cmd), Err(TryRecvError::Empty) => None, Err(TryRecvError::Disconnected) => return, } } else { match self.commands.recv_timeout(Duration::from_secs(5)) { Ok(cmd) => Some(cmd), Err(RecvTimeoutError::Timeout) => None, Err(RecvTimeoutError::Disconnected) => return, } } } else { match self.commands.recv() { Ok(cmd) => Some(cmd), Err(RecvError) => return, } }; if let Some(cmd) = cmd { self.handle_command(cmd); } if self.state.is_playing() && ! self.sink_running { self.start_sink(); } if self.sink_running { let packet = if let PlayerState::Playing { ref mut decoder, .. } = self.state { Some(decoder.next_packet().expect("Vorbis error")) } else { None }; if let Some(packet) = packet { self.handle_packet(packet); } } } } fn start_sink(&mut self) { match self.sink.start() { Ok(()) => self.sink_running = true, Err(err) => error!("Could not start audio: {}", err), } } fn stop_sink_if_running(&mut self) { if self.sink_running { self.stop_sink(); } } fn stop_sink(&mut self) { self.sink.stop().unwrap(); self.sink_running = false; } fn handle_packet(&mut self, packet: Option) { match packet { Some(mut packet) => { if let Some(ref editor) = self.audio_filter { editor.modify_stream(&mut packet.data_mut()) }; if let Err(err) = self.sink.write(&packet.data()) { error!("Could not write audio: {}", err); self.stop_sink(); } } None => { self.stop_sink(); self.run_onstop(); let old_state = mem::replace(&mut self.state, PlayerState::Stopped); old_state.signal_end_of_track(); } } } fn handle_command(&mut self, cmd: PlayerCommand) { debug!("command={:?}", cmd); match cmd { PlayerCommand::Load(track_id, play, position, end_of_track) => { if self.state.is_playing() { self.stop_sink_if_running(); } match self.load_track(track_id, position as i64) { Some(decoder) => { if play { if !self.state.is_playing() { self.run_onstart(); } self.start_sink(); self.state = PlayerState::Playing { decoder: decoder, end_of_track: end_of_track, }; } else { if self.state.is_playing() { self.run_onstop(); } self.state = PlayerState::Paused { decoder: decoder, end_of_track: end_of_track, }; } } None => { let _ = end_of_track.send(()); if self.state.is_playing() { self.run_onstop(); } } } } PlayerCommand::Seek(position) => { if let Some(decoder) = self.state.decoder() { match decoder.seek(position as i64) { Ok(_) => (), Err(err) => error!("Vorbis error: {:?}", err), } } else { warn!("Player::seek called from invalid state"); } } PlayerCommand::Play => { if let PlayerState::Paused { .. } = self.state { self.state.paused_to_playing(); self.run_onstart(); self.start_sink(); } else { warn!("Player::play called from invalid state"); } } PlayerCommand::Pause => { if let PlayerState::Playing { .. } = self.state { self.state.playing_to_paused(); self.stop_sink_if_running(); self.run_onstop(); } else { warn!("Player::pause called from invalid state"); } } PlayerCommand::Stop => { match self.state { PlayerState::Playing { .. } => { self.stop_sink_if_running(); self.run_onstop(); self.state = PlayerState::Stopped; } PlayerState::Paused { .. } => { self.state = PlayerState::Stopped; }, PlayerState::Stopped => { warn!("Player::stop called from invalid state"); } PlayerState::Invalid => panic!("invalid state"), } } } } fn run_onstart(&self) { if let Some(ref program) = self.config.onstart { run_program(program) } } fn run_onstop(&self) { if let Some(ref program) = self.config.onstop { run_program(program) } } fn find_available_alternative<'a>(&self, track: &'a Track) -> Option> { if track.available { Some(Cow::Borrowed(track)) } else { let alternatives = track.alternatives .iter() .map(|alt_id| { Track::get(&self.session, *alt_id) }); let alternatives = future::join_all(alternatives).wait().unwrap(); alternatives.into_iter().find(|alt| alt.available).map(Cow::Owned) } } fn load_track(&self, track_id: SpotifyId, position: i64) -> Option { let track = Track::get(&self.session, track_id).wait().unwrap(); info!("Loading track \"{}\"", track.name); let track = match self.find_available_alternative(&track) { Some(track) => track, None => { warn!("Track \"{}\" is not available", track.name); return None; } }; let format = match self.config.bitrate { Bitrate::Bitrate96 => FileFormat::OGG_VORBIS_96, Bitrate::Bitrate160 => FileFormat::OGG_VORBIS_160, Bitrate::Bitrate320 => FileFormat::OGG_VORBIS_320, }; let file_id = match track.files.get(&format) { Some(&file_id) => file_id, None => { warn!("Track \"{}\" is not available in format {:?}", track.name, format); return None; } }; let key = self.session.audio_key().request(track.id, file_id).wait().unwrap(); let encrypted_file = AudioFile::open(&self.session, file_id).wait().unwrap(); let audio_file = Subfile::new(AudioDecrypt::new(key, encrypted_file), 0xa7); let mut decoder = VorbisDecoder::new(audio_file).unwrap(); match decoder.seek(position) { Ok(_) => (), Err(err) => error!("Vorbis error: {:?}", err), } info!("Track \"{}\" loaded", track.name); Some(decoder) } } impl Drop for PlayerInternal { fn drop(&mut self) { debug!("drop Player[{}]", self.session.session_id()); } } impl ::std::fmt::Debug for PlayerCommand { fn fmt(&self, f: &mut ::std::fmt::Formatter) -> ::std::fmt::Result { match *self { PlayerCommand::Load(track, play, position, _) => { f.debug_tuple("Load") .field(&track) .field(&play) .field(&position) .finish() } PlayerCommand::Play => { f.debug_tuple("Play").finish() } PlayerCommand::Pause => { f.debug_tuple("Pause").finish() } PlayerCommand::Stop => { f.debug_tuple("Stop").finish() } PlayerCommand::Seek(position) => { f.debug_tuple("Seek") .field(&position) .finish() } } } } struct Subfile { stream: T, offset: u64, } impl Subfile { pub fn new(mut stream: T, offset: u64) -> Subfile { stream.seek(SeekFrom::Start(offset)).unwrap(); Subfile { stream: stream, offset: offset, } } } impl Read for Subfile { fn read(&mut self, buf: &mut [u8]) -> Result { self.stream.read(buf) } } impl Seek for Subfile { fn seek(&mut self, mut pos: SeekFrom) -> Result { pos = match pos { SeekFrom::Start(offset) => SeekFrom::Start(offset + self.offset), x => x, }; let newpos = try!(self.stream.seek(pos)); if newpos > self.offset { Ok(newpos - self.offset) } else { Ok(0) } } } fn run_program(program: &str) { info!("Running {}", program); let mut v: Vec<&str> = program.split_whitespace().collect(); let status = Command::new(&v.remove(0)) .args(&v) .status() .expect("program failed to start"); info!("Exit status: {}", status); }