From 7bddc77ba936aa23e15c7debdc0b27a9e6c16fe4 Mon Sep 17 00:00:00 2001 From: Howie Benefiel <5847863+howinator@users.noreply.github.com> Date: Sat, 30 Nov 2024 10:58:27 -0600 Subject: [PATCH] Update main.rs to handle multiple token types --- src/main.rs | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/main.rs b/src/main.rs index d5fe4f3..1b830ac 100644 --- a/src/main.rs +++ b/src/main.rs @@ -99,7 +99,9 @@ async fn build_pr_stack( async fn main() -> Result<(), Box> { dotenv::from_filename(".gh-stack").ok(); - let token = env::var("GHSTACK_OAUTH_TOKEN").expect("You didn't pass `GHSTACK_OAUTH_TOKEN`"); + let token = env::var("GHSTACK_OAUTH_TOKEN") + .or(env::var("GH_TOKEN")) + .expect("You didn't pass `GHSTACK_OAUTH_TOKEN or GH_TOKEN`"); let credentials = Credentials::new(&token); let cli = Cli::parse();