team_api.rs

use crate::sync::utils::ResponseExt;
use log::{debug, trace};
use std::borrow::Cow;
use std::path::PathBuf;

/// Determines how do we get access to the ground-truth data from `rust-lang/team`.
pub enum TeamApi {
    /// Access the live data from the published production REST API.
    Production,
    /// Directly access a directory with prebuilt JSON data.
    Prebuilt(PathBuf),
}

impl TeamApi {
    pub(crate) async fn get_teams(&self) -> anyhow::Result<Vec<rust_team_data::v1::Team>> {
        debug!("loading teams list from the Team API");
        Ok(self
            .req::<rust_team_data::v1::Teams>("teams.json")
            .await?
            .teams
            .into_iter()
            .map(|(_k, v)| v)
            .collect())
    }

    pub(crate) async fn get_repos(&self) -> anyhow::Result<Vec<rust_team_data::v1::Repo>> {
        debug!("loading teams list from the Team API");
        Ok(self
            .req::<rust_team_data::v1::Repos>("repos.json")
            .await?
            .repos
            .into_iter()
            .flat_map(|(_k, v)| v)
            .collect())
    }

    pub(crate) async fn get_lists(&self) -> anyhow::Result<rust_team_data::v1::Lists> {
        debug!("loading email lists list from the Team API");
        self.req::<rust_team_data::v1::Lists>("lists.json").await
    }

    pub(crate) async fn get_zulip_groups(&self) -> anyhow::Result<rust_team_data::v1::ZulipGroups> {
        debug!("loading Zulip groups from the Team API");
        self.req::<rust_team_data::v1::ZulipGroups>("zulip-groups.json")
            .await
    }

    pub(crate) async fn get_zulip_map(&self) -> anyhow::Result<rust_team_data::v1::ZulipMapping> {
        debug!("loading GitHub id to Zulip id map from the Team API");
        self.req::<rust_team_data::v1::ZulipMapping>("zulip-map.json")
            .await
    }

    pub(crate) async fn get_zulip_streams(
        &self,
    ) -> anyhow::Result<rust_team_data::v1::ZulipStreams> {
        debug!("loading Zulip streams from the Team API");
        self.req::<rust_team_data::v1::ZulipStreams>("zulip-streams.json")
            .await
    }

    async fn req<T: serde::de::DeserializeOwned>(&self, url: &str) -> anyhow::Result<T> {
        match self {
            TeamApi::Production => {
                let base = std::env::var("TEAM_DATA_BASE_URL")
                    .map(Cow::Owned)
                    .unwrap_or_else(|_| Cow::Borrowed(rust_team_data::v1::BASE_URL));
                let url = format!("{base}/{url}");
                trace!("http request: GET {url}");
                Ok(reqwest::get(&url)
                    .await?
                    .error_for_status()?
                    .json_annotated()
                    .await?)
            }
            TeamApi::Prebuilt(directory) => {
                let contents = std::fs::read(directory.join("v1").join(url))?;
                Ok(serde_json::from_slice(&contents)?)
            }
        }
    }
}