use super::Pending;
use crate::{
client::{self, Client},
endpoints,
error::Result,
model::{Bot, SearchResponse},
};
use std::{
collections::HashMap,
fmt::Display,
future::Future,
pin::Pin,
task::{Context, Poll},
};
pub struct SearchBots<'a> {
client: &'a Client,
fut: Option<Pending<'a, Result<SearchResponse<Bot>>>>,
params: HashMap<&'static str, String>,
}
impl<'a> SearchBots<'a> {
pub(crate) fn new(client: &'a Client) -> Self {
Self {
client,
fut: None,
params: HashMap::new(),
}
}
pub fn limit(mut self, mut limit: u16) -> Self {
if limit > 500 {
limit = 500;
}
self.params.insert("limit", limit.to_string());
self
}
pub fn offset(mut self, offset: u64) -> Self {
self.params.insert("offset", offset.to_string());
self
}
pub fn search(self, query: impl Into<String>) -> Self {
self._search(query.into())
}
fn _search(mut self, query: String) -> Self {
self.params.insert("search", query);
self
}
pub fn sort(mut self, field: impl Display, ascending: bool) -> Self {
let prefix = if ascending { "" } else { "-" };
self.params.insert("sort", format!("{}{}", prefix, field));
self
}
fn start(&mut self) -> Result<()> {
let url = client::url(endpoints::bots())?;
self.fut.replace(Box::pin(self.client.get(url)));
Ok(())
}
}
impl Future for SearchBots<'_> {
type Output = Result<SearchResponse<Bot>>;
fn poll(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
loop {
if let Some(fut) = self.fut.as_mut() {
return fut.as_mut().poll(cx);
} else if let Err(why) = self.start() {
return Poll::Ready(Err(why));
}
}
}
}