mirror of
https://github.com/denoland/deno.git
synced 2024-11-27 16:10:57 -05:00
50 lines
1.4 KiB
Rust
50 lines
1.4 KiB
Rust
// Copyright 2018-2024 the Deno authors. All rights reserved. MIT license.
|
|
|
|
use crate::CancelHandle;
|
|
use crate::CancelableResponseFuture;
|
|
use crate::FetchHandler;
|
|
|
|
use deno_core::futures::FutureExt;
|
|
use deno_core::futures::TryFutureExt;
|
|
use deno_core::futures::TryStreamExt;
|
|
use deno_core::url::Url;
|
|
use deno_core::CancelFuture;
|
|
use deno_core::OpState;
|
|
use http::StatusCode;
|
|
use http_body_util::BodyExt;
|
|
use std::rc::Rc;
|
|
use tokio_util::io::ReaderStream;
|
|
|
|
/// An implementation which tries to read file URLs from the file system via
|
|
/// tokio::fs.
|
|
#[derive(Clone)]
|
|
pub struct FsFetchHandler;
|
|
|
|
impl FetchHandler for FsFetchHandler {
|
|
fn fetch_file(
|
|
&self,
|
|
_state: &mut OpState,
|
|
url: &Url,
|
|
) -> (CancelableResponseFuture, Option<Rc<CancelHandle>>) {
|
|
let cancel_handle = CancelHandle::new_rc();
|
|
let path_result = url.to_file_path();
|
|
let response_fut = async move {
|
|
let path = path_result?;
|
|
let file = tokio::fs::File::open(path).map_err(|_| ()).await?;
|
|
let stream = ReaderStream::new(file)
|
|
.map_ok(hyper::body::Frame::data)
|
|
.map_err(Into::into);
|
|
let body = http_body_util::StreamBody::new(stream).boxed();
|
|
let response = http::Response::builder()
|
|
.status(StatusCode::OK)
|
|
.body(body)
|
|
.map_err(|_| ())?;
|
|
Ok::<_, ()>(response)
|
|
}
|
|
.map_err(move |_| super::FetchError::NetworkError)
|
|
.or_cancel(&cancel_handle)
|
|
.boxed_local();
|
|
|
|
(response_fut, Some(cancel_handle))
|
|
}
|
|
}
|