mirror of
https://git.alemi.dev/http-debugger.git
synced 2024-11-21 15:04:49 +01:00
feat: sorted keys, parse into types, add time
This commit is contained in:
parent
1ba596291e
commit
b2fbac1193
2 changed files with 66 additions and 22 deletions
|
|
@ -1,6 +1,6 @@
|
|||
[package]
|
||||
name = "http-debugger"
|
||||
version = "0.1.0"
|
||||
version = "0.2.0"
|
||||
edition = "2021"
|
||||
|
||||
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
|
||||
|
|
|
|||
86
src/main.rs
86
src/main.rs
|
|
@ -1,53 +1,97 @@
|
|||
use std::collections::HashMap;
|
||||
use std::collections::{HashMap, BTreeMap};
|
||||
use std::net::SocketAddr;
|
||||
use std::time::{SystemTime, UNIX_EPOCH, Duration};
|
||||
|
||||
use hyper::http::HeaderValue;
|
||||
use hyper::header::HeaderName;
|
||||
use hyper::http::{HeaderValue};
|
||||
use hyper::server::conn::Http;
|
||||
use hyper::service::service_fn;
|
||||
use hyper::{Request, Response, Body};
|
||||
use hyper::{Request, Response, Body, StatusCode, body};
|
||||
use serde_json::Value;
|
||||
use tokio::net::TcpListener;
|
||||
|
||||
use serde::Serialize;
|
||||
use serde::{Serialize, Serializer};
|
||||
|
||||
fn str_or_repr(v: &HeaderValue) -> String {
|
||||
match v.to_str() {
|
||||
/// excessive parsing of headers, because I want to use all json types...
|
||||
fn parse_header(k: &HeaderName, v: &HeaderValue) -> (String, Value) {
|
||||
let key = k.to_string();
|
||||
let value = match v.to_str() {
|
||||
Ok(str) => {
|
||||
str.to_string()
|
||||
match key.as_str() {
|
||||
"accept" | "accept-encoding" |
|
||||
"accept-language" => { // lists
|
||||
Value::from(str.split(",").map(|x| x.trim()).collect::<Vec<&str>>())
|
||||
},
|
||||
"upgrade-insecure-requests" => { // booleans
|
||||
Value::from(
|
||||
str == "1" || str.to_lowercase() == "true" ||
|
||||
str.to_lowercase() == "t"
|
||||
)
|
||||
},
|
||||
"x-real-port" => { // numbers
|
||||
if let Ok(int) = str.parse::<i64>() {
|
||||
Value::from(int)
|
||||
} else if let Ok(float) = str.parse::<f64>() {
|
||||
Value::from(float)
|
||||
} else {
|
||||
Value::from(str.to_string())
|
||||
}
|
||||
},
|
||||
_ => {
|
||||
Value::from(str.to_string())
|
||||
},
|
||||
}
|
||||
},
|
||||
Err(e) => {
|
||||
format!("{:?} ({})", v, e)
|
||||
Value::from(format!("{:?} ({})", v, e))
|
||||
},
|
||||
}
|
||||
};
|
||||
|
||||
(key, value)
|
||||
}
|
||||
|
||||
/// yoinked off stackoverflow #42723065
|
||||
fn ordered_map<S:Serializer>(value: &HashMap<String, Value>, serializer: S) -> Result<S::Ok, S::Error> {
|
||||
value.iter().collect::<BTreeMap<_, _>>().serialize(serializer)
|
||||
}
|
||||
|
||||
#[derive(Serialize)]
|
||||
struct InspectRequest {
|
||||
pub host: String,
|
||||
pub path: String,
|
||||
pub method: String,
|
||||
pub headers: HashMap<String, String>,
|
||||
pub body: Option<String>,
|
||||
pub version: String,
|
||||
#[serde(serialize_with = "ordered_map")]
|
||||
pub headers: HashMap<String, Value>,
|
||||
pub body: Option<String>,
|
||||
pub time: u128,
|
||||
}
|
||||
|
||||
async fn hello(req: Request<Body>) -> Result<Response<Body>, serde_json::error::Error> {
|
||||
let response = InspectRequest {
|
||||
host: req.uri().to_string(),
|
||||
async fn echo_json(mut req: Request<Body>) -> Result<Response<Body>, hyper::http::Error> {
|
||||
let inspect = InspectRequest {
|
||||
path: req.uri().to_string(),
|
||||
method: req.method().to_string(),
|
||||
version: format!("{:?}", req.version()),
|
||||
headers: req.headers().iter().map(|x| (x.0.to_string(), str_or_repr(x.1))).collect(),
|
||||
body: Some(format!("{:?}", req.body())),
|
||||
headers: req.headers().iter().map(|(k, v)| parse_header(k, v)).collect(),
|
||||
body: String::from_utf8(body::to_bytes(req.body_mut()).await.unwrap().to_vec()).ok(),
|
||||
time: SystemTime::now().duration_since(UNIX_EPOCH).unwrap_or(Duration::from_secs(0)).as_millis(),
|
||||
};
|
||||
|
||||
println!(" * {}", serde_json::to_string_pretty(&response)?);
|
||||
let serialized = serde_json::to_string(&inspect).unwrap_or("[]".to_string());
|
||||
|
||||
Ok(Response::new(serde_json::to_string(&response)?.into()))
|
||||
// println!(" * {}", serde_json::to_string_pretty(&response)?);
|
||||
println!(" * {}", serialized);
|
||||
|
||||
Response::builder()
|
||||
.status(StatusCode::OK)
|
||||
.header("Content-Type", "application/json")
|
||||
.body(serialized.into())
|
||||
}
|
||||
|
||||
#[tokio::main]
|
||||
pub async fn main() -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
|
||||
pretty_env_logger::init();
|
||||
|
||||
let addr: SocketAddr = ([127, 0, 0, 1], 3000).into();
|
||||
let addr: SocketAddr = ([127, 0, 0, 1], 10000).into();
|
||||
|
||||
let listener = TcpListener::bind(addr).await?;
|
||||
println!("Listening on http://{}", addr);
|
||||
|
|
@ -56,7 +100,7 @@ pub async fn main() -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
|
|||
|
||||
tokio::task::spawn(async move {
|
||||
if let Err(err) = Http::new()
|
||||
.serve_connection(stream, service_fn(hello))
|
||||
.serve_connection(stream, service_fn(echo_json))
|
||||
.await
|
||||
{
|
||||
println!("Error serving connection: {:?}", err);
|
||||
|
|
|
|||
Loading…
Reference in a new issue