antennas endpoints working

kairo-core/src/antennas.rs

@@ -0,0 +1,63 @@
+use diesel::prelude::*;
+use rocket::{http::Status, serde::json::Json, State};
+
+use kairo_common::{postgres, schema::antennas, Antenna, NewAntenna};
+
+#[rocket::get("/id/<id>")]
+pub fn get_by_id(db_pool: &State<postgres::DbPool>, id: String) -> Option<Json<Antenna>> {
+    let mut db = db_pool.get().unwrap();
+
+    let res = antennas::table
+        .select(antennas::all_columns)
+        .find(id)
+        .get_result::<Antenna>(&mut db);
+
+    match res {
+        Ok(v) => Some(rocket::serde::json::Json(v)),
+        _ => None,
+    }
+}
+
+#[rocket::get("/")]
+pub fn get_list(db_pool: &State<postgres::DbPool>) -> Json<Vec<Antenna>> {
+    let mut db = db_pool.get().unwrap();
+
+    let res = antennas::table
+        .select(antennas::all_columns)
+        .load::<Antenna>(&mut db);
+
+    match res {
+        Ok(v) => rocket::serde::json::Json(v),
+        _ => rocket::serde::json::Json(vec![]),
+    }
+}
+
+#[rocket::post("/new", format = "json", data = "<antenna>")]
+pub fn new(db_pool: &State<postgres::DbPool>, antenna: Json<NewAntenna<'_>>) -> Status {
+    let mut db = db_pool.get().unwrap();
+
+    let res = diesel::insert_into(antennas::table)
+        .values(*antenna)
+        .execute(&mut db);
+
+    match res {
+        Ok(_) => Status::Ok,
+        _ => Status::NotAcceptable,
+    }
+}
+
+#[rocket::patch("/update", format = "json", data = "<antenna>")]
+pub fn update(db_pool: &State<postgres::DbPool>, antenna: Json<NewAntenna<'_>>) -> Status {
+    let mut db = db_pool.get().unwrap();
+
+    let res = diesel::update(antennas::table)
+        .filter(antennas::id.eq(antenna.id))
+        .set(*antenna)
+        .execute(&mut db);
+
+    match res {
+        Ok(0) => Status::NotModified,
+        Ok(1) => Status::Ok,
+        _ => Status::BadRequest,
+    }
+}

kairo-core/src/main.rs

@@ -1,3 +1,30 @@
-fn main() {
-    println!("Hello, world!");
+#[macro_use]
+extern crate rocket;
+
+mod antennas;
+
+use std::path::{Path, PathBuf};
+
+use rocket::fs::NamedFile;
+use rocket::response::status::NotFound;
+
+use kairo_common::postgres;
+
+#[get("/<file..>")]
+async fn serve_file(file: PathBuf) -> Result<NamedFile, NotFound<String>> {
+    let path = Path::new("static/").join(file);
+    NamedFile::open(&path)
+        .await
+        .map_err(|e| NotFound(e.to_string()))
+}
+
+#[launch]
+fn rocket() -> _ {
+    rocket::build()
+        .manage(postgres::init_pool())
+        .mount("/static", routes![serve_file])
+        .mount("/antennas/", routes![antennas::get_by_id])
+        .mount("/antennas/", routes![antennas::get_list])
+        .mount("/antennas/", routes![antennas::update])
+        .mount("/antennas/", routes![antennas::new])
 }