use self::{
channel::CoinbaseChannel, market::CoinbaseMarket, subscription::CoinbaseSubResponse,
trade::CoinbaseTrade,
};
use crate::{
ExchangeWsStream, NoInitialSnapshots,
exchange::{Connector, ExchangeSub, StreamSelector},
instrument::InstrumentData,
subscriber::{WebSocketSubscriber, validator::WebSocketSubValidator},
subscription::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 channel;
pub mod market;
pub mod subscription;
pub mod trade;
pub const BASE_URL_COINBASE: &str = "wss://ws-feed.exchange.coinbase.com";
pub type CoinbaseWsStream<Transformer> = ExchangeWsStream<WebSocketSerdeParser, Transformer>;
#[derive(
Copy,
Clone,
Eq,
PartialEq,
Ord,
PartialOrd,
Hash,
Debug,
Default,
Display,
DeExchange,
SerExchange,
)]
pub struct Coinbase;
impl Connector for Coinbase {
const ID: ExchangeId = ExchangeId::Coinbase;
type Channel = CoinbaseChannel;
type Market = CoinbaseMarket;
type Subscriber = WebSocketSubscriber;
type SubValidator = WebSocketSubValidator;
type SubResponse = CoinbaseSubResponse;
fn url() -> Result<Url, url::ParseError> {
Url::parse(BASE_URL_COINBASE)
}
fn requests(exchange_subs: Vec<ExchangeSub<Self::Channel, Self::Market>>) -> Vec<WsMessage> {
exchange_subs
.into_iter()
.map(|ExchangeSub { channel, market }| {
WsMessage::text(
json!({
"type": "subscribe",
"product_ids": [market.as_ref()],
"channels": [channel.as_ref()],
})
.to_string(),
)
})
.collect()
}
}
impl<Instrument> StreamSelector<Instrument, PublicTrades> for Coinbase
where
Instrument: InstrumentData,
{
type SnapFetcher = NoInitialSnapshots;
type Stream =
CoinbaseWsStream<StatelessTransformer<Self, Instrument::Key, PublicTrades, CoinbaseTrade>>;
}