Expose control over log level

dholroyd · May 14, 2022 · 752632d · 752632d
1 parent 5e6adce
commit 752632d
Showing 1 changed file with 26 additions and 0 deletions.
diff --git a/src/lib.rs b/src/lib.rs
@@ -37,6 +37,32 @@ pub fn startup() -> Result<()> {
     }
 }
 
+pub mod log {
+    pub enum Level {
+        Crit,
+        Err,
+        Warning,
+        Notice,
+        Info,
+        Debug,
+    }
+    impl Level {
+        fn as_cint(&self) -> std::os::raw::c_int {
+            match self {
+                Level::Crit => 2,
+                Level::Err => 3,
+                Level::Warning => 4,
+                Level::Notice => 5,
+                Level::Info => 6,
+                Level::Debug => 7,
+            }
+        }
+    }
+    pub fn set_level(level: Level) {
+        unsafe { super::srt::srt_setloglevel(level.as_cint()) };
+    }
+}
+
 pub fn cleanup() -> Result<()> {
     let result = unsafe { srt::srt_cleanup() };
     error::handle_result((), result)