Merge pull request #5 from tsirysndr/feat/GetTransportInfo

feat: add support for `GetTransportInfo` action

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "upnp-client"
-version = "0.1.5"
+version = "0.1.6"
 edition = "2021"
 repository = "https://github.com/tsirysndr/upnp-client-rs"
 license = "MIT"

src/media_renderer.rs

@@ -7,8 +7,11 @@ use xml_builder::{XMLBuilder, XMLElement};
 
 use crate::{
     device_client::DeviceClient,
-    parser::{parse_duration, parse_position, parse_supported_protocols, parse_volume},
-    types::{Event, LoadOptions, Metadata, ObjectClass},
+    parser::{
+        parse_duration, parse_position, parse_supported_protocols, parse_transport_info,
+        parse_volume,
+    },
+    types::{Event, LoadOptions, Metadata, ObjectClass, TransportInfo},
     BROADCAST_EVENT,
 };
 
@@ -262,6 +265,16 @@ impl MediaRendererClient {
             }
         }
     }
+
+    pub async fn get_transport_info(&self) -> Result<TransportInfo, Error> {
+        let mut params = HashMap::new();
+        params.insert("InstanceID".to_string(), "0".to_string());
+        let response = self
+            .device_client
+            .call_action("AVTransport", "GetTransportInfo", params)
+            .await?;
+        Ok(parse_transport_info(response.as_str())?)
+    }
 }
 
 fn build_metadata(m: Metadata, media_type: ObjectClass) -> String {

src/parser.rs

@@ -1,6 +1,6 @@
 use std::time::Duration;
 
-use crate::types::{Action, Argument, Container, Device, Item, Metadata, Service};
+use crate::types::{Action, Argument, Container, Device, Item, Metadata, Service, TransportInfo};
 use anyhow::Error;
 use elementtree::Element;
 use surf::{http::Method, Client, Config, Url};
@@ -707,3 +707,51 @@ pub fn deserialize_content_directory(
     }
     Ok((containers, items))
 }
+
+pub fn parse_transport_info(xml: &str) -> Result<TransportInfo, Error> {
+    let parser = EventReader::from_str(xml);
+    let mut in_transport_state = false;
+    let mut in_transport_status = false;
+    let mut in_transport_play_speed = false;
+    let mut transport_info = TransportInfo::default();
+
+    for e in parser {
+        match e {
+            Ok(XmlEvent::StartElement { name, .. }) => {
+                if name.local_name == "CurrentTransportState" {
+                    in_transport_state = true;
+                }
+                if name.local_name == "CurrentTransportStatus" {
+                    in_transport_status = true;
+                }
+                if name.local_name == "CurrentSpeed" {
+                    in_transport_play_speed = true;
+                }
+            }
+            Ok(XmlEvent::EndElement { name }) => {
+                if name.local_name == "CurrentTransportState" {
+                    in_transport_state = false;
+                }
+                if name.local_name == "CurrentTransportStatus" {
+                    in_transport_status = false;
+                }
+                if name.local_name == "CurrentSpeed" {
+                    in_transport_play_speed = false;
+                }
+            }
+            Ok(XmlEvent::Characters(value)) => {
+                if in_transport_state {
+                    transport_info.current_transport_state = value.clone();
+                }
+                if in_transport_status {
+                    transport_info.current_transport_status = value.clone();
+                }
+                if in_transport_play_speed {
+                    transport_info.current_speed = value.clone();
+                }
+            }
+            _ => {}
+        }
+    }
+    Ok(transport_info)
+}

src/types.rs

@@ -212,3 +212,10 @@ pub struct Item {
     pub duration: Option<String>,
     pub object_class: Option<ObjectClass>,
 }
+
+#[derive(Debug, Clone, Default)]
+pub struct TransportInfo {
+    pub current_transport_state: String,
+    pub current_transport_status: String,
+    pub current_speed: String,
+}