turn an AssetMap into JSON

Author: jcnelson

src/vm/contexts.rs

@@ -91,6 +91,59 @@ pub struct AssetMap {
     asset_map: HashMap<PrincipalData, HashMap<AssetIdentifier, Vec<Value>>>,
 }
 
+impl AssetMap {
+    pub fn to_json(&self) -> serde_json::Value {
+        let stx : serde_json::map::Map<_, _> = self.stx_map
+            .iter()
+            .map(|(principal, amount)| (format!("{}", principal), serde_json::value::Value::String(format!("{}", amount))))
+            .collect();
+
+        let burns : serde_json::map::Map<_, _> = self.burn_map
+            .iter()
+            .map(|(principal, amount)| (format!("{}", principal), serde_json::value::Value::String(format!("{}", amount))))
+            .collect();
+
+        let tokens : serde_json::map::Map<_, _> = self.token_map
+            .iter()
+            .map(|(principal, token_map)| {
+                let token_json : serde_json::map::Map<_, _> = token_map
+                    .iter()
+                    .map(|(asset_id, amount)| (format!("{}", asset_id), serde_json::value::Value::String(format!("{}", amount))))
+                    .collect();
+
+                (format!("{}", principal), serde_json::value::Value::Object(token_json))
+            })
+            .collect();
+
+        let assets : serde_json::map::Map<_, _> = self.asset_map
+            .iter()
+            .map(|(principal, nft_map)| {
+                let nft_json : serde_json::map::Map<_, _> = nft_map
+                    .iter()
+                    .map(|(asset_id, nft_values)| {
+                        let nft_array = nft_values
+                            .iter()
+                            .map(|nft_value| serde_json::value::Value::String(format!("{}", nft_value)))
+                            .collect();
+
+                        (format!("{}", asset_id), serde_json::value::Value::Array(nft_array))
+                    })
+                    .collect();
+
+                (format!("{}", principal), serde_json::value::Value::Object(nft_json))
+            })
+            .collect();
+
+        json!({
+            "stx": stx,
+            "burns": burns,
+            "tokens": tokens,
+            "assets": assets
+        })
+    }
+}
+
+
 #[derive(Debug, Clone)]
 pub struct EventBatch {
     pub events: Vec<StacksTransactionEvent>,
@@ -640,6 +693,10 @@ impl<'a> OwnedEnvironment<'a> {
         Ok((asset_map, event_batch))
     }
 
+    pub fn get_cost_total(&self) -> ExecutionCost {
+        self.context.cost_track.get_total()
+    }
+
     /// Destroys this environment, returning ownership of its database reference.
     ///  If the context wasn't top-level (i.e., it had uncommitted data), return None,
     ///   because the database is not guaranteed to be in a sane state.