decryption success flag

Odebe · Odebe · commit 5888f44d6a21 · 2023-07-23T20:47:06.000+04:00
diff --git a/Gemfile.lock b/Gemfile.lock
@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    rb_mumble_protocol (0.1.0)
+    rb_mumble_protocol (0.1.1)
 
 GEM
   remote: https://rubygems.org/
diff --git a/ext/rb_mumble_protocol/src/lib.rs b/ext/rb_mumble_protocol/src/lib.rs
@@ -1,6 +1,11 @@
 use std::cell::RefCell;
 
-use magnus::{class, define_module, method, function, prelude::*, Error, RHash};
+use magnus::{
+    class, define_module,
+    method, function, prelude::*,
+    Error, RHash,
+    value::{QTRUE, QFALSE}, Value
+};
 
 use bytes::BytesMut;
 
@@ -66,13 +71,17 @@ impl CryptStateRef {
         buffer.to_vec()
     }
 
-    pub fn decrypt(&self, encrypted: Vec<u8>) -> Vec<u8> {
+    pub fn decrypt(&self, encrypted: Vec<u8>) -> (Value, Vec<u8>) {
         let mut buffer = BytesMut::new();
         buffer.extend_from_slice(&encrypted);
 
-        self.0.try_borrow_mut().unwrap().decrypt(&mut buffer).unwrap();
+        let result = self.0.try_borrow_mut().unwrap().decrypt(&mut buffer);
 
-        buffer.to_vec()
+        // TODO: raise error
+        match result {
+            Ok(_)  => (*QTRUE, buffer.to_vec()),
+            Err(_) => (*QFALSE, vec![]),
+        }
     }
 }
 
diff --git a/lib/rb_mumble_protocol/version.rb b/lib/rb_mumble_protocol/version.rb
@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module RbMumbleProtocol
-  VERSION = "0.1.0"
+  VERSION = "0.1.1"
 end
