pyrite/torn-api.rs
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

changed executor typings

Browse source
This commit is contained in:
TotallyNot 2023-02-27 00:14:37 +01:00
parent 6e76f74fd3
commit 9109720d15
7 changed files with 106 additions and 80 deletions
Download patch file Download diff file Expand all files Collapse all files

18
torn-api/src/lib.rs
View file

@ -96,7 +96,7 @@ impl ApiResponse {
}
}
pub trait ApiSelection {
pub trait ApiSelection: Send + Sync {
fn raw_value(&self) -> &'static str;
fn category() -> &'static str;
@ -137,7 +137,7 @@ where
#[derive(Debug)]
pub struct ApiRequest<A>
where
A: ApiCategoryResponse,
A: ApiSelection,
{
pub selections: Vec<&'static str>,
pub from: Option<DateTime<Utc>>,
@ -148,7 +148,7 @@ where
impl<A> std::default::Default for ApiRequest<A>
where
A: ApiCategoryResponse,
A: ApiSelection,
{
fn default() -> Self {
Self {
@ -163,10 +163,10 @@ where
impl<A> ApiRequest<A>
where
A: ApiCategoryResponse,
A: ApiSelection,
{
pub fn url(&self, key: &str, id: Option<&str>) -> String {
let mut url = format!("https://api.torn.com/{}/", A::Selection::category());
let mut url = format!("https://api.torn.com/{}/", A::category());
if let Some(id) = id {
write!(url, "{}", id).unwrap();
@ -192,7 +192,7 @@ where
pub struct ApiRequestBuilder<A>
where
A: ApiCategoryResponse,
A: ApiSelection,
{
request: ApiRequest<A>,
id: Option<String>,
@ -200,7 +200,7 @@ where
impl<A> Default for ApiRequestBuilder<A>
where
A: ApiCategoryResponse,
A: ApiSelection,
{
fn default() -> Self {
Self {
@ -212,10 +212,10 @@ where
impl<A> ApiRequestBuilder<A>
where
A: ApiCategoryResponse,
A: ApiSelection,
{
#[must_use]
pub fn selections(mut self, selections: &[A::Selection]) -> Self {
pub fn selections(mut self, selections: &[A]) -> Self {
self.request
.selections
.append(&mut selections.iter().map(ApiSelection::raw_value).collect());

67
torn-api/src/local.rs
View file

@ -2,7 +2,9 @@ use std::collections::HashMap;
use async_trait::async_trait;
use crate::{ApiCategoryResponse, ApiClientError, ApiRequest, ApiResponse, DirectExecutor};
use crate::{
ApiCategoryResponse, ApiClientError, ApiRequest, ApiResponse, ApiSelection, DirectExecutor,
};
pub struct ApiProvider<'a, C, E>
where
@ -28,8 +30,8 @@ where
pub async fn user<F>(&self, build: F) -> Result<crate::user::Response, E::Error>
where
F: FnOnce(
crate::ApiRequestBuilder<crate::user::Response>,
) -> crate::ApiRequestBuilder<crate::user::Response>,
crate::ApiRequestBuilder<crate::user::Selection>,
) -> crate::ApiRequestBuilder<crate::user::Selection>,
{
let mut builder = crate::ApiRequestBuilder::default();
builder = build(builder);
@ -37,6 +39,7 @@ where
self.executor
.execute(self.client, builder.request, builder.id)
.await
.map(crate::user::Response::from_response)
}
#[cfg(feature = "user")]
@ -47,8 +50,8 @@ where
) -> HashMap<I, Result<crate::user::Response, E::Error>>
where
F: FnOnce(
crate::ApiRequestBuilder<crate::user::Response>,
) -> crate::ApiRequestBuilder<crate::user::Response>,
crate::ApiRequestBuilder<crate::user::Selection>,
) -> crate::ApiRequestBuilder<crate::user::Selection>,
I: ToString + std::hash::Hash + std::cmp::Eq,
L: IntoIterator<Item = I>,
{
@ -58,14 +61,17 @@ where
self.executor
.execute_many(self.client, builder.request, Vec::from_iter(ids))
.await
.into_iter()
.map(|(k, v)| (k, v.map(crate::user::Response::from_response)))
.collect()
}
#[cfg(feature = "faction")]
pub async fn faction<F>(&self, build: F) -> Result<crate::faction::Response, E::Error>
where
F: FnOnce(
crate::ApiRequestBuilder<crate::faction::Response>,
) -> crate::ApiRequestBuilder<crate::faction::Response>,
crate::ApiRequestBuilder<crate::faction::Selection>,
) -> crate::ApiRequestBuilder<crate::faction::Selection>,
{
let mut builder = crate::ApiRequestBuilder::default();
builder = build(builder);
@ -73,6 +79,7 @@ where
self.executor
.execute(self.client, builder.request, builder.id)
.await
.map(crate::faction::Response::from_response)
}
#[cfg(feature = "faction")]
@ -83,8 +90,8 @@ where
) -> HashMap<I, Result<crate::faction::Response, E::Error>>
where
F: FnOnce(
crate::ApiRequestBuilder<crate::faction::Response>,
) -> crate::ApiRequestBuilder<crate::faction::Response>,
crate::ApiRequestBuilder<crate::faction::Selection>,
) -> crate::ApiRequestBuilder<crate::faction::Selection>,
I: ToString + std::hash::Hash + std::cmp::Eq,
L: IntoIterator<Item = I>,
{
@ -94,14 +101,17 @@ where
self.executor
.execute_many(self.client, builder.request, Vec::from_iter(ids))
.await
.into_iter()
.map(|(k, v)| (k, v.map(crate::faction::Response::from_response)))
.collect()
}
#[cfg(feature = "torn")]
pub async fn torn<F>(&self, build: F) -> Result<crate::torn::Response, E::Error>
where
F: FnOnce(
crate::ApiRequestBuilder<crate::torn::Response>,
) -> crate::ApiRequestBuilder<crate::torn::Response>,
crate::ApiRequestBuilder<crate::torn::Selection>,
) -> crate::ApiRequestBuilder<crate::torn::Selection>,
{
let mut builder = crate::ApiRequestBuilder::default();
builder = build(builder);
@ -109,6 +119,7 @@ where
self.executor
.execute(self.client, builder.request, builder.id)
.await
.map(crate::torn::Response::from_response)
}
#[cfg(feature = "torn")]
@ -119,8 +130,8 @@ where
) -> HashMap<I, Result<crate::torn::Response, E::Error>>
where
F: FnOnce(
crate::ApiRequestBuilder<crate::torn::Response>,
) -> crate::ApiRequestBuilder<crate::torn::Response>,
crate::ApiRequestBuilder<crate::torn::Selection>,
) -> crate::ApiRequestBuilder<crate::torn::Selection>,
I: ToString + std::hash::Hash + std::cmp::Eq,
L: IntoIterator<Item = I>,
{
@ -130,14 +141,17 @@ where
self.executor
.execute_many(self.client, builder.request, Vec::from_iter(ids))
.await
.into_iter()
.map(|(k, v)| (k, v.map(crate::torn::Response::from_response)))
.collect()
}
#[cfg(feature = "key")]
pub async fn key<F>(&self, build: F) -> Result<crate::key::Response, E::Error>
where
F: FnOnce(
crate::ApiRequestBuilder<crate::key::Response>,
) -> crate::ApiRequestBuilder<crate::key::Response>,
crate::ApiRequestBuilder<crate::key::Selection>,
) -> crate::ApiRequestBuilder<crate::key::Selection>,
{
let mut builder = crate::ApiRequestBuilder::default();
builder = build(builder);
@ -145,6 +159,7 @@ where
self.executor
.execute(self.client, builder.request, builder.id)
.await
.map(crate::key::Response::from_response)
}
}
@ -160,18 +175,18 @@ where
client: &C,
request: ApiRequest<A>,
id: Option<String>,
) -> Result<A, Self::Error>
) -> Result<ApiResponse, Self::Error>
where
A: ApiCategoryResponse;
A: ApiSelection;
async fn execute_many<A, I>(
&self,
client: &C,
request: ApiRequest<A>,
ids: Vec<I>,
) -> HashMap<I, Result<A, Self::Error>>
) -> HashMap<I, Result<ApiResponse, Self::Error>>
where
A: ApiCategoryResponse,
A: ApiSelection,
I: ToString + std::hash::Hash + std::cmp::Eq;
}
@ -187,15 +202,15 @@ where
client: &C,
request: ApiRequest<A>,
id: Option<String>,
) -> Result<A, Self::Error>
) -> Result<ApiResponse, Self::Error>
where
A: ApiCategoryResponse,
A: ApiSelection,
{
let url = request.url(&self.key, id.as_deref());
let value = client.request(url).await.map_err(ApiClientError::Client)?;
Ok(A::from_response(ApiResponse::from_value(value)?))
Ok(ApiResponse::from_value(value)?)
}
async fn execute_many<A, I>(
@ -203,9 +218,9 @@ where
client: &C,
request: ApiRequest<A>,
ids: Vec<I>,
) -> HashMap<I, Result<A, Self::Error>>
) -> HashMap<I, Result<ApiResponse, Self::Error>>
where
A: ApiCategoryResponse,
A: ApiSelection,
I: ToString + std::hash::Hash + std::cmp::Eq,
{
let request_ref = &request;
@ -217,9 +232,7 @@ where
(
i,
value
.and_then(|v| ApiResponse::from_value(v).map_err(Into::into))
.map(A::from_response),
value.and_then(|v| ApiResponse::from_value(v).map_err(Into::into)),
)
}))
.await;

67
torn-api/src/send.rs
View file

@ -2,7 +2,9 @@ use std::collections::HashMap;
use async_trait::async_trait;
use crate::{ApiCategoryResponse, ApiClientError, ApiRequest, ApiResponse, DirectExecutor};
use crate::{
ApiCategoryResponse, ApiClientError, ApiRequest, ApiResponse, ApiSelection, DirectExecutor,
};
pub struct ApiProvider<'a, C, E>
where
@ -28,8 +30,8 @@ where
pub async fn user<F>(&self, build: F) -> Result<crate::user::Response, E::Error>
where
F: FnOnce(
crate::ApiRequestBuilder<crate::user::Response>,
) -> crate::ApiRequestBuilder<crate::user::Response>,
crate::ApiRequestBuilder<crate::user::Selection>,
) -> crate::ApiRequestBuilder<crate::user::Selection>,
{
let mut builder = crate::ApiRequestBuilder::default();
builder = build(builder);
@ -37,6 +39,7 @@ where
self.executor
.execute(self.client, builder.request, builder.id)
.await
.map(crate::user::Response::from_response)
}
#[cfg(feature = "user")]
@ -47,8 +50,8 @@ where
) -> HashMap<I, Result<crate::user::Response, E::Error>>
where
F: FnOnce(
crate::ApiRequestBuilder<crate::user::Response>,
) -> crate::ApiRequestBuilder<crate::user::Response>,
crate::ApiRequestBuilder<crate::user::Selection>,
) -> crate::ApiRequestBuilder<crate::user::Selection>,
I: ToString + std::hash::Hash + std::cmp::Eq + Send + Sync,
L: IntoIterator<Item = I>,
{
@ -58,14 +61,17 @@ where
self.executor
.execute_many(self.client, builder.request, Vec::from_iter(ids))
.await
.into_iter()
.map(|(k, v)| (k, v.map(crate::user::Response::from_response)))
.collect()
}
#[cfg(feature = "faction")]
pub async fn faction<F>(&self, build: F) -> Result<crate::faction::Response, E::Error>
where
F: FnOnce(
crate::ApiRequestBuilder<crate::faction::Response>,
) -> crate::ApiRequestBuilder<crate::faction::Response>,
crate::ApiRequestBuilder<crate::faction::Selection>,
) -> crate::ApiRequestBuilder<crate::faction::Selection>,
{
let mut builder = crate::ApiRequestBuilder::default();
builder = build(builder);
@ -73,6 +79,7 @@ where
self.executor
.execute(self.client, builder.request, builder.id)
.await
.map(crate::faction::Response::from_response)
}
#[cfg(feature = "faction")]
@ -83,8 +90,8 @@ where
) -> HashMap<I, Result<crate::faction::Response, E::Error>>
where
F: FnOnce(
crate::ApiRequestBuilder<crate::faction::Response>,
) -> crate::ApiRequestBuilder<crate::faction::Response>,
crate::ApiRequestBuilder<crate::faction::Selection>,
) -> crate::ApiRequestBuilder<crate::faction::Selection>,
I: ToString + std::hash::Hash + std::cmp::Eq + Send + Sync,
L: IntoIterator<Item = I>,
{
@ -94,14 +101,17 @@ where
self.executor
.execute_many(self.client, builder.request, Vec::from_iter(ids))
.await
.into_iter()
.map(|(k, v)| (k, v.map(crate::faction::Response::from_response)))
.collect()
}
#[cfg(feature = "torn")]
pub async fn torn<F>(&self, build: F) -> Result<crate::torn::Response, E::Error>
where
F: FnOnce(
crate::ApiRequestBuilder<crate::torn::Response>,
) -> crate::ApiRequestBuilder<crate::torn::Response>,
crate::ApiRequestBuilder<crate::torn::Selection>,
) -> crate::ApiRequestBuilder<crate::torn::Selection>,
{
let mut builder = crate::ApiRequestBuilder::default();
builder = build(builder);
@ -109,6 +119,7 @@ where
self.executor
.execute(self.client, builder.request, builder.id)
.await
.map(crate::torn::Response::from_response)
}
#[cfg(feature = "torn")]
@ -119,8 +130,8 @@ where
) -> HashMap<I, Result<crate::torn::Response, E::Error>>
where
F: FnOnce(
crate::ApiRequestBuilder<crate::torn::Response>,
) -> crate::ApiRequestBuilder<crate::torn::Response>,
crate::ApiRequestBuilder<crate::torn::Selection>,
) -> crate::ApiRequestBuilder<crate::torn::Selection>,
I: ToString + std::hash::Hash + std::cmp::Eq + Send + Sync,
L: IntoIterator<Item = I>,
{
@ -130,14 +141,17 @@ where
self.executor
.execute_many(self.client, builder.request, Vec::from_iter(ids))
.await
.into_iter()
.map(|(k, v)| (k, v.map(crate::torn::Response::from_response)))
.collect()
}
#[cfg(feature = "key")]
pub async fn key<F>(&self, build: F) -> Result<crate::key::Response, E::Error>
where
F: FnOnce(
crate::ApiRequestBuilder<crate::key::Response>,
) -> crate::ApiRequestBuilder<crate::key::Response>,
crate::ApiRequestBuilder<crate::key::Selection>,
) -> crate::ApiRequestBuilder<crate::key::Selection>,
{
let mut builder = crate::ApiRequestBuilder::default();
builder = build(builder);
@ -145,6 +159,7 @@ where
self.executor
.execute(self.client, builder.request, builder.id)
.await
.map(crate::key::Response::from_response)
}
}
@ -160,18 +175,18 @@ where
client: &C,
request: ApiRequest<A>,
id: Option<String>,
) -> Result<A, Self::Error>
) -> Result<ApiResponse, Self::Error>
where
A: ApiCategoryResponse;
A: ApiSelection;
async fn execute_many<A, I>(
&self,
client: &C,
request: ApiRequest<A>,
ids: Vec<I>,
) -> HashMap<I, Result<A, Self::Error>>
) -> HashMap<I, Result<ApiResponse, Self::Error>>
where
A: ApiCategoryResponse,
A: ApiSelection,
I: ToString + std::hash::Hash + std::cmp::Eq + Send + Sync;
}
@ -187,15 +202,15 @@ where
client: &C,
request: ApiRequest<A>,
id: Option<String>,
) -> Result<A, Self::Error>
) -> Result<ApiResponse, Self::Error>
where
A: ApiCategoryResponse,
A: ApiSelection,
{
let url = request.url(&self.key, id.as_deref());
let value = client.request(url).await.map_err(ApiClientError::Client)?;
Ok(A::from_response(ApiResponse::from_value(value)?))
Ok(ApiResponse::from_value(value)?)
}
async fn execute_many<A, I>(
@ -203,9 +218,9 @@ where
client: &C,
request: ApiRequest<A>,
ids: Vec<I>,
) -> HashMap<I, Result<A, Self::Error>>
) -> HashMap<I, Result<ApiResponse, Self::Error>>
where
A: ApiCategoryResponse,
A: ApiSelection,
I: ToString + std::hash::Hash + std::cmp::Eq + Send + Sync,
{
let request_ref = &request;
@ -217,9 +232,7 @@ where
(
i,
value
.and_then(|v| ApiResponse::from_value(v).map_err(Into::into))
.map(A::from_response),
value.and_then(|v| ApiResponse::from_value(v).map_err(Into::into)),
)
}))
.await;