use polyoxide_core::{HttpClient, QueryBuilder, Request};
use crate::{
error::GammaError,
types::{Market, Tag},
};
#[derive(Clone)]
pub struct Markets {
pub(crate) http_client: HttpClient,
}
impl Markets {
pub fn get(&self, id: impl Into<String>) -> GetMarket {
GetMarket {
request: Request::new(
self.http_client.clone(),
format!("/markets/{}", urlencoding::encode(&id.into())),
),
}
}
pub fn get_by_slug(&self, slug: impl Into<String>) -> GetMarket {
GetMarket {
request: Request::new(
self.http_client.clone(),
format!("/markets/slug/{}", urlencoding::encode(&slug.into())),
),
}
}
pub fn list(&self) -> ListMarkets {
ListMarkets {
request: Request::new(self.http_client.clone(), "/markets"),
}
}
pub fn tags(&self, id: impl Into<String>) -> Request<Vec<Tag>, GammaError> {
Request::new(
self.http_client.clone(),
format!("/markets/{}/tags", urlencoding::encode(&id.into())),
)
}
}
pub struct GetMarket {
request: Request<Market, GammaError>,
}
impl GetMarket {
pub fn include_tag(mut self, include: bool) -> Self {
self.request = self.request.query("include_tag", include);
self
}
pub async fn send(self) -> Result<Market, GammaError> {
self.request.send().await
}
}
pub struct ListMarkets {
request: Request<Vec<Market>, GammaError>,
}
impl ListMarkets {
pub fn limit(mut self, limit: u32) -> Self {
self.request = self.request.query("limit", limit);
self
}
pub fn offset(mut self, offset: u32) -> Self {
self.request = self.request.query("offset", offset);
self
}
pub fn order(mut self, order: impl Into<String>) -> Self {
self.request = self.request.query("order", order.into());
self
}
pub fn ascending(mut self, ascending: bool) -> Self {
self.request = self.request.query("ascending", ascending);
self
}
pub fn id(mut self, ids: impl IntoIterator<Item = i64>) -> Self {
self.request = self.request.query_many("id", ids);
self
}
pub fn slug(mut self, slugs: impl IntoIterator<Item = impl ToString>) -> Self {
self.request = self.request.query_many("slug", slugs);
self
}
pub fn clob_token_ids(mut self, token_ids: impl IntoIterator<Item = impl ToString>) -> Self {
self.request = self.request.query_many("clob_token_ids", token_ids);
self
}
pub fn condition_ids(mut self, condition_ids: impl IntoIterator<Item = impl ToString>) -> Self {
self.request = self.request.query_many("condition_ids", condition_ids);
self
}
pub fn market_maker_address(
mut self,
addresses: impl IntoIterator<Item = impl ToString>,
) -> Self {
self.request = self.request.query_many("market_maker_address", addresses);
self
}
pub fn liquidity_num_min(mut self, min: f64) -> Self {
self.request = self.request.query("liquidity_num_min", min);
self
}
pub fn liquidity_num_max(mut self, max: f64) -> Self {
self.request = self.request.query("liquidity_num_max", max);
self
}
pub fn volume_num_min(mut self, min: f64) -> Self {
self.request = self.request.query("volume_num_min", min);
self
}
pub fn volume_num_max(mut self, max: f64) -> Self {
self.request = self.request.query("volume_num_max", max);
self
}
pub fn start_date_min(mut self, date: impl Into<String>) -> Self {
self.request = self.request.query("start_date_min", date.into());
self
}
pub fn start_date_max(mut self, date: impl Into<String>) -> Self {
self.request = self.request.query("start_date_max", date.into());
self
}
pub fn end_date_min(mut self, date: impl Into<String>) -> Self {
self.request = self.request.query("end_date_min", date.into());
self
}
pub fn end_date_max(mut self, date: impl Into<String>) -> Self {
self.request = self.request.query("end_date_max", date.into());
self
}
pub fn tag_id(mut self, tag_id: i64) -> Self {
self.request = self.request.query("tag_id", tag_id);
self
}
pub fn related_tags(mut self, include: bool) -> Self {
self.request = self.request.query("related_tags", include);
self
}
pub fn cyom(mut self, cyom: bool) -> Self {
self.request = self.request.query("cyom", cyom);
self
}
pub fn uma_resolution_status(mut self, status: impl Into<String>) -> Self {
self.request = self.request.query("uma_resolution_status", status.into());
self
}
pub fn game_id(mut self, game_id: impl Into<String>) -> Self {
self.request = self.request.query("game_id", game_id.into());
self
}
pub fn sports_market_types(mut self, types: impl IntoIterator<Item = impl ToString>) -> Self {
self.request = self.request.query_many("sports_market_types", types);
self
}
pub fn rewards_min_size(mut self, min: f64) -> Self {
self.request = self.request.query("rewards_min_size", min);
self
}
pub fn question_ids(mut self, question_ids: impl IntoIterator<Item = impl ToString>) -> Self {
self.request = self.request.query_many("question_ids", question_ids);
self
}
pub fn include_tag(mut self, include: bool) -> Self {
self.request = self.request.query("include_tag", include);
self
}
pub fn closed(mut self, closed: bool) -> Self {
self.request = self.request.query("closed", closed);
self
}
pub fn open(mut self, open: bool) -> Self {
self.request = self.request.query("closed", !open);
self
}
pub fn archived(mut self, archived: bool) -> Self {
self.request = self.request.query("archived", archived);
self
}
pub async fn send(self) -> Result<Vec<Market>, GammaError> {
self.request.send().await
}
}
#[cfg(test)]
mod tests {
use crate::Gamma;
fn gamma() -> Gamma {
Gamma::new().unwrap()
}
#[test]
fn test_list_markets_full_chain() {
let _list = gamma()
.markets()
.list()
.limit(25)
.offset(50)
.order("volume")
.ascending(false)
.id(vec![1i64, 2, 3])
.slug(vec!["slug-a"])
.clob_token_ids(vec!["token-1"])
.condition_ids(vec!["cond-1"])
.market_maker_address(vec!["0xaddr"])
.liquidity_num_min(1000.0)
.liquidity_num_max(50000.0)
.volume_num_min(100.0)
.volume_num_max(10000.0)
.start_date_min("2024-01-01")
.start_date_max("2025-01-01")
.end_date_min("2024-06-01")
.end_date_max("2025-12-31")
.tag_id(42)
.related_tags(true)
.cyom(false)
.uma_resolution_status("resolved")
.game_id("game-1")
.sports_market_types(vec!["moneyline"])
.rewards_min_size(10.0)
.question_ids(vec!["q1"])
.include_tag(true)
.closed(false)
.archived(false);
}
#[test]
fn test_open_and_closed_are_inverse() {
let _open = gamma().markets().list().open(true);
let _closed = gamma().markets().list().closed(false);
}
#[test]
fn test_get_market_accepts_string_and_str() {
let _req1 = gamma().markets().get("12345");
let _req2 = gamma().markets().get(String::from("12345"));
}
#[test]
fn test_get_by_slug_accepts_string_and_str() {
let _req1 = gamma().markets().get_by_slug("my-slug");
let _req2 = gamma().markets().get_by_slug(String::from("my-slug"));
}
#[test]
fn test_get_market_with_include_tag() {
let _req = gamma().markets().get("12345").include_tag(true);
}
#[test]
fn test_market_tags_accepts_str_and_string() {
let _req1 = gamma().markets().tags("12345");
let _req2 = gamma().markets().tags(String::from("12345"));
}
}