Files
tycho-protocol-sdk/substreams/crates/tycho-substreams/src/attributes.rs
tvinagre 3f1beeab7c fix: Make get_block_storage_changes public (#214)
* fix: Make get_block_storage_changes public

* Make RpcTraceData caller optional

* feat: Create function to decode list of addresses
2025-06-16 21:40:39 +00:00

68 lines
2.1 KiB
Rust

use serde_json::Value;
use std::fmt::Debug;
use substreams::prelude::BigInt;
/// Encodes a value to bytes using json.
///
/// ## Panics
/// In case the serialisation to json fails.
pub fn json_serialize_value<T: serde::Serialize + Debug>(v: T) -> Vec<u8> {
serde_json::to_value(v)
.unwrap_or_else(|e| panic!("Failed to encode value as json {e}"))
.to_string()
.as_bytes()
.to_vec()
}
/// Encodes a list of addresses (in byte representation) into json.
///
/// Converts each address to a 0x prefixed hex string and then serializes
/// the list of strings as a json.
///
/// ## Panics
/// In case the serialisation to json fails.
pub fn json_serialize_address_list(addresses: &[Vec<u8>]) -> Vec<u8> {
json_serialize_value(
addresses
.iter()
.map(|a| format!("0x{}", hex::encode(a)))
.collect::<Vec<_>>(),
)
}
/// Decodes a JSON-encoded list of 0x-prefixed hex strings into a list of addresses (in byte
/// representation). This function is the inverse of `json_serialize_address_list`.
///
/// ## Panics
/// Panics if the input is not valid JSON, not an array, or contains invalid hex strings.
pub fn json_deserialize_address_list(json_bytes: &[u8]) -> Vec<Vec<u8>> {
let value: Value =
serde_json::from_slice(json_bytes).expect("Failed to parse JSON for address list");
value
.as_array()
.expect("Expected a JSON array")
.iter()
.map(|v| {
let s = v.as_str().expect("Expected a string");
let s = s.strip_prefix("0x").unwrap_or(s);
hex::decode(s).expect("Invalid hex in address list")
})
.collect()
}
/// Encodes a list of BigInt values into json.
///
/// Converts each integer to a 0x prefixed hex string and then serializes
/// the list of strings as a json.
///
/// ## Panics
/// In case the serialisation to json fails.
pub fn json_serialize_bigint_list(values: &[BigInt]) -> Vec<u8> {
json_serialize_value(
values
.iter()
.map(|v| format!("0x{}", hex::encode(v.to_signed_bytes_be())))
.collect::<Vec<_>>(),
)
}