diff options
Diffstat (limited to 'src/service.rs')
-rw-r--r-- | src/service.rs | 27 |
1 files changed, 24 insertions, 3 deletions
diff --git a/src/service.rs b/src/service.rs index 35471ea..b7b204c 100644 --- a/src/service.rs +++ b/src/service.rs | |||
@@ -16,6 +16,26 @@ fn get_host(req: &Request<Body>) -> Option<HeaderValue> { | |||
16 | return Some(host); | 16 | return Some(host); |
17 | } | 17 | } |
18 | 18 | ||
19 | fn welcome(req: Request<Body>) -> Response<Body> { | ||
20 | let _h = get_host(&req); | ||
21 | let host = _h.as_ref().map(|h| h.as_bytes()).unwrap_or(b""); | ||
22 | let text = format!( | ||
23 | " | ||
24 | This URL shortening services is powered by hedge. | ||
25 | |||
26 | github.com/nerdypepper/hedge | ||
27 | |||
28 | To shorten urls: | ||
29 | curl -F'shorten=https://shorten.some/long/url' {} | ||
30 | ", | ||
31 | String::from_utf8_lossy(host) | ||
32 | ); | ||
33 | return Response::builder() | ||
34 | .header("content-type", "text/plain") | ||
35 | .body(Body::from(text)) | ||
36 | .unwrap(); | ||
37 | } | ||
38 | |||
19 | fn respond_with_shortlink<S: AsRef<[u8]>>(shortlink: S, host: &[u8]) -> Response<Body> { | 39 | fn respond_with_shortlink<S: AsRef<[u8]>>(shortlink: S, host: &[u8]) -> Response<Body> { |
20 | let url = [host, b"/", shortlink.as_ref()].concat(); | 40 | let url = [host, b"/", shortlink.as_ref()].concat(); |
21 | info!("Successfully generated shortlink"); | 41 | info!("Successfully generated shortlink"); |
@@ -102,8 +122,8 @@ async fn process_form(req: Request<Body>, conn: &mut Connection) -> Result<Respo | |||
102 | } | 122 | } |
103 | 123 | ||
104 | pub async fn shortner_service(req: Request<Body>, mut conn: Connection) -> Result<Response<Body>> { | 124 | pub async fn shortner_service(req: Request<Body>, mut conn: Connection) -> Result<Response<Body>> { |
105 | match req.method() { | 125 | match (req.method(), req.uri().path()) { |
106 | &Method::POST => { | 126 | (&Method::POST, "/") => { |
107 | let boundary = req | 127 | let boundary = req |
108 | .headers() | 128 | .headers() |
109 | .get(CONTENT_TYPE) | 129 | .get(CONTENT_TYPE) |
@@ -117,7 +137,8 @@ pub async fn shortner_service(req: Request<Body>, mut conn: Connection) -> Resul | |||
117 | trace!("Attempting to parse multipart request"); | 137 | trace!("Attempting to parse multipart request"); |
118 | return process_multipart(req, boundary.unwrap(), &mut conn).await; | 138 | return process_multipart(req, boundary.unwrap(), &mut conn).await; |
119 | } | 139 | } |
120 | &Method::GET => { | 140 | (&Method::GET, "/") => Ok(welcome(req)), |
141 | (&Method::GET, _) => { | ||
121 | trace!("GET: {}", req.uri()); | 142 | trace!("GET: {}", req.uri()); |
122 | let shortlink = req.uri().path().to_string(); | 143 | let shortlink = req.uri().path().to_string(); |
123 | let link = get_link(&shortlink[1..], &conn); | 144 | let link = get_link(&shortlink[1..], &conn); |