use super::multipart::{Node, Part, FilePart};
use http::header::{HeaderMap, HeaderValue, CONTENT_DISPOSITION, CONTENT_TYPE};
use crate::http::headers::{ContentDisposition};
use mime::Mime;
use super::error::Error;
#[derive(Clone, Debug, PartialEq)]
pub struct FormData {
pub fields: Vec<(String, String)>,
pub files: Vec<(String, FilePart)>,
}
impl FormData {
pub fn new() -> FormData {
FormData { fields: vec![], files: vec![] }
}
pub fn to_multipart(&self) -> Result<Vec<Node>, Error> {
let mut nodes: Vec<Node> = Vec::with_capacity(self.fields.len() + self.files.len());
for &(ref name, ref value) in &self.fields {
let mut h = HeaderMap::new();
h.append(CONTENT_TYPE, HeaderValue::from_static("text/plain"));
h.append(CONTENT_DISPOSITION, HeaderValue::from_str(&format!("form-data; name={}", name)).unwrap());
nodes.push( Node::Part( Part {
headers: h,
body: value.as_bytes().to_owned(),
}));
}
for &(ref name, ref filepart) in &self.files {
let mut filepart = filepart.clone();
filepart.headers.remove(CONTENT_DISPOSITION);
let filename = match filepart.path.file_name() {
Some(fname) => fname.to_string_lossy().into_owned(),
None => return Err(Error::NotAFile),
};
filepart.headers.append(CONTENT_DISPOSITION, HeaderValue::from_str(&format!("form-data; name={}; filename={}", name, filename)).unwrap());
nodes.push( Node::File( filepart ) );
}
Ok(nodes)
}
pub fn get_fields<T: AsRef<str>>(&self, key: T) -> Vec<&String>{
let mut result: Vec<&String> = vec![];
for (ref k, ref v) in &self.fields {
if key.as_ref() == k {
result.push(&v);
}
}
result
}
pub fn get_one_field<T: AsRef<str>>(&self, key: T) -> Option<&String>{
for (ref k, ref v) in &self.fields {
if key.as_ref() == k {
return Some(v);
}
}
None
}
}