use self::{
book::l1::KrakenOrderBookL1, channel::KrakenChannel, market::KrakenMarket,
message::KrakenMessage, subscription::KrakenSubResponse, trade::KrakenTrades,
};
use crate::{
ExchangeWsStream, NoInitialSnapshots,
exchange::{Connector, ExchangeSub, StreamSelector},
instrument::InstrumentData,
subscriber::{WebSocketSubscriber, validator::WebSocketSubValidator},
subscription::{book::OrderBooksL1, trade::PublicTrades},
transformer::stateless::StatelessTransformer,
};
use barter_instrument::exchange::ExchangeId;
use barter_integration::protocol::websocket::{WebSocketSerdeParser, WsMessage};
use barter_macro::{DeExchange, SerExchange};
use derive_more::Display;
use serde_json::json;
use url::Url;
pub mod book;
pub mod channel;
pub mod market;
pub mod message;
pub mod subscription;
pub mod trade;
pub const BASE_URL_KRAKEN: &str = "wss://ws.kraken.com/";
pub type KrakenWsStream<Transformer> = ExchangeWsStream<WebSocketSerdeParser, Transformer>;
#[derive(
Copy,
Clone,
Eq,
PartialEq,
Ord,
PartialOrd,
Hash,
Debug,
Default,
Display,
DeExchange,
SerExchange,
)]
pub struct Kraken;
impl Connector for Kraken {
const ID: ExchangeId = ExchangeId::Kraken;
type Channel = KrakenChannel;
type Market = KrakenMarket;
type Subscriber = WebSocketSubscriber;
type SubValidator = WebSocketSubValidator;
type SubResponse = KrakenSubResponse;
fn url() -> Result<Url, url::ParseError> {
Url::parse(BASE_URL_KRAKEN)
}
fn requests(exchange_subs: Vec<ExchangeSub<Self::Channel, Self::Market>>) -> Vec<WsMessage> {
exchange_subs
.into_iter()
.map(|ExchangeSub { channel, market }| {
WsMessage::text(
json!({
"event": "subscribe",
"pair": [market.as_ref()],
"subscription": {
"name": channel.as_ref()
}
})
.to_string(),
)
})
.collect()
}
}
impl<Instrument> StreamSelector<Instrument, PublicTrades> for Kraken
where
Instrument: InstrumentData,
{
type SnapFetcher = NoInitialSnapshots;
type Stream =
KrakenWsStream<StatelessTransformer<Self, Instrument::Key, PublicTrades, KrakenTrades>>;
}
impl<Instrument> StreamSelector<Instrument, OrderBooksL1> for Kraken
where
Instrument: InstrumentData,
{
type SnapFetcher = NoInitialSnapshots;
type Stream = KrakenWsStream<
StatelessTransformer<Self, Instrument::Key, OrderBooksL1, KrakenOrderBookL1>,
>;
}