Return correct URL from the API, not one pointing to localhost

src/params.rs

@@ -0,0 +1,35 @@
+use rocket::Request;
+use rocket::request::{FromRequest, Outcome};
+
+/// Holds a value that determines whether or not this request wanted a plaintext response.
+///
+/// We assume anything with the text/plain Accept or Content-Type headers want plaintext,
+/// and also anything calling us from the console or that we can't identify.
+pub struct IsPlaintextRequest(pub bool);
+
+impl<'a, 'r> FromRequest<'a, 'r> for IsPlaintextRequest {
+    type Error = ();
+
+    fn from_request(request: &'a Request<'r>) -> Outcome<IsPlaintextRequest, ()> {
+        if let Some(format) = request.format() {
+            if format.is_plain() {
+                return Outcome::Success(IsPlaintextRequest(true));
+            }
+        }
+
+        match request.headers().get_one("User-Agent").and_then(|u| u.splitn(2, '/').next()) {
+            None | Some("Wget") | Some("curl") | Some("HTTPie") => Outcome::Success(IsPlaintextRequest(true)),
+            _ => Outcome::Success(IsPlaintextRequest(false))
+        }
+    }
+}
+
+pub struct HostHeader(pub Option<String>);
+
+impl<'a, 'r> FromRequest<'a, 'r> for HostHeader {
+    type Error = ();
+
+    fn from_request(request: &'a Request<'r>) -> Outcome<HostHeader, ()> {
+        Outcome::Success(HostHeader(request.headers().get_one("Host").map(|h| h.to_string())))
+    }
+}