Add 'Staff view' links

Author: illicitonion

src/frontend.rs

@@ -6,8 +6,9 @@ use axum::{
     extract::{OriginalUri, Path, Query, State},
     response::{Html, IntoResponse, Response},
 };
-use futures::future::join_all;
+use futures::future::{join, join_all};
 use http::{header::CONTENT_TYPE, StatusCode, Uri};
+use octocrab::Octocrab;
 use serde::Deserialize;
 use tower_sessions::Session;
 
@@ -18,7 +19,7 @@ use crate::{
         TraineeStatus,
     },
     google_groups::{get_groups, groups_client, GoogleGroup},
-    octocrab::octocrab,
+    octocrab::{all_pages, octocrab},
     prs::{MaybeReviewerStaffOnlyDetails, PrState, ReviewerInfo},
     reviewer_staff_info::get_reviewer_staff_info,
     sheets::sheets_client,
@@ -42,6 +43,11 @@ pub async fn list_courses(
     .await
     .into_iter()
     .collect::<Result<Vec<_>, _>>()?;
+
+    let is_staff = is_staff(&octocrab, &server_state.config.github_org)
+        .await
+        .unwrap_or(false);
+
     let courses_with_batch_metadata = courses
         .keys()
         .zip(batch_metadata)
@@ -75,16 +81,39 @@ pub async fn list_courses(
     Ok(Html(
         ListCoursesTemplate {
             courses_with_batch_metadata,
+            is_staff,
         }
         .render()
         .unwrap(),
     ))
 }
 
+async fn is_staff(octocrab: &Octocrab, github_org: &str) -> Result<bool, Error> {
+    let team_future = all_pages("staff team members", octocrab, async || {
+        octocrab
+            .teams(github_org)
+            .members("cyf-staff-team")
+            .send()
+            .await
+    });
+    let current = octocrab.current();
+    let self_future = current.user();
+    let (team_members, user) = join(team_future, self_future).await;
+    let team_members = team_members?;
+    let user = user.context("Failed to get current user")?;
+    for team_member in team_members {
+        if team_member.login == user.login {
+            return Ok(true);
+        }
+    }
+    Ok(false)
+}
+
 #[derive(Template)]
 #[template(path = "list-courses.html")]
 struct ListCoursesTemplate {
     pub courses_with_batch_metadata: Vec<CourseScheduleWithBatchMetadata>,
+    pub is_staff: bool,
 }
 
 struct CourseScheduleWithBatchMetadata {

templates/list-courses.html

@@ -8,7 +8,12 @@ <h2>{{ cwbm.course.name }}</h2>
             {% for batch in cwbm.batch_metadata %}
             <li><a href="/courses/{{ cwbm.course.name }}/batches/{{ batch.github_team_slug }}">{{ batch.name }}</a></li>
             {% endfor %}
-            <li><a href="/courses/{{ cwbm.course.name }}/reviewers">Reviewers</a></li>
+            <li>
+                <a href="/courses/{{ cwbm.course.name }}/reviewers">Reviewers</a>
+                {% if is_staff %}
+                (<a href="/courses/{{ cwbm.course.name }}/reviewers?staff=true">Staff view</a>)
+                {% endif %}
+            </li>
         </ul>
         {% endfor %}
     </body>