use futures::sync::oneshot; use futures::{future, Future}; use std::borrow::Cow; use std::io::{Read, Seek}; use std::mem; use std::sync::mpsc::{RecvError, TryRecvError}; use std::thread; use std; use vorbis::{self, VorbisError}; use audio_backend::Sink; use audio_decrypt::AudioDecrypt; use audio_file::AudioFile; use metadata::{FileFormat, Track}; use session::Session; use mixer::AudioFilter; use util::{self, SpotifyId, Subfile}; use config::{Bitrate, PlayerConfig}; #[derive(Clone)] pub struct Player { commands: std::sync::mpsc::Sender, } struct PlayerInternal { session: Session, config: PlayerConfig, commands: std::sync::mpsc::Receiver, state: PlayerState, sink: Box, 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(); 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(), audio_filter: audio_filter, }; internal.run(); }); Player { commands: cmd_tx, } } fn command(&self, cmd: PlayerCommand) { self.commands.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)); } } type Decoder = vorbis::Decoder>>; 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, .. } => { end_of_track.complete(()) } 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() { match self.commands.try_recv() { Ok(cmd) => Some(cmd), Err(TryRecvError::Empty) => None, Err(TryRecvError::Disconnected) => return, } } else { match self.commands.recv() { Ok(cmd) => Some(cmd), Err(RecvError) => return, } }; if let Some(cmd) = cmd { self.handle_command(cmd); } let packet = if let PlayerState::Playing { ref mut decoder, .. } = self.state { Some(decoder.packets().next()) } else { None }; if let Some(packet) = packet { self.handle_packet(packet); } } } fn handle_packet(&mut self, packet: Option>) { match packet { Some(Ok(mut packet)) => { if let Some(ref editor) = self.audio_filter { editor.modify_stream(&mut packet.data) }; self.sink.write(&packet.data).unwrap(); } Some(Err(vorbis::VorbisError::Hole)) => (), Some(Err(e)) => panic!("Vorbis error {:?}", e), None => { self.sink.stop().unwrap(); 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.sink.stop().unwrap(); } match self.load_track(track_id, position as i64) { Some(decoder) => { if play { if !self.state.is_playing() { self.run_onstart(); } self.sink.start().unwrap(); 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 => { end_of_track.complete(()); if self.state.is_playing() { self.run_onstop(); } } } } PlayerCommand::Seek(position) => { if let Some(decoder) = self.state.decoder() { match vorbis_time_seek_ms(decoder, 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.sink.start().unwrap(); } else { warn!("Player::play called from invalid state"); } } PlayerCommand::Pause => { if let PlayerState::Playing { .. } = self.state { self.state.playing_to_paused(); self.sink.stop().unwrap(); self.run_onstop(); } else { warn!("Player::pause called from invalid state"); } } PlayerCommand::Stop => { match self.state { PlayerState::Playing { .. } => { self.sink.stop().unwrap(); 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 { util::run_program(program) } } fn run_onstop(&self) { if let Some(ref program) = self.config.onstop { util::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| { self.session.metadata().get::(*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 = self.session.metadata().get::(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 open = self.session.audio_file().open(file_id); let encrypted_file = open.wait().unwrap(); let audio_file = Subfile::new(AudioDecrypt::new(key, encrypted_file), 0xa7); let mut decoder = vorbis::Decoder::new(audio_file).unwrap(); match vorbis_time_seek_ms(&mut decoder, 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()); } } #[cfg(not(feature = "with-tremor"))] fn vorbis_time_seek_ms(decoder: &mut vorbis::Decoder, ms: i64) -> Result<(), vorbis::VorbisError> where R: Read + Seek { decoder.time_seek(ms as f64 / 1000f64) } #[cfg(feature = "with-tremor")] fn vorbis_time_seek_ms(decoder: &mut vorbis::Decoder, ms: i64) -> Result<(), vorbis::VorbisError> where R: Read + Seek { decoder.time_seek(ms) } 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() } } } }