chore: add support for next page leaderboard

Author: hackermondev

Diff for: .dockerignore

@@ -1 +1,2 @@
-target/
+target/
+.git/

Diff for: graphql/TeamYearThankQuery.graphql

@@ -1,6 +1,7 @@
 query TeamYearThankQuery(
   $selectedHandle: String!
   $year: Int
+  $cursor: String!
 ) {
   selectedTeam: team(handle: $selectedHandle) {
     ...TeamThanksSelectedTeam
@@ -11,8 +12,11 @@ fragment TeamThanksSelectedTeam on Team {
   name
   state
   handle
-  participants(first: 100, year: $year) {
-    year_range
+  participants(first: 999, year: $year, after: $cursor) {
+    pageInfo {
+			hasNextPage
+			endCursor
+		}
     edges {
       node {
         id

Diff for: src/sexurity-discord/src/subscriptions/reputation.rs

@@ -95,9 +95,9 @@ fn build_embed_data(diff: Vec<models::RepData>, handle: &str) -> Option<Embed> {
     let old = &diff[0];
     let new = &diff[1];
 
-    if old.rank == -1 {
+    if old.reputation == -1 {
         // new user added to leaderboard
-        let text = format!(
+        let mut text = format!(
             "[**``{}``**]({}) was added to [**``{}``**]({}) with **{} reputation** (rank: #{})",
             new.user_name,
             format!("https://hackerone.com/{}", new.user_name),
@@ -107,15 +107,26 @@ fn build_embed_data(diff: Vec<models::RepData>, handle: &str) -> Option<Embed> {
             new.rank
         );
 
+        if new.rank == -1 {
+            text = format!(
+                "[**``{}``**]({}) was added to [**``{}``**]({}) with **{} reputation** (rank: <100)",
+                new.user_name,
+                format!("https://hackerone.com/{}", new.user_name),
+                handle,
+                format!("https://hackerone.com/{}", handle),
+                new.reputation
+            );
+        }
+
         let embed = EmbedBuilder::new()
             .description(text)
             .color(models::embed_colors::POSTIVE)
             .build();
         return Some(embed);
-    } else if new.rank == -1 {
+    } else if new.reputation == -1 {
         // user removed from leaderboard
         let text = format!(
-            "[**``{}``**]({}) was removed from top 100",
+            "[**``{}``**]({}) was removed from the leaderboard",
             old.user_name,
             format!("https://hackerone.com/{}", old.user_name),
         );

Diff for: src/sexurity-poller/src/main.rs

@@ -76,6 +76,7 @@ fn ensure_args(client: &HackerOneClient, args: &Arguments) -> bool {
     let variables = hackerone::team_year_thank_query::Variables {
         selected_handle: args.hackerone_handle.clone(),
         year: Some(now.year().into()),
+        cursor: "".into(),
     };
 
     let query = hackerone::TeamYearThankQuery::build_query(variables);

Diff for: src/sexurity-poller/src/polls/reputation.rs

@@ -28,7 +28,7 @@ pub fn run_poll(config: &PollConfiguration) -> Result<(), Box<dyn std::error::Er
         .arg(models::redis_keys::REPUTATION_QUEUE_LAST_RUN_TIME)
         .query(&mut redis_conn)?;
     let mut last_rep_data = get_old_reputation_data(&mut redis_conn);
-    let rep_data = get_reputation_data(&config.team_handle, &config.hackerone).unwrap();
+    let rep_data = get_reputation_data(&config.team_handle, &config.hackerone, None, None).unwrap();
 
     debug!(
         "reputation poll event: last_run_time {}",
@@ -150,10 +150,14 @@ fn set_last_run_time_now(conn: &mut redis::Connection) {
 fn get_reputation_data(
     handle: &str,
     client: &HackerOneClient,
+    previous_data: Option<Vec<models::RepData>>,
+    next_cursor: Option<String>,
 ) -> Result<Vec<models::RepData>, Box<dyn std::error::Error>> {
+    debug!("get reputation data, cursor: {}", next_cursor.as_ref().unwrap_or(&String::from("")));
     let variables = hackerone::team_year_thank_query::Variables {
         selected_handle: handle.to_string(),
         year: None,
+        cursor: next_cursor.unwrap_or(String::from("")),
     };
 
     let query = hackerone::TeamYearThankQuery::build_query(variables);
@@ -164,16 +168,35 @@ fn get_reputation_data(
         .send()?;
 
     let mut result: Vec<models::RepData> = vec![];
-    // (TODO): find a better way to do this?
+    if previous_data.is_some() {
+        result = previous_data.unwrap();
+    }
+
     let data = response.json::<graphql_client::Response<<hackerone::TeamYearThankQuery as GraphQLQuery>::ResponseData>>().unwrap();
+    let page_info = &data
+        .data
+        .as_ref()
+        .unwrap()
+        .selected_team
+        .as_ref()
+        .unwrap()
+        .participants
+        .as_ref()
+        .unwrap()
+        .page_info;
+
     let researchers = data
         .data
+        .as_ref()
         .unwrap()
         .selected_team
+        .as_ref()
         .unwrap()
         .participants
+        .as_ref()
         .unwrap()
         .edges
+        .as_ref()
         .unwrap();
     for researcher in researchers {
         let user = researcher.as_ref().unwrap().node.as_ref().unwrap();
@@ -196,6 +219,11 @@ fn get_reputation_data(
         result.push(data);
     }
 
+    if page_info.has_next_page {
+        let end_cursor = page_info.end_cursor.as_ref().unwrap();
+        return Ok(get_reputation_data(handle, client, Some(result), Some(end_cursor.clone())).unwrap());
+    }
+
     Ok(result)
 }