more inspection tools

1 files changed, 30 insertions, 1 deletions

diff --git a/src/object.rs b/src/object.rs
index be76c18..061f58e 100644
--- a/src/object.rs
+++ b/src/object.rs
@@ -2,10 +2,11 @@ use crate::helper::{AlignExt, Endianness, ReadExt};
 use crate::serialized_file::TypeTreeNode;
 use anyhow::{Result, bail};
 use log::trace;
+use serde::Serialize;
 use std::io::Seek;
 use std::{collections::BTreeMap, io::Read};
 
-#[derive(Debug)]
+#[derive(Debug, Clone, Serialize)]
 pub enum Value {
     Bool(bool),
     U8(u8),
@@ -121,6 +122,34 @@ pub fn read_value(
 }
 
 impl Value {
+    pub fn as_class(self, name: &str) -> Option<BTreeMap<String, Value>> {
+        if let Value::Object { class, fields } = self {
+            if class == name { Some(fields) } else { None }
+        } else {
+            None
+        }
+    }
+    pub fn as_string(self) -> Option<String> {
+        if let Value::String(s) = self {
+            Some(s)
+        } else {
+            None
+        }
+    }
+    pub fn as_i64(&self) -> Option<i64> {
+        if let Value::I64(s) = self {
+            Some(*s)
+        } else {
+            None
+        }
+    }
+    pub fn as_i32(&self) -> Option<i32> {
+        if let Value::I32(s) = self {
+            Some(*s)
+        } else {
+            None
+        }
+    }
     pub fn to_json(self) -> serde_json::Value {
         match self {
             Value::Bool(x) => serde_json::Value::Bool(x),