reduce number of functions that return Result

This commit is contained in:
panicbit 2020-11-14 01:55:47 +01:00
parent 1a145c922c
commit af614a06aa
3 changed files with 55 additions and 36 deletions

View File

@ -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()
} }

View File

@ -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 {

View File

@ -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 {