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 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151
use { crate::{asset, util}, std::{ collections::HashMap, fmt, fs, io::{self, Read}, path::Path, }, }; pub struct Response { pub code: usize, pub body: String, pub headers: HashMap<String, String>, pub buf: Vec<u8>, pub content_type: String, } impl fmt::Debug for Response { fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { f.debug_struct("Response") .field("code", &self.code) .field("content_type", &self.content_type) .field("body", &self.body) .finish() } } impl Default for Response { fn default() -> Response { Response { code: 200, body: String::new(), buf: Vec::new(), headers: HashMap::new(), content_type: "text/html; charset=utf8".to_string(), } } } impl Response { pub fn new() -> Response { Response::default() } pub fn headers(&self) -> &HashMap<String, String> { &self.headers } pub fn header(&mut self, key: &str, value: &str) { self.headers.insert(key.to_string(), value.to_string()); } pub fn from<T: Into<Response>>(from: T) -> Response { from.into() } pub fn from_file(path: &str) -> Response { Response::default().with_file(path) } pub fn with_code(mut self, code: usize) -> Response { self.code = code; self } pub fn with_body(mut self, body: &str) -> Response { self.body.clear(); self.body.push_str(body); self } pub fn with_file(mut self, path: &str) -> Response { match fs::File::open(asset::normalize_path(path)) { Ok(mut file) => { self.content_type.clear(); self.content_type.push_str(util::content_type(path)); self.header("ETag", &asset::hash(path)); file.read_to_end(&mut self.buf); self } Err(e) => self .with_body(&format!("<h1>500 Internal Error</h1><pre>{:?}", e)) .with_code(500), } } pub fn len(&self) -> usize { if self.buf.is_empty() { self.body.len() } else { self.buf.len() } } } impl fmt::Display for Response { fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { if !self.buf.is_empty() { write!(f, "{}", String::from_utf8_lossy(&self.buf)) } else { write!(f, "{}", self.body) } } } impl From<&str> for Response { fn from(s: &str) -> Response { Response { body: s.to_string(), ..Response::default() } } } impl From<&String> for Response { fn from(s: &String) -> Response { Response { body: s.clone(), ..Response::default() } } } impl From<String> for Response { fn from(body: String) -> Response { Response { body, ..Response::default() } } } impl From<usize> for Response { fn from(i: usize) -> Response { Response { code: i.into(), ..Response::default() } } } impl From<std::borrow::Cow<'_, [u8]>> for Response { fn from(i: std::borrow::Cow<'_, [u8]>) -> Response { Response { body: String::from_utf8_lossy(&i).to_string(), ..Response::default() } } }