| 1 |
|
| 2 |
|
| 3 |
|
| 4 |
|
| 5 |
|
| 6 |
|
| 7 |
|
| 8 |
|
| 9 |
|
| 10 |
|
| 11 |
|
| 12 |
|
| 13 |
|
| 14 |
|
| 15 |
|
| 16 |
|
| 17 |
|
| 18 |
|
| 19 |
|
| 20 |
|
| 21 |
|
| 22 |
|
| 23 |
|
| 24 |
|
| 25 |
|
| 26 |
|
| 27 |
|
| 28 |
|
| 29 |
|
| 30 |
|
| 31 |
|
| 32 |
|
| 33 |
|
| 34 |
|
| 35 |
|
| 36 |
|
| 37 |
|
| 38 |
|
| 39 |
|
| 40 |
use quasi_router::{ |
| 41 |
Action, Address, Destination, Method, Node, Outcome, Params, Request, Response, RouteError, |
| 42 |
}; |
| 43 |
|
| 44 |
pub mod htmx; |
| 45 |
pub mod serves; |
| 46 |
|
| 47 |
pub use crate::serves::Serves; |
| 48 |
|
| 49 |
|
| 50 |
|
| 51 |
|
| 52 |
|
| 53 |
|
| 54 |
|
| 55 |
pub const DEFAULT_BODY_LIMIT: usize = 256 * 1024; |
| 56 |
|
| 57 |
|
| 58 |
|
| 59 |
|
| 60 |
|
| 61 |
|
| 62 |
|
| 63 |
pub const ALLOWED_METHODS: &str = "GET, POST, DELETE, PUT"; |
| 64 |
|
| 65 |
|
| 66 |
#[derive(Debug, Clone, PartialEq, Eq)] |
| 67 |
pub struct Incoming { |
| 68 |
|
| 69 |
pub method: Method, |
| 70 |
|
| 71 |
pub path: String, |
| 72 |
|
| 73 |
pub payload: Params, |
| 74 |
|
| 75 |
pub carried: Params, |
| 76 |
} |
| 77 |
|
| 78 |
|
| 79 |
|
| 80 |
|
| 81 |
|
| 82 |
|
| 83 |
|
| 84 |
#[derive(Debug, Clone, PartialEq, Eq)] |
| 85 |
pub struct Asked { |
| 86 |
|
| 87 |
pub method: Method, |
| 88 |
|
| 89 |
pub url: String, |
| 90 |
} |
| 91 |
|
| 92 |
impl Asked { |
| 93 |
|
| 94 |
#[must_use] |
| 95 |
pub fn new(incoming: &Incoming) -> Self { |
| 96 |
Self { |
| 97 |
method: incoming.method, |
| 98 |
url: route_url(&incoming.path, &incoming.carried), |
| 99 |
} |
| 100 |
} |
| 101 |
} |
| 102 |
|
| 103 |
impl From<Incoming> for Request { |
| 104 |
fn from(incoming: Incoming) -> Self { |
| 105 |
Self { |
| 106 |
method: incoming.method, |
| 107 |
path: incoming.path, |
| 108 |
captures: Params::new(), |
| 109 |
payload: incoming.payload, |
| 110 |
carried: incoming.carried, |
| 111 |
} |
| 112 |
} |
| 113 |
} |
| 114 |
|
| 115 |
|
| 116 |
|
| 117 |
|
| 118 |
|
| 119 |
|
| 120 |
#[derive(Debug, Clone, Copy, PartialEq, Eq)] |
| 121 |
pub enum Refusal { |
| 122 |
|
| 123 |
Method, |
| 124 |
|
| 125 |
TooLarge, |
| 126 |
|
| 127 |
Malformed, |
| 128 |
} |
| 129 |
|
| 130 |
impl Refusal { |
| 131 |
|
| 132 |
#[must_use] |
| 133 |
pub const fn status(self) -> u16 { |
| 134 |
match self { |
| 135 |
Self::Method => 405, |
| 136 |
Self::TooLarge => 413, |
| 137 |
Self::Malformed => 400, |
| 138 |
} |
| 139 |
} |
| 140 |
} |
| 141 |
|
| 142 |
|
| 143 |
|
| 144 |
|
| 145 |
|
| 146 |
|
| 147 |
|
| 148 |
|
| 149 |
|
| 150 |
|
| 151 |
|
| 152 |
|
| 153 |
|
| 154 |
|
| 155 |
|
| 156 |
|
| 157 |
|
| 158 |
|
| 159 |
|
| 160 |
|
| 161 |
pub fn decode( |
| 162 |
method: &http::Method, |
| 163 |
uri: &http::Uri, |
| 164 |
headers: &http::HeaderMap, |
| 165 |
body: &[u8], |
| 166 |
body_limit: usize, |
| 167 |
) -> Result<Incoming, Refusal> { |
| 168 |
let method = translate(method).ok_or(Refusal::Method)?; |
| 169 |
|
| 170 |
let mut payload = Params::new(); |
| 171 |
if method.mutates() && is_form(headers) { |
| 172 |
if body.len() > body_limit { |
| 173 |
return Err(Refusal::TooLarge); |
| 174 |
} |
| 175 |
let text = std::str::from_utf8(body).map_err(|_| Refusal::Malformed)?; |
| 176 |
payload.absorb(decode_pairs(text)); |
| 177 |
} |
| 178 |
|
| 179 |
Ok(Incoming { |
| 180 |
method, |
| 181 |
path: uri.path().to_owned(), |
| 182 |
payload, |
| 183 |
carried: decode_pairs(uri.query().unwrap_or_default()), |
| 184 |
}) |
| 185 |
} |
| 186 |
|
| 187 |
|
| 188 |
|
| 189 |
|
| 190 |
|
| 191 |
|
| 192 |
|
| 193 |
|
| 194 |
|
| 195 |
|
| 196 |
|
| 197 |
pub fn respond<R: Serves + ?Sized>( |
| 198 |
render: &R, |
| 199 |
outcome: Result<Response, RouteError>, |
| 200 |
asked: &Asked, |
| 201 |
) -> http::Response<Vec<u8>> { |
| 202 |
match outcome { |
| 203 |
Ok(answer) => { |
| 204 |
|
| 205 |
|
| 206 |
|
| 207 |
|
| 208 |
let address = placement(&answer, asked); |
| 209 |
|
| 210 |
|
| 211 |
let trigger = answer |
| 212 |
.notice |
| 213 |
.as_ref() |
| 214 |
.map(|notice| htmx::notice_trigger(notice.kind, notice.tone, ¬ice.text)); |
| 215 |
let mut response = match answer.outcome { |
| 216 |
|
| 217 |
|
| 218 |
|
| 219 |
|
| 220 |
|
| 221 |
|
| 222 |
Outcome::Screen(screen) => body(render, 200, render.screen(&screen), None), |
| 223 |
Outcome::Fragment { region, node } => { |
| 224 |
|
| 225 |
|
| 226 |
|
| 227 |
|
| 228 |
|
| 229 |
|
| 230 |
|
| 231 |
|
| 232 |
|
| 233 |
|
| 234 |
let mut markup = render.fragment(&node); |
| 235 |
for stale in &answer.invalidates { |
| 236 |
markup.push_str(&render.invalidated(&stale.region, &stale.node)); |
| 237 |
} |
| 238 |
body(render, 200, markup, Some(format!("#{region}"))) |
| 239 |
} |
| 240 |
|
| 241 |
|
| 242 |
Outcome::Goto(action) => redirect(&action), |
| 243 |
|
| 244 |
|
| 245 |
|
| 246 |
|
| 247 |
|
| 248 |
Outcome::Over(screen) => { |
| 249 |
let target = render.overlay_target().map(|id| format!("#{id}")); |
| 250 |
body(render, 200, render.overlay(&screen), target) |
| 251 |
} |
| 252 |
}; |
| 253 |
if let Some(trigger) = trigger |
| 254 |
&& let Ok(value) = http::HeaderValue::from_str(&trigger) |
| 255 |
{ |
| 256 |
response.headers_mut().insert(htmx::TRIGGER, value); |
| 257 |
} |
| 258 |
if let Some((header, url)) = address |
| 259 |
&& let Ok(value) = http::HeaderValue::from_str(&url) |
| 260 |
{ |
| 261 |
response.headers_mut().insert(header, value); |
| 262 |
} |
| 263 |
response |
| 264 |
} |
| 265 |
Err(error) => { |
| 266 |
let node = Node::Notice { |
| 267 |
kind: error.notice, |
| 268 |
tone: error.tone(), |
| 269 |
text: error.message.clone(), |
| 270 |
}; |
| 271 |
body( |
| 272 |
render, |
| 273 |
error.class.http_status(), |
| 274 |
render.fragment(&node), |
| 275 |
None, |
| 276 |
) |
| 277 |
} |
| 278 |
} |
| 279 |
} |
| 280 |
|
| 281 |
|
| 282 |
|
| 283 |
|
| 284 |
|
| 285 |
|
| 286 |
|
| 287 |
|
| 288 |
|
| 289 |
|
| 290 |
|
| 291 |
|
| 292 |
|
| 293 |
|
| 294 |
|
| 295 |
|
| 296 |
|
| 297 |
fn placement(answer: &Response, asked: &Asked) -> Option<(&'static str, String)> { |
| 298 |
match &answer.address { |
| 299 |
Some(Address::Enters(url)) => return Some((htmx::PUSH_URL, url.clone())), |
| 300 |
Some(Address::Replaces(url)) => return Some((htmx::REPLACE_URL, url.clone())), |
| 301 |
Some(Address::Unchanged) => return None, |
| 302 |
None => {} |
| 303 |
} |
| 304 |
|
| 305 |
match answer.outcome { |
| 306 |
Outcome::Screen(_) if asked.method == Method::Get => { |
| 307 |
Some((htmx::PUSH_URL, asked.url.clone())) |
| 308 |
} |
| 309 |
_ => None, |
| 310 |
} |
| 311 |
} |
| 312 |
|
| 313 |
|
| 314 |
|
| 315 |
|
| 316 |
|
| 317 |
|
| 318 |
|
| 319 |
|
| 320 |
|
| 321 |
fn redirect(action: &Action) -> http::Response<Vec<u8>> { |
| 322 |
let (header, address) = match &action.destination { |
| 323 |
|
| 324 |
|
| 325 |
|
| 326 |
Destination::Route(path) => (htmx::LOCATION, route_url(path, &action.carried)), |
| 327 |
Destination::External(address) => (htmx::REDIRECT, address.clone()), |
| 328 |
}; |
| 329 |
let mut builder = http::Response::builder().status(200); |
| 330 |
if let Ok(value) = http::HeaderValue::from_str(&address) { |
| 331 |
builder = builder.header(header, value); |
| 332 |
} |
| 333 |
builder |
| 334 |
.body(Vec::new()) |
| 335 |
.expect("a response with no body and one checked header is always valid") |
| 336 |
} |
| 337 |
|
| 338 |
|
| 339 |
|
| 340 |
|
| 341 |
|
| 342 |
|
| 343 |
|
| 344 |
|
| 345 |
|
| 346 |
|
| 347 |
|
| 348 |
|
| 349 |
#[must_use] |
| 350 |
pub fn route_url(path: &str, params: &Params) -> String { |
| 351 |
if params.is_empty() { |
| 352 |
return path.to_owned(); |
| 353 |
} |
| 354 |
let query = form_urlencoded::Serializer::new(String::new()) |
| 355 |
.extend_pairs(params.iter()) |
| 356 |
.finish(); |
| 357 |
let joiner = if path.contains('?') { '&' } else { '?' }; |
| 358 |
format!("{path}{joiner}{query}") |
| 359 |
} |
| 360 |
|
| 361 |
|
| 362 |
|
| 363 |
|
| 364 |
|
| 365 |
|
| 366 |
#[must_use] |
| 367 |
pub fn refuse(refusal: Refusal) -> http::Response<Vec<u8>> { |
| 368 |
let mut builder = http::Response::builder().status(refusal.status()); |
| 369 |
if refusal == Refusal::Method { |
| 370 |
builder = builder.header(http::header::ALLOW, ALLOWED_METHODS); |
| 371 |
} |
| 372 |
builder |
| 373 |
.body(Vec::new()) |
| 374 |
.expect("a response with no body and a static header is always valid") |
| 375 |
} |
| 376 |
|
| 377 |
|
| 378 |
fn body<R: Serves + ?Sized>( |
| 379 |
render: &R, |
| 380 |
status: u16, |
| 381 |
rendered: String, |
| 382 |
retarget: Option<String>, |
| 383 |
) -> http::Response<Vec<u8>> { |
| 384 |
let mut builder = http::Response::builder() |
| 385 |
.status(status) |
| 386 |
.header(http::header::CONTENT_TYPE, render.content_type()); |
| 387 |
if let Some(target) = retarget { |
| 388 |
builder = builder.header(htmx::RETARGET, target); |
| 389 |
} |
| 390 |
builder.body(rendered.into_bytes()).unwrap_or_else(|_| { |
| 391 |
|
| 392 |
|
| 393 |
http::Response::builder() |
| 394 |
.status(500) |
| 395 |
.body(Vec::new()) |
| 396 |
.expect("a response with no body and no headers is always valid") |
| 397 |
}) |
| 398 |
} |
| 399 |
|
| 400 |
|
| 401 |
fn translate(method: &http::Method) -> Option<Method> { |
| 402 |
match *method { |
| 403 |
http::Method::GET => Some(Method::Get), |
| 404 |
http::Method::POST => Some(Method::Post), |
| 405 |
http::Method::DELETE => Some(Method::Delete), |
| 406 |
http::Method::PUT => Some(Method::Put), |
| 407 |
_ => None, |
| 408 |
} |
| 409 |
} |
| 410 |
|
| 411 |
|
| 412 |
|
| 413 |
|
| 414 |
|
| 415 |
|
| 416 |
|
| 417 |
fn is_form(headers: &http::HeaderMap) -> bool { |
| 418 |
headers |
| 419 |
.get(http::header::CONTENT_TYPE) |
| 420 |
.and_then(|value| value.to_str().ok()) |
| 421 |
.is_some_and(|value| { |
| 422 |
value.split(';').next().is_some_and(|kind| { |
| 423 |
kind.trim() |
| 424 |
.eq_ignore_ascii_case("application/x-www-form-urlencoded") |
| 425 |
}) |
| 426 |
}) |
| 427 |
} |
| 428 |
|
| 429 |
|
| 430 |
|
| 431 |
|
| 432 |
|
| 433 |
|
| 434 |
fn decode_pairs(encoded: &str) -> Params { |
| 435 |
form_urlencoded::parse(encoded.as_bytes()) |
| 436 |
.map(|(name, value)| (name.into_owned(), value.into_owned())) |
| 437 |
.collect() |
| 438 |
} |
| 439 |
|
| 440 |
#[cfg(test)] |
| 441 |
mod tests; |
| 442 |
|