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

// Copyright (c) Aptos
// SPDX-License-Identifier: Apache-2.0

use crate::{Error, LedgerInfo};

use anyhow::Result;
use serde::Serialize;
use warp::http::header::{HeaderValue, CONTENT_TYPE};

pub const X_APTOS_CHAIN_ID: &str = "X-Aptos-Chain-Id";
pub const X_APTOS_EPOCH: &str = "X-Aptos-Epoch";
pub const X_APTOS_LEDGER_VERSION: &str = "X-Aptos-Ledger-Version";
pub const X_APTOS_LEDGER_TIMESTAMP: &str = "X-Aptos-Ledger-TimestampUsec";

pub struct Response {
    pub ledger_info: LedgerInfo,
    pub body: Vec<u8>,
}

impl Response {
    pub fn new<T: Serialize>(ledger_info: LedgerInfo, body: &T) -> Result<Self, Error> {
        Ok(Self {
            ledger_info,
            body: serde_json::to_vec(body)?,
        })
    }
}

impl warp::Reply for Response {
    fn into_response(self) -> warp::reply::Response {
        let mut res = warp::reply::Response::new(self.body.into());
        let headers = res.headers_mut();

        headers.insert(CONTENT_TYPE, HeaderValue::from_static("application/json"));
        headers.insert(X_APTOS_CHAIN_ID, (self.ledger_info.chain_id as u16).into());
        headers.insert(
            X_APTOS_LEDGER_VERSION,
            self.ledger_info.ledger_version.into(),
        );
        headers.insert(
            X_APTOS_LEDGER_TIMESTAMP,
            self.ledger_info.ledger_timestamp.into(),
        );
        headers.insert(X_APTOS_EPOCH, self.ledger_info.epoch.into());

        res
    }
}