anttp 0.26.0

AntTP is an HTTP server for the Autonomi Network
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

use actix_http::header;
use actix_web::{web, HttpRequest, HttpResponse};
use actix_web::http::header::{ContentLength, ContentType};
use actix_web::web::{Data, Payload};
use ant_core::data::Wallet;
use log::debug;
use crate::error::public_data_error::PublicDataError;
use crate::controller::get_store_type;
use crate::error::CreateError;
use crate::service::public_data_service::{PublicData, PublicDataService};
use crate::service::chunk_service::Chunk;

#[utoipa::path(
    post,
    path = "/anttp-0/public_data/{address}",
    responses(
        (status = OK, description = "Public data pushed successfully", body = Chunk)
    ),
    params(
        ("address" = String, Path, description = "Public data address"),
        ("x-store-type", Header, description = "Target store type (memory|disk|network)", example = "network"),
    ),
)]
pub async fn push_public_data(
    path: web::Path<String>,
    public_data_service: Data<PublicDataService>,
    evm_wallet_data: Data<Wallet>,
    request: HttpRequest,
) -> Result<HttpResponse, PublicDataError> {
    let address = path.into_inner();

    debug!("Pushing public data [{}] to target store type [{:?}]", address, get_store_type(&request));
    Ok(HttpResponse::Ok().json(
        public_data_service.push_public_data(address, get_store_type(&request)).await?
    ))
}

#[utoipa::path(
    post,
    path = "/anttp-0/binary/public_data",
    request_body(
        content = PublicData,
        content_type = "application/octet-stream"
    ),
    responses(
        (status = CREATED, description = "Public data uploaded successfully", body = PublicData),
    ),
    params(
        ("x-store-type", Header, description = "Only persist to cache and do not publish (memory|disk|none)",
        example = "memory"),
    ),
)]
pub async fn post_public_data(
    public_data_service: Data<PublicDataService>,
    evm_wallet_data: Data<Wallet>,
    payload: Payload,
    request: HttpRequest
) -> Result<HttpResponse, PublicDataError> {
    debug!("Creating new public data");
    match payload.to_bytes().await {
        Ok(bytes) => {
            Ok(HttpResponse::Created().json(
                public_data_service.create_public_data(bytes, get_store_type(&request)).await?
            ))
        }
        Err(e) => {
            Err(CreateError::InvalidData(e.to_string()).into())
        }
    }
}

#[utoipa::path(
    get,
    path = "/anttp-0/binary/public_data/{address}",
    responses(
        (status = 200, description = "Public data found successfully", content_type = "application/octet-stream"),
        (status = NOT_FOUND, description = "Public data was not found")
    ),
    params(
        ("address" = String, Path, description = "Public data address"),
    )
)]
pub async fn get_public_data(
    path: web::Path<String>,
    public_data_service: Data<PublicDataService>,
) -> Result<HttpResponse, PublicDataError> {
    let address = path.into_inner();

    debug!("Getting public data at [{}]", address);
    let bytes = public_data_service.get_public_data_binary(address).await?;
    // todo: add caching headers (etag, etc)
    Ok(HttpResponse::Ok()
        .insert_header(ContentType::octet_stream())
        .insert_header(ContentLength(bytes.len()))
        .insert_header((header::SERVER, format!("{}/{}", env!("CARGO_PKG_NAME"), env!("CARGO_PKG_VERSION"))))
        .body(bytes))
}