rsmycqu 0.4.4

A Rust library for interacting with Chonqing University services, including SSO authentication, campus card management, and more.
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
99
100
101
102
103
104
105
106
107
108

//! 重庆大学单点登录（SSO）相关模块

use reqwest::{Response, StatusCode};
use snafu::ensure;

use crate::{
    errors,
    errors::{
        ApiError,
        sso::{SSOError, SSOResult},
    },
    session::{Client, Session},
    sso::tools::{LoginPageResponse, get_login_request_data, launch_login_data},
    utils::{
        consts::{SSO_LOGIN_URL, SSO_LOGOUT_URL},
        get_response_header,
    },
};

mod encrypt;
mod tools;

#[cfg(test)]
mod tests;

/// 退出账号登陆
pub async fn logout(client: &Client, session: &mut Session) -> SSOResult<()> {
    session
        .execute(client.get(SSO_LOGOUT_URL))
        .await
        .map_err(|_| SSOError::LogoutError)?;
    session.is_login = false;

    Ok(())
}

/// 登陆结果
#[derive(Debug, Eq, PartialEq)]
pub enum LoginResult {
    /// 登陆成功
    Success,

    /// 账号或密码错误
    IncorrectLoginCredentials,
}

/// 登陆账号
pub async fn login(
    client: &Client,
    session: &mut Session,
    auth: impl AsRef<str>,
    password: impl AsRef<str>,
    force_relogin: bool,
) -> SSOResult<LoginResult> {
    let request_data = get_login_request_data(client, session, force_relogin).await?;

    match request_data {
        LoginPageResponse::HasLogin { login_url } => {
            session.execute(client.get(login_url)).await?;
            session.is_login = true;
            Ok(LoginResult::Success)
        }

        LoginPageResponse::NormalLogin { login_page_data } => {
            let login_data = launch_login_data(auth, password, &login_page_data)?;
            let res = session
                .execute(client.post(SSO_LOGIN_URL).form(&login_data))
                .await?;

            match res.status() {
                StatusCode::FOUND => {
                    let url =
                        get_response_header(&res, "Location").ok_or(ApiError::location_error())?;
                    session.execute(client.get(url)).await?;
                    session.is_login = true;
                    Ok(LoginResult::Success)
                }
                StatusCode::UNAUTHORIZED => Ok(LoginResult::IncorrectLoginCredentials),
                other => Err(SSOError::UnknownSSOError {
                    msg: format!("Unexpected status code: {}", other),
                }
                .into()),
            }
        }
    }
}

#[cfg(feature = "mycqu")]
/// 使用登陆了统一身份认证的账号获取指定服务许可
pub(super) async fn access_services(
    client: &Client,
    session: &Session,
    service: impl AsRef<str>,
) -> SSOResult<Response> {
    let res = session
        .execute(
            client
                .get(SSO_LOGIN_URL)
                .query(&[("service", service.as_ref())]),
        )
        .await?;

    ensure!(res.status() == StatusCode::FOUND, errors::NotLoginSnafu {});

    let jump_url = get_response_header(&res, "Location").ok_or(ApiError::location_error())?;

    Ok(session.execute(client.get(jump_url)).await?)
}