use std::io::Read;
use serde::{de::DeserializeOwned, Deserialize, Serialize};
use crate::{response_error::ResponseError, Error, Id, Version};
pub trait Response: DeserializeOwned + Sized {
fn from_string(response: impl AsRef<[u8]>) -> Result<Self, Error> {
let wrapper: Wrapper<Self> =
serde_json::from_slice(response.as_ref()).map_err(Error::serde)?;
wrapper.into_result()
}
fn from_reader(reader: impl Read) -> Result<Self, Error> {
let wrapper: Wrapper<Self> = serde_json::from_reader(reader).map_err(Error::serde)?;
wrapper.into_result()
}
}
#[derive(Debug, Deserialize, Serialize, Clone)]
pub struct Wrapper<R> {
jsonrpc: Version,
id: Id,
result: Option<R>,
error: Option<ResponseError>,
}
impl<R> Wrapper<R> {
pub fn version(&self) -> &Version {
&self.jsonrpc
}
#[allow(dead_code)]
pub fn id(&self) -> &Id {
&self.id
}
pub fn into_error(self) -> Option<Error> {
self.error.map(Error::response)
}
pub fn into_result(self) -> Result<R, Error> {
self.version().ensure_supported()?;
if let Some(e) = self.error {
Err(Error::response(e))
} else if let Some(result) = self.result {
Ok(result)
} else {
Err(Error::malformed_json())
}
}
pub fn new_with_id(id: Id, result: Option<R>, error: Option<ResponseError>) -> Self {
Self {
jsonrpc: Version::current(),
id,
result,
error,
}
}
}