🏡 index : ~doyle/sonos.rs.git

extern crate ssdp;

use self::ssdp::FieldMap;
use self::ssdp::header::{HeaderMut, HeaderRef, Man, MX, ST};
use self::ssdp::message::{Multicast, SearchRequest, SearchResponse};
use device::Device;
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<Device>> {
    let mut request = SearchRequest::new();

    request.set(Man); // required header for discovery
    request.set(MX(2)); // set maximum wait to 2 seconds
    request.set(ST::Target(FieldMap::URN(String::from(SONOS_URN)))); // we're only looking for sonos

    let mut devices: Vec<Device> = 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;
        }

        devices.push(Device::from_ip(src.ip())
            .chain_err(|| "Failed to get device information")?)
    }

    Ok(devices)
}