Mercurial > lbo > hg > async-google-apis
changeset 56:7ba99d02bf39
Add multipart module.
author | Lewin Bormann <lbo@spheniscida.de> |
---|---|
date | Mon, 19 Oct 2020 19:27:51 +0200 |
parents | a2558cbd853c |
children | f6150c39a944 |
files | async-google-apis-common/src/multipart.rs |
diffstat | 1 files changed, 34 insertions(+), 0 deletions(-) [+] |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/async-google-apis-common/src/multipart.rs Mon Oct 19 19:27:51 2020 +0200 @@ -0,0 +1,34 @@ +//! Format and stream a multipart/related request, for uploads. + +use hyper::{self, body::Bytes}; +use radix64::STD; +use serde::Serialize; +use std::io::Write; + +pub const MIME_BOUNDARY: &'static str = "PB0BHe6XN3O6Q4bpnWQgS1pKfMfglTZdifFvh8YIc2APj4Cz3C"; + +pub fn format_multipart<Req: Serialize>(req: &Req, data: Bytes) -> anyhow::Result<Bytes> { + let meta = serde_json::to_string(req)?; + let mut buf = Vec::with_capacity(meta.len() + (1.5 * (data.len() as f64)) as usize); + + // Write metadata. + buf.write(format!("--{}\n", MIME_BOUNDARY).as_bytes()) + .unwrap(); + buf.write("Content-Type: application/json; charset=UTF-8\n\n".as_bytes()) + .unwrap(); + buf.write(meta.as_bytes())?; + + buf.write(format!("\n\n--{}\n", MIME_BOUNDARY).as_bytes()) + .unwrap(); + buf.write("Content-Transfer-Encoding: base64\n\n".as_bytes()) + .unwrap(); + + // Write data contents. + let mut ew = radix64::io::EncodeWriter::new(STD, buf); + ew.write(data.as_ref())?; + + let mut buf = ew.finish()?; + buf.write(format!("\n\n--{}--\n", MIME_BOUNDARY).as_bytes()) + .unwrap(); + Ok(Bytes::from(buf)) +}