1 files changed, 104 insertions, 0 deletions
diff --git a/src/msg_util.rs b/src/msg_util.rs
index d61be23f7..8634e2282 100644
--- a/src/msg_util.rs
+++ b/src/msg_util.rs
@@ -1,7 +1,18 @@
 // Copyright 2018 the Deno authors. All rights reserved. MIT license.
 // Helpers for serialization.
 use flatbuffers;
+use http::header::HeaderName;
+use http::uri::Uri;
+use http::Method;
+use hyper::header::HeaderMap;
+use hyper::header::HeaderValue;
+use hyper::Body;
+use hyper::Request;
+use hyper::Response;
 use msg;
+use std::str::FromStr;
+
+type Headers = HeaderMap<HeaderValue>;
 
 pub fn serialize_key_value<'bldr>(
   builder: &mut flatbuffers::FlatBufferBuilder<'bldr>,
@@ -15,6 +26,99 @@ pub fn serialize_key_value<'bldr>(
     &msg::KeyValueArgs {
       key: Some(key),
       value: Some(value),
+      ..Default::default()
+    },
+  )
+}
+
+pub fn serialize_request_header<'bldr>(
+  builder: &mut flatbuffers::FlatBufferBuilder<'bldr>,
+  r: &Request<Body>,
+) -> flatbuffers::WIPOffset<msg::HttpHeader<'bldr>> {
+  let method = builder.create_string(r.method().as_str());
+  let url = builder.create_string(r.uri().to_string().as_ref());
+
+  let mut fields = Vec::new();
+  for (key, val) in r.headers().iter() {
+    let kv = serialize_key_value(builder, key.as_ref(), val.to_str().unwrap());
+    fields.push(kv);
+  }
+  let fields = builder.create_vector(fields.as_ref());
+
+  msg::HttpHeader::create(
+    builder,
+    &msg::HttpHeaderArgs {
+      is_request: true,
+      method: Some(method),
+      url: Some(url),
+      fields: Some(fields),
+      ..Default::default()
+    },
+  )
+}
+
+pub fn serialize_fields<'bldr>(
+  builder: &mut flatbuffers::FlatBufferBuilder<'bldr>,
+  headers: &Headers,
+) -> flatbuffers::WIPOffset<
+  flatbuffers::Vector<
+    'bldr,
+    flatbuffers::ForwardsUOffset<msg::KeyValue<'bldr>>,
+  >,
+> {
+  let mut fields = Vec::new();
+  for (key, val) in headers.iter() {
+    let kv = serialize_key_value(builder, key.as_ref(), val.to_str().unwrap());
+    fields.push(kv);
+  }
+  return builder.create_vector(fields.as_ref());
+}
+
+// Not to be confused with serialize_response which has nothing to do with HTTP.
+pub fn serialize_http_response<'bldr>(
+  builder: &mut flatbuffers::FlatBufferBuilder<'bldr>,
+  r: &Response<Body>,
+) -> flatbuffers::WIPOffset<msg::HttpHeader<'bldr>> {
+  let status = r.status().as_u16();
+  let fields = serialize_fields(builder, r.headers());
+  msg::HttpHeader::create(
+    builder,
+    &msg::HttpHeaderArgs {
+      is_request: false,
+      status,
+      fields: Some(fields),
+      ..Default::default()
     },
   )
 }
+
+pub fn deserialize_request(
+  header_msg: msg::HttpHeader,
+  body: Body,
+) -> Request<Body> {
+  let mut r = Request::new(body);
+
+  assert!(header_msg.is_request() == true);
+
+  let url = header_msg.url().unwrap();
+  let uri = Uri::from_str(url).unwrap();
+  *r.uri_mut() = uri;
+
+  if let Some(method) = header_msg.method() {
+    let method = Method::from_str(method).unwrap();
+    *r.method_mut() = method;
+  }
+
+  if let Some(fields) = header_msg.fields() {
+    let headers = r.headers_mut();
+    for i in 0..fields.len() {
+      let kv = fields.get(i);
+      let key = kv.key().unwrap();
+      let name = HeaderName::from_bytes(key.as_bytes()).unwrap();
+      let value = kv.value().unwrap();
+      let v = HeaderValue::from_str(value).unwrap();
+      headers.insert(name, v);
+    }
+  }
+  r
+}