1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
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));
}
}
}
}