feat: Add bundles features flag, and BundleApi

Author: caleb-leinz

Cargo.toml

@@ -28,7 +28,7 @@ moka = { version = "0.12.10", features = ["future"], optional = true }
 
 # ATLAS internal dependencies
 freedom-config = { version = "1.0.0", features = ["serde"] }
-freedom-models = { version = "2.2.0", features = ["serde"] }
+freedom-models = { git = "https://github.com/ATLAS-Space-Operations/rust-freedom-models", branch = "dev", features = ["serde"] }
 
 [dev-dependencies]
 futures = { version = "0.3.30" }
@@ -38,6 +38,7 @@ tokio-test = { version = "0.4.4"}
 tracing-test = { version = "0.2.4" }
 
 [features]
+bundles = ["freedom-models/bundles"]
 caching = ["dep:moka", "serde/rc"]
 
 [[example]]

src/api.rs

@@ -31,6 +31,8 @@ use futures_core::Stream;
 
 use crate::error::Error;
 
+#[cfg(feature = "bundles")]
+pub(crate) mod bundle;
 pub(crate) mod post;
 
 /// A super trait containing all the requirements for Freedom API Values

src/api/bundle.rs

@@ -0,0 +1,33 @@
+use freedom_models::bundle::fps_task;
+use time::{OffsetDateTime, format_description::well_known::Iso8601};
+
+use super::{Api, Error};
+
+/// Adds additional functionality by exposing bundle endpoints.
+///
+/// These are primarily used internally by the FPS and Gateway but exist in the public API, and are
+/// thus included here. In general, use of these should be avoided for customers
+pub trait BundleApi: Api {
+    /// Produces a list of [`fps_task::Bundle`]s within the designated window
+    ///
+    /// See [`get`](Api::get) documentation for more details about the process and return type
+    fn get_fps_task_bundle(
+        &self,
+        start: OffsetDateTime,
+        end: OffsetDateTime,
+    ) -> impl Future<Output = Result<Self::Container<Vec<fps_task::Bundle>>, Error>> + Send + Sync
+    {
+        async move {
+            let start = start.format(&Iso8601::DEFAULT).map_err(Error::from)?;
+            let end = end.format(&Iso8601::DEFAULT).map_err(Error::from)?;
+
+            let mut uri = self.path_to_url("fpstaskbundle/search/findByOverlapping");
+            uri.set_query(Some(&format!("start={}&end={}", start, end)));
+
+            self.get_json_map::<Self::Container<Vec<fps_task::Bundle>>>(uri)
+                .await
+        }
+    }
+}
+
+impl<T> BundleApi for T where T: Api {}

src/lib.rs

@@ -8,6 +8,8 @@ pub mod error;
 pub mod extensions;
 mod utils;
 
+#[cfg(feature = "bundles")]
+pub use self::api::bundle::BundleApi;
 pub use self::{
     api::{Api, Container, Inner, PaginatedStream, Value},
     client::Client,