playa_ffmpeg/util/dictionary/
iter.rs

1use std::{
2    ffi::{CStr, CString},
3    marker::PhantomData,
4    ptr,
5    str::from_utf8_unchecked,
6};
7
8use crate::ffi::*;
9
10pub struct Iter<'a> {
11    ptr: *const AVDictionary,
12    cur: *mut AVDictionaryEntry,
13
14    _marker: PhantomData<&'a ()>,
15}
16
17impl<'a> Iter<'a> {
18    pub fn new(dictionary: *const AVDictionary) -> Self {
19        Iter { ptr: dictionary, cur: ptr::null_mut(), _marker: PhantomData }
20    }
21}
22
23impl<'a> Iterator for Iter<'a> {
24    type Item = (&'a str, &'a str);
25
26    fn next(&mut self) -> Option<<Self as Iterator>::Item> {
27        unsafe {
28            let empty = CString::new("").unwrap();
29            let entry = av_dict_get(self.ptr, empty.as_ptr(), self.cur, AV_DICT_IGNORE_SUFFIX);
30
31            if !entry.is_null() {
32                let key = from_utf8_unchecked(CStr::from_ptr((*entry).key).to_bytes());
33                let val = from_utf8_unchecked(CStr::from_ptr((*entry).value).to_bytes());
34
35                self.cur = entry;
36
37                Some((key, val))
38            } else {
39                None
40            }
41        }
42    }
43}