media_info 0.7.3

A tool to extract media information from a media source.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144

use ffmpeg_next::{self as ffmpeg};
use std::path::Path;

#[derive(Debug)]
pub struct VideoInfo {
    pub creation_date: String,
    pub major_brand: String,
    pub minor_version: String,
    pub compatible_brands: String,
    pub encoder: String,
    pub comment: String,
    pub duration_in_secs: i64,
    pub bit_rate: i64,

    pub location: String,
    pub make: String,
    pub model: String,
    pub software: String,
}

/// Creates a new `VideoInfo` instance by extracting metadata from the given video file path.
///
/// # Arguments
///
/// * `path` - A reference to a `Path` representing the location of the video file.
///
/// # Returns
///
/// * `Ok(VideoInfo)` containing the extracted metadata if successful.
/// * `Err(String)` with an error message if the file does not exist or if metadata extraction fails.
///
/// # Errors
///
/// Returns an error if the file does not exist at the specified path or if FFmpeg fails to initialize.
///
/// # Metadata Extraction
///
/// Attempts to extract the following metadata fields, preferring QuickTime-specific tags when available:
/// - `creation_date`
/// - `location`
/// - `make`
/// - `model`
/// - `software`
///
/// Also extracts:
/// - `major_brand`
/// - `minor_version`
/// - `compatible_brands`
/// - `encoder`
/// - `comment`
/// - `duration_in_secs`
/// - `bit_rate`
///
/// # Panics
///
/// Panics if FFmpeg fails to initialize.
impl VideoInfo {
    pub fn new(path: &Path) -> Result<Self, String> {
        if !path.exists() {
            return Err(format!("File does not exist: {:?}", path));
        }

        ffmpeg::init().expect("could not initialize ffmpeg");

        let input_context = ffmpeg::format::input(path).unwrap();
        let video_metadata = input_context.metadata().to_owned();

        macro_rules! get_video_metadata {
            ($tag:expr) => {
                video_metadata
                    .get($tag)
                    .unwrap_or(&"".to_string())
                    .to_string()
            };
        }

        let creation_date = if !get_video_metadata!("com.apple.quicktime.creationdate").is_empty()
        {
            get_video_metadata!("com.apple.quicktime.creationdate")
        } else {
            get_video_metadata!("creation_time")
        };
        let location = if !get_video_metadata!("com.apple.quicktime.location").is_empty() {
            get_video_metadata!("com.apple.quicktime.location")
        } else {
            get_video_metadata!("location")
        };
        let make = if !get_video_metadata!("com.apple.quicktime.make").is_empty() {
            get_video_metadata!("com.apple.quicktime.make")
        } else {
            get_video_metadata!("make")
        };
        let model = if !get_video_metadata!("com.apple.quicktime.model").is_empty() {
            get_video_metadata!("com.apple.quicktime.model")
        } else {
            get_video_metadata!("model")
        };
        let software = if !get_video_metadata!("com.apple.quicktime.software").is_empty() {
            get_video_metadata!("com.apple.quicktime.software")
        } else {
            get_video_metadata!("software")
        };

        Ok(VideoInfo {
            creation_date,
            major_brand: get_video_metadata!("major_brand"),
            minor_version: get_video_metadata!("minor_version"),
            compatible_brands: get_video_metadata!("compatible_brands"),
            encoder: get_video_metadata!("encoder"),
            comment: get_video_metadata!("comment"),
            duration_in_secs: input_context.duration() / 1_000_000,
            bit_rate: input_context.bit_rate(),
            location,
            make,
            model,
            software,
        })
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn can_read_video_info() {
        let raw_path_str = "../test-media/corgi_race.mp4";
        let path = Path::new(raw_path_str);
        let video_info = VideoInfo::new(path).unwrap();

        println!("{:?}", video_info);

        assert_eq!(video_info.creation_date.contains("2024-10-20"), true);
        assert_eq!(video_info.major_brand.contains("isom"), true);
        assert_eq!(video_info.minor_version.contains("512"), true);
        assert_eq!(
            video_info.compatible_brands.contains("isomiso2avc1mp41"),
            true
        );
        assert_eq!(video_info.encoder.contains(""), true);
        assert_eq!(video_info.comment.contains(""), true);
        assert_eq!(video_info.duration_in_secs > 0, true);
    }
}