use headers_map

This commit is contained in:
Toshit Chawda 2024-01-12 17:00:14 -08:00
parent 85b57b8019
commit 713987f63d
No known key found for this signature in database
GPG key ID: 91480ED99E2B3D9D
2 changed files with 39 additions and 31 deletions

View file

@ -14,7 +14,7 @@ use bytes::Bytes;
use http::{uri, HeaderName, HeaderValue, Request, Response};
use hyper::{
body::Incoming,
client::conn::{self as hyper_conn, http1::Builder},
client::conn::http1::Builder,
Uri,
};
use js_sys::{Array, Object, Reflect, Uint8Array};
@ -245,8 +245,15 @@ impl WsTcp {
let mut builder = Request::builder().uri(uri.clone()).method(req_method);
let headers_map = builder.headers_mut().replace_err("Failed to get headers")?;
headers_map.insert("Connection", HeaderValue::from_str("close")?);
headers_map.insert("User-Agent", HeaderValue::from_str(&self.useragent)?);
headers_map.insert("Host", HeaderValue::from_str(uri_host)?);
if body_bytes.is_empty() {
headers_map.insert("Content-Length", HeaderValue::from_str("0")?);
}
if let Some(headers) = headers {
let headers_map = builder.headers_mut().replace_err("Failed to get headers")?;
for hdr in headers {
headers_map.insert(
HeaderName::from_bytes(hdr[0].as_bytes())
@ -257,15 +264,6 @@ impl WsTcp {
}
}
builder = builder
// this breaks a shit ton of things
// .header("Host", uri_host)
// .header("User-Agent", self.useragent.clone())
.header("Connection", "close");
if body_bytes.len() == 0 {
builder = builder.header("Content-Length", 0);
}
let request = builder
.body(HttpBody::new(body_bytes))
.replace_err("Failed to make request")?;