WIP: content type

rnestler · rnestler · commit 31020eb6b791 · 2022-08-21T16:33:57.000+02:00
diff --git a/src/server/handlers.rs b/src/server/handlers.rs
@@ -1,20 +1,21 @@
 //! Handlers for the server.
 
+use hyper::header::{ACCESS_CONTROL_ALLOW_ORIGIN, CONTENT_TYPE};
 use std::convert::Infallible;
 // use iron::modifiers::Header;
 // use iron::prelude::*;
 // use iron::{headers, middleware, status};
-use log::{debug, error, info, warn};
 use hyper::{Body, Request, Response, Server, StatusCode};
+use log::{debug, error, info, warn};
 use routerify::prelude::*;
 // use router::Router;
 use serde::ser::{Serialize, SerializeMap, Serializer};
 
 use crate::api;
 //use crate::modifiers;
 use crate::sensors;
-use crate::types::RedisPool;
 use crate::server::SpaceapiServer;
+use crate::types::RedisPool;
 
 #[derive(Debug)]
 struct ErrorResponse {
@@ -59,7 +60,7 @@ pub async fn json_response_handler(req: Request<Body>) -> Result<Response<Body>,
                 warn!(
                     "Could not retrieve key '{}' from Redis, omiting the sensor",
                     &sensor_spec.data_key
-                    );
+                );
                 match err {
                     sensors::SensorError::Redis(e) => debug!("Error: {:?}", e),
                     sensors::SensorError::R2d2(e) => debug!("Error: {:?}", e),
@@ -74,10 +75,16 @@ pub async fn json_response_handler(req: Request<Body>) -> Result<Response<Body>,
     }
 
     // Serialize to JSON
-    Ok(Response::new(Body::from(serde_json::to_string(&status_copy).expect(
-        "Status object could not be serialized to JSON. \
-             Please open an issue at https://github.com/spaceapi-community/spaceapi-server-rs/issues",
-             ))))
+    let response = Response::builder()
+        .status(StatusCode::OK)
+        .header(CONTENT_TYPE, "application/json; charset=utf-8")
+        .header(ACCESS_CONTROL_ALLOW_ORIGIN, "*")
+        .body(Body::from(serde_json::to_string(&status_copy).expect(
+            "Status object could not be serialized to JSON. \
+         Please open an issue at https://github.com/spaceapi-community/spaceapi-server-rs/issues",
+        )))
+        .unwrap();
+    Ok(response)
 }
 
 /*
