зеркало из https://github.com/mozilla/gecko-dev.git
85 строки
2.9 KiB
Rust
85 строки
2.9 KiB
Rust
/* 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<u8>), 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)));
|
|
}
|
|
}
|
|
}
|
|
}
|