Merge pull request #45 from tomhoule/github-example

GitHub CLI example

Author: tomhoule

examples/github/.gitignore

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

examples/github/Cargo.toml

@@ -10,3 +10,7 @@ serde = "1.0"
 serde_derive = "1.0"
 serde_json = "1.0"
 reqwest = "*"
+prettytable-rs = "0.7.0"
+structopt = "0.2.10"
+dotenv = "0.13.0"
+envy = "0.3.2"

examples/github/README.md

@@ -1,3 +1,13 @@
 # graphql-client GitHub API examples
 
 The schema is taken from [this repo](https://raw.githubusercontent.com/octokit/graphql-schema/master/schema.graphql).
+
+## How to run it
+
+The example expects to find a valid GitHub API Token in the environment (`GITHUB_API_TOKEN`). See the [GitHub documentation](https://developer.github.com/v4/guides/forming-calls/#authenticating-with-graphql) on how to generate one.
+
+Then just run the example with a repository name as argument. For example:
+
+```bash
+cargo run -- tomhoule/graphql-client
+```

examples/github/src/main.rs

@@ -1,3 +1,6 @@
+extern crate dotenv;
+extern crate envy;
+#[macro_use]
 extern crate failure;
 #[macro_use]
 extern crate graphql_client;
@@ -6,14 +9,103 @@ extern crate serde;
 extern crate serde_json;
 #[macro_use]
 extern crate serde_derive;
+#[macro_use]
+extern crate structopt;
+#[macro_use]
+extern crate prettytable;
 
 use graphql_client::*;
+use structopt::StructOpt;
 
 #[derive(GraphQLQuery)]
 #[gql(schema_path = "src/schema.graphql", query_path = "src/query_1.graphql")]
 struct Query1;
 
+#[derive(StructOpt)]
+struct Command {
+    #[structopt(name = "repository")]
+    repo: String,
+}
+
+#[derive(Deserialize, Debug)]
+struct Env {
+    github_api_token: String,
+}
+
+fn parse_repo_name(repo_name: &str) -> Result<(&str, &str), failure::Error> {
+    let mut parts = repo_name.split("/");
+    match (parts.next(), parts.next()) {
+        (Some(owner), Some(name)) => Ok((owner, name)),
+        _ => Err(format_err!("wrong format for the repository name param (we expect something like facebook/graphql)"))
+    }
+}
+
 fn main() -> Result<(), failure::Error> {
-    let _q = Query1::build_query(query1::Variables);
+    dotenv::dotenv().ok();
+
+    let config: Env = envy::from_env()?;
+
+    let args = Command::from_args();
+
+    let repo = args.repo;
+    let (owner, name) = parse_repo_name(&repo).unwrap_or(("tomhoule", "graphql-client"));
+
+    let q = Query1::build_query(query1::Variables {
+        owner: owner.to_string(),
+        name: name.to_string(),
+    });
+
+    let client = reqwest::Client::new();
+
+    let mut res = client
+        .post("https://api.github.com/graphql")
+        .header(reqwest::header::Authorization(format!(
+            "bearer {}",
+            config.github_api_token
+        )))
+        .json(&q)
+        .send()?;
+
+    let response_body: GraphQLResponse<query1::ResponseData> = res.json()?;
+    let response_data: query1::ResponseData = response_body.data.expect("missing response data");
+
+    let stars: Option<i64> = response_data
+        .repository
+        .as_ref()
+        .map(|repo| repo.stargazers.total_count);
+
+    println!("{}/{} - 🌟 {}", owner, name, stars.unwrap_or(0),);
+
+    let mut table = prettytable::Table::new();
+
+    table.add_row(row!(b => "issue", "comments"));
+
+    for issue in &response_data
+        .repository
+        .expect("missing repository")
+        .issues
+        .nodes
+        .expect("issue nodes is null")
+    {
+        if let Some(issue) = issue {
+            table.add_row(row!(issue.title, issue.comments.total_count));
+        }
+    }
+
+    table.printstd();
     Ok(())
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn parse_repo_name_works() {
+        assert_eq!(
+            parse_repo_name("tomhoule/graphql-client").unwrap(),
+            ("tomhoule", "graphql-client")
+        );
+        assert!(parse_repo_name("abcd").is_err());
+    }
+}

examples/github/src/query_1.graphql

@@ -1,17 +1,21 @@
-query OctocatRepos {
-  repository(owner: "octocat", name: "Hello-World") {
-    issues(last: 20, states: CLOSED) {
-      edges {
-        node {
-          title
-          url
-          labels(first: 5) {
-            edges {
-              node {
-                name
-              }
-            }
-          }
+query RepoView($owner: String!, $name: String!) {
+  repository(owner: $owner, name: $name) {
+    stargazers {
+      totalCount
+    }
+    issues(first: 20, states: OPEN) {
+      nodes {
+        title
+        comments {
+          totalCount
+        }
+      }
+    }
+    pullRequests(first: 20, states: OPEN) {
+      nodes {
+        title
+        commits {
+          totalCount
         }
       }
     }