Read GHSTACK_OAUTH_TOKEN from ~/.gh-stack

timothyandrew · Dec 4, 2020 · d092b05 · d092b05
1 parent 45e54e9
commit d092b05
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 7 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -23,3 +23,4 @@ git2 = "0.13"
 dialoguer = "0.6.2"
 clap = "2.33"
 console = "0.11"
+dotenv = "0.15"
diff --git a/src/main.rs b/src/main.rs
@@ -1,5 +1,4 @@
 use git2::Repository;
-use std::collections::HashMap;
 use std::env;
 use console::style;
 use std::error::Error;
@@ -84,13 +83,10 @@ async fn build_pr_stack(pattern: &str, credentials: &Credentials) -> Result<Flat
 
 #[tokio::main]
 async fn main() -> Result<(), Box<dyn Error>> {
-    let env: HashMap<String, String> = env::vars().collect();
+    dotenv::from_filename(".gh-stack").ok();
 
-    let token = env
-        .get("GHSTACK_OAUTH_TOKEN")
-        .expect("You didn't pass `GHSTACK_OAUTH_TOKEN`");
-
-    let credentials = Credentials::new(token);
+    let token = env::var("GHSTACK_OAUTH_TOKEN").expect("You didn't pass `GHSTACK_OAUTH_TOKEN`");
+    let credentials = Credentials::new(&token);
     let matches = clap().get_matches();
 
     match matches.subcommand() {