get cookie from env

Alextopher · Nov 27, 2023 · e97b1ae · e97b1ae
1 parent 707cdf0
commit e97b1ae
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/.github/workflows/rust.yml b/.github/workflows/rust.yml
@@ -11,7 +11,7 @@ jobs:
   build:
     runs-on: ubuntu-latest
     env:
-      TEST_COOKIE: ${{ secrets.TEST_COOKIE }}
+      AOC_COOKIE: ${{ secrets.AOC_COOKIE }}
     steps:
     - uses: actions/checkout@v3
     - name: Build

diff --git a/aoc-macro/src/lib.rs b/aoc-macro/src/lib.rs
@@ -41,8 +41,14 @@ pub fn get_input(input: TokenStream) -> TokenStream {
     let input_file = std::path::Path::new(&input_file);
 
     if !input_file.exists() {
-        // Step 3: include the cookie file
-        let cookie = include_str!("../../inputs/cookie.txt");
+        // Step 3: get the session from the cookie file, or from an environment variable
+        let mut cookie = std::env::var("AOC_COOKIE").ok();
+
+        if cookie.is_none() {
+            cookie = std::fs::read_to_string("inputs/cookie.txt").ok();
+        }
+
+        let cookie = cookie.expect("Failed to get cookie");
 
         // Step 4: download the input file
         let client = reqwest::blocking::Client::new();