fix: change data_type to enum for matching

Author: icecreammatt

src/media_info.rs

@@ -49,23 +49,26 @@ pub fn get_media_info(input: &str) -> OutputMetadata {
     };
 
     for track in media.track {
-        // println!("{:#?}", track);
-        if track.data_type == "Video" {
-            metadata.frame_count = track.frame_count.parse().unwrap();
-            let duration = DurationSeconds {
-                duration_seconds: track.duration.parse().unwrap(),
-            };
-            metadata.duration_seconds = duration.duration_seconds;
-            metadata.duration_human = duration.duration_human();
-            metadata.frame_rate = track.frame_rate.parse().unwrap();
-
-            if let Some(width) = track.width {
-                metadata.width = width.parse().unwrap()
-            }
-
-            if let Some(height) = track.height {
-                metadata.height = height.parse().unwrap()
+        match track.data_type {
+            TrackType::Video => {
+                metadata.frame_count = track.frame_count.parse().unwrap();
+                let duration = DurationSeconds {
+                    duration_seconds: track.duration.parse().unwrap(),
+                };
+                metadata.duration_seconds = duration.duration_seconds;
+                metadata.duration_human = duration.duration_human();
+                metadata.frame_rate = track.frame_rate.parse().unwrap();
+
+                if let Some(width) = track.width {
+                    metadata.width = width.parse().unwrap()
+                }
+
+                if let Some(height) = track.height {
+                    metadata.height = height.parse().unwrap()
+                }
             }
+            TrackType::Audio => (),
+            TrackType::General => (),
         }
     }
 

src/metadata.rs

@@ -45,10 +45,17 @@ pub struct CreatingLibrary {
     url: String,
 }
 
+#[derive(Serialize, Deserialize, Debug)]
+pub enum TrackType {
+    Audio,
+    General,
+    Video,
+}
+
 #[derive(Serialize, Deserialize, Debug)]
 pub struct TrackGeneral {
     #[serde(rename = "@type")]
-    pub data_type: String,
+    pub data_type: TrackType,
     #[serde(rename = "FrameCount")]
     pub frame_count: String,
     #[serde(rename = "FrameRate")]