gws: add diff for google accounts

Author: ubiratansoares

rust_team_data/src/v1.rs

@@ -26,6 +26,15 @@ pub struct Team {
     pub github: Option<TeamGitHub>,
     pub website_data: Option<TeamWebsite>,
     pub roles: Vec<MemberRole>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub google_workspace_saml_group: Option<bool>,
+}
+
+#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
+pub struct GoogleWorkspace {
+    pub first_name: String,
+    pub last_name: String,
+    pub account_handle: String,
 }
 
 #[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
@@ -36,6 +45,8 @@ pub struct TeamMember {
     pub is_lead: bool,
     #[serde(skip_serializing_if = "Vec::is_empty", default)]
     pub roles: Vec<String>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub google_workspace: Option<GoogleWorkspace>,
 }
 
 #[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]

src/main.rs

@@ -10,7 +10,13 @@ mod static_api;
 mod sync;
 mod validate;
 
-const AVAILABLE_SERVICES: &[&str] = &["github", "mailgun", "zulip", "crates-io"];
+const AVAILABLE_SERVICES: &[&str] = &[
+    "github",
+    "google-workspace",
+    "mailgun",
+    "zulip",
+    "crates-io",
+];
 
 const USER_AGENT: &str = "https://github.com/rust-lang/team (infra@rust-lang.org)";
 

src/schema.rs

@@ -76,13 +76,22 @@ pub(crate) struct Funding {
     github_sponsors: bool,
 }
 
-#[allow(dead_code)]
-#[derive(serde::Deserialize, Debug)]
+#[derive(serde::Deserialize, Debug, Clone)]
 #[serde(deny_unknown_fields, rename_all = "kebab-case")]
 pub(crate) struct GoogleWorkspace {
-    first_name: String,
-    last_name: String,
-    account_handle: String,
+    pub first_name: String,
+    pub last_name: String,
+    pub account_handle: String,
+}
+
+impl From<&GoogleWorkspace> for rust_team_data::v1::GoogleWorkspace {
+    fn from(gws: &GoogleWorkspace) -> Self {
+        rust_team_data::v1::GoogleWorkspace {
+            first_name: gws.first_name.clone(),
+            last_name: gws.last_name.clone(),
+            account_handle: gws.account_handle.clone(),
+        }
+    }
 }
 
 #[allow(dead_code)]
@@ -157,8 +166,8 @@ impl Person {
         &self.permissions
     }
 
-    pub(crate) fn google_workspace(&self) -> &Option<GoogleWorkspace> {
-        &self.google_workspace
+    pub(crate) fn google_workspace(&self) -> Option<&GoogleWorkspace> {
+        self.google_workspace.as_ref()
     }
 
     pub(crate) fn validate(&self) -> Result<(), Error> {
@@ -194,7 +203,6 @@ impl std::fmt::Display for TeamKind {
     }
 }
 
-#[allow(dead_code)]
 #[derive(serde::Deserialize, Debug)]
 #[serde(deny_unknown_fields, rename_all = "kebab-case")]
 pub(crate) struct Team {

src/static_api.rs

@@ -7,7 +7,9 @@ use anyhow::{Context as _, Error, ensure};
 use indexmap::IndexMap;
 use log::info;
 use rust_team_data::v1;
-use rust_team_data::v1::{BranchProtectionMode, Crate, CrateTeamOwner, RepoMember};
+use rust_team_data::v1::{
+    BranchProtectionMode, Crate, CrateTeamOwner, GoogleWorkspace, RepoMember,
+};
 use std::collections::HashMap;
 use std::path::Path;
 
@@ -539,6 +541,7 @@ fn convert_teams<'a>(
                     github_id: person.github_id(),
                     is_lead: leads.contains(github_name),
                     roles: website_roles.get(*github_name).cloned().unwrap_or_default(),
+                    google_workspace: person.google_workspace().map(GoogleWorkspace::from),
                 });
             }
         }
@@ -557,6 +560,7 @@ fn convert_teams<'a>(
                         .get(alum.github.as_str())
                         .cloned()
                         .unwrap_or_default(),
+                    google_workspace: person.google_workspace().map(GoogleWorkspace::from),
                 });
             }
         }
@@ -606,6 +610,7 @@ fn convert_teams<'a>(
                     description: role.description.clone(),
                 })
                 .collect(),
+            google_workspace_saml_group: team.google_workspace_saml_group(),
         };
         team_map.insert(team.name().into(), team_data);
     }

src/sync/github/tests/test_utils.rs

@@ -287,6 +287,7 @@ impl From<TeamData> for v1::Team {
             github: (!gh_teams.is_empty()).then_some(TeamGitHub { teams: gh_teams }),
             website_data: None,
             roles: vec![],
+            google_workspace_saml_group: None,
         }
     }
 }

src/sync/gws/api.rs

@@ -0,0 +1,41 @@
+use crate::sync::gws::RUST_LANG_GWS_DOMAIN;
+use async_trait::async_trait;
+use rust_team_data::v1::GoogleWorkspace;
+
+/// https://developers.google.com/workspace/admin/directory/reference/rest/v1/groups
+#[derive(Clone, Debug, PartialEq)]
+pub(crate) struct Group {
+    pub name: String,
+    pub email: String,
+}
+
+/// https://developers.google.com/workspace/admin/directory/reference/rest/v1/users#UserName
+#[derive(Clone, Debug, Eq, Hash, Ord, PartialEq, PartialOrd)]
+pub(crate) struct UserName {
+    pub given_name: String,
+    pub family_name: String,
+}
+
+///https://developers.google.com/workspace/admin/directory/reference/rest/v1/users
+#[derive(Clone, Debug, Eq, Hash, Ord, PartialEq, PartialOrd)]
+pub(crate) struct User {
+    pub name: UserName,
+    pub primary_email: String,
+}
+
+impl From<&GoogleWorkspace> for User {
+    fn from(gws: &GoogleWorkspace) -> Self {
+        Self {
+            primary_email: format!("{}@{}", gws.account_handle, RUST_LANG_GWS_DOMAIN),
+            name: UserName {
+                given_name: gws.first_name.to_string(),
+                family_name: gws.last_name.to_string(),
+            },
+        }
+    }
+}
+
+#[async_trait]
+pub(crate) trait GoogleWorkspaceApiClient {
+    async fn get_users(&self) -> anyhow::Result<Vec<User>>;
+}