main.rs

mod model;

use crate::model::BinResponse;
use axum::extract::{OriginalUri, Query};
use axum::http::HeaderMap;
use axum::response::{IntoResponse, Response};
use axum::{Json, Router};
use std::collections::HashMap;
use tokio::net::TcpListener;

#[cfg(flavor = "native")]
const FLAVOR: &str = "native";

#[cfg(flavor = "embed")]
const FLAVOR: &str = "embed";

#[cfg(flavor = "runtime")]
const FLAVOR: &str = "runtime";

#[tokio::main(flavor = "current_thread")]
async fn main() {
    let app = Router::new().route("/get", axum::routing::get(get));
    let addr = "0.0.0.0:3000";
    let tcp_listener = TcpListener::bind(addr).await.unwrap();
    println!("listening on {}", addr);
    axum::Server::from_tcp(tcp_listener.into_std().unwrap())
        .unwrap()
        .serve(app.into_make_service())
        .await
        .unwrap();
}

async fn get(
    OriginalUri(uri): OriginalUri,
    Query(params): Query<HashMap<String, String>>,
    header_map: HeaderMap,
) -> Response {
    let headers = header_map
        .iter()
        .map(|(k, v)| (k.to_string(), v.to_str().unwrap().to_string()))
        .collect();
    let bin_response = BinResponse::new(FLAVOR, params, headers, uri.to_string());
    Json(bin_response).into_response()
}