2022-02-21 17:46:23 +01:00
|
|
|
#![doc = include_str!("../README.md")]
|
2022-02-12 12:25:05 +01:00
|
|
|
#![warn(
|
|
|
|
clippy::all,
|
|
|
|
missing_debug_implementations,
|
|
|
|
rust_2018_idioms,
|
|
|
|
rustdoc::broken_intra_doc_links
|
|
|
|
)]
|
|
|
|
#![deny(missing_docs)]
|
|
|
|
|
2022-02-13 11:16:19 +01:00
|
|
|
use std::sync::{Arc, Mutex};
|
2022-02-12 21:35:58 +01:00
|
|
|
|
2022-02-12 15:58:56 +01:00
|
|
|
use color_eyre::Result;
|
2022-02-15 17:05:12 +01:00
|
|
|
use rocket::http::ContentType;
|
|
|
|
use rocket::response::content::Custom;
|
2022-02-12 12:25:05 +01:00
|
|
|
use rocket::serde::json::Json;
|
2022-02-12 15:58:56 +01:00
|
|
|
use rocket::tokio::{self, select};
|
2022-02-15 13:14:01 +01:00
|
|
|
use rocket::{get, routes, State};
|
2022-02-12 15:58:56 +01:00
|
|
|
|
2022-02-15 14:15:59 +01:00
|
|
|
pub(crate) use self::forecast::Metric;
|
|
|
|
use self::forecast::{forecast, Forecast};
|
2022-02-22 12:49:42 +01:00
|
|
|
pub(crate) use self::maps::{mark_map, Maps, MapsHandle};
|
2022-02-15 14:15:59 +01:00
|
|
|
use self::position::{resolve_address, Position};
|
2022-02-12 15:58:56 +01:00
|
|
|
|
2022-02-15 13:14:01 +01:00
|
|
|
pub(crate) mod forecast;
|
2022-02-13 12:45:27 +01:00
|
|
|
pub(crate) mod maps;
|
2022-02-15 14:15:59 +01:00
|
|
|
pub(crate) mod position;
|
2022-02-13 12:45:27 +01:00
|
|
|
pub(crate) mod providers;
|
2022-02-12 12:25:05 +01:00
|
|
|
|
|
|
|
/// Handler for retrieving the forecast for an address.
|
|
|
|
#[get("/forecast?<address>&<metrics>")]
|
2022-02-12 21:35:58 +01:00
|
|
|
async fn forecast_address(
|
|
|
|
address: String,
|
|
|
|
metrics: Vec<Metric>,
|
|
|
|
maps_handle: &State<MapsHandle>,
|
|
|
|
) -> Option<Json<Forecast>> {
|
2022-02-15 14:15:59 +01:00
|
|
|
let position = resolve_address(address).await?;
|
|
|
|
let forecast = forecast(position, metrics, maps_handle).await;
|
2022-02-12 12:25:05 +01:00
|
|
|
|
|
|
|
Some(Json(forecast))
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Handler for retrieving the forecast for a geocoded position.
|
|
|
|
#[get("/forecast?<lat>&<lon>&<metrics>", rank = 2)]
|
2022-02-12 21:35:58 +01:00
|
|
|
async fn forecast_geo(
|
|
|
|
lat: f64,
|
|
|
|
lon: f64,
|
|
|
|
metrics: Vec<Metric>,
|
|
|
|
maps_handle: &State<MapsHandle>,
|
|
|
|
) -> Json<Forecast> {
|
2022-02-15 14:15:59 +01:00
|
|
|
let position = Position::new(lat, lon);
|
|
|
|
let forecast = forecast(position, metrics, maps_handle).await;
|
2022-02-12 12:25:05 +01:00
|
|
|
|
|
|
|
Json(forecast)
|
|
|
|
}
|
|
|
|
|
2022-02-18 23:18:50 +01:00
|
|
|
/// Handler for showing the current map with the geocoded position of an address for a specific
|
|
|
|
/// metric.
|
2022-02-15 17:05:12 +01:00
|
|
|
///
|
|
|
|
/// Note: This handler is mosly used for debugging purposes!
|
2022-02-17 22:25:13 +01:00
|
|
|
#[get("/map?<address>&<metric>")]
|
2022-02-22 12:49:42 +01:00
|
|
|
async fn map_address(
|
2022-02-17 22:25:13 +01:00
|
|
|
address: String,
|
2022-02-15 17:05:12 +01:00
|
|
|
metric: Metric,
|
|
|
|
maps_handle: &State<MapsHandle>,
|
|
|
|
) -> Option<Custom<Vec<u8>>> {
|
2022-02-18 23:18:50 +01:00
|
|
|
let position = resolve_address(address).await?;
|
2022-02-22 12:49:42 +01:00
|
|
|
let image_data = mark_map(position, metric, maps_handle).await;
|
2022-02-18 23:18:50 +01:00
|
|
|
|
|
|
|
image_data.map(|id| Custom(ContentType::PNG, id))
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Handler for showing the current map with the geocoded position for a specific metric.
|
|
|
|
///
|
|
|
|
/// Note: This handler is mosly used for debugging purposes!
|
|
|
|
#[get("/map?<lat>&<lon>&<metric>", rank = 2)]
|
2022-02-22 12:49:42 +01:00
|
|
|
async fn map_geo(
|
2022-02-18 23:18:50 +01:00
|
|
|
lat: f64,
|
|
|
|
lon: f64,
|
|
|
|
metric: Metric,
|
|
|
|
maps_handle: &State<MapsHandle>,
|
|
|
|
) -> Option<Custom<Vec<u8>>> {
|
|
|
|
let position = Position::new(lat, lon);
|
2022-02-22 12:49:42 +01:00
|
|
|
let image_data = mark_map(position, metric, maps_handle).await;
|
2022-02-18 23:18:50 +01:00
|
|
|
|
|
|
|
image_data.map(|id| Custom(ContentType::PNG, id))
|
|
|
|
}
|
|
|
|
|
2022-02-12 15:58:56 +01:00
|
|
|
/// Starts the main maps refresh loop and sets up and launches Rocket.
|
2022-02-13 11:22:22 +01:00
|
|
|
///
|
|
|
|
/// See [`maps::run`] for the maps refresh loop.
|
2022-02-12 15:58:56 +01:00
|
|
|
#[rocket::main]
|
|
|
|
async fn main() -> Result<()> {
|
|
|
|
color_eyre::install()?;
|
|
|
|
|
2022-02-12 21:35:58 +01:00
|
|
|
let maps = Maps::new();
|
|
|
|
let maps_handle = Arc::new(Mutex::new(maps));
|
|
|
|
let maps_updater = tokio::spawn(maps::run(Arc::clone(&maps_handle)));
|
|
|
|
|
2022-02-12 15:58:56 +01:00
|
|
|
let rocket = rocket::build()
|
2022-02-12 21:35:58 +01:00
|
|
|
.manage(maps_handle)
|
2022-02-18 23:18:50 +01:00
|
|
|
.mount(
|
|
|
|
"/",
|
2022-02-22 12:49:42 +01:00
|
|
|
routes![forecast_address, forecast_geo, map_address, map_geo],
|
2022-02-18 23:18:50 +01:00
|
|
|
)
|
2022-02-12 15:58:56 +01:00
|
|
|
.ignite()
|
|
|
|
.await?;
|
|
|
|
let shutdown = rocket.shutdown();
|
|
|
|
|
|
|
|
select! {
|
|
|
|
result = rocket.launch() => {
|
|
|
|
result?
|
|
|
|
}
|
|
|
|
result = maps_updater => {
|
|
|
|
shutdown.notify();
|
|
|
|
result?
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
Ok(())
|
2022-02-12 12:25:05 +01:00
|
|
|
}
|