extern crate ssdp;
use self::ssdp::FieldMap;
use self::ssdp::header::{HeaderMut, HeaderRef, Man, MX, ST};
use self::ssdp::message::{Multicast, SearchRequest, SearchResponse};
use std::collections::HashMap;
use device::Speaker;
use error::*;
const SONOS_URN: &str = "schemas-upnp-org:device:ZonePlayer:1";
fn get_header(msg: &SearchResponse, header: &str) -> Result<String> {
let bytes = msg.get_raw(header)
.chain_err(|| "Failed to get header from discovery response")?;
String::from_utf8(bytes.get(0).unwrap().to_vec())
.chain_err(|| "Failed to convert header to UTF-8")
}
pub fn discover() -> Result<Vec<Speaker>> {
let mut request = SearchRequest::new();
request.set(Man);
request.set(MX(2));
request.set(ST::Target(FieldMap::URN(String::from(SONOS_URN))));
let mut speakers = Vec::new();
for (msg, src) in request.multicast().unwrap() {
let usn = get_header(&msg, "USN")?;
if !usn.contains(SONOS_URN) {
error!("Misbehaving client responded to our discovery ({})", usn);
continue;
}
speakers.push(Speaker::from_ip(src.ip()).chain_err(|| "Failed to get device information")?);
}
Ok(speakers)
}