reduce number of functions that return Result
This commit is contained in:
parent
1a145c922c
commit
af614a06aa
|
@ -32,7 +32,7 @@ fn handle_request(users: Arc<RwLock<HashMap<CertBytes, String>>>, request: Reque
|
||||||
if let Some(user) = users_read.get(cert_bytes) {
|
if let Some(user) = users_read.get(cert_bytes) {
|
||||||
// The user has already registered
|
// The user has already registered
|
||||||
Ok(
|
Ok(
|
||||||
Response::success(&GEMINI_MIME)?
|
Response::success(&GEMINI_MIME)
|
||||||
.with_body(format!("Welcome {}!", user))
|
.with_body(format!("Welcome {}!", user))
|
||||||
)
|
)
|
||||||
} else {
|
} else {
|
||||||
|
@ -44,7 +44,7 @@ fn handle_request(users: Arc<RwLock<HashMap<CertBytes, String>>>, request: Reque
|
||||||
let mut users_write = users.write().await;
|
let mut users_write = users.write().await;
|
||||||
users_write.insert(cert_bytes.clone(), username.to_owned());
|
users_write.insert(cert_bytes.clone(), username.to_owned());
|
||||||
Ok(
|
Ok(
|
||||||
Response::success(&GEMINI_MIME)?
|
Response::success(&GEMINI_MIME)
|
||||||
.with_body(format!(
|
.with_body(format!(
|
||||||
"Your account has been created {}! Welcome!",
|
"Your account has been created {}! Welcome!",
|
||||||
username
|
username
|
||||||
|
@ -57,7 +57,7 @@ fn handle_request(users: Arc<RwLock<HashMap<CertBytes, String>>>, request: Reque
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
// The user didn't provide a certificate
|
// The user didn't provide a certificate
|
||||||
Response::client_certificate_required()
|
Ok(Response::client_certificate_required())
|
||||||
}
|
}
|
||||||
}.boxed()
|
}.boxed()
|
||||||
}
|
}
|
||||||
|
|
75
src/types.rs
75
src/types.rs
|
@ -88,11 +88,18 @@ impl ResponseHeader {
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn success(mime: &Mime) -> Result<Self> {
|
pub fn input_lossy(prompt: impl AsRef<str> + Into<String>) -> Self {
|
||||||
Ok(Self {
|
Self {
|
||||||
|
status: Status::INPUT,
|
||||||
|
meta: Meta::new_lossy(prompt),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn success(mime: &Mime) -> Self {
|
||||||
|
Self {
|
||||||
status: Status::SUCCESS,
|
status: Status::SUCCESS,
|
||||||
meta: Meta::new(mime.to_string())?,
|
meta: Meta::new_lossy(mime.to_string()),
|
||||||
})
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn server_error(reason: impl AsRef<str> + Into<String>) -> Result<Self> {
|
pub fn server_error(reason: impl AsRef<str> + Into<String>) -> Result<Self> {
|
||||||
|
@ -102,25 +109,32 @@ impl ResponseHeader {
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn not_found() -> Result<Self> {
|
pub fn server_error_lossy(reason: impl AsRef<str> + Into<String>) -> Self {
|
||||||
Ok(Self {
|
Self {
|
||||||
|
status: Status::PERMANENT_FAILURE,
|
||||||
|
meta: Meta::new_lossy(reason),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn not_found() -> Self {
|
||||||
|
Self {
|
||||||
status: Status::NOT_FOUND,
|
status: Status::NOT_FOUND,
|
||||||
meta: Meta::new("Not found")?,
|
meta: Meta::new_lossy("Not found"),
|
||||||
})
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn client_certificate_required() -> Result<Self> {
|
pub fn client_certificate_required() -> Self {
|
||||||
Ok(Self {
|
Self {
|
||||||
status: Status::CLIENT_CERTIFICATE_REQUIRED,
|
status: Status::CLIENT_CERTIFICATE_REQUIRED,
|
||||||
meta: Meta::new("No certificate provided")?,
|
meta: Meta::new_lossy("No certificate provided"),
|
||||||
})
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn certificate_not_authorized() -> Result<Self> {
|
pub fn certificate_not_authorized() -> Self {
|
||||||
Ok(Self {
|
Self {
|
||||||
status: Status::CERTIFICATE_NOT_AUTHORIZED,
|
status: Status::CERTIFICATE_NOT_AUTHORIZED,
|
||||||
meta: Meta::new("Your certificate is not authorized to view this content")?,
|
meta: Meta::new_lossy("Your certificate is not authorized to view this content"),
|
||||||
})
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn status(&self) -> &Status {
|
pub fn status(&self) -> &Status {
|
||||||
|
@ -272,9 +286,14 @@ impl Response {
|
||||||
Ok(Self::new(header))
|
Ok(Self::new(header))
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn success(mime: &Mime) -> Result<Self> {
|
pub fn input_lossy(prompt: impl AsRef<str> + Into<String>) -> Self {
|
||||||
let header = ResponseHeader::success(&mime)?;
|
let header = ResponseHeader::input_lossy(prompt);
|
||||||
Ok(Self::new(header))
|
Self::new(header)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn success(mime: &Mime) -> Self {
|
||||||
|
let header = ResponseHeader::success(&mime);
|
||||||
|
Self::new(header)
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn server_error(reason: impl AsRef<str> + Into<String>) -> Result<Self> {
|
pub fn server_error(reason: impl AsRef<str> + Into<String>) -> Result<Self> {
|
||||||
|
@ -282,19 +301,19 @@ impl Response {
|
||||||
Ok(Self::new(header))
|
Ok(Self::new(header))
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn not_found() -> Result<Self> {
|
pub fn not_found() -> Self {
|
||||||
let header = ResponseHeader::not_found()?;
|
let header = ResponseHeader::not_found();
|
||||||
Ok(Self::new(header))
|
Self::new(header)
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn client_certificate_required() -> Result<Self> {
|
pub fn client_certificate_required() -> Self {
|
||||||
let header = ResponseHeader::client_certificate_required()?;
|
let header = ResponseHeader::client_certificate_required();
|
||||||
Ok(Self::new(header))
|
Self::new(header)
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn certificate_not_authorized() -> Result<Self> {
|
pub fn certificate_not_authorized() -> Self {
|
||||||
let header = ResponseHeader::certificate_not_authorized()?;
|
let header = ResponseHeader::certificate_not_authorized();
|
||||||
Ok(Self::new(header))
|
Self::new(header)
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn with_body(mut self, body: impl Into<Body>) -> Self {
|
pub fn with_body(mut self, body: impl Into<Body>) -> Self {
|
||||||
|
|
10
src/util.rs
10
src/util.rs
|
@ -15,12 +15,12 @@ pub async fn serve_file<P: AsRef<Path>>(path: P, mime: &Mime) -> Result<Response
|
||||||
let file = match File::open(path).await {
|
let file = match File::open(path).await {
|
||||||
Ok(file) => file,
|
Ok(file) => file,
|
||||||
Err(err) => match err.kind() {
|
Err(err) => match err.kind() {
|
||||||
io::ErrorKind::NotFound => return Ok(Response::not_found()?),
|
io::ErrorKind::NotFound => return Ok(Response::not_found()),
|
||||||
_ => return Err(err.into()),
|
_ => return Err(err.into()),
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
Ok(Response::success(&mime)?.with_body(file))
|
Ok(Response::success(&mime).with_body(file))
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn serve_dir<D: AsRef<Path>, P: AsRef<Path>>(dir: D, virtual_path: &[P]) -> Result<Response> {
|
pub async fn serve_dir<D: AsRef<Path>, P: AsRef<Path>>(dir: D, virtual_path: &[P]) -> Result<Response> {
|
||||||
|
@ -35,7 +35,7 @@ pub async fn serve_dir<D: AsRef<Path>, P: AsRef<Path>>(dir: D, virtual_path: &[P
|
||||||
let path = path.canonicalize()?;
|
let path = path.canonicalize()?;
|
||||||
|
|
||||||
if !path.starts_with(&dir) {
|
if !path.starts_with(&dir) {
|
||||||
return Ok(Response::not_found()?);
|
return Ok(Response::not_found());
|
||||||
}
|
}
|
||||||
|
|
||||||
if !path.is_dir() {
|
if !path.is_dir() {
|
||||||
|
@ -52,7 +52,7 @@ async fn serve_dir_listing<P: AsRef<Path>, B: AsRef<Path>>(path: P, virtual_path
|
||||||
let mut dir = match fs::read_dir(path).await {
|
let mut dir = match fs::read_dir(path).await {
|
||||||
Ok(dir) => dir,
|
Ok(dir) => dir,
|
||||||
Err(err) => match err.kind() {
|
Err(err) => match err.kind() {
|
||||||
io::ErrorKind::NotFound => return Ok(Response::not_found()?),
|
io::ErrorKind::NotFound => return Ok(Response::not_found()),
|
||||||
_ => return Err(err.into()),
|
_ => return Err(err.into()),
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
@ -82,7 +82,7 @@ async fn serve_dir_listing<P: AsRef<Path>, B: AsRef<Path>>(path: P, virtual_path
|
||||||
)?;
|
)?;
|
||||||
}
|
}
|
||||||
|
|
||||||
Ok(Response::success(&GEMINI_MIME)?.with_body(listing))
|
Ok(Response::success(&GEMINI_MIME).with_body(listing))
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn guess_mime_from_path<P: AsRef<Path>>(path: P) -> Mime {
|
pub fn guess_mime_from_path<P: AsRef<Path>>(path: P) -> Mime {
|
||||||
|
|
Loading…
Reference in a new issue