Add the possibility to get only top-level metrics in the web api

Author: calixteman

src/web/metrics.rs

@@ -10,6 +10,7 @@ pub struct WebMetricsPayload {
     pub id: String,
     pub file_name: String,
     pub code: String,
+    pub unit: bool,
 }
 
 #[derive(Debug, Serialize)]
@@ -21,13 +22,15 @@ pub struct WebMetricsResponse<'a> {
 #[derive(Debug, Deserialize)]
 pub struct WebMetricsInfo {
     pub file_name: String,
+    pub unit: Option<String>,
 }
 
 pub struct WebMetricsCallback {}
 
 pub struct WebMetricsCfg {
     pub id: String,
     pub path: PathBuf,
+    pub unit: bool,
 }
 
 impl Callback for WebMetricsCallback {
@@ -36,6 +39,17 @@ impl Callback for WebMetricsCallback {
 
     fn call<T: TSParserTrait>(cfg: Self::Cfg, parser: &T) -> Self::Res {
         let spaces = metrics(parser, &cfg.path);
+        let spaces = if cfg.unit {
+            if let Some(mut spaces) = spaces {
+                spaces.spaces.clear();
+                Some(spaces)
+            } else {
+                None
+            }
+        } else {
+            spaces
+        };
+
         serde_json::to_value(WebMetricsResponse { id: cfg.id, spaces }).unwrap()
     }
 }

src/web/server.rs

@@ -93,6 +93,7 @@ fn metrics_json(item: web::Json<WebMetricsPayload>, _req: HttpRequest) -> HttpRe
         let cfg = WebMetricsCfg {
             id: payload.id,
             path,
+            unit: payload.unit,
         };
         HttpResponse::Ok().json(action::<WebMetricsCallback>(
             &language,
@@ -116,6 +117,10 @@ fn metrics_plain(code: Bytes, info: Query<WebMetricsInfo>) -> HttpResponse {
         let cfg = WebMetricsCfg {
             id: "".to_string(),
             path,
+            unit: info
+                .unit
+                .as_ref()
+                .map_or(false, |s| s == "1" || s == "true"),
         };
         HttpResponse::Ok().json(action::<WebMetricsCallback>(
             &language,
@@ -519,10 +524,11 @@ mod tests {
         );
         let req = test::TestRequest::post()
             .uri("/metrics")
-            .set_json(&WebCommentPayload {
+            .set_json(&WebMetricsPayload {
                 id: "1234".to_string(),
                 file_name: "test.py".to_string(),
                 code: "def foo():\n    pass\n".to_string(),
+                unit: false,
             })
             .to_request();
 
@@ -569,6 +575,47 @@ mod tests {
         assert_eq!(res, expected);
     }
 
+    #[test]
+    fn test_web_metrics_json_unit() {
+        let mut app = test::init_service(
+            App::new().service(web::resource("/metrics").route(web::post().to(metrics_json))),
+        );
+        let req = test::TestRequest::post()
+            .uri("/metrics")
+            .set_json(&WebMetricsPayload {
+                id: "1234".to_string(),
+                file_name: "test.py".to_string(),
+                code: "def foo():\n    pass\n".to_string(),
+                unit: true,
+            })
+            .to_request();
+
+        let res: Value = test::read_response_json(&mut app, req);
+        let expected = json!({
+            "id": "1234",
+            "spaces": {"kind": "unit",
+                       "line": 1,
+                       "metrics": {"cyclomatic": 1.0,
+                                   "halstead": {"bugs": 0.000_942_552_557_372_941_4,
+                                                "difficulty": 1.0,
+                                                "effort": 4.754_887_502_163_468,
+                                                "length": 3.0,
+                                                "level": 1.0,
+                                                "operands": 1.0,
+                                                "operators": 2.0,
+                                                "size": 3.0,
+                                                "time": 0.264_160_416_786_859_36,
+                                                "unique_operands": 1.0,
+                                                "unique_operators": 2.0,
+                                                "volume": 4.754_887_502_163_468},
+                                   "loc": {"lloc": 2.0, "sloc": 3.0}},
+                       "name": "test.py",
+                       "spaces": []}
+        });
+
+        assert_eq!(res, expected);
+    }
+
     #[test]
     fn test_web_metrics_plain() {
         let mut app = test::init_service(