polymarket-client-sdk 0.4.4

//! Core traits for generic WebSocket infrastructure.

use secrecy::ExposeSecret as _;
use serde::Serialize;
use serde::de::DeserializeOwned;
use serde_json::{Value, json};

use crate::auth::Credentials;

/// Message parser trait for converting raw bytes to messages.
///
/// This abstracts the different parsing strategies:
/// - CLOB/WS: Interest-based filtering, peeking at `event_type` before full deserialization
/// - RTDS: Simple parse, no filtering
///
/// # Example
///
/// ```ignore
/// pub struct SimpleParser;
///
/// impl MessageParser<MyMessage> for SimpleParser {
///     fn parse(&self, bytes: &[u8]) -> crate::Result<Vec<MyMessage>> {
///         let msg: MyMessage = serde_json::from_slice(bytes)?;
///         Ok(vec![msg])
///     }
/// }
/// ```
pub trait MessageParser<M: DeserializeOwned>: Send + Sync + 'static {
    /// Parse incoming bytes into messages.
    ///
    /// May return empty vec if messages are filtered out based on interest or other criteria.
    /// Handles both single objects and arrays of messages.
    fn parse(&self, bytes: &[u8]) -> crate::Result<Vec<M>>;
}

pub trait WithCredentials: Serialize + Sized {
    fn as_authenticated(&self, credentials: &Credentials) -> Result<String, serde_json::Error> {
        let mut payload_json = serde_json::to_value(self)?;
        let auth = json!({
            "apiKey": credentials.key.to_string(),
            "secret": credentials.secret.expose_secret(),
            "passphrase": credentials.passphrase.expose_secret(),
        });

        if let Value::Object(ref mut obj) = payload_json {
            obj.insert("auth".to_owned(), auth);
        }

        serde_json::to_string(&payload_json)
    }
}