use self::{channel::GateioChannel, market::GateioMarket, subscription::GateioSubResponse};
use crate::{
ExchangeWsStream,
exchange::{Connector, ExchangeServer, subscription::ExchangeSub},
subscriber::{WebSocketSubscriber, validator::WebSocketSubValidator},
};
use barter_instrument::exchange::ExchangeId;
use barter_integration::protocol::websocket::{WebSocketSerdeParser, WsMessage};
use serde_json::json;
use std::{fmt::Debug, marker::PhantomData};
use url::Url;
pub mod channel;
pub mod spot;
pub mod future;
pub mod perpetual;
pub mod option;
pub mod market;
pub mod message;
pub mod subscription;
pub type GateiotWsStream<Transformer> = ExchangeWsStream<WebSocketSerdeParser, Transformer>;
#[derive(Copy, Clone, Eq, PartialEq, Ord, PartialOrd, Hash, Debug, Default)]
pub struct Gateio<Server> {
server: PhantomData<Server>,
}
impl<Server> Connector for Gateio<Server>
where
Server: ExchangeServer,
{
const ID: ExchangeId = Server::ID;
type Channel = GateioChannel;
type Market = GateioMarket;
type Subscriber = WebSocketSubscriber;
type SubValidator = WebSocketSubValidator;
type SubResponse = GateioSubResponse;
fn url() -> Result<Url, url::ParseError> {
Url::parse(Server::websocket_url())
}
fn requests(exchange_subs: Vec<ExchangeSub<Self::Channel, Self::Market>>) -> Vec<WsMessage> {
exchange_subs
.into_iter()
.map(|ExchangeSub { channel, market }| {
WsMessage::text(
json!({
"time": chrono::Utc::now().timestamp_millis(),
"channel": channel.as_ref(),
"event": "subscribe",
"payload": [market.as_ref()]
})
.to_string(),
)
})
.collect()
}
}
impl<'de, Server> serde::Deserialize<'de> for Gateio<Server>
where
Server: ExchangeServer,
{
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
where
D: serde::de::Deserializer<'de>,
{
let input = <String as serde::Deserialize>::deserialize(deserializer)?;
if input.as_str() == Self::ID.as_str() {
Ok(Self::default())
} else {
Err(serde::de::Error::invalid_value(
serde::de::Unexpected::Str(input.as_str()),
&Self::ID.as_str(),
))
}
}
}
impl<Server> serde::Serialize for Gateio<Server>
where
Server: ExchangeServer,
{
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
where
S: serde::ser::Serializer,
{
serializer.serialize_str(Self::ID.as_str())
}
}