mirror of
https://github.com/denoland/deno.git
synced 2024-11-29 16:30:56 -05:00
b2bacd969c
Originally landed in
f6fd6619e7.
Reverted in https://github.com/denoland/deno/pull/24574.
This reland contains a fix that sends "Accept: */*" header for calls made
from "FileFetcher". Absence of this header made downloading source code
from JSR broken. This is tested by ensuring this header is present in the
test server that servers JSR packages.
---------
Co-authored-by: Sean McArthur <sean@seanmonstar.com>
53 lines
1.5 KiB
Rust
53 lines
1.5 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::error::type_error;
|
|
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 |_| {
|
|
type_error("NetworkError when attempting to fetch resource.")
|
|
})
|
|
.or_cancel(&cancel_handle)
|
|
.boxed_local();
|
|
|
|
(response_fut, Some(cancel_handle))
|
|
}
|
|
}
|