119 lines
3.0 KiB
Rust
119 lines
3.0 KiB
Rust
use std::convert::TryInto;
|
|
use std::borrow::Borrow;
|
|
|
|
use anyhow::*;
|
|
use uriparse::URIReference;
|
|
use crate::types::{ResponseHeader, Body, Document};
|
|
use crate::util::Cowy;
|
|
|
|
pub struct Response {
|
|
header: ResponseHeader,
|
|
body: Option<Body>,
|
|
}
|
|
|
|
impl Response {
|
|
pub const fn new(header: ResponseHeader) -> Self {
|
|
Self {
|
|
header,
|
|
body: None,
|
|
}
|
|
}
|
|
|
|
#[deprecated(
|
|
since = "0.4.0",
|
|
note = "Deprecated in favor of Response::success_gemini() or Document::into()"
|
|
)]
|
|
pub fn document(document: impl Borrow<Document>) -> Self {
|
|
Self::success_gemini(document)
|
|
}
|
|
|
|
pub fn input(prompt: impl Cowy<str>) -> Result<Self> {
|
|
let header = ResponseHeader::input(prompt)?;
|
|
Ok(Self::new(header))
|
|
}
|
|
|
|
pub fn input_lossy(prompt: impl Cowy<str>) -> Self {
|
|
let header = ResponseHeader::input_lossy(prompt);
|
|
Self::new(header)
|
|
}
|
|
|
|
pub fn redirect_temporary_lossy<'a>(location: impl TryInto<URIReference<'a>>) -> Self {
|
|
let header = ResponseHeader::redirect_temporary_lossy(location);
|
|
Self::new(header)
|
|
}
|
|
|
|
/// Create a successful response with a given body and MIME
|
|
pub fn success(mime: impl ToString, body: impl Into<Body>) -> Self {
|
|
Self {
|
|
header: ResponseHeader::success(mime),
|
|
body: Some(body.into()),
|
|
}
|
|
}
|
|
|
|
/// Create a successful response with a `text/gemini` MIME
|
|
pub fn success_gemini(body: impl Into<Body>) -> Self {
|
|
Self::success("text/gemini", body)
|
|
}
|
|
|
|
/// Create a successful response with a `text/plain` MIME
|
|
pub fn success_plain(body: impl Into<Body>) -> Self {
|
|
Self::success("text/plain", body)
|
|
}
|
|
|
|
pub fn server_error(reason: impl Cowy<str>) -> Result<Self> {
|
|
let header = ResponseHeader::server_error(reason)?;
|
|
Ok(Self::new(header))
|
|
}
|
|
|
|
pub fn not_found() -> Self {
|
|
let header = ResponseHeader::not_found();
|
|
Self::new(header)
|
|
}
|
|
|
|
pub fn bad_request_lossy(reason: impl Cowy<str>) -> Self {
|
|
let header = ResponseHeader::bad_request_lossy(reason);
|
|
Self::new(header)
|
|
}
|
|
|
|
pub fn client_certificate_required() -> Self {
|
|
let header = ResponseHeader::client_certificate_required();
|
|
Self::new(header)
|
|
}
|
|
|
|
pub fn certificate_not_authorized() -> Self {
|
|
let header = ResponseHeader::certificate_not_authorized();
|
|
Self::new(header)
|
|
}
|
|
|
|
pub fn with_body(mut self, body: impl Into<Body>) -> Self {
|
|
self.body = Some(body.into());
|
|
self
|
|
}
|
|
|
|
pub const fn header(&self) -> &ResponseHeader {
|
|
&self.header
|
|
}
|
|
|
|
pub fn take_body(&mut self) -> Option<Body> {
|
|
self.body.take()
|
|
}
|
|
}
|
|
|
|
impl AsRef<Option<Body>> for Response {
|
|
fn as_ref(&self) -> &Option<Body> {
|
|
&self.body
|
|
}
|
|
}
|
|
|
|
impl AsMut<Option<Body>> for Response {
|
|
fn as_mut(&mut self) -> &mut Option<Body> {
|
|
&mut self.body
|
|
}
|
|
}
|
|
|
|
impl<D: Borrow<Document>> From<D> for Response {
|
|
fn from(doc: D) -> Self {
|
|
Self::success_gemini(doc)
|
|
}
|
|
}
|