Clean up before merging.

This commit is contained in:
Simon Persson 2016-02-19 00:02:41 +01:00
parent c98ff9bd54
commit 3ccb2e2a10

View file

@ -117,32 +117,28 @@ impl SpircManager {
pub fn send_play(&mut self, recipient: &str) { pub fn send_play(&mut self, recipient: &str) {
let mut internal = self.0.lock().unwrap(); let mut internal = self.0.lock().unwrap();
CommandSender::new(&mut *internal, CommandSender::new(&mut *internal, MessageType::kMessageTypePlay)
protocol::spirc::MessageType::kMessageTypePlay)
.recipient(recipient) .recipient(recipient)
.send(); .send();
} }
pub fn send_pause(&mut self, recipient: &str) { pub fn send_pause(&mut self, recipient: &str) {
let mut internal = self.0.lock().unwrap(); let mut internal = self.0.lock().unwrap();
CommandSender::new(&mut *internal, CommandSender::new(&mut *internal, MessageType::kMessageTypePause)
protocol::spirc::MessageType::kMessageTypePause)
.recipient(recipient) .recipient(recipient)
.send(); .send();
} }
pub fn send_prev(&mut self, recipient: &str) { pub fn send_prev(&mut self, recipient: &str) {
let mut internal = self.0.lock().unwrap(); let mut internal = self.0.lock().unwrap();
CommandSender::new(&mut *internal, CommandSender::new(&mut *internal, MessageType::kMessageTypePrev)
protocol::spirc::MessageType::kMessageTypePrev)
.recipient(recipient) .recipient(recipient)
.send(); .send();
} }
pub fn send_next(&mut self, recipient: &str) { pub fn send_next(&mut self, recipient: &str) {
let mut internal = self.0.lock().unwrap(); let mut internal = self.0.lock().unwrap();
CommandSender::new(&mut *internal, CommandSender::new(&mut *internal, MessageType::kMessageTypeNext)
protocol::spirc::MessageType::kMessageTypeNext)
.recipient(recipient) .recipient(recipient)
.send(); .send();
} }
@ -152,8 +148,7 @@ impl SpircManager {
track_ids: I) { track_ids: I) {
let state = track_ids_to_state(track_ids); let state = track_ids_to_state(track_ids);
let mut internal = self.0.lock().unwrap(); let mut internal = self.0.lock().unwrap();
CommandSender::new(&mut *internal, CommandSender::new(&mut *internal, MessageType::kMessageTypeReplace)
protocol::spirc::MessageType::kMessageTypeReplace)
.recipient(recipient) .recipient(recipient)
.state(state) .state(state)
.send(); .send();
@ -164,8 +159,7 @@ impl SpircManager {
track_ids: I) { track_ids: I) {
let state = track_ids_to_state(track_ids); let state = track_ids_to_state(track_ids);
let mut internal = self.0.lock().unwrap(); let mut internal = self.0.lock().unwrap();
CommandSender::new(&mut *internal, CommandSender::new(&mut *internal, MessageType::kMessageTypeLoad)
protocol::spirc::MessageType::kMessageTypeLoad)
.recipient(recipient) .recipient(recipient)
.state(state) .state(state)
.send(); .send();
@ -188,7 +182,7 @@ impl SpircInternal {
} }
} }
fn handle(&mut self, mut frame: protocol::spirc::Frame) { fn handle(&mut self, frame: protocol::spirc::Frame) {
if frame.get_ident() == self.ident || if frame.get_ident() == self.ident ||
(frame.get_recipient().len() > 0 && !frame.get_recipient().contains(&self.ident)) { (frame.get_recipient().len() > 0 && !frame.get_recipient().contains(&self.ident)) {
return; return;
@ -200,14 +194,15 @@ impl SpircInternal {
} }
if frame.has_ident() && !frame.has_goodbye() && frame.has_device_state() { if frame.has_ident() && !frame.has_goodbye() && frame.has_device_state() {
self.devices.insert(frame.take_ident(), frame.take_device_state().take_name()); self.devices.insert(frame.get_ident().into(),
frame.get_device_state().get_name().into());
} }
match frame.get_typ() { match frame.get_typ() {
protocol::spirc::MessageType::kMessageTypeHello => { MessageType::kMessageTypeHello => {
self.notify(false, Some(frame.get_ident())); self.notify(false, Some(frame.get_ident()));
} }
protocol::spirc::MessageType::kMessageTypeLoad => { MessageType::kMessageTypeLoad => {
if !self.is_active { if !self.is_active {
self.is_active = true; self.is_active = true;
self.became_active_at = util::now_ms(); self.became_active_at = util::now_ms();
@ -220,41 +215,42 @@ impl SpircInternal {
let position = frame.get_state().get_position_ms(); let position = frame.get_state().get_position_ms();
self.player.load(track, play, position); self.player.load(track, play, position);
} }
protocol::spirc::MessageType::kMessageTypePlay => { MessageType::kMessageTypePlay => {
self.player.play(); self.player.play();
} }
protocol::spirc::MessageType::kMessageTypePause => { MessageType::kMessageTypePause => {
self.player.pause(); self.player.pause();
} }
protocol::spirc::MessageType::kMessageTypeNext => { MessageType::kMessageTypeNext => {
self.index = (self.index + 1) % self.tracks.len() as u32; self.index = (self.index + 1) % self.tracks.len() as u32;
let track = self.tracks[self.index as usize]; let track = self.tracks[self.index as usize];
self.player.load(track, true, 0); self.player.load(track, true, 0);
} }
protocol::spirc::MessageType::kMessageTypePrev => { MessageType::kMessageTypePrev => {
self.index = (self.index - 1) % self.tracks.len() as u32; self.index = (self.index - 1) % self.tracks.len() as u32;
let track = self.tracks[self.index as usize]; let track = self.tracks[self.index as usize];
self.player.load(track, true, 0); self.player.load(track, true, 0);
} }
protocol::spirc::MessageType::kMessageTypeSeek => { MessageType::kMessageTypeSeek => {
self.player.seek(frame.get_position()); self.player.seek(frame.get_position());
} }
protocol::spirc::MessageType::kMessageTypeReplace => { MessageType::kMessageTypeReplace => {
self.reload_tracks(&frame); self.reload_tracks(&frame);
} }
protocol::spirc::MessageType::kMessageTypeNotify => { MessageType::kMessageTypeNotify => {
if self.is_active && frame.get_device_state().get_is_active() { if self.is_active && frame.get_device_state().get_is_active() {
self.is_active = false; self.is_active = false;
self.player.stop(); self.player.stop();
} }
if frame.has_ident() && frame.has_goodbye() {
self.devices.remove(&frame.take_ident());
}
} }
protocol::spirc::MessageType::kMessageTypeVolume => { MessageType::kMessageTypeVolume => {
self.player.volume(frame.get_volume() as u16); self.player.volume(frame.get_volume() as u16);
} }
MessageType::kMessageTypeGoodbye => {
if frame.has_ident() {
self.devices.remove(frame.get_ident());
}
}
_ => (), _ => (),
} }
} }
@ -270,37 +266,33 @@ impl SpircInternal {
} }
fn notify(&mut self, hello: bool, recipient: Option<&str>) { fn notify(&mut self, hello: bool, recipient: Option<&str>) {
let cs = CommandSender::new(self, let mut cs = CommandSender::new(self,
if hello { if hello {
MessageType::kMessageTypeHello MessageType::kMessageTypeHello
} else { } else {
MessageType::kMessageTypeNotify MessageType::kMessageTypeNotify
}); });
if let Some(s) = recipient { if let Some(s) = recipient {
cs.recipient(&s) cs = cs.recipient(&s);
.send()
} else {
cs.send()
} }
cs.send();
} }
fn notify_with_player_state(&mut self, fn notify_with_player_state(&mut self,
hello: bool, hello: bool,
recipient: Option<&str>, recipient: Option<&str>,
player_state: &PlayerState) { player_state: &PlayerState) {
let cs = CommandSender::new(self, let mut cs = CommandSender::new(self,
if hello { if hello {
MessageType::kMessageTypeHello MessageType::kMessageTypeHello
} else { } else {
MessageType::kMessageTypeNotify MessageType::kMessageTypeNotify
}) })
.player_state(player_state); .player_state(player_state);
if let Some(s) = recipient { if let Some(s) = recipient {
cs.recipient(&s) cs = cs.recipient(&s);
.send()
} else {
cs.send()
} }
cs.send();
} }
fn spirc_state(&self, player_state: &PlayerState) -> protocol::spirc::State { fn spirc_state(&self, player_state: &PlayerState) -> protocol::spirc::State {
@ -395,16 +387,14 @@ impl SpircInternal {
struct CommandSender<'a> { struct CommandSender<'a> {
spirc_internal: &'a mut SpircInternal, spirc_internal: &'a mut SpircInternal,
cmd: protocol::spirc::MessageType, cmd: MessageType,
recipient: Option<&'a str>, recipient: Option<&'a str>,
player_state: Option<&'a PlayerState>, player_state: Option<&'a PlayerState>,
state: Option<protocol::spirc::State>, state: Option<protocol::spirc::State>,
} }
impl<'a> CommandSender<'a> { impl<'a> CommandSender<'a> {
fn new(spirc_internal: &'a mut SpircInternal, fn new(spirc_internal: &'a mut SpircInternal, cmd: MessageType) -> CommandSender {
cmd: protocol::spirc::MessageType)
-> CommandSender {
CommandSender { CommandSender {
spirc_internal: spirc_internal, spirc_internal: spirc_internal,
cmd: cmd, cmd: cmd,
@ -430,34 +420,24 @@ impl<'a> CommandSender<'a> {
} }
fn send(self) { fn send(self) {
let internal_player_state = self.spirc_internal.player.state();
let s = self.player_state.unwrap_or(&*internal_player_state);
let mut pkt = protobuf_init!(protocol::spirc::Frame::new(), { let mut pkt = protobuf_init!(protocol::spirc::Frame::new(), {
version: 1, version: 1,
ident: self.spirc_internal.ident.clone(), ident: self.spirc_internal.ident.clone(),
protocol_version: "2.0.0".to_owned(), protocol_version: "2.0.0".to_owned(),
seq_nr: { self.spirc_internal.seq_nr += 1; self.spirc_internal.seq_nr }, seq_nr: { self.spirc_internal.seq_nr += 1; self.spirc_internal.seq_nr },
typ: self.cmd, typ: self.cmd,
recipient: protobuf::RepeatedField::from_vec( recipient: RepeatedField::from_vec(
self.recipient.map(|r| vec![r.to_owned()] ).unwrap_or(vec![]) self.recipient.map(|r| vec![r.to_owned()] ).unwrap_or(vec![])
), ),
device_state: self.spirc_internal.device_state(s),
state_update_id: s.update_time() as i64,
}); });
if let Some(s) = self.player_state { if self.spirc_internal.is_active {
pkt.set_device_state(self.spirc_internal.device_state(s)); pkt.set_state(self.spirc_internal.spirc_state(s));
pkt.set_state_update_id(s.update_time() as i64);
if self.spirc_internal.is_active {
pkt.set_state(self.spirc_internal.spirc_state(s));
}
} else {
let s = &*self.spirc_internal.player.state();
pkt.set_device_state(self.spirc_internal.device_state(s));
pkt.set_state_update_id(s.update_time() as i64);
if self.spirc_internal.is_active {
pkt.set_state(self.spirc_internal.spirc_state(s));
}
}
if let Some(s) = self.state {
pkt.set_state(s);
} }
self.spirc_internal self.spirc_internal