post agent errors on github

zeegomo · zeegomo · commit 30656d6ac150 · 2019-01-22T14:39:05.000+01:00
diff --git a/src/server/routes/agent.rs b/src/server/routes/agent.rs
@@ -53,6 +53,14 @@ pub fn routes(
         .and(auth_filter(data.clone(), TokenType::Agent))
         .map(endpoint_heartbeat);
 
+    let error = warp::post2()
+        .and(warp::path("error"))
+        .and(warp::path::end())
+        .and(warp::body::json())
+        .and(data_filter.clone())
+        .and(auth_filter(data.clone(), TokenType::Agent))
+        .map(endpoint_error);
+
     warp::any()
         .and(
             config
@@ -63,6 +71,8 @@ pub fn routes(
                 .or(record_progress)
                 .unify()
                 .or(heartbeat)
+                .unify()
+                .or(error)
                 .unify(),
         )
         .map(handle_results)
@@ -146,6 +156,25 @@ fn endpoint_heartbeat(data: Arc<Data>, auth: AuthDetails) -> Fallible<Response<B
     Ok(ApiResponse::Success { result: true }.into_response()?)
 }
 
+fn endpoint_error(error: String, data: Arc<Data>, auth: AuthDetails) -> Fallible<Response<Body>> {
+    let ex = Experiment::run_by(&data.db, &Assignee::Agent(auth.name.clone()))?
+        .ok_or_else(|| err_msg("no experiment run by this agent"))?;
+
+    if let Some(ref github_issue) = ex.github_issue {
+        Message::new()
+            .line(
+                "exclamation",
+                format!(
+                    "Experiment **`{}`** **running** on agent `{}` has encountered an error",
+                    ex.name, auth.name,
+                ),
+            )
+            .line("", format!("error: {}", error,))
+            .send(&github_issue.api_url, &data)?;
+    }
+    Ok(ApiResponse::Success { result: true }.into_response()?)
+}
+
 fn handle_results(resp: Fallible<Response<Body>>) -> Response<Body> {
     match resp {
         Ok(resp) => resp,
