Add issueStat query and openIssueCount field

Closes #145

Author: danbi2990

CHANGELOG.md

@@ -11,6 +11,10 @@ this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.htm
 - Tracing with a filter set by `RUST_LOG` environment variable.
 - Added support for passing the SSH passphrase through the `SSH_PASSPHRASE`
   environment variable.
+- Added new GraphQL API: `issueStat` query. Users can filter issues by
+  `assignee`, `author`, `repo`(repository name), `begin` and `end` (creation
+  date range). The query returns the `openIssueCount` field, indicating the
+  number of open issues.
 
 ### Changed
 

Cargo.lock

@@ -5,11 +5,11 @@ edition = "2021"
 
 [dependencies]
 anyhow = "1"
-async-graphql = "7"
+async-graphql = { version = "7", features = ["chrono"] }
 async-graphql-warp = "7"
 base64 = "0.22"
 bincode = "1"
-chrono = "0.4"
+chrono = { version = "0.4", features = ["serde"] }
 clap = { version = "4", features = ["derive"] }
 config = { version = "0.15", features = ["toml"], default-features = false }
 directories = "6"

Cargo.toml

@@ -7,7 +7,7 @@ use sled::{Db, Tree};
 
 use crate::{
     github::{GitHubIssue, GitHubPullRequests},
-    graphql::{Issue, PullRequest},
+    graphql::{issue::Issue, pull_request::PullRequest},
 };
 const ISSUE_TREE_NAME: &str = "issues";
 const PULL_REQUEST_TREE_NAME: &str = "pull_requests";
@@ -66,11 +66,7 @@ impl Database {
     ) -> Result<()> {
         for item in resp {
             let keystr: String = format!("{owner}/{name}#{}", item.number);
-            Database::insert(
-                &keystr,
-                (&item.title, &item.author, &item.closed_at),
-                &self.issue_tree,
-            )?;
+            Database::insert(&keystr, item, &self.issue_tree)?;
         }
         Ok(())
     }

src/database.rs

@@ -4,12 +4,12 @@ use anyhow::{bail, Context, Result};
 use chrono::Utc;
 use graphql_client::{GraphQLQuery, QueryBody, Response as GraphQlResponse};
 use reqwest::{Client, RequestBuilder, Response};
-use serde::Serialize;
+use serde::{Deserialize, Serialize};
 use tokio::time;
 use tracing::error;
 
 use crate::{database::Database, github::{
-    issues::IssuesRepositoryIssuesNodesAuthor::User as userName,
+    issues::{IssueState, IssuesRepositoryIssuesNodesAuthor::User as userName},
     pull_requests::PullRequestsRepositoryPullRequestsNodesReviewRequestsNodesRequestedReviewer::User,
 }, settings::Repository as RepoInfo};
 
@@ -18,15 +18,15 @@ const GITHUB_URL: &str = "https://api.github.com/graphql";
 const APP_USER_AGENT: &str = concat!(env!("CARGO_PKG_NAME"), "/", env!("CARGO_PKG_VERSION"),);
 const INIT_TIME: &str = "1992-06-05T00:00:00Z";
 
-type DateTime = String;
+type DateTime = chrono::DateTime<Utc>;
 
 #[derive(GraphQLQuery)]
 #[graphql(
     schema_path = "src/github/schema.graphql",
     query_path = "src/github/issues.graphql",
-    response_derives = "Debug"
+    response_derives = "Debug, Clone"
 )]
-struct Issues;
+pub(crate) struct Issues;
 
 #[derive(GraphQLQuery)]
 #[graphql(
@@ -35,12 +35,22 @@ struct Issues;
 )]
 struct PullRequests;
 
-#[derive(Debug)]
+#[allow(clippy::derivable_impls)]
+impl Default for IssueState {
+    fn default() -> Self {
+        IssueState::OPEN
+    }
+}
+
+#[derive(Debug, Default, Serialize, Deserialize)]
 pub(super) struct GitHubIssue {
     pub(super) number: i64,
     pub(super) title: String,
     pub(super) author: String,
     pub(super) closed_at: Option<DateTime>,
+    pub(super) created_at: DateTime,
+    pub(super) state: IssueState,
+    pub(super) assignees: Vec<String>,
 }
 
 #[derive(Debug)]
@@ -133,7 +143,7 @@ async fn send_github_issue_query(
             last: None,
             before: None,
             after: end_cur,
-            since: Some(since.to_string()),
+            since: Some(since.parse::<DateTime>()?),
         };
         let resp_body: GraphQlResponse<issues::ResponseData> =
             send_query::<Issues>(token, var).await?.json().await?;
@@ -147,11 +157,18 @@ async fn send_github_issue_query(
                         if let userName(on_user) = author_ref {
                             author.clone_from(&on_user.login.clone());
                         }
+                        let assignees =
+                            issue.assignees.nodes.as_ref().map_or(Vec::new(), |nodes| {
+                                nodes.iter().flatten().map(|a| a.login.clone()).collect()
+                            });
                         issues.push(GitHubIssue {
                             number: issue.number,
                             title: issue.title.to_string(),
                             author,
-                            closed_at: issue.closed_at.clone(),
+                            closed_at: issue.closed_at,
+                            created_at: issue.created_at,
+                            state: issue.state.clone(),
+                            assignees,
                         });
                     }
                     if !repository.issues.page_info.has_next_page {

src/github.rs

@@ -35,6 +35,11 @@ query Issues(
             login
           }
         }
+        assignees(first: 10) {
+          nodes {
+            login
+          }
+        }
       }
     }
   }

src/github/issues.graphql

@@ -1,5 +1,6 @@
-mod issue;
-mod pull_request;
+pub(crate) mod issue;
+pub(crate) mod issue_stat;
+pub(crate) mod pull_request;
 
 use std::fmt::Display;
 
@@ -9,8 +10,6 @@ use async_graphql::{
 };
 use base64::{engine::general_purpose, Engine as _};
 
-pub(crate) use self::issue::Issue;
-pub(crate) use self::pull_request::PullRequest;
 use crate::database::Database;
 
 /// The default page size for connections when neither `first` nor `last` is
@@ -21,7 +20,11 @@ const DEFAULT_PAGE_SIZE: usize = 100;
 ///
 /// This is exposed only for [`Schema`], and not used directly.
 #[derive(Default, MergedObject)]
-pub(crate) struct Query(issue::IssueQuery, pull_request::PullRequestQuery);
+pub(crate) struct Query(
+    issue::IssueQuery,
+    pull_request::PullRequestQuery,
+    issue_stat::IssueStatQuery,
+);
 
 pub(crate) type Schema = async_graphql::Schema<Query, EmptyMutation, EmptySubscription>;
 

src/graphql.rs

@@ -3,10 +3,16 @@ use std::fmt;
 use anyhow::Context as AnyhowContext;
 use async_graphql::{
     connection::{query, Connection, EmptyFields},
-    Context, Object, Result, SimpleObject,
+    scalar, Context, Object, Result, SimpleObject,
 };
+use chrono::{DateTime, Utc};
 
-use crate::database::{self, Database, TryFromKeyValue};
+use crate::{
+    database::{self, Database, TryFromKeyValue},
+    github::{issues::IssueState, GitHubIssue},
+};
+
+scalar!(IssueState);
 
 #[derive(SimpleObject)]
 pub(crate) struct Issue {
@@ -15,19 +21,32 @@ pub(crate) struct Issue {
     pub(crate) number: i32,
     pub(crate) title: String,
     pub(crate) author: String,
+    pub(crate) created_at: DateTime<Utc>,
+    pub(crate) state: IssueState,
+    pub(crate) assignees: Vec<String>,
 }
 
 impl TryFromKeyValue for Issue {
     fn try_from_key_value(key: &[u8], value: &[u8]) -> anyhow::Result<Self> {
         let (owner, repo, number) = database::parse_key(key)
             .with_context(|| format!("invalid key in database: {key:02x?}"))?;
-        let (title, author, _) = bincode::deserialize::<(String, String, Option<String>)>(value)?;
+        let GitHubIssue {
+            title,
+            author,
+            created_at,
+            state,
+            assignees,
+            ..
+        } = bincode::deserialize::<GitHubIssue>(value)?;
         let issue = Issue {
             title,
             author,
             owner,
             repo,
             number: i32::try_from(number).unwrap_or(i32::MAX),
+            created_at,
+            state,
+            assignees,
         };
         Ok(issue)
     }
@@ -69,6 +88,15 @@ impl IssueQuery {
 mod tests {
     use crate::{github::GitHubIssue, graphql::TestSchema};
 
+    fn create_issues(n: usize) -> Vec<GitHubIssue> {
+        (1..=n)
+            .map(|i| GitHubIssue {
+                number: i64::try_from(i).unwrap(),
+                ..Default::default()
+            })
+            .collect()
+    }
+
     #[tokio::test]
     async fn issues_empty() {
         let schema = TestSchema::new();
@@ -89,26 +117,7 @@ mod tests {
     #[tokio::test]
     async fn issues_first() {
         let schema = TestSchema::new();
-        let issues = vec![
-            GitHubIssue {
-                number: 1,
-                title: "issue 1".to_string(),
-                author: "author 1".to_string(),
-                closed_at: None,
-            },
-            GitHubIssue {
-                number: 2,
-                title: "issue 2".to_string(),
-                author: "author 2".to_string(),
-                closed_at: None,
-            },
-            GitHubIssue {
-                number: 3,
-                title: "issue 3".to_string(),
-                author: "author 3".to_string(),
-                closed_at: None,
-            },
-        ];
+        let issues = create_issues(3);
         schema.db.insert_issues(issues, "owner", "name").unwrap();
 
         let query = r"
@@ -148,26 +157,7 @@ mod tests {
     #[tokio::test]
     async fn issues_last() {
         let schema = TestSchema::new();
-        let issues = vec![
-            GitHubIssue {
-                number: 1,
-                title: "issue 1".to_string(),
-                author: "author 1".to_string(),
-                closed_at: None,
-            },
-            GitHubIssue {
-                number: 2,
-                title: "issue 2".to_string(),
-                author: "author 2".to_string(),
-                closed_at: None,
-            },
-            GitHubIssue {
-                number: 3,
-                title: "issue 3".to_string(),
-                author: "author 3".to_string(),
-                closed_at: None,
-            },
-        ];
+        let issues = create_issues(3);
         schema.db.insert_issues(issues, "owner", "name").unwrap();
 
         let query = r"