From 95f30751c6008a8dc1edc12cc2bdfb7c93d62a46 Mon Sep 17 00:00:00 2001
From: Paul van Tilburg <paul@luon.net>
Date: Thu, 24 Feb 2022 20:24:08 +0100
Subject: [PATCH] Add the combined provider

---
 src/providers.rs          |   1 +
 src/providers/combined.rs | 112 ++++++++++++++++++++++++++++++++++++++
 2 files changed, 113 insertions(+)
 create mode 100644 src/providers/combined.rs

diff --git a/src/providers.rs b/src/providers.rs
index f1bf8cc..5ff1493 100644
--- a/src/providers.rs
+++ b/src/providers.rs
@@ -3,4 +3,5 @@
 //! Data is either provided via a direct (JSON) API or via looking up values on maps.
 
 pub(crate) mod buienradar;
+pub(crate) mod combined;
 pub(crate) mod luchtmeetnet;
diff --git a/src/providers/combined.rs b/src/providers/combined.rs
new file mode 100644
index 0000000..8608694
--- /dev/null
+++ b/src/providers/combined.rs
@@ -0,0 +1,112 @@
+//! The combined data provider.
+//!
+//! This combines and collates data using the other providers.
+
+use cached::proc_macro::cached;
+use chrono::serde::ts_seconds;
+use chrono::{DateTime, Utc};
+use rocket::serde::Serialize;
+
+pub(crate) use super::buienradar::{self, Sample as BuienradarSample};
+pub(crate) use super::luchtmeetnet::{self, Item as LuchtmeetnetItem};
+use crate::maps::MapsHandle;
+use crate::position::Position;
+use crate::Metric;
+
+/// The combined data item.
+#[derive(Clone, Debug, Serialize)]
+#[serde(crate = "rocket::serde")]
+pub(crate) struct Item {
+    /// The time(stamp) of the forecast.
+    #[serde(serialize_with = "ts_seconds::serialize")]
+    time: DateTime<Utc>,
+
+    /// The forecasted value.
+    value: f32,
+}
+
+/// Merges pollen samples and AQI items into combined items.
+///
+/// This drops items from either the pollen samples or from the AQI items if they are not stamped
+/// with half an hour of the first item of the latest stating series, thus lining them before they
+/// are combined.
+///
+/// Returns [`None`] if there are no pollen samples, if  there are no AQI items, or if
+/// lining them up fails.
+fn merge(
+    pollen_samples: Vec<BuienradarSample>,
+    aqi_items: Vec<LuchtmeetnetItem>,
+) -> Option<Vec<Item>> {
+    let mut pollen_samples = pollen_samples;
+    let mut aqi_items = aqi_items;
+
+    // Align the iterators based on the (hourly) timestamps!
+    let pollen_first_time = pollen_samples.first()?.time;
+    let aqi_first_time = aqi_items.first()?.time;
+    if pollen_first_time < aqi_first_time {
+        // Drain one or more pollen samples to line up.
+        let idx = pollen_samples.iter().position(|smp| {
+            smp.time
+                .signed_duration_since(aqi_first_time)
+                .num_seconds()
+                .abs()
+                < 1800
+        })?;
+        pollen_samples.drain(..idx);
+    } else {
+        // Drain one or more AQI items to line up.
+        let idx = aqi_items.iter().position(|item| {
+            item.time
+                .signed_duration_since(pollen_first_time)
+                .num_seconds()
+                .abs()
+                < 1800
+        })?;
+        aqi_items.drain(..idx);
+    }
+
+    // Combine the samples with items by taking the maximum of pollen sample score and AQI item
+    // value.
+    let items = pollen_samples
+        .into_iter()
+        .zip(aqi_items.into_iter())
+        .map(|(pollen_sample, aqi_item)| {
+            let time = pollen_sample.time;
+            let value = (pollen_sample.score as f32).max(aqi_item.value);
+
+            Item { time, value }
+        })
+        .collect();
+
+    Some(items)
+}
+
+/// Retrieves the combined forecasted items for the provided position and metric.
+///
+/// It supports the following metric:
+/// * [`Metric::PAQI`]
+///
+/// Returns [`None`] if retrieving data from either the Buienradar or the Luchtmeetnet provider
+/// fails or if they cannot be combined.
+///
+/// If the result is [`Some`] it will be cached for 30 minutes for the the given position and
+/// metric.
+#[cached(
+    time = 1800,
+    key = "(Position, Metric)",
+    convert = r#"{ (position, metric) }"#,
+    option = true
+)]
+pub(crate) async fn get(
+    position: Position,
+    metric: Metric,
+    maps_handle: &MapsHandle,
+) -> Option<Vec<Item>> {
+    if metric != Metric::PAQI {
+        return None;
+    };
+    let pollen_items = buienradar::get_samples(position, Metric::Pollen, maps_handle).await;
+    let aqi_items = luchtmeetnet::get(position, Metric::AQI).await;
+
+    merge(pollen_items?, aqi_items?)
+}