/* This Source Code Form is subject to the terms of the Mozilla Public * License, v. 2.0. If a copy of the MPL was not distributed with this * file, You can obtain one at http://mozilla.org/MPL/2.0/. */ use filemanager_thread::FileManager; use hyper::header::{Charset, ContentLength, ContentType, Headers}; use hyper::header::{ContentDisposition, DispositionParam, DispositionType}; use ipc_channel::ipc; use mime::{Attr, Mime}; use net_traits::NetworkError; use net_traits::blob_url_store::parse_blob_url; use net_traits::filemanager_thread::ReadFileProgress; use servo_url::ServoUrl; // TODO: Check on GET // https://w3c.github.io/FileAPI/#requestResponseModel /// https://fetch.spec.whatwg.org/#concept-basic-fetch (partial) // TODO: make async. pub fn load_blob_sync (url: ServoUrl, filemanager: FileManager) -> Result<(Headers, Vec), NetworkError> { let (id, origin) = match parse_blob_url(&url) { Ok((id, origin)) => (id, origin), Err(()) => { let e = format!("Invalid blob URL format {:?}", url); return Err(NetworkError::Internal(e)); } }; let (sender, receiver) = ipc::channel().unwrap(); let check_url_validity = true; filemanager.read_file(sender, id, check_url_validity, origin); let blob_buf = match receiver.recv().unwrap() { Ok(ReadFileProgress::Meta(blob_buf)) => blob_buf, Ok(_) => { return Err(NetworkError::Internal("Invalid filemanager reply".to_string())); } Err(e) => { return Err(NetworkError::Internal(format!("{:?}", e))); } }; let content_type: Mime = blob_buf.type_string.parse().unwrap_or(mime!(Text / Plain)); let charset = content_type.get_param(Attr::Charset); let mut headers = Headers::new(); if let Some(name) = blob_buf.filename { let charset = charset.and_then(|c| c.as_str().parse().ok()); headers.set(ContentDisposition { disposition: DispositionType::Inline, parameters: vec![ DispositionParam::Filename(charset.unwrap_or(Charset::Us_Ascii), None, name.as_bytes().to_vec()) ] }); } // Basic fetch, Step 4. headers.set(ContentLength(blob_buf.size as u64)); // Basic fetch, Step 5. headers.set(ContentType(content_type.clone())); let mut bytes = blob_buf.bytes; loop { match receiver.recv().unwrap() { Ok(ReadFileProgress::Partial(ref mut new_bytes)) => { bytes.append(new_bytes); } Ok(ReadFileProgress::EOF) => { return Ok((headers, bytes)); } Ok(_) => { return Err(NetworkError::Internal("Invalid filemanager reply".to_string())); } Err(e) => { return Err(NetworkError::Internal(format!("{:?}", e))); } } } }