blah/blahrs
1
0
Fork 0
mirror of https://github.com/Blah-IM/blahrs.git synced 2025-06-30 19:45:34 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat(blahd): set Cache-Control

Browse source
This commit is contained in:
oxalica 2024-09-25 15:14:43 -04:00
parent cd7d0f773f
commit 1a4980ebba
2 changed files with 20 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

27
blahd/src/lib.rs
View file

@ -50,6 +50,12 @@ pub use middleware::ApiError;
pub(crate) const SERVER_AND_VERSION: &str = concat!("blahd/", env!("CARGO_PKG_VERSION"));
const SERVER_SRC_URL: Option<&str> = option_env!("CFG_SRC_URL");
const HEADER_PUBLIC_NO_CACHE: (HeaderName, HeaderValue) = (
header::CACHE_CONTROL,
HeaderValue::from_static("public, no-cache"),
);
const DEFAULT_CACHE_CONTROL: HeaderValue = HeaderValue::from_static("private, no-cache");
#[serde_inline_default]
#[derive(Debug, Clone, Deserialize)]
#[serde(deny_unknown_fields)]
@ -158,6 +164,12 @@ pub fn router(st: Arc<AppState>) -> Router {
.layer(tower_http::limit::RequestBodyLimitLayer::new(
st.config.max_request_len,
))
.layer(
tower_http::set_header::SetResponseHeaderLayer::if_not_present(
header::CACHE_CONTROL,
DEFAULT_CACHE_CONTROL,
),
)
// NB. This comes at last (outmost layer), so inner errors will still be wrapped with
// correct CORS headers. Also `Authorization` must be explicitly included besides `*`.
.layer(
@ -176,15 +188,14 @@ type RE<T> = R<T, ApiError>;
async fn handle_server_metadata(State(st): ArcState) -> Response {
let (json, etag) = st.server_metadata.clone();
(
[(
let headers = [
(
header::CONTENT_TYPE,
const { HeaderValue::from_static("application/json") },
)],
etag,
json,
)
.into_response()
),
HEADER_PUBLIC_NO_CACHE,
];
(headers, etag, json).into_response()
}
async fn handle_ws(State(st): ArcState, ws: WebSocketUpgrade) -> Response {
@ -495,7 +506,7 @@ async fn room_get_feed<FT: feed::FeedType>(
self_url,
next_url,
});
Ok((ETag(Some(ret_etag)), resp).into_response())
Ok(([HEADER_PUBLIC_NO_CACHE], ETag(Some(ret_etag)), resp).into_response())
}
/// Get room messages with pagination parameters,