vn-core 0.11.1

VNDB for Rust
Documentation 
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

use super::request_json;
use crate::error::{Error, Result};
use crate::http::{Endpoint, FieldSet, UrlQueryParams};
use crate::make_request;
use crate::model::prelude::*;
use crate::vndb::{Token, Vndb};
use http::Method;
use serde::de::DeserializeOwned;
use std::sync::Weak;
use tokio::sync::Semaphore;
use tokio::time::Duration;

pub struct Get {
  vndb: Weak<Vndb>,
}

impl Get {
  pub fn new(vndb: Weak<Vndb>) -> Self {
    Self { vndb }
  }

  pub async fn auth_info(&self) -> Result<AuthInfo> {
    let vndb = Vndb::upgrade(&self.vndb)?;
    if vndb.token.is_none() {
      return Err(Error::Unauthorized);
    }

    make_request!(vndb, get_json(Endpoint::AuthInfo))
  }

  pub async fn schema(&self) -> Result<Schema> {
    let vndb = Vndb::upgrade(&self.vndb)?;
    make_request!(vndb, get_json(Endpoint::Schema))
  }

  pub async fn stats(&self) -> Result<Stats> {
    let vndb = Vndb::upgrade(&self.vndb)?;
    make_request!(vndb, get_json(Endpoint::Stats))
  }

  /// Search for a user by their id or username.
  pub async fn user<UserQuery, Field>(&self, user: UserQuery, fields: Field) -> Result<Users>
  where
    UserQuery: Into<UserUrlQuery>,
    Field: Into<FieldSet<UserField>>,
  {
    let mut query = user.into().into_query();
    if query.is_empty() {
      return Ok(Users::default());
    }

    let fields = fields.into().into_url_query();
    query.extend(fields);

    let vndb = Vndb::upgrade(&self.vndb)?;
    make_request!(vndb, get_json(Endpoint::User).query(query))
  }
}

impl Clone for Get {
  fn clone(&self) -> Self {
    Self { vndb: Weak::clone(&self.vndb) }
  }
}

#[bon::builder]
async fn get_json<Json>(
  #[builder(start_fn)] endpoint: Endpoint,
  semaphore: Weak<Semaphore>,
  query: Option<UrlQueryParams>,
  token: Option<&Token>,
  delay: Option<Duration>,
  timeout: Option<Duration>,
  user_agent: Option<&str>,
) -> Result<Json>
where
  Json: DeserializeOwned,
{
  request_json::<(), _>(endpoint)
    .method(Method::GET)
    .semaphore(semaphore)
    .maybe_query(query)
    .maybe_token(token)
    .maybe_delay(delay)
    .maybe_timeout(timeout)
    .maybe_user_agent(user_agent)
    .call()
    .await
}