jquants-api-client 0.1.0

A Rust client for the J-Quants API, providing seamless access to financial data.
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
90
91
92
93
94
95
96
97
98

//! This module contains the authentication API.

use reqwest::Client;

use crate::{
    api::build_url, IdTokenRequest, IdTokenResponse, JQuantsError, JQuantsErrorResponse,
    RefreshTokenRequest, RefreshTokenResponse,
};

pub mod id_token;
pub mod refresh_token;

/// Get a refresh token from the Refresh Token (/token/auth_user) API.
pub(crate) async fn get_refresh_token_from_api(
    client: &Client,
    mail_address: &str,
    password: &str,
) -> Result<String, JQuantsError> {
    let url = build_url("token/auth_user");
    let request_body = RefreshTokenRequest {
        mail_address: mail_address.to_string(),
        password: password.to_string(),
    };

    let response = client.post(&url).json(&request_body).send().await?;
    let status = response.status();
    let status_code = status.as_u16();
    let text = response.text().await.unwrap_or_default();
    if status == reqwest::StatusCode::OK {
        match serde_json::from_str::<RefreshTokenResponse>(&text) {
            Ok(data) => Ok(data.refresh_token),
            Err(_) => Err(JQuantsError::InvalidResponseFormat {
                status_code,
                body: text,
            }),
        }
    } else {
        match serde_json::from_str::<JQuantsErrorResponse>(&text) {
            Ok(error_response) => match status {
                reqwest::StatusCode::BAD_REQUEST | reqwest::StatusCode::FORBIDDEN => {
                    Err(JQuantsError::InvalidCredentials {
                        body: error_response,
                        status_code,
                    })
                }
                _ => Err(JQuantsError::ApiError {
                    body: error_response,
                    status_code,
                }),
            },
            Err(_) => Err(JQuantsError::InvalidResponseFormat {
                status_code,
                body: text,
            }),
        }
    }
}

/// リフレッシュトークンを使用してAPI経由でIDトークンを取得
pub(crate) async fn get_id_token_from_api(
    client: &Client,
    refresh_token: &str,
) -> Result<String, JQuantsError> {
    let url = build_url("token/auth_refresh");
    let request_body = IdTokenRequest {
        refresh_token: refresh_token.to_string(),
    };
    let response = client.post(&url).query(&request_body).send().await?;
    let status = response.status();
    let status_code = status.as_u16();
    let text = response.text().await.unwrap_or_default();
    if status == reqwest::StatusCode::OK {
        match serde_json::from_str::<IdTokenResponse>(&text) {
            Ok(data) => Ok(data.id_token),
            Err(_) => Err(JQuantsError::InvalidResponseFormat {
                status_code,
                body: text,
            }),
        }
    } else {
        match serde_json::from_str::<JQuantsErrorResponse>(&text) {
            Ok(error_response) => match status {
                reqwest::StatusCode::FORBIDDEN => Err(JQuantsError::IdTokenInvalidOrExpired {
                    body: error_response,
                    status_code,
                }),
                _ => Err(JQuantsError::ApiError {
                    body: error_response,
                    status_code,
                }),
            },
            Err(_) => Err(JQuantsError::InvalidResponseFormat {
                status_code,
                body: text,
            }),
        }
    }
}