alpaca-mock 0.26.0

Executable mock server for alpaca-rust trading and market-data flows
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

use axum::Json;
use axum::http::{HeaderMap, StatusCode, header::HeaderName};
use axum::response::{IntoResponse, Response};
use serde_json::json;

const APCA_API_KEY_ID: HeaderName = HeaderName::from_static("apca-api-key-id");
const APCA_API_SECRET_KEY: HeaderName = HeaderName::from_static("apca-api-secret-key");

#[derive(Debug, Clone, PartialEq, Eq)]
pub(crate) struct MockAuth {
    pub(crate) api_key: String,
    pub(crate) secret_key: String,
}

#[derive(Debug, Clone, PartialEq, Eq)]
pub(crate) struct AuthenticatedAccount {
    pub(crate) api_key: String,
}

#[derive(Debug, Clone, PartialEq, Eq)]
pub(crate) struct MockHttpError {
    status: StatusCode,
    message: String,
}

impl MockHttpError {
    pub(crate) fn unauthorized(message: impl Into<String>) -> Self {
        Self {
            status: StatusCode::UNAUTHORIZED,
            message: message.into(),
        }
    }

    pub(crate) fn bad_request(message: impl Into<String>) -> Self {
        Self {
            status: StatusCode::BAD_REQUEST,
            message: message.into(),
        }
    }

    pub(crate) fn not_found(message: impl Into<String>) -> Self {
        Self {
            status: StatusCode::NOT_FOUND,
            message: message.into(),
        }
    }

    pub(crate) fn conflict(message: impl Into<String>) -> Self {
        Self {
            status: StatusCode::UNPROCESSABLE_ENTITY,
            message: message.into(),
        }
    }

    pub(crate) fn internal(message: impl Into<String>) -> Self {
        Self {
            status: StatusCode::INTERNAL_SERVER_ERROR,
            message: message.into(),
        }
    }

    pub(crate) fn with_status(status: StatusCode, message: impl Into<String>) -> Self {
        Self {
            status,
            message: message.into(),
        }
    }
}

impl IntoResponse for MockHttpError {
    fn into_response(self) -> Response {
        (
            self.status,
            Json(json!({
                "code": self.status.as_u16(),
                "message": self.message,
            })),
        )
            .into_response()
    }
}

pub(crate) fn extract_auth(headers: &HeaderMap) -> Result<MockAuth, MockHttpError> {
    Ok(MockAuth {
        api_key: required_header(headers, &APCA_API_KEY_ID)?,
        secret_key: required_header(headers, &APCA_API_SECRET_KEY)?,
    })
}

fn required_header(headers: &HeaderMap, name: &HeaderName) -> Result<String, MockHttpError> {
    let value = headers
        .get(name)
        .ok_or_else(|| MockHttpError::unauthorized(format!("missing {} header", name.as_str())))?;
    let value = value
        .to_str()
        .map_err(|_| MockHttpError::unauthorized(format!("invalid {} header", name.as_str())))?;
    let value = value.trim();
    if value.is_empty() {
        return Err(MockHttpError::unauthorized(format!(
            "missing {} header",
            name.as_str()
        )));
    }

    Ok(value.to_owned())
}